package clog

import (
	"fmt"
	"os"
	"path"
	"time"

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

// 日志文件路徑和文件名
var (
	logFilePath = "./log/"
	logFileName = "system.log"
)

// Logger 是日志記錄器實(shí)例
var Logger *logrus.Logger

// init 函數(shù)在包被導(dǎo)入時(shí)執(zhí)行，用于初始化日志設(shè)置
func init() {
	// 初始化日志記錄器
	Logger = logrus.New()

	// 設(shè)置日志級(jí)別為 Debug
	Logger.SetLevel(logrus.DebugLevel)

	// 設(shè)置輸出到文件
	fileName := path.Join(logFilePath, logFileName)
	src, err := os.OpenFile(fileName, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
	if err != nil {
		fmt.Println("err", err)
	}
	Logger.Out = src

	// 設(shè)置 rotatelogs，實(shí)現(xiàn)日志文件輪轉(zhuǎn)
	logWriter, _ := rotatelogs.New(
		fileName+".%Y%m%d.log",
		rotatelogs.WithLinkName(fileName),
		rotatelogs.WithMaxAge(15*24*time.Hour),
		rotatelogs.WithRotationTime(24*time.Hour),
	)

	// 配置 lfshook，將不同日志級(jí)別的日志寫入同一個(gè)文件
	writeMap := lfshook.WriterMap{
		logrus.InfoLevel:  logWriter,
		logrus.FatalLevel: logWriter,
		logrus.DebugLevel: logWriter,
		logrus.WarnLevel:  logWriter,
		logrus.ErrorLevel: logWriter,
		logrus.PanicLevel: logWriter,
	}

	// 添加 lfshook 到 Logger
	Logger.AddHook(lfshook.NewHook(writeMap, &logrus.TextFormatter{
		TimestampFormat: "2006-01-02 15:04:05",
	}))
}

// Add 函數(shù)用于添加日志記錄
func Add(requestId, info string, err error) {
	if err != nil {
		// 如果有錯(cuò)誤，記錄 Error 級(jí)別的日志
		Logger.WithFields(logrus.Fields{
			"request_id": requestId,
			"info":       info,
			"error":      err.Error(),
		}).Error()
	} else {
		// 如果沒(méi)有錯(cuò)誤，記錄 Info 級(jí)別的日志
		Logger.WithFields(logrus.Fields{
			"request_id": requestId,
			"info":       info,
			"error":      "",
		}).Info()
	}
}

package auth

import (
	"encoding/json"
	"fmt"
	"time"

	"go-yinhe-cloud/middleware/clog"
	"go-yinhe-cloud/models"
	"go-yinhe-cloud/pkg/util"

	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"
)

// LoggerMiddleware 返回一個(gè) Gin 中間件，用于記錄請(qǐng)求日志
func LoggerMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 記錄請(qǐng)求開(kāi)始時(shí)間
		startTime := time.Now()

		// 處理請(qǐng)求
		c.Next()

		// 記錄請(qǐng)求結(jié)束時(shí)間
		endTime := time.Now()

		// 計(jì)算請(qǐng)求執(zhí)行時(shí)間
		latencyTime := endTime.Sub(startTime)

		// 獲取請(qǐng)求相關(guān)信息
		reqMethod := c.Request.Method        // 請(qǐng)求方式
		reqUrl := c.Request.RequestURI       // 請(qǐng)求路由
		statusCode := c.Writer.Status()      // 狀態(tài)碼
		clientIP := c.ClientIP()             // 請(qǐng)求IP
		header := c.GetHeader("User-Agent") + "@@" + c.GetHeader("token") // 請(qǐng)求頭
		_ = c.Request.ParseMultipartForm(128)
		reqForm := c.Request.Form
		var reqJsonStr string

		// 將請(qǐng)求數(shù)據(jù)轉(zhuǎn)為 JSON 字符串
		if len(reqForm) > 0 {
			reqJsonByte, _ := json.Marshal(reqForm)
			reqJsonStr = string(reqJsonByte)
		}

		// 獲取請(qǐng)求中的 requestId
		requestId := c.GetString("requestId")

		// 獲取 Gin 中間件中的錯(cuò)誤信息
		var errorsStr string
		for _, err := range c.Errors.Errors() {
			errorsStr += err + "; "
		}

		// 記錄請(qǐng)求日志到文件
		clog.Logger.WithFields(logrus.Fields{
			"request_id":   requestId,
			"status_code":  statusCode,
			"latency_time": latencyTime,
			"client_ip":    clientIP,
			"req_method":   reqMethod,
			"ua_token":     header,
			"req_uri":      reqUrl,
			"body":         reqJsonStr,
			"err_str":		errorsStr,
		}).Info()
	}
}