feat: 添加 dump request 中间件和处理程序 curl

middlewares/dump/req.go

@@ -0,0 +1,147 @@
+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()
+}