SimpleCloudNotifier/scnserver/init.go

package server

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"os"
	"path"
	"path/filepath"
	"runtime"
)

var callerRoot = ""

func init() {
	_, file, _, ok := runtime.Caller(0)
	if !ok {
		return
	}

	callerRoot = path.Dir(file)
}

func Init(cfg Config) {
	cw := zerolog.ConsoleWriter{
		Out:          os.Stdout,
		TimeFormat:   "2006-01-02 15:04:05 Z07:00",
		FormatCaller: formatCaller,
	}

	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
	multi := zerolog.MultiLevelWriter(cw)
	logger := zerolog.New(multi).With().
		Timestamp().
		Caller().
		Logger()

	log.Logger = logger

	if cfg.GinDebug {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}

	zerolog.SetGlobalLevel(cfg.LogLevel)

	log.Debug().Msg("Initialized")
}

func formatCaller(i any) string {
	const (
		colorBlack = iota + 30
		colorRed
		colorGreen
		colorYellow
		colorBlue
		colorMagenta
		colorCyan
		colorWhite

		colorBold     = 1
		colorDarkGray = 90
	)

	var c string
	if cc, ok := i.(string); ok {
		c = cc
	}
	if len(c) > 0 {
		if rel, err := filepath.Rel(callerRoot, c); err == nil {
			c = rel
		}
		c = colorize(c, colorBold, false) + colorize(" >", colorCyan, false)
	}
	return c
}

func colorize(s interface{}, c int, disabled bool) string {
	if disabled {
		return fmt.Sprintf("%s", s)
	}
	return fmt.Sprintf("\x1b[%dm%v\x1b[0m", c, s)
}