chatgpt-dingtalk/pkg/logger/logger.go

package logger

import (
	"os"
	"sync"

	"github.com/charmbracelet/log"
)

var Logger *log.Logger
var once sync.Once

func InitLogger(level string) {
	once.Do(func() {
		Logger = log.NewWithOptions(os.Stderr, log.Options{ReportTimestamp: true})
	})
	if level == "debug" {
		Logger.SetLevel(log.DebugLevel)
	} else {
		Logger.SetLevel(log.InfoLevel)
	}
}

func Info(args ...interface{}) {
	Logger.Info(args)
}

func Warning(args ...interface{}) {
	Logger.Warn(args)
}

func Debug(args ...interface{}) {
	Logger.Debug(args)
}

func Error(args ...interface{}) {
	Logger.Error(args)
}

func Fatal(args ...interface{}) {
	Logger.Fatal(args)
}