package logging

import (
	"fmt"
	"gitee.com/kristas/booting-go/framework/common/util/lang"
	"gitee.com/kristas/booting-go/framework/core/config_loader"
	"github.com/sirupsen/logrus"
	"strconv"
	"strings"
)

type logrusAdapter struct {
}

func (r *logrusAdapter) CreateLog() *logrus.Logger {
	logger := logrus.New()
	logger.SetReportCaller(true)
	loggingLevel := config_loader.ParserString("logging.level").(string)
	level, err := logrus.ParseLevel(loggingLevel)
	if err != nil {
		level, _ = logrus.ParseLevel("INFO")
	}
	logger.SetLevel(level)
	logger.SetReportCaller(true)
	logger.SetFormatter(new(FrameFormatter))
	return logger
}

type FrameFormatter struct {
}

func (f *FrameFormatter) Format(entry *logrus.Entry) ([]byte, error) {
	data := entry.Data
	s := fmt.Sprintf("[%s][%-5s][%s][%s] - %s ",
		entry.Time.Format("2006-01-02 15:04:05.000"),
		strings.ToLower(entry.Level.String()),
		packageParser(entry.Caller.Function),
		entry.Caller.File[strings.LastIndex(entry.Caller.File, "/")+1:]+":"+strconv.Itoa(entry.Caller.Line),
		entry.Message)
	var skipField = []string{"file", "line", "packageName", "funcName"}
	for k, v := range data {
		if !contains(skipField, k) {
			s += fmt.Sprintf("[%s = %s]", k, v)
		}
	}
	return append([]byte(s), '\n'), nil
}

func packageParser(pkg string) string {
	split := lang.NewString(pkg).ReplaceAll("/", ".").Split(".")
	pkgArr := lang.TransferSliceToSimple(split)
	var tmp []string
	for i, s := range pkgArr {
		if i < 4 {
			tmp = append(tmp, s[:1])
		} else {
			tmp = append(tmp, s)
		}
	}
	return strings.Join(tmp, ".")
}

func contains(arr []string, str string) bool {
	for i := range arr {
		if arr[i] == str {
			return true
		}
	}
	return false
}