mirror of https://github.com/databricks/cli.git
82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package root
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"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,
|
|
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
ctx := cmd.Context()
|
|
|
|
// Configure our user agent with the command that's about to be executed.
|
|
ctx = withCommandInUserAgent(ctx, cmd)
|
|
ctx = withUpstreamInUserAgent(ctx)
|
|
cmd.SetContext(ctx)
|
|
|
|
if Verbose {
|
|
logLevel = append(logLevel, "[DEBUG]")
|
|
}
|
|
log.SetOutput(&logLevel)
|
|
},
|
|
|
|
// Uncomment the following line if your bare application
|
|
// has an action associated with it:
|
|
}
|
|
|
|
// TODO: replace with zerolog
|
|
type levelWriter []string
|
|
|
|
var logLevel = levelWriter{"[INFO]", "[ERROR]", "[WARN]"}
|
|
|
|
// Verbose means additional debug information, like API logs
|
|
var Verbose bool
|
|
|
|
func (lw *levelWriter) Write(p []byte) (n int, err error) {
|
|
a := string(p)
|
|
for _, l := range *lw {
|
|
if strings.Contains(a, l) {
|
|
return os.Stderr.Write(p)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// 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)
|
|
// flags available for every child command
|
|
RootCmd.PersistentFlags().BoolVarP(&Verbose, "verbose", "v", false, "print debug logs")
|
|
}
|