build.go

package main

import (
	"flag"
	"log"
	"os"
	"os/exec"
	"path/filepath"

	"github.com/sirupsen/logrus"
)

func main() {
	flag.Parse()

	for _, cmd := range flag.Args() {
		switch cmd {
		case "generate":
			generate()
		case "install":
			install()
		default:
			log.Fatalf("Unknown command %q", cmd)
		}
	}
}

func runGo(args ...string) error {
	cmd := exec.Command("go", args...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	cmd.Env = os.Environ()
	return cmd.Run()
}

func generate() {
	if err := runGo("generate", "./tools/..."); err != nil {
		logrus.WithError(err).Fatal("generate")
	}
}

func install() {
	generate()

	goPath := os.Getenv("GOPATH")
	if goPath == "" {
		logrus.Fatal("GOPATH was not set")
	}
	binPath := filepath.Join(goPath, "bin", "koci")

	args := []string{"build", "-o", binPath, "github.com/bryanl/koci/cmd/koci"}
	if err := runGo(args...); err != nil {
		logrus.WithError(err).Fatal("build and install koci")
	}
}