98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
package logger
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/rs/zerolog"
|
|
"github.com/sirrobot01/decypharr/internal/config"
|
|
"gopkg.in/natefinch/lumberjack.v2"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
once sync.Once
|
|
logger zerolog.Logger
|
|
)
|
|
|
|
func GetLogPath() string {
|
|
cfg := config.Get()
|
|
logsDir := filepath.Join(cfg.Path, "logs")
|
|
|
|
if _, err := os.Stat(logsDir); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(logsDir, 0755); err != nil {
|
|
panic(fmt.Sprintf("Failed to create logs directory: %v", err))
|
|
}
|
|
}
|
|
|
|
return filepath.Join(logsDir, "decypharr.log")
|
|
}
|
|
|
|
func New(prefix string) zerolog.Logger {
|
|
|
|
level := config.Get().LogLevel
|
|
|
|
rotatingLogFile := &lumberjack.Logger{
|
|
Filename: GetLogPath(),
|
|
MaxSize: 10,
|
|
MaxAge: 15,
|
|
Compress: true,
|
|
}
|
|
|
|
consoleWriter := zerolog.ConsoleWriter{
|
|
Out: os.Stdout,
|
|
TimeFormat: "2006-01-02 15:04:05",
|
|
NoColor: false, // Set to true if you don't want colors
|
|
FormatLevel: func(i interface{}) string {
|
|
return strings.ToUpper(fmt.Sprintf("| %-6s|", i))
|
|
},
|
|
FormatMessage: func(i interface{}) string {
|
|
return fmt.Sprintf("[%s] %v", prefix, i)
|
|
},
|
|
}
|
|
|
|
fileWriter := zerolog.ConsoleWriter{
|
|
Out: rotatingLogFile,
|
|
TimeFormat: "2006-01-02 15:04:05",
|
|
NoColor: true, // No colors in file output
|
|
FormatLevel: func(i interface{}) string {
|
|
return strings.ToUpper(fmt.Sprintf("| %-6s|", i))
|
|
},
|
|
FormatMessage: func(i interface{}) string {
|
|
return fmt.Sprintf("[%s] %v", prefix, i)
|
|
},
|
|
}
|
|
|
|
multi := zerolog.MultiLevelWriter(consoleWriter, fileWriter)
|
|
|
|
logger := zerolog.New(multi).
|
|
With().
|
|
Timestamp().
|
|
Logger().
|
|
Level(zerolog.InfoLevel)
|
|
|
|
// Set the log level
|
|
level = strings.ToLower(level)
|
|
switch level {
|
|
case "debug":
|
|
logger = logger.Level(zerolog.DebugLevel)
|
|
case "info":
|
|
logger = logger.Level(zerolog.InfoLevel)
|
|
case "warn":
|
|
logger = logger.Level(zerolog.WarnLevel)
|
|
case "error":
|
|
logger = logger.Level(zerolog.ErrorLevel)
|
|
case "trace":
|
|
logger = logger.Level(zerolog.TraceLevel)
|
|
}
|
|
return logger
|
|
}
|
|
|
|
func GetDefaultLogger() zerolog.Logger {
|
|
once.Do(func() {
|
|
logger = New("decypharr")
|
|
})
|
|
return logger
|
|
}
|