mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
2ea2293745
* chore: add package * feat: get size by api for ptp and btn * feat: download and parse torrent if not api * feat: bypass tls check and load meta from file * fix: no invite command needed for btn * feat: add ggn api * feat: imrpove logging * feat: build request url * feat: improve err logging
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package logger
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
|
|
"github.com/r3labs/sse/v2"
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/rs/zerolog/pkgerrors"
|
|
"gopkg.in/natefinch/lumberjack.v2"
|
|
)
|
|
|
|
func Setup(cfg domain.Config, sse *sse.Server) {
|
|
zerolog.TimeFieldFormat = time.RFC3339
|
|
zerolog.ErrorStackMarshaler = pkgerrors.MarshalStack
|
|
|
|
switch cfg.LogLevel {
|
|
case "INFO":
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
case "DEBUG":
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
case "ERROR":
|
|
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
|
|
case "WARN":
|
|
zerolog.SetGlobalLevel(zerolog.WarnLevel)
|
|
case "TRACE":
|
|
zerolog.SetGlobalLevel(zerolog.TraceLevel)
|
|
default:
|
|
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
|
|
}
|
|
|
|
// setup console writer
|
|
consoleWriter := zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339}
|
|
|
|
writers := io.MultiWriter(consoleWriter)
|
|
|
|
// if logPath set create file writer
|
|
if cfg.LogPath != "" {
|
|
fileWriter := &lumberjack.Logger{
|
|
Filename: cfg.LogPath,
|
|
MaxSize: 100, // megabytes
|
|
MaxBackups: 3,
|
|
}
|
|
|
|
// overwrite writers
|
|
writers = io.MultiWriter(consoleWriter, fileWriter)
|
|
}
|
|
|
|
log.Logger = log.Hook(&ServerSentEventHook{sse: sse})
|
|
log.Logger = log.Output(writers)
|
|
}
|