pye/internal/logging/log.go

package logging

import (
	"io"
	"log"
	"log/slog"
	"os"

	"git.a71.su/Andrew71/pye/internal/config"
	"github.com/go-chi/chi/middleware"
)

// Load makes slog output to both os.Stdout and a file if needed, and sets slog.LevelDebug if enabled.
func Load(debugMode bool) {
	var w io.Writer
	if config.Cfg.LogToFile {
		f, err := os.OpenFile(config.Cfg.LogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
		if err != nil {
			slog.Error("error opening log file, logging to stdout only", "path", config.Cfg.LogFile, "error", err)
			return
		}
		// No defer f.Close() because that breaks the MultiWriter
		w = io.MultiWriter(f, os.Stdout)
	} else {
		w = os.Stdout
	}

	// Make slog use intended format
	var opts *slog.HandlerOptions
	if debugMode {
		opts = &slog.HandlerOptions{Level: slog.LevelDebug}
	}
	slog.SetDefault(slog.New(slog.NewTextHandler(w, opts)))
	middleware.DefaultLogger = middleware.RequestLogger(&middleware.DefaultLogFormatter{Logger: log.Default(), NoColor: true})
	slog.Debug("debug mode active")
}