first commit

internal/middleware/auth_ma.go

@@ -0,0 +1,60 @@
+// Package middleware 提供 HTTP 中间件。
+package middleware
+
+import (
+	"github.com/gin-gonic/gin"
+
+	"portal-mini-go/internal/pkg/errors"
+	"portal-mini-go/internal/repository/miniapp"
+)
+
+const (
+	appIDCtxKey  = "app_id"
+	miniIDCtxKey = "mini_id"
+)
+
+// headerAppID 是小程序在请求头中携带的微信 AppID 字段名。
+const HeaderAppID = "X-App-ID"
+
+// MiniAppAuth 校验请求头中的 app_id 是否对应一个已启用且未删除的被纳管小程序，
+// 通过则将 app_id 与小程序的数据库主键注入 context，供后续 handler/service 使用。
+// 用户身份最终由 (mini_id, open_id) 唯一确定。
+func MiniAppAuth(repo *miniapp.Repository) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		appID := c.GetHeader(HeaderAppID)
+		if appID == "" {
+			c.Error(errors.ErrBadRequest)
+			c.Abort()
+			return
+		}
+		mp, err := repo.GetActiveByAppID(c.Request.Context(), appID)
+		if err != nil {
+			c.Error(errors.ErrNotFound)
+			c.Abort()
+			return
+		}
+		c.Set(appIDCtxKey, appID)
+		c.Set(miniIDCtxKey, mp.ID)
+		c.Next()
+	}
+}
+
+// AppID 从 gin.Context 取出已校验的 appID。
+func AppID(c *gin.Context) string {
+	if v, ok := c.Get(appIDCtxKey); ok {
+		if s, ok := v.(string); ok {
+			return s
+		}
+	}
+	return ""
+}
+
+// MiniID 从 gin.Context 取出已校验的小程序数据库主键。
+func MiniID(c *gin.Context) uint64 {
+	if v, ok := c.Get(miniIDCtxKey); ok {
+		if id, ok := v.(uint64); ok {
+			return id
+		}
+	}
+	return 0
+}