135 lines
3.3 KiB
Go
135 lines
3.3 KiB
Go
// vagrant implements the packer.PostProcessor interface and adds a
|
|
// post-processor that turns artifacts of known builders into Vagrant
|
|
// boxes.
|
|
package vagrant
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/mitchellh/packer/common"
|
|
"github.com/mitchellh/packer/packer"
|
|
"log"
|
|
)
|
|
|
|
var builtins = map[string]string{
|
|
"mitchellh.amazonebs": "aws",
|
|
"mitchellh.virtualbox": "virtualbox",
|
|
"mitchellh.vmware": "vmware",
|
|
}
|
|
|
|
type Config struct {
|
|
common.PackerConfig `mapstructure:",squash"`
|
|
|
|
OutputPath string `mapstructure:"output"`
|
|
}
|
|
|
|
type PostProcessor struct {
|
|
config Config
|
|
premade map[string]packer.PostProcessor
|
|
rawConfigs []interface{}
|
|
}
|
|
|
|
func (p *PostProcessor) Configure(raws ...interface{}) error {
|
|
// Store the raw configs for usage later
|
|
p.rawConfigs = raws
|
|
|
|
_, err := common.DecodeConfig(&p.config, raws...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tpl, err := packer.NewConfigTemplate()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tpl.UserVars = p.config.PackerUserVars
|
|
|
|
// Defaults
|
|
ppExtraConfig := make(map[string]interface{})
|
|
if p.config.OutputPath == "" {
|
|
p.config.OutputPath = "packer_{{ .BuildName }}_{{.Provider}}.box"
|
|
ppExtraConfig["output"] = p.config.OutputPath
|
|
}
|
|
|
|
// Accumulate any errors
|
|
errs := new(packer.MultiError)
|
|
if err := tpl.Validate(p.config.OutputPath); err != nil {
|
|
errs = packer.MultiErrorAppend(
|
|
errs, fmt.Errorf("Error parsing output template: %s", err))
|
|
}
|
|
|
|
// Store the extra configuration for post-processors
|
|
p.rawConfigs = append(p.rawConfigs, ppExtraConfig)
|
|
|
|
// TODO(mitchellh): Properly handle multiple raw configs
|
|
var mapConfig map[string]interface{}
|
|
if err := mapstructure.Decode(raws[0], &mapConfig); err != nil {
|
|
errs = packer.MultiErrorAppend(errs,
|
|
fmt.Errorf("Failed to decode config: %s", err))
|
|
return errs
|
|
}
|
|
|
|
p.premade = make(map[string]packer.PostProcessor)
|
|
for k, raw := range mapConfig {
|
|
pp := keyToPostProcessor(k)
|
|
if pp == nil {
|
|
continue
|
|
}
|
|
|
|
// Create the proper list of configurations
|
|
ppConfigs := make([]interface{}, len(p.rawConfigs), len(p.rawConfigs)+1)
|
|
copy(ppConfigs, p.rawConfigs)
|
|
ppConfigs = append(ppConfigs, raw)
|
|
|
|
if err := pp.Configure(ppConfigs...); err != nil {
|
|
errs = packer.MultiErrorAppend(errs, err)
|
|
}
|
|
|
|
p.premade[k] = pp
|
|
}
|
|
|
|
if len(errs.Errors) > 0 {
|
|
return errs
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *PostProcessor) PostProcess(ui packer.Ui, artifact packer.Artifact) (packer.Artifact, bool, error) {
|
|
ppName, ok := builtins[artifact.BuilderId()]
|
|
if !ok {
|
|
return nil, false, fmt.Errorf("Unknown artifact type, can't build box: %s", artifact.BuilderId())
|
|
}
|
|
|
|
// Use the premade PostProcessor if we have one. Otherwise, we
|
|
// create it and configure it here.
|
|
pp, ok := p.premade[ppName]
|
|
if !ok {
|
|
log.Printf("Premade post-processor for '%s' not found. Creating.", ppName)
|
|
pp = keyToPostProcessor(ppName)
|
|
if pp == nil {
|
|
return nil, false, fmt.Errorf("Vagrant box post-processor not found: %s", ppName)
|
|
}
|
|
|
|
if err := pp.Configure(p.rawConfigs...); err != nil {
|
|
return nil, false, err
|
|
}
|
|
}
|
|
|
|
ui.Say(fmt.Sprintf("Creating Vagrant box for '%s' provider", ppName))
|
|
return pp.PostProcess(ui, artifact)
|
|
}
|
|
|
|
func keyToPostProcessor(key string) packer.PostProcessor {
|
|
switch key {
|
|
case "aws":
|
|
return new(AWSBoxPostProcessor)
|
|
case "virtualbox":
|
|
return new(VBoxBoxPostProcessor)
|
|
case "vmware":
|
|
return new(VMwareBoxPostProcessor)
|
|
default:
|
|
return nil
|
|
}
|
|
}
|