52655916c6
Our logger wasn't entirely compatible with Windows escape sequences when not in TTY mode. This fix changes the stderr to one that automatically transforms the escape sequences to some Windows compatible ones. Signed-off-by: guillaume <guillaume.derouville@gmail.com>
55 lines
1.0 KiB
Go
55 lines
1.0 KiB
Go
// Logger utilities for the CLI
|
|
//
|
|
// These utilities rely on command line flags to set up the logger, therefore
|
|
// they are tightly coupled with the CLI and should not be used outside.
|
|
|
|
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/mattn/go-colorable"
|
|
"github.com/rs/zerolog"
|
|
"github.com/spf13/viper"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
func New() zerolog.Logger {
|
|
logger := zerolog.
|
|
New(os.Stderr).
|
|
With().
|
|
Timestamp().
|
|
Logger()
|
|
|
|
if !jsonLogs() {
|
|
logger = logger.Output(&PlainOutput{Out: colorable.NewColorableStderr()})
|
|
} else {
|
|
logger = logger.With().Timestamp().Caller().Logger()
|
|
}
|
|
|
|
level := viper.GetString("log-level")
|
|
lvl, err := zerolog.ParseLevel(level)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return logger.Level(lvl)
|
|
}
|
|
|
|
func jsonLogs() bool {
|
|
switch f := viper.GetString("log-format"); f {
|
|
case "json":
|
|
return true
|
|
case "plain":
|
|
return false
|
|
case "tty":
|
|
return false
|
|
case "auto":
|
|
return !term.IsTerminal(int(os.Stdout.Fd()))
|
|
default:
|
|
fmt.Fprintf(os.Stderr, "invalid --log-format %q\n", f)
|
|
os.Exit(1)
|
|
}
|
|
return false
|
|
}
|