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