chore(trdl-vault): add flag to control log level

Signed-off-by: Evgeniy Frolov <evgeniy.frolov@flant.com>

Author: Fral738

release/cmd/trdl-vault/main.go

@@ -48,6 +48,7 @@ func main() {
 	common.SetupRetry(&commonCmdData, cmd)
 	common.SetupMaxAttemps(&commonCmdData, cmd)
 	common.SetupDelay(&commonCmdData, cmd)
+	common.SetupLogLevel(&commonCmdData, cmd)
 
 	cmd.AddCommand(publishCmd)
 	cmd.AddCommand(releaseCmd)
@@ -66,6 +67,7 @@ func publishCmd(c *common.CmdData, projectName string) error {
 		Retry:        *c.Retry,
 		MaxAttempts:  *c.MaxAttempts,
 		Delay:        *c.Delay,
+		LogLevel:     c.GetLogLevel(),
 	})
 	if err != nil {
 		return fmt.Errorf("unable to create client: %w", err)
@@ -83,6 +85,7 @@ func releaseCmd(c *common.CmdData, projectName, gitTag string) error {
 		Retry:        *c.Retry,
 		MaxAttempts:  *c.MaxAttempts,
 		Delay:        *c.Delay,
+		LogLevel:     c.GetLogLevel(),
 	})
 	if err != nil {
 		return fmt.Errorf("unable to create client: %w", err)

release/common/cmd_data.go

@@ -8,4 +8,5 @@ type CmdData struct {
 	Retry        *bool
 	MaxAttempts  *int
 	Delay        *time.Duration
+	LogLevel     *string
 }

release/common/common.go

@@ -1,11 +1,13 @@
 package common
 
 import (
+	"log/slog"
 	"net/url"
 	"os"
 	"time"
 
 	"github.com/spf13/cobra"
+	"github.com/werf/trdl/release/pkg/logger"
 )
 
 func SetupVaultAddress(cmdData *CmdData, cmd *cobra.Command) {
@@ -54,3 +56,11 @@ func SetupDelay(cmdData *CmdData, cmd *cobra.Command) {
 	cmd.PersistentFlags().DurationVarP(cmdData.Delay, "delay", "", 10*time.Second, "Set max delay between retries")
 
 }
+func SetupLogLevel(cmdData *CmdData, cmd *cobra.Command) {
+	cmdData.LogLevel = new(string)
+	cmd.PersistentFlags().StringVarP(cmdData.LogLevel, "log-level", "", "", "Set log level (debug, info, warn, error)")
+}
+
+func (c *CmdData) GetLogLevel() slog.Level {
+	return logger.ParseLogLevel(*c.LogLevel)
+}

release/pkg/client/main.go

@@ -36,10 +36,11 @@ type NewTrdlVaultClientOpts struct {
 	Retry        bool
 	MaxAttempts  int
 	Delay        time.Duration
+	LogLevel     slog.Level
 }
 
 func NewTrdlVaultClient(opts NewTrdlVaultClientOpts) (*Client, error) {
-	log := logger.NewLogger(slog.LevelInfo) // TODO: add logger level to opts
+	log := logger.NewLogger(opts.LogLevel)
 	trdlClient, err := vault.NewTrdlClient(vault.NewTrdlClientOpts{
 		VaultAddress: opts.VaultAddress,
 		VaultToken:   opts.VaultToken,

release/pkg/logger/logger.go

@@ -15,6 +15,19 @@ func NewLogger(level slog.Level) *Logger {
 	return &Logger{logger: slog.New(handler)}
 }
 
+func ParseLogLevel(level string) slog.Level {
+	switch level {
+	case "debug":
+		return slog.LevelDebug
+	case "warn":
+		return slog.LevelWarn
+	case "error":
+		return slog.LevelError
+	default:
+		return slog.LevelInfo
+	}
+}
+
 func (l *Logger) Debug(taskID, msg string) {
 	l.log(context.Background(), slog.LevelDebug, taskID, msg)
 }