aboutsummaryrefslogtreecommitdiff
path: root/internal/slog.go
blob: f54e19e3e03b84411baeb3c157a5ad4f702fbcf2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package internal

import (
	"flag"
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"os"
)

var (
	slogLevel = flag.String("slog-level", "INFO", "log level")

	leveler *slog.LevelVar
)

func Slog() {
	var programLevel slog.Level
	if err := (&programLevel).UnmarshalText([]byte(*slogLevel)); err != nil {
		fmt.Fprintf(os.Stderr, "invalid log level %s: %v, using info\n", *slogLevel, err)
		programLevel = slog.LevelInfo
	}

	leveler = &slog.LevelVar{}
	leveler.Set(programLevel)

	h := slog.NewJSONHandler(os.Stderr, &slog.HandlerOptions{
		AddSource: true,
		Level:     leveler,
	})
	slog.SetDefault(slog.New(h))

	http.HandleFunc("/.within/debug/slog-level", func(w http.ResponseWriter, r *http.Request) {
		var level, old slog.Level
		old = leveler.Level()

		if r.Method == http.MethodPost {
			data, err := io.ReadAll(http.MaxBytesReader(w, r.Body, 64))
			defer r.Body.Close()
			if err != nil {
				http.Error(w, err.Error(), http.StatusBadRequest)
				return
			}

			if err := (&level).UnmarshalText(data); err != nil {
				http.Error(w, err.Error(), http.StatusBadRequest)
				return
			}

			leveler.Set(level)
			slog.Info("changed level", "from", old, "to", level)
			fmt.Fprintln(w, level)
		} else {
			fmt.Fprintln(w, old)
		}
	})
}