packer-cn/post-processor/vsphere/post-processor.go

158 lines
4.0 KiB
Go
Raw Normal View History

2013-11-11 03:56:51 -05:00
package vsphere
import (
2013-11-18 18:57:31 -05:00
"bytes"
"fmt"
"github.com/mitchellh/packer/common"
"github.com/mitchellh/packer/packer"
"os/exec"
"strings"
2013-11-11 03:56:51 -05:00
)
var builtins = map[string]string{
2013-11-18 18:57:31 -05:00
"mitchellh.vmware": "vmware",
2013-11-11 03:56:51 -05:00
}
type Config struct {
2013-11-18 18:57:31 -05:00
common.PackerConfig `mapstructure:",squash"`
Insecure bool `mapstructure:"insecure"`
2013-11-29 11:33:26 -05:00
Cluster string `mapstructure:"cluster"`
Datacenter string `mapstructure:"datacenter"`
Datastore string `mapstructure:"datastore"`
2013-11-29 11:33:26 -05:00
Debug bool `mapstructure:"debug"`
Host string `mapstructure:"host"`
Password string `mapstructure:"password"`
2013-11-29 11:33:26 -05:00
ResourcePool string `mapstructure:"resource_pool"`
Username string `mapstructure:"username"`
VMFolder string `mapstructure:"vm_folder"`
VMName string `mapstructure:"vm_name"`
2013-11-18 19:23:28 -05:00
VMNetwork string `mapstructure:"vm_network"`
tpl *packer.ConfigTemplate
2013-11-11 03:56:51 -05:00
}
type PostProcessor struct {
2013-11-18 18:57:31 -05:00
config Config
2013-11-11 03:56:51 -05:00
}
func (p *PostProcessor) Configure(raws ...interface{}) error {
2013-11-18 18:57:31 -05:00
_, err := common.DecodeConfig(&p.config, raws...)
if err != nil {
return err
}
p.config.tpl, err = packer.NewConfigTemplate()
2013-11-18 18:57:31 -05:00
if err != nil {
return err
}
p.config.tpl.UserVars = p.config.PackerUserVars
2013-11-18 18:57:31 -05:00
// Accumulate any errors
errs := new(packer.MultiError)
if _, err := exec.LookPath("ovftool"); err != nil {
2013-11-18 18:57:31 -05:00
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("ovftool not found: %s", err))
2013-11-18 18:57:31 -05:00
}
validates := map[string]*string{
2013-11-29 11:33:26 -05:00
"cluster": &p.config.Cluster,
"datacenter": &p.config.Datacenter,
"datastore": &p.config.Datastore,
"host": &p.config.Host,
"vm_network": &p.config.VMNetwork,
"password": &p.config.Password,
"resource_pool": &p.config.ResourcePool,
"username": &p.config.Username,
"vm_folder": &p.config.VMFolder,
"vm_name": &p.config.VMName,
2013-11-18 18:57:31 -05:00
}
for n := range validates {
if *validates[n] == "" {
errs = packer.MultiErrorAppend(
errs, fmt.Errorf("%s must be set", n))
2013-11-18 18:57:31 -05:00
}
}
if len(errs.Errors) > 0 {
return errs
}
return nil
2013-11-11 03:56:51 -05:00
}
2013-11-18 18:57:31 -05:00
func (p *PostProcessor) PostProcess(ui packer.Ui, artifact packer.Artifact) (packer.Artifact, bool, error) {
if _, ok := builtins[artifact.BuilderId()]; !ok {
2013-11-18 18:57:31 -05:00
return nil, false, fmt.Errorf("Unknown artifact type, can't build box: %s", artifact.BuilderId())
}
vmx := ""
for _, path := range artifact.Files() {
if strings.HasSuffix(path, ".vmx") {
vmx = path
break
2013-11-18 18:57:31 -05:00
}
}
if vmx == "" {
return nil, false, fmt.Errorf("VMX file not found")
}
// Get user variables from template
vm_name, err := p.config.tpl.Process(p.config.VMName, p.config.PackerUserVars)
if err != nil {
return nil, false, fmt.Errorf("Failed: %s", err)
}
username, err := p.config.tpl.Process(p.config.Username, p.config.PackerUserVars)
if err != nil {
return nil, false, fmt.Errorf("Failed: %s", err)
}
password, err := p.config.tpl.Process(p.config.Password, p.config.PackerUserVars)
if err != nil {
return nil, false, fmt.Errorf("Failed: %s", err)
}
datastore, err := p.config.tpl.Process(p.config.Datastore, p.config.PackerUserVars)
if err != nil {
return nil, false, fmt.Errorf("Failed: %s", err)
}
ui.Message(fmt.Sprintf("Uploading %s to vSphere", vmx))
args := []string{
fmt.Sprintf("--noSSLVerify=%t", p.config.Insecure),
"--acceptAllEulas",
fmt.Sprintf("--name=%s", vm_name),
fmt.Sprintf("--datastore=%s", datastore),
fmt.Sprintf("--network=%s", p.config.VMNetwork),
fmt.Sprintf("--vmFolder=%s", p.config.VMFolder),
2013-11-29 11:33:26 -05:00
fmt.Sprintf("%s", vmx),
fmt.Sprintf("vi://%s:%s@%s/%s/host/%s/Resources/%s",
username,
password,
p.config.Host,
p.config.Datacenter,
2013-11-29 11:33:26 -05:00
p.config.Cluster,
p.config.ResourcePool),
}
2013-11-29 11:33:26 -05:00
if p.config.Debug {
ui.Message(fmt.Sprintf("DEBUG: %s", args))
}
2013-11-18 18:57:31 -05:00
var out bytes.Buffer
cmd := exec.Command("ovftool", args...)
2013-11-18 18:57:31 -05:00
cmd.Stdout = &out
if err := cmd.Run(); err != nil {
return nil, false, fmt.Errorf("Failed: %s\nStdout: %s", err, out.String())
2013-11-18 18:57:31 -05:00
}
ui.Message(fmt.Sprintf("%s", out.String()))
return artifact, false, nil
2013-11-11 03:56:51 -05:00
}