2022-05-14 17:54:35 +00:00
|
|
|
package root
|
2022-05-13 13:30:22 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2022-05-13 15:43:54 +00:00
|
|
|
"log"
|
2022-05-13 13:30:22 +00:00
|
|
|
"os"
|
2022-05-13 15:43:54 +00:00
|
|
|
"strings"
|
2022-05-13 13:30:22 +00:00
|
|
|
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
)
|
|
|
|
|
2022-05-14 17:54:35 +00:00
|
|
|
// RootCmd represents the base command when called without any subcommands
|
|
|
|
var RootCmd = &cobra.Command{
|
2022-05-13 13:30:22 +00:00
|
|
|
Use: "bricks",
|
|
|
|
Short: "Databricks project lifecycle management",
|
|
|
|
Long: `Where's "data"? Secured by the unity catalog. Projects build lifecycle is secured by bricks`,
|
2022-05-14 17:54:35 +00:00
|
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
|
|
if Verbose {
|
|
|
|
logLevel = append(logLevel, "[DEBUG]")
|
|
|
|
}
|
|
|
|
log.SetOutput(&logLevel)
|
|
|
|
},
|
2022-05-13 13:30:22 +00:00
|
|
|
// Uncomment the following line if your bare application
|
|
|
|
// has an action associated with it:
|
2022-05-13 15:43:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: replace with zerolog
|
|
|
|
type levelWriter []string
|
|
|
|
|
|
|
|
var logLevel = levelWriter{"[INFO]", "[ERROR]", "[WARN]"}
|
2022-05-14 17:54:35 +00:00
|
|
|
|
|
|
|
// Verbose means additional debug information, like API logs
|
|
|
|
var Verbose bool
|
2022-05-13 15:43:54 +00:00
|
|
|
|
|
|
|
func (lw *levelWriter) Write(p []byte) (n int, err error) {
|
|
|
|
a := string(p)
|
|
|
|
for _, l := range *lw {
|
|
|
|
if strings.Contains(a, l) {
|
|
|
|
return os.Stdout.Write(p)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return
|
2022-05-13 13:30:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// 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() {
|
2022-05-14 17:54:35 +00:00
|
|
|
// TODO: deferred panic recovery
|
|
|
|
ctx := context.Background()
|
|
|
|
err := RootCmd.ExecuteContext(ctx)
|
2022-05-13 13:30:22 +00:00
|
|
|
if err != nil {
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
2022-05-14 17:54:35 +00:00
|
|
|
RootCmd.PersistentFlags().BoolVarP(&Verbose, "verbose", "v", false, "print debug logs")
|
2022-05-13 13:30:22 +00:00
|
|
|
}
|