context.go RawHistoryFolder

1package logging
2
3import (
4	"context"
5
6	"github.com/rs/zerolog"
7)
8
9// FromContext extracts the logger from context
10// If no logger is found, returns a disabled logger (no-op)
11func FromContext(ctx context.Context) *zerolog.Logger {
12	return zerolog.Ctx(ctx)
13}
14
15// WithContext returns a new context with the logger attached
16func WithContext(ctx context.Context, logger zerolog.Logger) context.Context {
17	return logger.WithContext(ctx)
18}
19
20// With creates a child logger with additional fields and returns a new context
21func With(ctx context.Context, fields map[string]any) context.Context {
22	logger := FromContext(ctx)
23	childCtx := logger.With()
24
25	for k, v := range fields {
26		childCtx = childCtx.Interface(k, v)
27	}
28
29	childLogger := childCtx.Logger()
30	return WithContext(ctx, childLogger)
31}
32
33// WithComponent creates a child logger with a component field
34func WithComponent(ctx context.Context, component string) context.Context {
35	logger := FromContext(ctx)
36	childLogger := logger.With().Str("component", component).Logger()
37	return WithContext(ctx, childLogger)
38}
39
40// WithPaneID creates a child logger with a pane_id field
41func WithPaneID(ctx context.Context, paneID string) context.Context {
42	logger := FromContext(ctx)
43	childLogger := logger.With().Str("pane_id", paneID).Logger()
44	return WithContext(ctx, childLogger)
45}
46
47// WithTabID creates a child logger with a tab_id field
48func WithTabID(ctx context.Context, tabID string) context.Context {
49	logger := FromContext(ctx)
50	childLogger := logger.With().Str("tab_id", tabID).Logger()
51	return WithContext(ctx, childLogger)
52}
53
54// WithURL creates a child logger with a redacted url field.
55func WithURL(ctx context.Context, url string) context.Context {
56	logger := FromContext(ctx)
57	childLogger := logger.With().Str("url", RedactURL(url)).Logger()
58	return WithContext(ctx, childLogger)
59}