Redmage/pkg/log/log.go

package log

import (
	"context"
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"os"
	"strings"
	"time"

	"github.com/mattn/go-colorable"
	"github.com/mattn/go-isatty"
	"github.com/tigorlazuardi/redmage/config"
	"github.com/tigorlazuardi/redmage/pkg/caller"
	"go.opentelemetry.io/otel/trace"

	slogmulti "github.com/samber/slog-multi"
)

var handler slog.Handler = NullHandler{}

func NewHandler(cfg *config.Config) slog.Handler {
	var handlers []slog.Handler

	if cfg.Bool("log.enable") {
		handlers = append(handlers, createStandardLogger(cfg))
	}

	if cfg.Bool("telemetry.openobserve.enable") && cfg.Bool("telemetry.openobserve.log.enable") {
		handlers = append(handlers, createO2Logger(cfg))
	}

	if len(handlers) == 0 {
		return NullHandler{}
	}

	return slogmulti.Fanout(handlers...)
}

func createStandardLogger(cfg *config.Config) slog.Handler {
	var output io.Writer

	if strings.ToLower(cfg.String("log.output")) == "stdout" {
		output = colorable.NewColorableStdout()
	} else {
		output = colorable.NewColorableStderr()
	}

	var lvl slog.Level
	_ = lvl.UnmarshalText(cfg.Bytes("log.level"))
	opts := &slog.HandlerOptions{
		AddSource: cfg.Bool("log.source"),
		Level:     lvl,
	}

	format := strings.ToLower(cfg.String("log.format"))
	if isatty.IsTerminal(os.Stdout.Fd()) && format == "pretty" {
		return NewPrettyHandler(output, opts)
	} else {
		return slog.NewJSONHandler(output, opts)
	}
}

func createO2Logger(cfg *config.Config) slog.Handler {
	var lvl slog.Level
	_ = lvl.UnmarshalText(cfg.Bytes("telemetry.openobserve.log.level"))
	opts := &slog.HandlerOptions{
		AddSource: cfg.Bool("telemetry.openobserve.log.source"),
		Level:     lvl,
	}
	return NewOpenObserveHandler(OpenObserveHandlerOptions{
		HandlerOptions: opts,
		BufferSize:     cfg.Int("telemetry.openobserve.log.buffer.size"),
		BufferTimeout:  cfg.Duration("telemetry.openobserve.log.buffer.timeout"),
		Concurrency:    cfg.Int("telemetry.openobserve.log.concurrency"),
		Endpoint:       cfg.String("telemetry.openobserve.log.endpoint"),
		HTTPClient:     http.DefaultClient,
		Username:       cfg.String("telemetry.openobserve.log.username"),
		Password:       cfg.String("telemetry.openobserve.log.password"),
	})
}

type Entry struct {
	ctx       context.Context
	handler   slog.Handler
	caller    caller.Caller
	time      time.Time
	err       error
	level     slog.Level
	withAttrs []slog.Attr
}

// New prepares a new entry to write logs.
func New(ctx context.Context) *Entry {
	h := FromContext(ctx)
	if h == nil {
		h = handler
	}
	return &Entry{ctx: ctx, handler: h, time: time.Now()}
}

func (entry *Entry) Caller(caller caller.Caller) *Entry {
	entry.caller = caller
	return entry
}

func (entry *Entry) Err(err error) *Entry {
	entry.err = err
	return entry
}

func (entry *Entry) Level(lvl slog.Level) *Entry {
	entry.level = lvl
	return entry
}

// With adds fields to top level of the log entry.
func (entry *Entry) With(fields ...any) *Entry {
	entry.withAttrs = append(entry.withAttrs, argsToAttrSlice(fields)...)
	return entry
}

const badKey = "!BADKEY"

func argsToAttrSlice(args []any) []slog.Attr {
	var (
		attr  slog.Attr
		attrs []slog.Attr
	)
	for len(args) > 0 {
		attr, args = argsToAttr(args)
		attrs = append(attrs, attr)
	}
	return attrs
}

func argsToAttr(args []any) (slog.Attr, []any) {
	switch x := args[0].(type) {
	case string:
		if len(args) == 1 {
			return slog.String(badKey, x), nil
		}
		return slog.Any(x, args[1]), args[2:]

	case slog.Attr:
		return x, args[1:]

	default:
		return slog.Any(badKey, x), args[1:]
	}
}

func (entry *Entry) Log(message string, fields ...any) {
	if !entry.handler.Enabled(entry.ctx, entry.level) {
		return
	}
	record := slog.NewRecord(entry.time, entry.level, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	record.AddAttrs(slog.Group("details", fields...))
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Info(message string, fields ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelInfo) {
		return
	}
	record := slog.NewRecord(entry.time, slog.LevelInfo, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	record.AddAttrs(slog.Group("details", fields...))
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Infof(format string, args ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelInfo) {
		return
	}
	message := fmt.Sprintf(format, args...)
	record := slog.NewRecord(entry.time, slog.LevelInfo, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Error(message string, fields ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelError) {
		return
	}
	record := slog.NewRecord(entry.time, slog.LevelError, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	record.AddAttrs(slog.Group("details", fields...))
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Errorf(format string, args ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelError) {
		return
	}
	message := fmt.Sprintf(format, args...)
	record := slog.NewRecord(entry.time, slog.LevelError, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	if entry.err != nil {
		record.AddAttrs(slog.Any("details", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Debug(message string, fields ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelDebug) {
		return
	}
	record := slog.NewRecord(entry.time, slog.LevelDebug, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	record.AddAttrs(slog.Group("details", fields...))
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Debugf(format string, args ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelDebug) {
		return
	}
	message := fmt.Sprintf(format, args...)
	record := slog.NewRecord(entry.time, slog.LevelDebug, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Warn(message string, fields ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelWarn) {
		return
	}
	record := slog.NewRecord(entry.time, slog.LevelWarn, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	record.AddAttrs(slog.Group("details", fields...))
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) Warnf(format string, args ...any) {
	if !entry.handler.Enabled(entry.ctx, slog.LevelWarn) {
		return
	}
	message := fmt.Sprintf(format, args...)
	record := slog.NewRecord(entry.time, slog.LevelWarn, message, entry.getCaller().PC)
	record.AddAttrs(entry.getExtra()...)
	if entry.err != nil {
		record.AddAttrs(slog.Any("error", entry.err))
	}
	_ = entry.handler.Handle(entry.ctx, record)
}

func (entry *Entry) getCaller() caller.Caller {
	if entry.caller.PC != 0 {
		return entry.caller
	}
	return caller.New(4)
}

func (entry *Entry) getExtra() []slog.Attr {
	out := make([]slog.Attr, 0, 4)
	if span := trace.SpanFromContext(entry.ctx); span.IsRecording() {
		out = append(out,
			slog.String("trace.id", span.SpanContext().TraceID().String()),
			slog.String("span.id", span.SpanContext().SpanID().String()),
		)
	}

	out = append(out, entry.withAttrs...)

	return out
}

func SetDefault(h slog.Handler) {
	handler = h
}