main.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"strings"
	"time"

	"github.com/docker/docker/pkg/term"
	"github.com/erikh/box/builder"
	"github.com/erikh/box/log"
	"github.com/erikh/box/repl"
	"github.com/fatih/color"
	"github.com/urfave/cli"
)

var (
	// Version is the version of the application
	Version = "0.3.2"
	// Name is the name of the application
	Name = "box"
	// Email is my email
	Email = "github@hollensbe.org"
	// Usage is the title of the application
	Usage = "Advanced mruby Container Image Builder"
	// Author is me
	Author = "Erik Hollensbe"

	// Copyright is the copyright, generated automatically for each year.
	Copyright = fmt.Sprintf("(C) %d %s - Licensed under MIT license", time.Now().Year(), Author)
	// UsageText is the description of how to use the program.
	UsageText = "box [options] filename"
)

func main() {
	app := cli.NewApp()

	app.Name = Name
	app.Email = Email
	app.Version = "0.3.1"
	app.Usage = Usage
	app.Author = Author
	app.Copyright = Copyright
	app.UsageText = UsageText
	app.HideHelp = true
	app.Flags = []cli.Flag{
		cli.BoolFlag{
			Name:  "no-cache, n",
			Usage: "Disable the build cache",
		},
		cli.BoolFlag{
			Name:  "no-tty",
			Usage: "Disable TTY features this run",
		},
		cli.BoolFlag{
			Name:  "force-tty",
			Usage: "Force TTY features this run",
		},
		cli.BoolFlag{
			Name:  "help, h",
			Usage: "Show the help",
		},
		cli.StringFlag{
			Name:  "tag, t",
			Usage: "Tag the last image with this name",
		},
		cli.StringSliceFlag{
			Name:  "omit, o",
			Usage: "Omit functions/verbs. One per option, repeatable.",
		},
	}

	app.Commands = []cli.Command{
		{
			Name:        "repl",
			Action:      runRepl,
			Description: "Run the read-eval-print loop to interactively work with box",
			Usage:       "Run the read-eval-print loop to interactively work with box",
			ArgsUsage:   " ",
		},
		{
			Name:        "shell",
			Action:      runRepl,
			Description: "Run the read-eval-print loop to interactively work with box",
			Usage:       "Run the read-eval-print loop to interactively work with box",
			ArgsUsage:   " ",
		},
	}

	app.Action = func(ctx *cli.Context) {
		if ctx.Bool("help") {
			cli.ShowAppHelp(ctx)
			os.Exit(0)
		}

		args := ctx.Args()

		tty := !ctx.Bool("no-tty")

		if !term.IsTerminal(0) {
			tty = ctx.Bool("force-tty")
		}

		b, err := builder.NewBuilder(tty, ctx.StringSlice("omit"))
		if err != nil {
			panic(err)
		}
		defer b.Close()

		var content []byte

		if len(args) == 1 {
			content, err = ioutil.ReadFile(args[0])
		} else {
			cli.ShowAppHelp(ctx)
			color.Red("!!! Please provide a filename to process!\n\n")
			os.Exit(1)
		}

		if err != nil {
			fmt.Printf("\n\n!!! Error: %v\n", err.Error())
			os.Exit(2)
		}

		if ctx.Bool("no-cache") {
			b.SetCache(false)
		}

		response, err := b.Run(string(content))
		if err != nil {
			fmt.Printf("\n\n!!! Error: %v\n", err.Error())
			os.Exit(1)
		}

		if response.String() != "" {
			log.EvalResponse(response.String())
		}

		tag := ctx.String("tag")

		if tag != "" {
			if err := b.Tag(tag); err != nil {
				fmt.Printf("!!! Can't tag with tag %q: %v\n", tag, err)
				os.Exit(1)
			}
			log.Tag(tag)
		}

		id := b.ImageID()

		if strings.Contains(id, ":") {
			id = strings.SplitN(id, ":", 2)[1]
		}

		log.Finish(id)
	}

	if err := app.Run(os.Args); err != nil {
		fmt.Fprintf(os.Stderr, "\n\n!!! Error: %v\n", err)
		os.Exit(1)
	}
}

func runRepl(ctx *cli.Context) {
	r, err := repl.NewRepl()
	if err != nil {
		fmt.Fprintf(os.Stderr, "\n\n!!! Error bootstrapping repl: %v\n", err)
		os.Exit(1)
	}

	if err := r.Loop(); err != nil {
		fmt.Printf("\n\n!!! Error: %v\n", err)
		os.Exit(1)
	}
}