builder/googlecompute: move config out to seperate file

This commit is contained in:
Mitchell Hashimoto 2013-12-12 16:41:44 -08:00
parent a78b10c06d
commit ee9ea170af
9 changed files with 202 additions and 161 deletions

View File

@ -3,8 +3,6 @@
package googlecompute package googlecompute
import ( import (
"errors"
"fmt"
"log" "log"
"time" "time"
@ -18,163 +16,19 @@ const BuilderId = "packer.googlecompute"
// Builder represents a Packer Builder. // Builder represents a Packer Builder.
type Builder struct { type Builder struct {
config config config *Config
runner multistep.Runner runner multistep.Runner
} }
// config holds the googlecompute builder configuration settings.
type config struct {
BucketName string `mapstructure:"bucket_name"`
ClientSecretsFile string `mapstructure:"client_secrets_file"`
ImageName string `mapstructure:"image_name"`
ImageDescription string `mapstructure:"image_description"`
MachineType string `mapstructure:"machine_type"`
Metadata map[string]string `mapstructure:"metadata"`
Network string `mapstructure:"network"`
Passphrase string `mapstructure:"passphrase"`
PrivateKeyFile string `mapstructure:"private_key_file"`
ProjectId string `mapstructure:"project_id"`
SourceImage string `mapstructure:"source_image"`
SSHUsername string `mapstructure:"ssh_username"`
SSHPort uint `mapstructure:"ssh_port"`
RawSSHTimeout string `mapstructure:"ssh_timeout"`
RawStateTimeout string `mapstructure:"state_timeout"`
Tags []string `mapstructure:"tags"`
Zone string `mapstructure:"zone"`
clientSecrets *clientSecrets
common.PackerConfig `mapstructure:",squash"`
instanceName string
privateKeyBytes []byte
sshTimeout time.Duration
stateTimeout time.Duration
tpl *packer.ConfigTemplate
}
// Prepare processes the build configuration parameters. // Prepare processes the build configuration parameters.
func (b *Builder) Prepare(raws ...interface{}) ([]string, error) { func (b *Builder) Prepare(raws ...interface{}) ([]string, error) {
// Load the packer config. c, warnings, errs := NewConfig(raws...)
md, err := common.DecodeConfig(&b.config, raws...) if errs != nil {
if err != nil { return warnings, errs
return nil, err
} }
b.config.tpl, err = packer.NewConfigTemplate() b.config = c
if err != nil {
return nil, err
}
b.config.tpl.UserVars = b.config.PackerUserVars
errs := common.CheckUnusedConfig(md) return warnings, nil
// Collect errors if any.
if err := common.CheckUnusedConfig(md); err != nil {
return nil, err
}
// Set defaults.
if b.config.Network == "" {
b.config.Network = "default"
}
if b.config.ImageDescription == "" {
b.config.ImageDescription = "Created by Packer"
}
if b.config.ImageName == "" {
// Default to packer-{{ unix timestamp (utc) }}
b.config.ImageName = "packer-{{timestamp}}"
}
if b.config.MachineType == "" {
b.config.MachineType = "n1-standard-1"
}
if b.config.RawSSHTimeout == "" {
b.config.RawSSHTimeout = "5m"
}
if b.config.RawStateTimeout == "" {
b.config.RawStateTimeout = "5m"
}
if b.config.SSHUsername == "" {
b.config.SSHUsername = "root"
}
if b.config.SSHPort == 0 {
b.config.SSHPort = 22
}
// Process Templates
templates := map[string]*string{
"bucket_name": &b.config.BucketName,
"client_secrets_file": &b.config.ClientSecretsFile,
"image_name": &b.config.ImageName,
"image_description": &b.config.ImageDescription,
"machine_type": &b.config.MachineType,
"network": &b.config.Network,
"passphrase": &b.config.Passphrase,
"private_key_file": &b.config.PrivateKeyFile,
"project_id": &b.config.ProjectId,
"source_image": &b.config.SourceImage,
"ssh_username": &b.config.SSHUsername,
"ssh_timeout": &b.config.RawSSHTimeout,
"state_timeout": &b.config.RawStateTimeout,
"zone": &b.config.Zone,
}
for n, ptr := range templates {
var err error
*ptr, err = b.config.tpl.Process(*ptr, nil)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Error processing %s: %s", n, err))
}
}
// Process required parameters.
if b.config.BucketName == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a bucket_name must be specified"))
}
if b.config.ClientSecretsFile == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a client_secrets_file must be specified"))
}
if b.config.PrivateKeyFile == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a private_key_file must be specified"))
}
if b.config.ProjectId == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a project_id must be specified"))
}
if b.config.SourceImage == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a source_image must be specified"))
}
if b.config.Zone == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a zone must be specified"))
}
// Process timeout settings.
sshTimeout, err := time.ParseDuration(b.config.RawSSHTimeout)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing ssh_timeout: %s", err))
}
b.config.sshTimeout = sshTimeout
stateTimeout, err := time.ParseDuration(b.config.RawStateTimeout)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing state_timeout: %s", err))
}
b.config.stateTimeout = stateTimeout
// Load the client secrets file.
cs, err := loadClientSecrets(b.config.ClientSecretsFile)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing client secrets file: %s", err))
}
b.config.clientSecrets = cs
// Load the private key.
b.config.privateKeyBytes, err = processPrivateKeyFile(b.config.PrivateKeyFile, b.config.Passphrase)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed loading private key file: %s", err))
}
// Check for any errors.
if errs != nil && len(errs.Errors) > 0 {
return nil, errs
}
return nil, nil
} }
// Run executes a googlecompute Packer build and returns a packer.Artifact // Run executes a googlecompute Packer build and returns a packer.Artifact

View File

@ -0,0 +1,187 @@
package googlecompute
import (
"errors"
"fmt"
"time"
"github.com/mitchellh/packer/common"
"github.com/mitchellh/packer/packer"
)
// Config is the configuration structure for the GCE builder. It stores
// both the publicly settable state as well as the privately generated
// state of the config object.
type Config struct {
common.PackerConfig `mapstructure:",squash"`
BucketName string `mapstructure:"bucket_name"`
ClientSecretsFile string `mapstructure:"client_secrets_file"`
ImageName string `mapstructure:"image_name"`
ImageDescription string `mapstructure:"image_description"`
MachineType string `mapstructure:"machine_type"`
Metadata map[string]string `mapstructure:"metadata"`
Network string `mapstructure:"network"`
Passphrase string `mapstructure:"passphrase"`
PrivateKeyFile string `mapstructure:"private_key_file"`
ProjectId string `mapstructure:"project_id"`
SourceImage string `mapstructure:"source_image"`
SSHUsername string `mapstructure:"ssh_username"`
SSHPort uint `mapstructure:"ssh_port"`
RawSSHTimeout string `mapstructure:"ssh_timeout"`
RawStateTimeout string `mapstructure:"state_timeout"`
Tags []string `mapstructure:"tags"`
Zone string `mapstructure:"zone"`
clientSecrets *clientSecrets
instanceName string
privateKeyBytes []byte
sshTimeout time.Duration
stateTimeout time.Duration
tpl *packer.ConfigTemplate
}
func NewConfig(raws ...interface{}) (*Config, []string, error) {
c := new(Config)
md, err := common.DecodeConfig(c, raws...)
if err != nil {
return nil, nil, err
}
c.tpl, err = packer.NewConfigTemplate()
if err != nil {
return nil, nil, err
}
c.tpl.UserVars = c.PackerUserVars
// Prepare the errors
errs := common.CheckUnusedConfig(md)
// Set defaults.
if c.Network == "" {
c.Network = "default"
}
if c.ImageDescription == "" {
c.ImageDescription = "Created by Packer"
}
if c.ImageName == "" {
c.ImageName = "packer-{{timestamp}}"
}
if c.MachineType == "" {
c.MachineType = "n1-standard-1"
}
if c.RawSSHTimeout == "" {
c.RawSSHTimeout = "5m"
}
if c.RawStateTimeout == "" {
c.RawStateTimeout = "5m"
}
if c.SSHUsername == "" {
c.SSHUsername = "root"
}
if c.SSHPort == 0 {
c.SSHPort = 22
}
// Process Templates
templates := map[string]*string{
"bucket_name": &c.BucketName,
"client_secrets_file": &c.ClientSecretsFile,
"image_name": &c.ImageName,
"image_description": &c.ImageDescription,
"machine_type": &c.MachineType,
"network": &c.Network,
"passphrase": &c.Passphrase,
"private_key_file": &c.PrivateKeyFile,
"project_id": &c.ProjectId,
"source_image": &c.SourceImage,
"ssh_username": &c.SSHUsername,
"ssh_timeout": &c.RawSSHTimeout,
"state_timeout": &c.RawStateTimeout,
"zone": &c.Zone,
}
for n, ptr := range templates {
var err error
*ptr, err = c.tpl.Process(*ptr, nil)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Error processing %s: %s", n, err))
}
}
// Process required parameters.
if c.BucketName == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a bucket_name must be specified"))
}
if c.ClientSecretsFile == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a client_secrets_file must be specified"))
}
if c.PrivateKeyFile == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a private_key_file must be specified"))
}
if c.ProjectId == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a project_id must be specified"))
}
if c.SourceImage == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a source_image must be specified"))
}
if c.Zone == "" {
errs = packer.MultiErrorAppend(
errs, errors.New("a zone must be specified"))
}
// Process timeout settings.
sshTimeout, err := time.ParseDuration(c.RawSSHTimeout)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing ssh_timeout: %s", err))
}
c.sshTimeout = sshTimeout
stateTimeout, err := time.ParseDuration(c.RawStateTimeout)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing state_timeout: %s", err))
}
c.stateTimeout = stateTimeout
// Load the client secrets file.
cs, err := loadClientSecrets(c.ClientSecretsFile)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed parsing client secrets file: %s", err))
}
c.clientSecrets = cs
// Load the private key.
c.privateKeyBytes, err = processPrivateKeyFile(c.PrivateKeyFile, c.Passphrase)
if err != nil {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("Failed loading private key file: %s", err))
}
// Check for any errors.
if errs != nil && len(errs.Errors) > 0 {
return nil, nil, errs
}
return c, nil, nil
}

View File

@ -10,14 +10,14 @@ import (
// sshAddress returns the ssh address. // sshAddress returns the ssh address.
func sshAddress(state multistep.StateBag) (string, error) { func sshAddress(state multistep.StateBag) (string, error) {
config := state.Get("config").(config) config := state.Get("config").(*Config)
ipAddress := state.Get("instance_ip").(string) ipAddress := state.Get("instance_ip").(string)
return fmt.Sprintf("%s:%d", ipAddress, config.SSHPort), nil return fmt.Sprintf("%s:%d", ipAddress, config.SSHPort), nil
} }
// sshConfig returns the ssh configuration. // sshConfig returns the ssh configuration.
func sshConfig(state multistep.StateBag) (*gossh.ClientConfig, error) { func sshConfig(state multistep.StateBag) (*gossh.ClientConfig, error) {
config := state.Get("config").(config) config := state.Get("config").(*Config)
privateKey := state.Get("ssh_private_key").(string) privateKey := state.Get("ssh_private_key").(string)
keyring := new(ssh.SimpleKeychain) keyring := new(ssh.SimpleKeychain)

View File

@ -18,7 +18,7 @@ type stepCreateImage int
// command on the running GCE instance. // command on the running GCE instance.
func (s *stepCreateImage) Run(state multistep.StateBag) multistep.StepAction { func (s *stepCreateImage) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
config = state.Get("config").(config) config = state.Get("config").(*Config)
comm = state.Get("communicator").(packer.Communicator) comm = state.Get("communicator").(packer.Communicator)
sudoPrefix = "" sudoPrefix = ""
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)

View File

@ -18,7 +18,7 @@ type stepCreateInstance struct {
func (s *stepCreateInstance) Run(state multistep.StateBag) multistep.StepAction { func (s *stepCreateInstance) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
client = state.Get("client").(*GoogleComputeClient) client = state.Get("client").(*GoogleComputeClient)
config = state.Get("config").(config) config = state.Get("config").(*Config)
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)
) )
ui.Say("Creating instance...") ui.Say("Creating instance...")
@ -106,7 +106,7 @@ func (s *stepCreateInstance) Run(state multistep.StateBag) multistep.StepAction
func (s *stepCreateInstance) Cleanup(state multistep.StateBag) { func (s *stepCreateInstance) Cleanup(state multistep.StateBag) {
var ( var (
client = state.Get("client").(*GoogleComputeClient) client = state.Get("client").(*GoogleComputeClient)
config = state.Get("config").(config) config = state.Get("config").(*Config)
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)
) )
if s.instanceName == "" { if s.instanceName == "" {

View File

@ -14,7 +14,7 @@ type stepInstanceInfo int
func (s *stepInstanceInfo) Run(state multistep.StateBag) multistep.StepAction { func (s *stepInstanceInfo) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
client = state.Get("client").(*GoogleComputeClient) client = state.Get("client").(*GoogleComputeClient)
config = state.Get("config").(config) config = state.Get("config").(*Config)
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)
) )
instanceName := state.Get("instance_name").(string) instanceName := state.Get("instance_name").(string)

View File

@ -14,7 +14,7 @@ type stepRegisterImage int
func (s *stepRegisterImage) Run(state multistep.StateBag) multistep.StepAction { func (s *stepRegisterImage) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
client = state.Get("client").(*GoogleComputeClient) client = state.Get("client").(*GoogleComputeClient)
config = state.Get("config").(config) config = state.Get("config").(*Config)
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)
) )
ui.Say("Adding image to the project...") ui.Say("Adding image to the project...")

View File

@ -19,7 +19,7 @@ type stepUpdateGsutil int
// prompt to update gsutil if a newer version is available. // prompt to update gsutil if a newer version is available.
func (s *stepUpdateGsutil) Run(state multistep.StateBag) multistep.StepAction { func (s *stepUpdateGsutil) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
config = state.Get("config").(config) config = state.Get("config").(*Config)
comm = state.Get("communicator").(packer.Communicator) comm = state.Get("communicator").(packer.Communicator)
sudoPrefix = "" sudoPrefix = ""
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)

View File

@ -13,7 +13,7 @@ type stepUploadImage int
// Run executes the Packer build step that uploads a GCE machine image. // Run executes the Packer build step that uploads a GCE machine image.
func (s *stepUploadImage) Run(state multistep.StateBag) multistep.StepAction { func (s *stepUploadImage) Run(state multistep.StateBag) multistep.StepAction {
var ( var (
config = state.Get("config").(config) config = state.Get("config").(*Config)
comm = state.Get("communicator").(packer.Communicator) comm = state.Get("communicator").(packer.Communicator)
sudoPrefix = "" sudoPrefix = ""
ui = state.Get("ui").(packer.Ui) ui = state.Get("ui").(packer.Ui)