mirror of https://github.com/databricks/cli.git
51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package root
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/databricks/bricks/libs/flags"
|
|
"github.com/databricks/bricks/libs/log"
|
|
"github.com/spf13/cobra"
|
|
"golang.org/x/exp/slog"
|
|
)
|
|
|
|
func initializeLogger(ctx context.Context, cmd *cobra.Command) (context.Context, error) {
|
|
opts := slog.HandlerOptions{}
|
|
opts.Level = logLevel.Level()
|
|
opts.AddSource = true
|
|
opts.ReplaceAttr = log.ReplaceLevelAttr
|
|
|
|
// Open the underlying log file if the user configured an actual file to log to.
|
|
err := logFile.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var handler slog.Handler
|
|
switch logOutput {
|
|
case flags.OutputJSON:
|
|
handler = opts.NewJSONHandler(logFile.Writer())
|
|
case flags.OutputText:
|
|
handler = opts.NewTextHandler(logFile.Writer())
|
|
default:
|
|
return nil, fmt.Errorf("invalid log output: %s", logOutput)
|
|
}
|
|
|
|
slog.SetDefault(slog.New(handler))
|
|
return log.NewContext(ctx, slog.Default()), nil
|
|
}
|
|
|
|
var logFile = flags.NewLogFileFlag()
|
|
var logLevel = flags.NewLogLevelFlag()
|
|
var logOutput = flags.OutputText
|
|
|
|
func init() {
|
|
RootCmd.PersistentFlags().Var(&logFile, "log-file", "file to write logs to")
|
|
RootCmd.PersistentFlags().Var(&logLevel, "log-level", "log level")
|
|
RootCmd.PersistentFlags().Var(&logOutput, "log-format", "log output format (text or json)")
|
|
RootCmd.RegisterFlagCompletionFunc("log-file", logFile.Complete)
|
|
RootCmd.RegisterFlagCompletionFunc("log-level", logLevel.Complete)
|
|
RootCmd.RegisterFlagCompletionFunc("log-format", logOutput.Complete)
|
|
}
|