package middleware

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
	"io"
	"strconv"
	"strings"
)

func LoggerMiddleWare() gin.HandlerFunc {
      
      
	return func(ctx *gin.Context) {
      
      
		//请求方式
		method := ctx.Request.Method
		//请求路由
		reqUrlList := strings.Split(ctx.Request.URL.String(), "?")
		//状态码
		statusCode := ctx.Writer.Status()
		//请求ip
		clientIP := ctx.ClientIP() 
		// 获取请求体数据
		var data map[string]interface{
      
      }
		body, err := io.ReadAll(ctx.Request.Body)
		// 等于拷贝一份往下传递,否则下接口的方法中拿不到请求体数据
		ctx.Request.Body = io.NopCloser(bytes.NewBuffer(body))
		if err != nil {
      
      
			fmt.Println(err, "????")
		}
		err = json.Unmarshal(body, &data)
		// 记录到数据库中,排除登录接口
		if reqUrlList[0] != "/api/v1/admin/login" {
      
      
			message := ""
			if method == "GET" {
      
      
				if len(reqUrlList) == 2 && reqUrlList[1] != "" {
      
      
					message = reqUrlList[1]
				}
			} else {
      
      
				if utils.MapToJson(data) != "null" {
      
      
					message = utils.MapToJson(data)
				}
			}
			// TODO 插入到数据库中
		}

		// 打印日志
		//loggerMap := map[string]interface{} {
      
      
		//	"status_code":statusCode,
		//	"client_ip": clientIP,
		//	"req_method":method,
		//	"req_uri": reqUrl,
		//}
		//marshal, _ := json.Marshal(loggerMap)
		loggerStr := fmt.Sprintf("status_code:%s,client_ip:%s,req_method:%s,req_uri:%s", strconv.Itoa(statusCode), clientIP, method, reqUrlList[0])
		global.Logger.Info("中间件本次请求", zap.String("http", loggerStr))
		ctx.Next()
	}
}