mirror of https://github.com/databricks/cli.git
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package root
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// RootCmd represents the base command when called without any subcommands
|
|
var RootCmd = &cobra.Command{
|
|
Use: "bricks",
|
|
Short: "Databricks project lifecycle management",
|
|
Long: `Where's "data"? Secured by the unity catalog. Projects build lifecycle is secured by bricks`,
|
|
|
|
// Cobra prints the usage string to stderr if a command returns an error.
|
|
// This usage string should only be displayed if an invalid combination of flags
|
|
// is specified and not when runtime errors occur (e.g. resource not found).
|
|
// The usage string is include in [flagErrorFunc] for flag errors only.
|
|
SilenceUsage: true,
|
|
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
ctx := cmd.Context()
|
|
|
|
// Configure default logger.
|
|
ctx, err := initializeLogger(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Configure progress logger
|
|
ctx, err = initializeProgressLogger(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Configure our user agent with the command that's about to be executed.
|
|
ctx = withCommandInUserAgent(ctx, cmd)
|
|
ctx = withUpstreamInUserAgent(ctx)
|
|
cmd.SetContext(ctx)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
// Wrap flag errors to include the usage string.
|
|
func flagErrorFunc(c *cobra.Command, err error) error {
|
|
return fmt.Errorf("%w\n\n%s", err, c.UsageString())
|
|
}
|
|
|
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
|
// This is called by main.main(). It only needs to happen once to the rootCmd.
|
|
func Execute() {
|
|
// TODO: deferred panic recovery
|
|
ctx := context.Background()
|
|
err := RootCmd.ExecuteContext(ctx)
|
|
if err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
RootCmd.SetFlagErrorFunc(flagErrorFunc)
|
|
|
|
// The VS Code extension passes `-v` in debug mode and must be changed
|
|
// to use the new flags in `./logger.go` prior to removing this flag.
|
|
RootCmd.PersistentFlags().BoolP("verbose", "v", false, "")
|
|
RootCmd.PersistentFlags().MarkHidden("verbose")
|
|
}
|