encoder.go

// Content managed by Project Forge, see [projectforge.md] for details.
package log

import (
	"fmt"
	"strings"

	"github.com/pkg/errors"
	"github.com/samber/lo"
	"go.uber.org/zap/buffer"
	"go.uber.org/zap/zapcore"

	"admini.dev/admini/app/util"
)

const timeFormat = "15:04:05.000000"

type customEncoder struct {
	zapcore.Encoder
	pool buffer.Pool
}

func newEncoder(cfg zapcore.EncoderConfig) *customEncoder {
	return &customEncoder{Encoder: zapcore.NewJSONEncoder(cfg), pool: buffer.NewPool()}
}

func (e *customEncoder) Clone() zapcore.Encoder {
	return &customEncoder{Encoder: e.Encoder.Clone(), pool: e.pool}
}

func (e *customEncoder) EncodeEntry(entry zapcore.Entry, fields []zapcore.Field) (*buffer.Buffer, error) {
	b, err := e.Encoder.EncodeEntry(entry, fields)
	if err != nil {
		return nil, errors.Wrap(err, "logging error")
	}
	out := b.Bytes()
	b.Free()

	data := util.ValueMap{}
	err = util.FromJSON(out, &data)
	if err != nil {
		return nil, errors.Wrap(err, "can't parse logging JSON")
	}

	ret := e.pool.Get()
	ret.AppendByte('\n')
	addLine := func(l string) {
		ret.AppendString(l)
		ret.AppendByte('\n')
	}

	lvl := fmt.Sprintf("%-5v", entry.Level.CapitalString())
	lvl = levelToColor[entry.Level.String()].Add(lvl)
	tm := entry.Time.Format(timeFormat)

	msg := entry.Message
	var msgLines []string
	if strings.Contains(msg, "\n") {
		msgLines = util.StringSplitLines(msg)
		msg = msgLines[0]
		msgLines = msgLines[1:]
	}

	addLine(fmt.Sprintf("[%s] %s %s", lvl, tm, Cyan.Add(msg)))
	lo.ForEach(msgLines, func(ml string, _ int) {
		if strings.Contains(ml, util.AppKey) {
			ml = Green.Add(ml)
		}
		addLine("  " + Cyan.Add(ml))
	})
	if len(data) > 0 {
		addLine("  " + util.ToJSONCompact(data))
	}
	caller := entry.Caller.String()
	if entry.Caller.Function != "" {
		caller += " (" + entry.Caller.Function + ")"
	}
	// idx := strings.Index(caller, "github.com/")
	// if idx > 0 {
	// 	caller = caller[idx:]
	// }
	addLine("  " + caller)

	if entry.Stack != "" {
		st := util.StringSplitLines(entry.Stack)
		lo.ForEach(st, func(stl string, _ int) {
			if strings.Contains(stl, util.AppKey) {
				stl = Green.Add(stl)
			}
			addLine("  " + stl)
		})
	}
	return ret, nil
}