packer-cn/packer.go

57 lines
1.2 KiB
Go

// This is the main package for the `packer` application.
package main
import (
"github.com/mitchellh/packer/packer"
"github.com/mitchellh/packer/packer/plugin"
"fmt"
"io/ioutil"
"log"
"os"
"os/exec"
)
func main() {
if os.Getenv("PACKER_LOG") == "" {
// If we don't have logging explicitly enabled, then disable it
log.SetOutput(ioutil.Discard)
} else {
// Logging is enabled, make sure it goes to stderr
log.SetOutput(os.Stderr)
}
defer plugin.CleanupClients()
commands := map[string]string {
"build": "packer-build",
}
commandKeys := make([]string, 0, len(commands))
for k, _ := range commands {
commandKeys = append(commandKeys, k)
}
envConfig := packer.DefaultEnvironmentConfig()
envConfig.Commands = commandKeys
envConfig.CommandFunc = func(n string) (packer.Command, error) {
log.Printf("Loading command: %s\n", n)
commandBin, ok := commands[n]
if !ok {
log.Printf("Command not found: %s\n", n)
return nil, nil
}
return plugin.Command(exec.Command(commandBin))
}
env, err := packer.NewEnvironment(envConfig)
if err != nil {
fmt.Fprintf(os.Stderr, "Packer initialization error: \n\n%s\n", err)
os.Exit(1)
}
exitCode, _ := env.Cli(os.Args[1:])
plugin.CleanupClients()
os.Exit(exitCode)
}