main.go

package main

import (
	"os"

	"github.com/segmentio/robo/cli"
	"github.com/segmentio/robo/config"
	"github.com/tj/docopt"
	analytics "gopkg.in/segmentio/analytics-go.v3"
)

// Set by linker flags
var version = "dev"

const analyticsWriteKey = "JX2IUwzrdfKoGilkImNmjRdjNFPmSMfe"

var (
	username        string
	analyticsClient analytics.Client
)

const usage = `
  Usage:
    robo [--config file]
    robo [--xtrace] <task> [<arg>...] [--config file]
    robo help [<task>] [--config file]
    robo variables [--config file]
    robo -h | --help
    robo --version

  Options:
    -c, --config file   config file to load
    -x, --xtrace        set xtrace when running scripts or commands
    -h, --help          output help information
    -v, --version       output version

  Examples:

    output tasks
    $ robo

    output task help
    $ robo help mytask

`

func main() {
	args, err := docopt.Parse(usage, nil, true, version, true)
	if err != nil {
		cli.Fatalf("error parsing arguments: %s", err)
	}

	analyticsClient, _ = analytics.NewWithConfig(analyticsWriteKey, analytics.Config{
		BatchSize: 1,
	})

	username = os.Getenv("USER")
	analyticsClient.Enqueue(analytics.Identify{
		UserId: username,
		Traits: analytics.NewTraits().
			Set("robo-version", version),
	})

	c := args["--config"]
	if c == nil {
		c = os.Getenv("ROBO_CONFIG")
		if c == nil || c == "" {
			cli.Fatalf("robo requires a config file passed via --config or the ROBO_CONFIG env var")
		}
	}

	file := c.(string)
	conf, err := config.New(file)
	if err != nil {
		cli.Fatalf("error loading configuration: %s", err)
	}

	switch {
	case args["help"].(bool):
		if name, ok := args["<task>"].(string); ok {
			cli.Help(conf, name)
		} else {
			cli.List(conf)
		}
	case args["variables"].(bool):
		cli.ListVariables(conf)
	default:
		if name, ok := args["<task>"].(string); ok {
			arguments := args["<arg>"].([]string)
			analyticsClient.Enqueue(analytics.Track{
				UserId: username,
				Event:  "Ran Command",
				Properties: analytics.NewProperties().
					Set("robo-version", version).
					Set("config", file).
					Set("task", name).
					Set("args", arguments),
			})

			xtrace, ok := args["--xtrace"].(bool)
			if xtrace && ok {
				cli.RunTrace(conf, name, arguments)
			} else {
				cli.Run(conf, name, arguments)
			}
		} else {
			cli.List(conf)
		}
	}

	analyticsClient.Close()
}