package logger

import (
	"os"
	"strings"
	"time"

	"github.com/natefinch/lumberjack"
	"go.uber.org/zap"
	"go.uber.org/zap/zapcore"
)

type BLog struct {
	logger *zap.SugaredLogger
}

var (
	blog        *BLog
	atomicLevel = zap.NewAtomicLevelAt(zapcore.DebugLevel)
)

func (l *BLog) Write(buf []byte) (n int, err error) {
	l.logger.Debug(strings.Replace(string(buf), "\n", "", -1))
	return len(buf), nil
}

func GetLogFile() *BLog {
	return blog
}

const (
	DebugLevel = iota - 1
	InfoLevel
	WarnLevel
	ErrorLevel
	DPanicLevel
	PanicLevel
	FatalLevel
)

func InitLogger(logPath string, logLevel int, maxSize int, maxBackups int, maxAge int) *zap.SugaredLogger {
	logdir := "./logger/"
	logName := ""
	if index := strings.LastIndex(logPath, "/"); index != -1 {
		logdir = logPath[0:index] + "/"
		if index < len(logPath)-1 {
			logName = logPath[index+1:]
		}
	} else {
		logName = logPath
	}
	if logName == "" {
		logName = "log.log"
	}
	fi, err := os.Stat(logdir)
	if !((err == nil || os.IsExist(err)) && fi.IsDir()) {
		os.MkdirAll(logdir, os.ModePerm)
	}
	if logLevel < DebugLevel || logLevel > FatalLevel {
		logLevel = DebugLevel
	}
	if maxSize <= 0 {
		maxSize = 128
	}
	if maxBackups <= 0 {
		maxBackups = 30
	}
	if maxAge <= 0 {
		maxAge = 15
	}

	hook := lumberjack.Logger{
		Filename:   logdir + logName, //日志文件的位置
		MaxSize:    maxSize,          //在进行切割之前，日志文件的最大大小（以MB为单位）
		MaxBackups: maxBackups,       //保留旧文件的最大个数
		MaxAge:     maxAge,           //保留旧文件的最大天数
		Compress:   true,             //是否压缩
		LocalTime:  true,
	}
	w := zapcore.AddSync(&hook)

	atomicLevel = zap.NewAtomicLevelAt(getZapLevel(logLevel))

	encoderConfig := zap.NewProductionEncoderConfig()
	encoderConfig.EncodeLevel = zapcore.CapitalColorLevelEncoder
	encoderConfig.CallerKey = "file"
	encoderConfig.EncodeCaller = zapcore.ShortCallerEncoder
	encoderConfig.EncodeTime = func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
		enc.AppendString(t.Format("2006-01-02 15:04:05"))
	}

	core := zapcore.NewCore(
		zapcore.NewConsoleEncoder(encoderConfig),
		w,
		atomicLevel,
	)

	log := zap.New(core, zap.AddCaller(), zap.AddCallerSkip(1))

	blog = &BLog{
		logger: log.Sugar(),
	}
	blog.logger.Info("init logger success")
	return blog.logger
}

func getZapLevel(logLevel int) zapcore.Level {
	var level zapcore.Level
	switch logLevel {
	case DebugLevel:
		level = zap.DebugLevel
	case InfoLevel:
		level = zap.InfoLevel
	case WarnLevel:
		level = zap.WarnLevel
	case ErrorLevel:
		level = zap.ErrorLevel
	case DPanicLevel:
		level = zap.DPanicLevel
	case PanicLevel:
		level = zap.PanicLevel
	case FatalLevel:
		level = zap.FatalLevel
	default:
		level = zap.InfoLevel
	}
	return level
}

func SetLevel(lv int) error {
	atomicLevel.SetLevel(getZapLevel(lv))
	return nil
}

func Debug(v ...interface{}) {
	blog.logger.Debug(v...)
}

func Debugf(template string, v ...interface{}) {
	blog.logger.Debugf(template, v...)
}

func Info(v ...interface{}) {
	blog.logger.Info(v...)
}

func Infof(template string, v ...interface{}) {
	blog.logger.Infof(template, v...)
}

func Warn(v ...interface{}) {
	blog.logger.Warn(v...)
}

func Warnf(template string, v ...interface{}) {
	blog.logger.Warnf(template, v...)
}

func Error(v ...interface{}) {
	blog.logger.Error(v...)
}

func Errorf(template string, v ...interface{}) {
	blog.logger.Errorf(template, v...)
}

func Fatal(v ...interface{}) {
	blog.logger.Fatal(v...)
}

func Fatalf(template string, v ...interface{}) {
	blog.logger.Fatalf(template, v...)
}