squadron/cmd/actions/root.go

package actions

import (
	"os"
	"path"

	"github.com/foomo/configurd"
	"github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

var (
	log     = logrus.New()
	rootCmd = &cobra.Command{
		Use: "configurd",
		PersistentPreRun: func(cmd *cobra.Command, args []string) {
			wdir, err := os.Getwd()
			if err != nil {
				log.Fatal(err)
			}
			if flagDir != "" {
				flagDir = path.Join(wdir, flagDir)
			} else {
				flagDir = wdir
			}
			if cmd.Name() == "help" || cmd.Name() == "init" {
				return
			}
		},
	}

	flagTag       string
	flagDir       string
	flagVerbose   bool
	flagNamespace string
)

func newConfigurd(log *logrus.Logger, tag, basePath string) (configurd.Configurd, error) {
	config := configurd.Config{
		Tag:      tag,
		BasePath: basePath,
		Log:      log,
	}

	return configurd.New(config)
}

func mustNewConfigurd(log *logrus.Logger, tag, basePath string) configurd.Configurd {
	cnf, err := newConfigurd(log, tag, basePath)
	if err != nil {
		log.Fatal(err)
	}
	return cnf
}

func init() {
	rootCmd.PersistentFlags().StringVarP(&flagTag, "tag", "t", "latest", "Specifies the image tag")
	rootCmd.PersistentFlags().StringVarP(&flagDir, "dir", "d", "", "Specifies working directory")
	rootCmd.PersistentFlags().BoolVarP(&flagVerbose, "verbose", "v", false, "Specifies should command output be displayed")
	rootCmd.AddCommand(buildCmd, installCmd, uninstallCmd, initCmd, versionCmd)
}

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		log.Fatal(err)
	}

}

func newLogger(verbose bool) *logrus.Logger {
	logger := logrus.New()
	if verbose {
		logger.SetLevel(logrus.TraceLevel)
	}
	return logger
}