address comment

client/cmd.go

@@ -5,7 +5,6 @@ import (
 	"crypto/tls"
 	"errors"
 	"fmt"
-	"os"
 	"slices"
 	"strings"
 
@@ -401,7 +400,7 @@ func GetLoggerFromCmd(cmd *cobra.Command) log.Logger {
 	v := cmd.Context().Value(corectx.LoggerContextKey)
 	logger, ok := v.(log.Logger)
 	if !ok {
-		return log.NewLogger(os.Stdout)
+		return log.NewLogger(cmd.OutOrStdout())
 	}
 	return logger
 }

core/context/server_context.go

@@ -1,8 +1,6 @@
 package context
 
-type ServerContextKey string
-
-const (
-	LoggerContextKey ServerContextKey = "server.logger"
-	ViperContextKey  ServerContextKey = "server.viper"
+var (
+	LoggerContextKey struct{}
+	ViperContextKey  struct{}
 )

server/util.go

@@ -200,7 +200,7 @@ func GetServerContextFromCmd(cmd *cobra.Command) *LegacyContext {
 		logger := v.(log.Logger)
 		serverCtx.Logger = logger
 	} else {
-		serverCtx.Logger = log.NewLogger(os.Stdout)
+		serverCtx.Logger = log.NewLogger(cmd.OutOrStdout())
 	}
 
 	return serverCtx