cherry-pick caller logprint fix from psycho-backend

scnserver/init.go

@@ -1,16 +1,32 @@
 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",
+		Out:          os.Stdout,
+		TimeFormat:   "2006-01-02 15:04:05 Z07:00",
+		FormatCaller: formatCaller,
 	}
 
 	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
@@ -32,3 +48,38 @@ func Init(cfg Config) {
 
 	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)
+}