injectlib/urldb
1
0
Fork 0
mirror of https://github.com/ctwj/urldb.git synced 2025-11-25 19:37:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

update: 系统配置重构

Browse Source
This commit is contained in:
Kerwin
2025-07-29 17:04:49 +08:00
parent ba7dd4d064
commit 4d3f4a082e
10 changed files with 528 additions and 218 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
middleware/public_api.go
View File

@@ -3,6 +3,7 @@ package middleware
import (
"net/http"
"github.com/ctwj/urldb/db/entity"
"github.com/ctwj/urldb/db/repo"
"github.com/gin-gonic/gin"
)
@@ -45,7 +46,8 @@ func PublicAPIAuth() gin.HandlerFunc {
return
}
config, err := repoManager.SystemConfigRepository.FindFirst()
// 验证API Token
apiTokenConfig, err := repoManager.SystemConfigRepository.GetConfigValue(entity.ConfigKeyApiToken)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{
"success": false,
@@ -56,7 +58,7 @@ func PublicAPIAuth() gin.HandlerFunc {
return
}
if config.ApiToken == "" {
if apiTokenConfig == "" {
c.JSON(http.StatusServiceUnavailable, gin.H{
"success": false,
"message": "API Token未配置",
@@ -66,7 +68,7 @@ func PublicAPIAuth() gin.HandlerFunc {
return
}
if config.ApiToken != apiToken {
if apiTokenConfig != apiToken {
c.JSON(http.StatusUnauthorized, gin.H{
"success": false,
"message": "API Token无效",
@@ -77,7 +79,18 @@ func PublicAPIAuth() gin.HandlerFunc {
}
// 检查维护模式
if config.MaintenanceMode {
maintenanceMode, err := repoManager.SystemConfigRepository.GetConfigBool(entity.ConfigKeyMaintenanceMode)
if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{
"success": false,
"message": "系统配置获取失败",
"code": 500,
})
c.Abort()
return
}
if maintenanceMode {
c.JSON(http.StatusServiceUnavailable, gin.H{
"success": false,
"message": "系统维护中，请稍后再试",