Changelog v0.4.0

common/config.go

@@ -22,7 +22,7 @@ type DebridConfig struct {
 type ProxyConfig struct {
 	Port       string `json:"port"`
 	Enabled    bool   `json:"enabled"`
-	Debug      bool   `json:"debug"`
+	LogLevel   string `json:"log_level"`
 	Username   string `json:"username"`
 	Password   string `json:"password"`
 	CachedOnly *bool  `json:"cached_only"`
@@ -32,7 +32,7 @@ type QBitTorrentConfig struct {
 	Username        string   `json:"username"`
 	Password        string   `json:"password"`
 	Port            string   `json:"port"`
-	Debug           bool     `json:"debug"`
+	LogLevel        string   `json:"log_level"`
 	DownloadFolder  string   `json:"download_folder"`
 	Categories      []string `json:"categories"`
 	RefreshInterval int      `json:"refresh_interval"`
@@ -51,6 +51,7 @@ type RepairConfig struct {
 }
 
 type Config struct {
+	LogLevel     string            `json:"log_level"`
 	Debrid       DebridConfig      `json:"debrid"`
 	Debrids      []DebridConfig    `json:"debrids"`
 	Proxy        ProxyConfig       `json:"proxy"`

common/logger.go

@@ -2,13 +2,40 @@ package common
 
 import (
 	"fmt"
-	"log"
+	"github.com/rs/zerolog"
 	"os"
+	"strings"
 )
 
-func NewLogger(prefix string, output *os.File) *log.Logger {
-	f := fmt.Sprintf("[%s] ", prefix)
-	return log.New(output, f, log.LstdFlags)
-}
+func NewLogger(prefix string, level string, output *os.File) zerolog.Logger {
+	writer := zerolog.ConsoleWriter{
+		Out:        output,
+		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)
+		},
+	}
 
-var Logger = NewLogger("Main", os.Stdout)
+	logger := zerolog.New(writer).
+		With().
+		Timestamp().
+		Logger().
+		Level(zerolog.InfoLevel)
+
+	// Set the log 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)
+	}
+	return logger
+}