116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/mitchellh/packer/packer"
|
|
"github.com/mitchellh/packer/packer/plugin"
|
|
"io"
|
|
"log"
|
|
"os/exec"
|
|
)
|
|
|
|
// This is the default, built-in configuration that ships with
|
|
// Packer.
|
|
const defaultConfig = `
|
|
{
|
|
"plugin_min_port": 10000,
|
|
"plugin_max_port": 25000,
|
|
|
|
"builders": {
|
|
"amazon-ebs": "packer-builder-amazon-ebs",
|
|
"digital-ocean": "packer-builder-digital-ocean",
|
|
"virtualbox": "packer-builder-virtualbox",
|
|
"vmware": "packer-builder-vmware"
|
|
},
|
|
|
|
"commands": {
|
|
"build": "packer-command-build",
|
|
"validate": "packer-command-validate"
|
|
},
|
|
|
|
"provisioners": {
|
|
"shell": "packer-provisioner-shell"
|
|
}
|
|
}
|
|
`
|
|
|
|
type config struct {
|
|
PluginMinPort uint
|
|
PluginMaxPort uint
|
|
|
|
Builders map[string]string
|
|
Commands map[string]string
|
|
Provisioners map[string]string
|
|
}
|
|
|
|
// Decodes configuration in JSON format from the given io.Reader into
|
|
// the config object pointed to.
|
|
func decodeConfig(r io.Reader, c *config) error {
|
|
decoder := json.NewDecoder(r)
|
|
return decoder.Decode(c)
|
|
}
|
|
|
|
// Returns an array of defined command names.
|
|
func (c *config) CommandNames() (result []string) {
|
|
result = make([]string, 0, len(c.Commands))
|
|
for name, _ := range c.Commands {
|
|
result = append(result, name)
|
|
}
|
|
return
|
|
}
|
|
|
|
// This is a proper packer.BuilderFunc that can be used to load packer.Builder
|
|
// implementations from the defined plugins.
|
|
func (c *config) LoadBuilder(name string) (packer.Builder, error) {
|
|
log.Printf("Loading builder: %s\n", name)
|
|
bin, ok := c.Builders[name]
|
|
if !ok {
|
|
log.Printf("Builder not found: %s\n", name)
|
|
return nil, nil
|
|
}
|
|
|
|
return c.pluginClient(bin).Builder()
|
|
}
|
|
|
|
// This is a proper packer.CommandFunc that can be used to load packer.Command
|
|
// implementations from the defined plugins.
|
|
func (c *config) LoadCommand(name string) (packer.Command, error) {
|
|
log.Printf("Loading command: %s\n", name)
|
|
bin, ok := c.Commands[name]
|
|
if !ok {
|
|
log.Printf("Command not found: %s\n", name)
|
|
return nil, nil
|
|
}
|
|
|
|
return c.pluginClient(bin).Command()
|
|
}
|
|
|
|
// This is a proper implementation of packer.HookFunc that can be used
|
|
// to load packer.Hook implementations from the defined plugins.
|
|
func (c *config) LoadHook(name string) (packer.Hook, error) {
|
|
log.Printf("Loading hook: %s\n", name)
|
|
return c.pluginClient(name).Hook()
|
|
}
|
|
|
|
// This is a proper packer.ProvisionerFunc that can be used to load
|
|
// packer.Provisioner implementations from defined plugins.
|
|
func (c *config) LoadProvisioner(name string) (packer.Provisioner, error) {
|
|
log.Printf("Loading provisioner: %s\n", name)
|
|
bin, ok := c.Provisioners[name]
|
|
if !ok {
|
|
log.Printf("Provisioner not found: %s\n", name)
|
|
return nil, nil
|
|
}
|
|
|
|
return c.pluginClient(bin).Provisioner()
|
|
}
|
|
|
|
func (c *config) pluginClient(path string) *plugin.Client {
|
|
var config plugin.ClientConfig
|
|
config.Cmd = exec.Command(path)
|
|
config.Managed = true
|
|
config.MinPort = c.PluginMinPort
|
|
config.MaxPort = c.PluginMaxPort
|
|
return plugin.NewClient(&config)
|
|
}
|