packer-cn/config.go

package main

import (
	"fmt"

	"github.com/hashicorp/packer/common"
	"github.com/hashicorp/packer/helper/communicator"
	"github.com/hashicorp/packer/helper/config"
	"github.com/hashicorp/packer/packer"
	"github.com/hashicorp/packer/template/interpolate"
	"strconv"
	"time"
)

type Config struct {
	common.PackerConfig `mapstructure:",squash"`
	communicator.Config `mapstructure:",squash"`

	VCenterHost string `mapstructure:"vcenter_host"`
	Username    string `mapstructure:"username"`
	Password    string `mapstructure:"password"`

	// Location
	Template     string `mapstructure:"template"`
	VMName       string `mapstructure:"vm_name"`
	FolderName   string `mapstructure:"folder"`
	Datacenter   string `mapstructure:"datacenter"`
	Host         string `mapstructure:"host"`
	ResourcePool string `mapstructure:"resource_pool"`
	Datastore    string `mapstructure:"datastore"`

	// Settings
	LinkedClone        bool   `mapstructure:"linked_clone"`
	ConvertToTemplate  bool   `mapstructure:"convert_to_template"`
	RawShutdownTimeout string `mapstructure:"shutdown_timeout"`
	ShutdownTimeout    time.Duration

	// Customization
	CPUs            string `mapstructure:"CPUs"`
	ShutdownCommand string `mapstructure:"shutdown_command"`
	Ram             string `mapstructure:"RAM"`
	CreateSnapshot  bool   `mapstructure:"create_snapshot"`


	ctx      interpolate.Context
}

func NewConfig(raws ...interface{}) (*Config, []string, error) {
	c := new(Config)
	err := config.Decode(c, &config.DecodeOpts{
		Interpolate:        true,
		InterpolateContext: &c.ctx,
	}, raws...)
	if err != nil {
		return nil, nil, err
	}

	// Accumulate any errors
	errs := new(packer.MultiError)
	var warnings []string

	// Prepare config(s)
	errs = packer.MultiErrorAppend(errs, c.Config.Prepare(&c.ctx)...)

	// Check the required params
	if c.VCenterHost == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("vCenter host required"))
	}
	if c.Username == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Username required"))
	}
	if c.Password == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Password required"))
	}
	if c.Template == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Template VM name required"))
	}
	if c.VMName == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Target VM name required"))
	}
	if c.Host == "" {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Target host required"))
	}

	// Verify numeric parameters if present
	if c.CPUs != "" {
		if _, err = strconv.Atoi(c.CPUs); err != nil {
			errs = packer.MultiErrorAppend(errs, fmt.Errorf("Invalid number of CPU sockets"))
		}
	}
	if c.Ram != "" {
		if _, err = strconv.Atoi(c.Ram); err != nil {
			errs = packer.MultiErrorAppend(errs, fmt.Errorf("Invalid number for Ram"))
		}
	}
	if c.RawShutdownTimeout == "" {
		c.RawShutdownTimeout = "5m"
	}
	c.ShutdownTimeout, err = time.ParseDuration(c.RawShutdownTimeout)
	if err != nil {
		errs = packer.MultiErrorAppend(errs, fmt.Errorf("Failed parsing shutdown_timeout: %s", err))
	}

	//if c.Datastore == "" {
	//	warnings = append(warnings, "Datastore is not specified, will try to find the default one")
	//}

	if len(errs.Errors) > 0 {
		return nil, warnings, errs
	}

	return c, warnings, nil
}