oiupoyt/discordo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: add log-level flag

Browse source
This commit is contained in:
ayn2op 2025-04-30 14:27:29 -04:00
parent 284853dc4a
commit 88f8604307
3 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
cmd/root.go
View file

@ -19,11 +19,23 @@ var (
var (
rootCmd = &cobra.Command{
RunE: func(cmd *cobra.Command, args []string) error {
if err := logger.Load(); err != nil {
return err
PreRunE: func(cmd *cobra.Command, args []string) error {
var level slog.Level
switch str, _ := cmd.Flags().GetString("log-level"); str {
case "debug":
level = slog.LevelDebug
case "info":
level = slog.LevelInfo
case "warn":
level = slog.LevelWarn
case "error":
level = slog.LevelError
}
return logger.Load(level)
},
RunE: func(cmd *cobra.Command, args []string) error {
token, _ := cmd.Flags().GetString("token")
if token == "" {
var err error
@ -63,5 +75,7 @@ var (
)
func init() {
rootCmd.Flags().StringP("token", "t", "", "the authentication token")
flags := rootCmd.Flags()
flags.StringP("token", "t", "", "the authentication token")
flags.StringP("log-level", "l", "info", "log level")
}

2
cmd/state.go
View file

@ -49,7 +49,7 @@ func openState(token string) error {
func (s *State) onRequest(r httpdriver.Request) error {
req, ok := r.(*httpdriver.DefaultRequest)
if ok {
slog.Info("new HTTP request", "method", req.Method, "path", req.URL.Path)
slog.Debug("new HTTP request", "method", req.Method, "url", req.URL, "headers", req.Header)
}
return nil

4
internal/logger/logger.go
View file

@ -11,7 +11,7 @@ import (
const fileName = "logs.txt"
// Opens the log file and configures default logger.
func Load() error {
func Load(level slog.Level) error {
path, err := os.UserCacheDir()
if err != nil {
return err
@ -28,7 +28,7 @@ func Load() error {
return err
}
l := slog.New(slog.NewTextHandler(file, &slog.HandlerOptions{AddSource: true}))
l := slog.New(slog.NewTextHandler(file, &slog.HandlerOptions{AddSource: true, Level: level}))
slog.SetDefault(l)
return nil
}