package log

import (
	"io/ioutil"
	"time"

	rotatelogs "github.com/lestrrat-go/file-rotatelogs"
	"github.com/rifflock/lfshook"
	"github.com/sirupsen/logrus"
)

var Log *logrus.Logger

func init() {
	Log = NewLogger()
	Log.SetLevel(logrus.ErrorLevel)
}

func SetLogLevel(level string){
	if level == "ErrorLevel" {
		Log.SetLevel(logrus.ErrorLevel)
	} else if level == "InfoLevel" {
		Log.SetLevel(logrus.InfoLevel)
	} else {
		Log.SetLevel(logrus.InfoLevel)
	}
}

func NewLogger() *logrus.Logger {
	if Log != nil {
		return Log
	}

	infopath := "logs/info.log"
	infowriter, _ := rotatelogs.New(
		infopath+".%Y%m%d%H%M",
		rotatelogs.WithLinkName(infopath),
		rotatelogs.WithMaxAge(30*24*time.Hour),
		rotatelogs.WithRotationTime(24*time.Hour),
	)

	errorpath := "logs/error.log"
	errorwriter, _ := rotatelogs.New(
		errorpath+".%Y%m%d%H%M",
		rotatelogs.WithLinkName(errorpath),
		rotatelogs.WithMaxAge(30*24*time.Hour),
		rotatelogs.WithRotationTime(24*time.Hour),
	)

	Log = logrus.New()
	Log.SetOutput(ioutil.Discard)
	Log.Hooks.Add(lfshook.NewHook(
		lfshook.WriterMap{
			logrus.InfoLevel:  infowriter,
			logrus.ErrorLevel: errorwriter,
		},
		&logrus.JSONFormatter{},
	))
	return Log
}