feat: 优化web

internal/auth/wire_provider.go

@@ -0,0 +1,43 @@
+package auth
+
+import (
+	"giter.top/smart/internal/auth/handler"
+	"giter.top/smart/internal/auth/middleware"
+	"giter.top/smart/internal/auth/oauth2"
+	"giter.top/smart/internal/auth/session"
+	"giter.top/smart/pkg/config"
+	"github.com/gin-gonic/gin"
+	"github.com/google/wire"
+)
+
+// ProviderSet Wire 注入。
+var ProviderSet = wire.NewSet(
+	session.NewStore,
+	oauth2.NewStore,
+	oauth2.NewService,
+	oauth2.NewHandler,
+	handler.NewLoginHandler,
+	ProvideBearer,
+	ProvideLoginRLimitWire,
+	ProvideTokenRLimitWire,
+	NewAuthRoutes,
+)
+
+// ProvideBearer 提供 Gin 中间件。
+func ProvideBearer(store *oauth2.Store) gin.HandlerFunc {
+	return middleware.NewBearer(store)
+}
+
+// LoginRateLimitWire、TokenRateLimitWire 用于 Wire 区分多个 gin.HandlerFunc 形参。
+type LoginRateLimitWire gin.HandlerFunc
+type TokenRateLimitWire gin.HandlerFunc
+
+// ProvideLoginRLimitWire 登录接口限流。
+func ProvideLoginRLimitWire(cfg *config.Config) LoginRateLimitWire {
+	return LoginRateLimitWire(middleware.PerIPMinute(cfg.Auth.RateLimit.Enabled, cfg.Auth.RateLimit.LoginPerMinute))
+}
+
+// ProvideTokenRLimitWire 令牌与自省端点限流。
+func ProvideTokenRLimitWire(cfg *config.Config) TokenRateLimitWire {
+	return TokenRateLimitWire(middleware.PerIPMinute(cfg.Auth.RateLimit.Enabled, cfg.Auth.RateLimit.TokenPerMinute))
+}