gitea-mcp/pkg/log/log.go

package log

import (
	"fmt"
	"os"
	"sync"
	"time"

	"gitea.com/gitea/gitea-mcp/pkg/flag"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
	"gopkg.in/natefinch/lumberjack.v2"
)

var (
	defaultLoggerOnce sync.Once
	defaultLogger     *zap.Logger
)

func Default() *zap.Logger {
	defaultLoggerOnce.Do(func() {
		if defaultLogger == nil {
			ec := zap.NewProductionEncoderConfig()
			ec.EncodeTime = zapcore.TimeEncoderOfLayout(time.DateTime)
			ec.EncodeLevel = zapcore.CapitalLevelEncoder

			var ws zapcore.WriteSyncer
			var wss []zapcore.WriteSyncer

			home, _ := os.UserHomeDir()
			if home == "" {
				home = os.TempDir()
			}

			logDir := fmt.Sprintf("%s/.gitea-mcp", home)
			if err := os.MkdirAll(logDir, 0o700); err != nil {
				// Fallback to temp directory if creation fails
				logDir = os.TempDir()
			}

			wss = append(wss, zapcore.AddSync(&lumberjack.Logger{
				Filename:   fmt.Sprintf("%s/gitea-mcp.log", logDir),
				MaxSize:    100,
				MaxBackups: 10,
				MaxAge:     30,
			}))

			if flag.Mode == "http" || flag.Mode == "sse" {
				wss = append(wss, zapcore.AddSync(os.Stdout))
			}

			ws = zapcore.NewMultiWriteSyncer(wss...)

			enc := zapcore.NewConsoleEncoder(ec)
			var level zapcore.Level
			if flag.Debug {
				level = zapcore.DebugLevel
			} else {
				level = zapcore.InfoLevel
			}
			core := zapcore.NewCore(enc, ws, level)
			options := []zap.Option{
				zap.AddStacktrace(zapcore.DPanicLevel),
				zap.AddCaller(),
				zap.AddCallerSkip(1),
			}
			defaultLogger = zap.New(core, options...)
		}
	})

	return defaultLogger
}

func SetDefault(logger *zap.Logger) {
	if logger != nil {
		defaultLogger = logger
	}
}

func Logger() *zap.Logger {
	return defaultLogger
}

func Debug(msg string, fields ...zap.Field) {
	Default().Debug(msg, fields...)
}

func Info(msg string, fields ...zap.Field) {
	Default().Info(msg, fields...)
}

func Warn(msg string, fields ...zap.Field) {
	Default().Warn(msg, fields...)
}

func Error(msg string, fields ...zap.Field) {
	Default().Error(msg, fields...)
}

func Panic(msg string, fields ...zap.Field) {
	Default().Panic(msg, fields...)
}

func Debugf(format string, args ...any) {
	Default().Sugar().Debugf(format, args...)
}

func Infof(format string, args ...any) {
	Default().Sugar().Infof(format, args...)
}

func Warnf(format string, args ...any) {
	Default().Sugar().Warnf(format, args...)
}

func Errorf(format string, args ...any) {
	Default().Sugar().Errorf(format, args...)
}

func Fatalf(format string, args ...any) {
	Default().Sugar().Fatalf(format, args...)
}