logging.go RawHistoryFolder

1package logging
2
3import (
4	"io"
5	"log/slog"
6	"os"
7	"path/filepath"
8	"runtime/debug"
9	"strings"
10	"sync"
11
12	"github.com/bnema/vev/pkg/safedir"
13)
14
15type Component string
16
17const (
18	Daemon Component = "daemon"
19	Client Component = "client"
20	Stdio  Component = "stdio"
21)
22
23const DefaultMaxBytes = 20 << 20
24
25type Config struct {
26	Dir             string
27	Component       Component
28	Level           slog.Level
29	MaxBytes        int64
30	RotateAtRuntime bool
31}
32
33func Setup(cfg Config) (*slog.Logger, io.Closer, error) {
34	if err := safedir.EnsurePrivate(cfg.Dir); err != nil {
35		return nil, nil, err
36	}
37
38	component := cfg.Component
39	logPath := filepath.Join(cfg.Dir, "vev-"+string(component)+".log")
40	writer, err := newRotatingWriter(logPath, cfg.MaxBytes, cfg.RotateAtRuntime)
41	if err != nil {
42		return nil, nil, err
43	}
44
45	crashPath := filepath.Join(cfg.Dir, "vev-"+string(component)+"-crash.log")
46	crashFile, err := os.OpenFile(crashPath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o600)
47	if err != nil {
48		_ = writer.Close()
49		return nil, nil, err
50	}
51	if err := debug.SetCrashOutput(crashFile, debug.CrashOptions{}); err != nil {
52		_ = crashFile.Close()
53		_ = writer.Close()
54		return nil, nil, err
55	}
56
57	logger := slog.New(slog.NewJSONHandler(writer, &slog.HandlerOptions{Level: cfg.Level})).With("component", string(component))
58	slog.SetDefault(logger)
59
60	return logger, &setupCloser{log: writer, crash: crashFile}, nil
61}
62
63func ParseLevel(s string) slog.Level {
64	switch strings.ToLower(strings.TrimSpace(s)) {
65	case "debug":
66		return slog.LevelDebug
67	case "warn", "warning":
68		return slog.LevelWarn
69	case "error":
70		return slog.LevelError
71	default:
72		return slog.LevelInfo
73	}
74}
75
76func EnvLevel() slog.Level {
77	return ParseLevel(os.Getenv("VEV_LOG"))
78}
79
80type setupCloser struct {
81	once  sync.Once
82	log   io.Closer
83	crash io.Closer
84	err   error
85}
86
87func (c *setupCloser) Close() error {
88	c.once.Do(func() {
89		if c.log != nil {
90			c.err = c.log.Close()
91		}
92		if c.crash != nil {
93			if err := c.crash.Close(); c.err == nil {
94				c.err = err
95			}
96		}
97	})
98	return c.err
99}