cli.go

package main

import (
	"fmt"
	"github.com/Flipboard/gcredstash/src/gcredstash"
	"github.com/Flipboard/gcredstash/src/gcredstash/command"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/dynamodb"
	"github.com/aws/aws-sdk-go/service/kms"
	"github.com/mitchellh/cli"
	"os"
)

func Run(args []string) int {
	// Meta-option for executables.
	// It defines output color and its stdout/stderr stream.

	awsSession := session.New()

	meta := &command.Meta{
		Ui: &cli.ColoredUi{
			InfoColor:  cli.UiColorBlue,
			ErrorColor: cli.UiColorRed,
			Ui: &cli.BasicUi{
				Writer:      os.Stdout,
				ErrorWriter: os.Stderr,
				Reader:      os.Stdin,
			},
		},
		Table:  os.Getenv("GCREDSTASH_TABLE"),
		KmsKey: os.Getenv("GCREDSTASH_KMS_KEY"),
		Driver: &gcredstash.Driver{
			Ddb: dynamodb.New(awsSession),
			Kms: kms.New(awsSession),
		},
	}

	if meta.Table == "" {
		meta.Table = "credential-store"
	}

	if meta.KmsKey == "" {
		meta.KmsKey = "alias/credstash"
	}

	return RunCustom(args, Commands(meta))
}

func RunCustom(args []string, commands map[string]cli.CommandFactory) int {
	cli := &cli.CLI{
		Args:       args,
		Commands:   commands,
		Version:    Version,
		HelpFunc:   cli.BasicHelpFunc(Name),
		HelpWriter: os.Stdout,
	}

	exitCode, err := cli.Run()
	if err != nil {
		fmt.Fprintf(os.Stderr, "Failed to execute: %s\n", err.Error())
	}

	return exitCode
}