package logger

import (
	"runtime/debug"
	"sync"

	zl "gitee.com/golden-go/golden-go/pkg/utils/zap_logger"
	"go.uber.org/zap"
)

var logger *zap.Logger
var Closer zl.Closer
var mu = new(sync.Mutex)

func init() {
	mu.Lock()
	defer mu.Unlock()
	logger, _ = zap.NewDevelopment()
	Closer = func() error {
		return logger.Sync()
	}
}

func LoggerInit(env, dir, service, when string) {
	mu.Lock()
	defer mu.Unlock()
	logger.Sync()
	var err error
	if env == "dev" || env == "local" {
		logger, Closer, err = zl.GetDevLogger(dir, service, when)
	} else {
		logger, Closer, err = zl.GetProdLogger(dir, service, when)
	}
	if err != nil {
		l, _ := zap.NewDevelopment()
		l.Sugar().Fatal("logger init fail!!!", zap.Error(err))
	}
	logger.Debug("logger init ok", zap.String("dir", dir))
}

func JsonLoggerInit(env string) {
	mu.Lock()
	defer mu.Unlock()
	logger.Sync()
	var err error
	if env == "dev" || env == "local" {
		logger, Closer, err = zl.GetDevJsonLogger()
	} else {
		logger, Closer, err = zl.GetDevJsonLogger()
	}
	if err != nil {
		l, _ := zap.NewDevelopment()
		l.Sugar().Fatal("logger init fail!!!", zap.Error(err))
	}
	logger.Debug("logger init ok")
}

func GetLogger() *zap.Logger {
	return logger
}

func SetLogger(logger2 *zap.Logger) {
	mu.Lock()
	defer mu.Unlock()
	logger.Sync()
	logger = logger2
	Closer = func() error {
		return logger.Sync()
	}
}

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

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

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

func Error(msg string, fields ...zap.Field) {
	logger.Error(msg+"\n---------------------------->stack:\n"+string(debug.Stack())+"\n<----------------------------stack", fields...)
}