packer-cn/step_connect.go

package main

import (
	"github.com/mitchellh/multistep"
	"fmt"
)

type ConnectConfig struct {
	VCenterServer      string `mapstructure:"vcenter_server"`
	Datacenter         string `mapstructure:"datacenter"`
	Username           string `mapstructure:"username"`
	Password           string `mapstructure:"password"`
	InsecureConnection bool   `mapstructure:"insecure_connection"`
}

func (c *ConnectConfig) Prepare() []error {
	var errs []error

	if c.VCenterServer == "" {
		errs = append(errs, fmt.Errorf("vCenter hostname is required"))
	}
	if c.Username == "" {
		errs = append(errs, fmt.Errorf("Username is required"))
	}
	if c.Password == "" {
		errs = append(errs, fmt.Errorf("Password is required"))
	}

	return errs
}

type StepConnect struct {
	config *ConnectConfig
}

func (s *StepConnect) Run(state multistep.StateBag) multistep.StepAction {
	driver, err := NewDriverVSphere(s.config)
	if err != nil {
		state.Put("error", err)
		return multistep.ActionHalt
	}
	state.Put("driver", driver)

	return multistep.ActionContinue
}

func (s *StepConnect) Cleanup(multistep.StateBag) {}