// 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, err := env.Cli(os.Args[1:])
	if err != nil {
		fmt.Fprintf(os.Stderr, "Error executing CLI: %s\n", err.Error())
		os.Exit(1)
	}

	plugin.CleanupClients()
	os.Exit(exitCode)
}