main.go

package main

import (
	"os"
	"time"

	"github.com/bruin-data/bruin/cmd"
	"github.com/bruin-data/bruin/pkg/telemetry"
	v "github.com/bruin-data/bruin/pkg/version"
	"github.com/fatih/color"
	"github.com/urfave/cli/v2"
)

var (
	version      = "dev"
	commit       = ""
	TelemetryKey string
)

func main() {
	isDebug := false
	color.NoColor = false
	var optOut bool
	if os.Getenv("TELEMETRY_OPTOUT") != "" {
		optOut = true
	}

	v.Version = version
	v.Commit = commit
	if TelemetryKey == "" {
		TelemetryKey = os.Getenv("TELEMETRY_KEY")
	}
	telemetry.TelemetryKey = TelemetryKey
	telemetry.OptOut = optOut
	telemetry.AppVersion = v.Version
	client := telemetry.Init()
	defer client.Close()

	versionCommand := cmd.VersionCmd(v.Commit)

	cli.VersionPrinter = func(cCtx *cli.Context) {
		err := versionCommand.Action(cCtx)
		if err != nil {
			panic(err)
		}
	}

	app := &cli.App{
		Name:           "bruin",
		Version:        version,
		Usage:          "The CLI used for managing Bruin-powered data pipelines",
		Compiled:       time.Now(),
		ExitErrHandler: telemetry.ErrorCommand,
		Flags: []cli.Flag{
			&cli.BoolFlag{
				Name:        "debug",
				Value:       false,
				Usage:       "show debug information",
				Destination: &isDebug,
			},
		},
		Commands: []*cli.Command{
			cmd.Lint(&isDebug),
			cmd.Run(&isDebug),
			cmd.Render(),
			cmd.Lineage(),
			cmd.CleanCmd(),
			cmd.Format(&isDebug),
			cmd.Docs(),
			cmd.Init(),
			cmd.Internal(),
			cmd.Environments(&isDebug),
			cmd.Connections(),
			cmd.Query(),
			versionCommand,
		},
	}

	err := app.Run(os.Args)

	if err != nil {
		cli.HandleExitCoder(err)
	}
}