Merge pull request #12 from zy84338719/feature/login-gates

Enforce setup gating and align admin UI theme

Author: zy84338719

internal/handlers/setup.go

@@ -166,6 +166,41 @@ func (h *SetupHandler) enableUserSystem(adminConfig AdminConfig) error {
 	})
 }
 
+func (h *SetupHandler) isSystemInitialized() (bool, error) {
+	return isSystemInitialized(h.manager, h.daoManager)
+}
+
+func isSystemInitialized(manager *config.ConfigManager, daoManager *repository.RepositoryManager) (bool, error) {
+	if daoManager != nil && daoManager.User != nil {
+		count, err := daoManager.User.CountAdminUsers()
+		if err != nil {
+			return false, err
+		}
+		return count > 0, nil
+	}
+
+	if manager == nil {
+		return false, nil
+	}
+
+	db := manager.GetDB()
+	if db == nil {
+		return false, nil
+	}
+
+	repo := repository.NewRepositoryManager(db)
+	if repo.User == nil {
+		return false, nil
+	}
+
+	count, err := repo.User.CountAdminUsers()
+	if err != nil {
+		return false, err
+	}
+
+	return count > 0, nil
+}
+
 // contains 检查字符串是否包含子字符串
 func contains(s, substr string) bool {
 	for i := 0; i <= len(s)-len(substr); i++ {
@@ -197,6 +232,17 @@ func InitializeNoDB(manager *config.ConfigManager) gin.HandlerFunc {
 			return
 		}
 		defer atomic.StoreInt32(&initInProgress, 0)
+
+		initialized, err := isSystemInitialized(manager, nil)
+		if err != nil {
+			logrus.WithError(err).Error("[InitializeNoDB] 检查系统初始化状态失败")
+			common.InternalServerErrorResponse(c, "检查系统初始化状态失败")
+			return
+		}
+		if initialized {
+			common.ForbiddenResponse(c, "系统已初始化，禁止重复初始化")
+			return
+		}
 		// 解析 JSON（仅接受嵌套结构），不再兼容 legacy 扁平字段
 		var req SetupRequest
 		if !utils.BindJSONWithValidation(c, &req) {
@@ -431,6 +477,17 @@ func (h *SetupHandler) Initialize(c *gin.Context) {
 		return
 	}
 
+	initialized, err := h.isSystemInitialized()
+	if err != nil {
+		logrus.WithError(err).Error("[SetupHandler.Initialize] 检查系统初始化状态失败")
+		common.InternalServerErrorResponse(c, "检查系统初始化状态失败")
+		return
+	}
+	if initialized {
+		common.ForbiddenResponse(c, "系统已初始化，禁止重复初始化")
+		return
+	}
+
 	var req SetupRequest
 	if !utils.BindJSONWithValidation(c, &req) {
 		return

internal/middleware/setup_guard.go

@@ -0,0 +1,109 @@
+package middleware
+
+import (
+	"net/http"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+)
+
+// SetupGuardConfig controls how the setup guard middleware behaves.
+type SetupGuardConfig struct {
+	// IsInitialized returns the current initialization status.
+	IsInitialized func() (bool, error)
+	// SetupPath denotes the setup entry path, defaults to /setup.
+	SetupPath string
+	// RedirectPath denotes the path to redirect to once initialized, defaults to /.
+	RedirectPath string
+	// AllowPaths lists exact paths that should remain accessible before initialization.
+	AllowPaths []string
+	// AllowPrefixes lists path prefixes that should remain accessible before initialization.
+	AllowPrefixes []string
+}
+
+// SetupGuard ensures only setup resources are accessible before initialization
+// and blocks setup routes after initialization is complete.
+func SetupGuard(cfg SetupGuardConfig) gin.HandlerFunc {
+	setupPath := cfg.SetupPath
+	if setupPath == "" {
+		setupPath = "/setup"
+	}
+	redirectPath := cfg.RedirectPath
+	if redirectPath == "" {
+		redirectPath = "/"
+	}
+
+	allowPaths := map[string]struct{}{
+		setupPath:       {},
+		setupPath + "/": {},
+	}
+
+	for _, p := range cfg.AllowPaths {
+		allowPaths[p] = struct{}{}
+	}
+
+	allowPrefixes := []string{setupPath + "/"}
+	allowPrefixes = append(allowPrefixes, cfg.AllowPrefixes...)
+
+	return func(c *gin.Context) {
+		initialized := false
+		if cfg.IsInitialized != nil {
+			var err error
+			initialized, err = cfg.IsInitialized()
+			if err != nil {
+				logrus.WithError(err).Warn("setup guard: failed to determine initialization state")
+				// Fail closed on error so users can still reach setup for recovery.
+				initialized = false
+			}
+		}
+
+		path := c.Request.URL.Path
+
+		if initialized {
+			if path == setupPath || strings.HasPrefix(path, setupPath+"/") {
+				switch c.Request.Method {
+				case http.MethodGet, http.MethodHead:
+					c.Redirect(http.StatusFound, redirectPath)
+				case http.MethodOptions:
+					c.Status(http.StatusNoContent)
+				default:
+					c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
+						"code":    http.StatusForbidden,
+						"message": "系统已初始化，禁止重新初始化",
+					})
+				}
+				c.Abort()
+				return
+			}
+
+			c.Next()
+			return
+		}
+
+		if _, ok := allowPaths[path]; ok {
+			c.Next()
+			return
+		}
+		for _, prefix := range allowPrefixes {
+			if strings.HasPrefix(path, prefix) {
+				c.Next()
+				return
+			}
+		}
+
+		switch c.Request.Method {
+		case http.MethodGet, http.MethodHead:
+			c.Redirect(http.StatusFound, setupPath)
+		case http.MethodOptions:
+			// Allow CORS preflight to complete without redirect loops.
+			c.Status(http.StatusNoContent)
+		default:
+			c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
+				"code":    http.StatusForbidden,
+				"message": "系统未初始化，请访问 /setup 完成初始化",
+			})
+		}
+		c.Abort()
+	}
+}

internal/routes/setup.go

@@ -90,6 +90,71 @@ func CreateAndSetupRouter(
 	router.Use(middleware.CORS())
 	router.Use(middleware.RateLimit(manager))
 
+	var cachedInitialized atomic.Bool
+	var cachedRepo atomic.Pointer[repository.RepositoryManager]
+
+	guardConfig := middleware.SetupGuardConfig{
+		SetupPath:    "/setup",
+		RedirectPath: "/",
+		AllowPaths: []string{
+			"/setup/initialize",
+			"/check-init",
+			"/user/system-info",
+			"/health",
+		},
+		AllowPrefixes: []string{
+			"/assets/",
+			"/css/",
+			"/js/",
+			"/components/",
+		},
+	}
+
+	guardConfig.IsInitialized = func() (bool, error) {
+		if cachedInitialized.Load() {
+			return true, nil
+		}
+
+		if daoManager != nil && daoManager.User != nil {
+			count, err := daoManager.User.CountAdminUsers()
+			if err != nil {
+				return false, err
+			}
+			if count > 0 {
+				cachedInitialized.Store(true)
+				return true, nil
+			}
+			return false, nil
+		}
+
+		db := manager.GetDB()
+		if db == nil {
+			return false, nil
+		}
+
+		repo := cachedRepo.Load()
+		if repo == nil || repo.DB() != db {
+			repo = repository.NewRepositoryManager(db)
+			cachedRepo.Store(repo)
+		}
+		if repo.User == nil {
+			return false, nil
+		}
+
+		count, err := repo.User.CountAdminUsers()
+		if err != nil {
+			return false, err
+		}
+
+		if count > 0 {
+			cachedInitialized.Store(true)
+			return true, nil
+		}
+		return false, nil
+	}
+
+	router.Use(middleware.SetupGuard(guardConfig))
+
 	// 如果 daoManager 为 nil，表示尚未初始化数据库，只注册基础和初始化相关的路由
 	if daoManager == nil {
 		// 基础路由（不传 userHandler）