NetworkAuth/middleware/install.go

package middleware

import (
	"NetworkAuth/services"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"
	"github.com/spf13/viper"
)

// InstallCheckMiddleware 检查系统是否已安装
func InstallCheckMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		path := c.Request.URL.Path

		isInstallRoute := path == "/api/install" || path == "/api/install/"

		// 获取系统的安装状态
		isInstalledStr := services.GetSettingsService().GetString("is_installed", "0")
		isInstalled := isInstalledStr == "1"

		// 如果设置服务没获取到（因为未连接数据库），再结合文件判断
		if !isInstalled {
			// 检查数据库文件是否存在（如果是 sqlite）
			dbType := viper.GetString("database.type")
			switch dbType {
			case "sqlite":
				dbPath := viper.GetString("database.sqlite.path")
				if dbPath == "" {
					dbPath = "./database.db"
				}
				if _, err := os.Stat(dbPath); os.IsNotExist(err) {
					isInstalled = false
				} else {
					isInstalled = true
				}
			case "mysql":
				// 如果是 mysql 且配置了 database，我们认为是已安装
				dbName := viper.GetString("database.mysql.database")
				if dbName != "" {
					isInstalled = true
				}
			}
		}

		// 如果未安装且不是访问安装接口，则返回 403 JSON
		if !isInstalled && !isInstallRoute {
			c.JSON(http.StatusForbidden, gin.H{
				"code": 403,
				"msg":  "系统未初始化，请先完成安装",
			})
			c.Abort()
			return
		}

		// 如果已安装但尝试访问安装接口，则返回 403 JSON
		if isInstalled && isInstallRoute {
			c.JSON(http.StatusForbidden, gin.H{
				"code": 403,
				"msg":  "系统已安装，请勿重复初始化",
			})
			c.Abort()
			return
		}

		c.Next()
	}
}