package logger

import (
	"fmt"
	"os"
	"pandax/base/config"
	"strings"
	"time"

	"github.com/sirupsen/logrus"
)

func InitLog(logConf *config.Log) *logrus.Logger {
	var Log = logrus.New()
	Log.SetFormatter(new(LogFormatter))
	Log.SetReportCaller(true)

	// 如果不存在日志配置信息，则默认debug级别
	if logConf == nil {
		Log.SetLevel(logrus.DebugLevel)
		return nil
	}

	// 根据配置文件设置日志级别
	if level := logConf.Level; level != "" {
		l, err := logrus.ParseLevel(level)
		if err != nil {
			panic(any(fmt.Sprintf("日志级别不存在: %s", level)))
		}
		Log.SetLevel(l)
	} else {
		Log.SetLevel(logrus.DebugLevel)
	}

	if logFile := logConf.File; logFile != nil {
		//写入文件
		file, err := os.OpenFile(logFile.GetFilename(), os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModeAppend|0666)
		if err != nil {
			panic(any(fmt.Sprintf("创建日志文件失败: %s", err.Error())))
		}

		Log.Out = file
	}
	return Log
}

type LogFormatter struct{}

func (l *LogFormatter) Format(entry *logrus.Entry) ([]byte, error) {
	timestamp := time.Now().Local().Format("2006-01-02 15:04:05.000")
	level := entry.Level
	logMsg := fmt.Sprintf("%s [%s]", timestamp, strings.ToUpper(level.String()))
	// 如果存在调用信息，且为error级别以上记录文件及行号
	if caller := entry.Caller; caller != nil {
		var fp string
		// 全路径切割，只获取项目相关路径，
		// 即/Users/hml/Desktop/project/go/pandax/test.go只获取/test.go
		ps := strings.Split(caller.File, "pandax/")
		if len(ps) >= 2 {
			fp = ps[1]
		} else {
			fp = ps[0]
		}
		logMsg = logMsg + fmt.Sprintf(" [%s:%d]", fp, caller.Line)
	}
	for k, v := range entry.Data {
		logMsg = logMsg + fmt.Sprintf(" [%s=%v]", k, v)
	}
	logMsg = logMsg + fmt.Sprintf(" : %s\n", entry.Message)
	return []byte(logMsg), nil
}