100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
const (
|
|
// BuilderId for the local artifacts
|
|
BuilderId = "mitchellh.vmware"
|
|
BuilderIdESX = "mitchellh.vmware-esx"
|
|
|
|
ArtifactConfFormat = "artifact.conf.format"
|
|
ArtifactConfKeepRegistered = "artifact.conf.keep_registered"
|
|
ArtifactConfSkipExport = "artifact.conf.skip_export"
|
|
)
|
|
|
|
// Artifact is the result of running the VMware builder, namely a set
|
|
// of files associated with the resulting machine.
|
|
type artifact struct {
|
|
builderId string
|
|
id string
|
|
dir OutputDir
|
|
f []string
|
|
config map[string]string
|
|
|
|
// StateData should store data such as GeneratedData
|
|
// to be shared with post-processors
|
|
StateData map[string]interface{}
|
|
}
|
|
|
|
func (a *artifact) BuilderId() string {
|
|
return a.builderId
|
|
}
|
|
|
|
func (a *artifact) Files() []string {
|
|
return a.f
|
|
}
|
|
|
|
func (a *artifact) Id() string {
|
|
return a.id
|
|
}
|
|
|
|
func (a *artifact) String() string {
|
|
return fmt.Sprintf("VM files in directory: %s", a.dir)
|
|
}
|
|
|
|
func (a *artifact) State(name string) interface{} {
|
|
if _, ok := a.StateData[name]; ok {
|
|
return a.StateData[name]
|
|
}
|
|
return a.config[name]
|
|
}
|
|
|
|
func (a *artifact) Destroy() error {
|
|
if a.dir != nil {
|
|
return a.dir.RemoveAll()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func NewArtifact(remoteType string, format string, exportOutputPath string, vmName string, skipExport bool, keepRegistered bool, state multistep.StateBag) (packer.Artifact, error) {
|
|
var files []string
|
|
var dir OutputDir
|
|
var err error
|
|
if remoteType != "" && !skipExport {
|
|
dir = new(LocalOutputDir)
|
|
dir.SetOutputDir(exportOutputPath)
|
|
} else {
|
|
dir = state.Get("dir").(OutputDir)
|
|
}
|
|
files, err = dir.ListFiles()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Set the proper builder ID
|
|
builderId := BuilderId
|
|
if remoteType != "" {
|
|
builderId = BuilderIdESX
|
|
}
|
|
|
|
config := make(map[string]string)
|
|
config[ArtifactConfKeepRegistered] = strconv.FormatBool(keepRegistered)
|
|
config[ArtifactConfFormat] = format
|
|
config[ArtifactConfSkipExport] = strconv.FormatBool(skipExport)
|
|
|
|
return &artifact{
|
|
builderId: builderId,
|
|
id: vmName,
|
|
dir: dir,
|
|
f: files,
|
|
config: config,
|
|
StateData: map[string]interface{}{"generated_data": state.Get("generated_data")},
|
|
}, nil
|
|
}
|