packer-cn/command/build/command.go

package build

import (
	"fmt"
	"github.com/mitchellh/packer/packer"
	"io/ioutil"
	"log"
	"sync"
)

type Command byte

func (Command) Run(env packer.Environment, args []string) int {
	if len(args) != 1 {
		env.Ui().Error("A single template argument is required.\n")
		return 1
	}

	// Read the file into a byte array so that we can parse the template
	log.Printf("Reading template: %s\n", args[0])
	tplData, err := ioutil.ReadFile(args[0])
	if err != nil {
		env.Ui().Error("Failed to read template file: %s\n", err.Error())
		return 1
	}

	// Parse the template into a machine-usable format
	log.Println("Parsing template...")
	tpl, err := packer.ParseTemplate(tplData)
	if err != nil {
		env.Ui().Error("Failed to parse template: %s\n", err.Error())
		return 1
	}

	// The component finder for our builds
	components := &packer.ComponentFinder{
		Builder: env.Builder,
		Hook:    env.Hook,
	}

	// Go through each builder and compile the builds that we care about
	buildNames := tpl.BuildNames()
	builds := make([]packer.Build, 0, len(buildNames))
	for _, buildName := range buildNames {
		log.Printf("Creating build: %s\n", buildName)
		build, err := tpl.Build(buildName, components)
		if err != nil {
			env.Ui().Error("Failed to create build '%s': \n\n%s\n", buildName, err.Error())
			return 1
		}

		builds = append(builds, build)
	}

	// Compile all the UIs for the builds
	buildUis := make(map[string]packer.Ui)
	for _, b := range builds {
		buildUis[b.Name()] = &packer.PrefixedUi{
			fmt.Sprintf("==> %s", b.Name()),
			env.Ui(),
		}
	}

	// Prepare all the builds
	for _, b := range builds {
		log.Printf("Preparing build: %s\n", b.Name())
		err := b.Prepare()
		if err != nil {
			env.Ui().Error("%s\n", err)
			return 1
		}
	}

	// Run all the builds in parallel and wait for them to complete
	var wg sync.WaitGroup
	for _, b := range builds {
		log.Printf("Starting build run: %s\n", b.Name())

		// Increment the waitgroup so we wait for this item to finish properly
		wg.Add(1)

		// Run the build in a goroutine
		go func() {
			defer wg.Done()
			b.Run(buildUis[b.Name()])
		}()
	}

	wg.Wait()
	return 0
}

func (Command) Synopsis() string {
	return "build image(s) from template"
}