package middleware

import (
	"bytes"
	"encoding/json"
	"io"
	"log"
	"strings"

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

// 打印请求参数和响应参数

func LoggerParams() gin.HandlerFunc {

	return func(ctx *gin.Context) {
		buf := &bytes.Buffer{}
		tea := io.TeeReader(ctx.Request.Body, buf)
		body, err := io.ReadAll(tea)
		if err != nil {
			log.Panicf("read body err:%+v", err)
		}
		ctx.Request.Body = io.NopCloser(buf)

		// 下面是响应参数缓存
		blw := &bodyParamsWriter{
			body:           bytes.NewBufferString(""),
			ResponseWriter: ctx.Writer,
		}
		ctx.Writer = blw

		ctx.Next()

		// 判断返回的值是否json
		contentType := 	ctx.Writer.Header().Get("Content-Type")

		if !strings.Contains(contentType, "application/json") {
			return
		}

		resp := blw.body.String()

		m := map[string]interface{}{
			"trace_id":       ctx.Keys["trace_id"],
			"method":         ctx.Request.Method,
			"path":           ctx.FullPath(),
			"client_ip":      ctx.ClientIP(),
			"body":           string(body),
			"response":       resp,
			"request_header": ctx.Request.Header,
			"response_code":  ctx.Writer.Status(),
		}
		b, _ := json.Marshal(m)

		log.Println(string(b))
	}

	// return gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {

	// 	buf := &bytes.Buffer{}
	// 	tea := io.TeeReader(param.Request.Body, buf)
	// 	body, err := ioutil.ReadAll(tea)
	// 	if err != nil {
	// 		log.Panicf("read body err:%+v", err)
	// 	}
	// 	param.Request.Body = io.NopCloser(buf)

	// 	fmt.Println("ddddd",string(body))

	// 	// 你的自定义格式
	// 	return fmt.Sprintf("[VISIT] trace_id:%s | 参数：%s | IP:%s | 时间：%s | 方法：%s | 路径：%s | 协议：%s | 状态码：%d | 耗时：%s | 代理头：%s | 错误信息：%s \n",
	// 		// trace_id
	// 		param.Keys["trace_id"],
	// 		body,
	// 		//客户端IP
	// 		param.ClientIP,
	// 		//时间格式
	// 		param.TimeStamp.Format("2006-01-02 15:04:05"),
	// 		//http请求方式 get post等
	// 		param.Method,
	// 		//客户端请求的路径
	// 		param.Path,
	// 		//http请求协议版本
	// 		param.Request.Proto,
	// 		//http请求状态码
	// 		param.StatusCode,
	// 		//请求耗时
	// 		param.Latency,
	// 		//http请求代理头
	// 		param.Request.UserAgent(),
	// 		//处理请求错误时设置错误消息
	// 		param.ErrorMessage,
	// 	)
	// })
}

type bodyParamsWriter struct {
	gin.ResponseWriter
	body *bytes.Buffer
}

// 真正输出的时候，它会调用这个输出方法
func (w bodyParamsWriter) Write(b []byte) (int, error) {
	// b = []byte(`{"code":0}`)
	w.body.Write(b)
	return w.ResponseWriter.Write(b)
}