// Config 日志配置
type Config struct {
	DirName       string   // 日志目录名称
	NamePrefix    string   // 日志前置名称
	RetentionDays int64    // 日志保留时间
	FieldsOrder   []string // 字段顺序
	IsWriteToFile bool     // 是否写入文件中
}

// clearExpiredLogs 清除过期日志任务
func clearExpiredLogs() {
	timeTemplate := "2006-01-2"
	entry, err := os.ReadDir(c.DirName)
	if err != nil {
		fmt.Println(err)
		return
	}

	retentionSeconds := c.RetentionDays * 24 * 60 * 60
	for _, v := range entry {
		if !v.IsDir() {
			name := v.Name()
			if strings.HasPrefix(name, c.NamePrefix) {
				timeStr := strings.Replace(name, c.NamePrefix+"_", "", 1)
				timeStr = strings.Replace(timeStr, ".log", "", 1)
				timeStr = strings.Replace(timeStr, "_", "-", 1)

				stamp, err := time.Parse(timeTemplate, timeStr)
				if err != nil {
					fmt.Println(err)
					continue
				}

				if time.Now().Unix()-stamp.Unix() >= retentionSeconds {
					os.Remove(fmt.Sprintf("./%s/%s", c.DirName, v.Name()))
				}
			}
		}
	}
}

func startRotatingTask() error {
	c := cron.New()

	_, err := c.AddFunc("@daily", func() {
		fmt.Println("Every day")
		clearExpiredLogs()
	}) //  每天运行一次，午夜 | 0 0 * * *
	clearExpiredLogs()

	if err != nil {
		return err
	}

	c.Start()

	return nil
}

// 日志实例
var Logger *logrus.Logger

// InitLog 初始化日志
func InitLog(conf *Config) {
	// 初始化配置
	setDirName(conf.DirName)
	setNamePrefix(conf.NamePrefix)
	setRetentionDays(conf.RetentionDays)
	setFieldsOrder(conf.FieldsOrder)
	setIsWriteToFile(conf.IsWriteToFile)

	// 创建日志目录
	if !isExistedDir(c.DirName) {
		if err := os.MkdirAll(c.DirName, 0777); err != nil {
			panic(err)
		}
	}

	// 开始日志清理任务
	if Logger != nil {
		err := startRotatingTask()
		if err != nil {
			panic(err)
		}
	}

	// 创建日志实体
	createLog()
}

func createLog() {
	f := &zt_formatter.ZtFormatter{
		CallerPrettyfier: func(f *runtime.Frame) (string, string) {
			filename := path.Base(f.File)
			return fmt.Sprintf("%s()", f.Function), fmt.Sprintf("%s:%d", filename, f.Line)
		},
		Formatter: nested.Formatter{
			FieldsOrder:     c.FieldsOrder,
			TimestampFormat: "2006-01-02T15:04:05.999999999Z07:00",
		},
	}

	Logger = logrus.New()

	if c.IsWriteToFile {
		ztLog.SetupLogsCanExpand(Logger, f, true, fmt.Sprintf("./%s/%s", c.DirName, c.NamePrefix), int(logrus.DebugLevel))
	} else {
		Logger.SetLevel(logrus.DebugLevel)
		Logger.SetReportCaller(true)
		Logger.SetFormatter(f)
	}
}

func main() {
	glog.InitLog(&glog.Config{
		IsWriteToFile: true,
	})

	l := glog.Logger
	l.Infof("this is %v demo", "glog")

	lWebServer := l.WithField("component", "web-server")
	lWebServer.Info("starting...")

	lWebServerReq := lWebServer.WithFields(logrus.Fields{
		"req":   "GET /api/stats",
		"reqId": "#1",
	})

	lWebServerReq.Info("params: startYear=2048")
	lWebServerReq.Error("response: 400 Bad Request")

	lDbConnector := l.WithField("category", "db-connector")
	lDbConnector.Info("connecting to db on 10.10.10.13...")
	lDbConnector.Warn("connection took 10s")

	l.Info("demo end.")
}