feat: 强制 User-Agent 中间件

.gitignore

@@ -4,7 +4,7 @@ gohub
 .DS_Store
 .history
 *.rdb
-
+output
 # Golang #
 ######################
 # `go test -c` 生成的二进制文件

app/http/middlewares/force_ua.go

@@ -0,0 +1,21 @@
+package middlewares
+
+import (
+	"errors"
+
+	"github.com/gin-gonic/gin"
+
+	"github.com/runningwater/gohub/pkg/response"
+)
+
+// ForceUA 强制设置UA， 请求必须附带 User-Agent 标志头
+func ForceUA() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		if c.Request.Header.Get("User-Agent") == "" {
+			response.BadRequest(c, errors.New("User-Agent is required"), "请求必须附带 User-Agent 标志头")
+			return
+		}
+
+		c.Next()
+	}
+}

bootstrap/route.go

@@ -6,6 +6,7 @@ import (
 	"strings"
 
 	"github.com/gin-gonic/gin"
+
 	"github.com/runningwater/gohub/app/http/middlewares"
 	"github.com/runningwater/gohub/routes"
 )
@@ -41,6 +42,7 @@ func registerGlobalMiddleWare(router *gin.Engine) {
 		middlewares.Logger(),
 		middlewares.Recovery(),
 		middlewares.Cors(),
+		middlewares.ForceUA(),
 		// middlewares.LimitIP("1000-H"),
 	)
 }