package dump

import (
	"bytes"
	"context"
	"gitea.loveuer.com/yizhisec/packages/logger"
	"gitea.loveuer.com/yizhisec/packages/tool"
	"github.com/gin-gonic/gin"
	"io"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"
)

type RequestHandler func(r *http.Request, body []byte) string

func Request(ctx context.Context, handler RequestHandler, writers ...io.Writer) gin.HandlerFunc {

	var (
		out     io.Writer = os.Stdout
		ch                = make(chan string, 128)
		builder           = strings.Builder{}
		buf               = make([]string, 0, 16)
	)

	if len(writers) > 0 && writers[0] != nil {
		out = writers[0]
	}

	do := func() {
		for _, item := range buf {
			builder.WriteString(item)
			builder.WriteRune('\n')
		}

		_, _ = out.Write(tool.StringToBytes(builder.String()))

		builder.Reset()
		buf = buf[:0]
	}

	go func() {

		ticker := time.NewTicker(time.Second)
		defer ticker.Stop()

		for {
			select {
			case <-ctx.Done():
				if len(buf) > 0 {
					do()
				}
				return
			case <-ticker.C:
				if len(buf) > 0 {
					do()
				}
			case msg, _ := <-ch:
				buf = append(buf, msg)
				if len(buf) >= 10 {
					do()
				}
			}
		}
	}()

	return func(c *gin.Context) {
		var (
			err           error
			contentType   = c.GetHeader("Content-Type")
			contentLength = c.GetHeader("Content-Length")
			cl            int
		)

		if contentLength == "" && (c.Request.Method == "GET" || c.Request.Method == "HEAD") {
			goto DUMP
		}

		if cl, err = strconv.Atoi(contentLength); err != nil {
			logger.WarnCtx(c.Request.Context(), "Request: convert Content-Length failed, err = %s", err.Error())
			c.Next()
			return
		}

		if cl > 0 && !strings.Contains(contentType, "application/json") {
			c.Next()
			return
		}

	DUMP:

		bodyBytes, err := io.ReadAll(c.Request.Body)
		if err != nil {
			logger.WarnCtx(c.Request.Context(), "读取请求体错误: %v", err)
			c.Next()
			return
		}

		c.Request.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))

		dumped := handler(c.Request, bodyBytes)

		ch <- dumped

		c.Next()
	}
}

func RequestHandlerCurl(r *http.Request, body []byte) string {
	var builder strings.Builder

	// 添加 curl 基础命令和方法
	builder.WriteString("curl -X " + r.Method)

	// 添加请求 URL
	url := getFullURL(r)
	builder.WriteString(" '" + url + "'")

	// 添加请求头
	for key, values := range r.Header {
		if strings.EqualFold(key, "Host") {
			continue // 跳过 Host 头
		}
		for _, value := range values {
			builder.WriteString(" -H '" + key + ": " + value + "'")
		}
	}

	// 添加 JSON 数据
	if len(body) > 0 {
		// 转义单引号防止命令中断
		escapedBody := strings.ReplaceAll(string(body), "'", `'\''`)
		builder.WriteString(" -d '" + escapedBody + "'")
	}

	return builder.String()
}

func getFullURL(r *http.Request) string {
	scheme := "http"
	if r.TLS != nil {
		scheme = "https"
	}
	return scheme + "://" + r.Host + r.URL.RequestURI()
}