mirror of
https://git.fightbot.fun/hxuanyu/FileRelay.git
synced 2026-02-15 06:01:43 +08:00
108 lines
3.6 KiB
Go
108 lines
3.6 KiB
Go
package admin
|
|
|
|
import (
|
|
"FileRelay/internal/bootstrap"
|
|
"FileRelay/internal/config"
|
|
"FileRelay/internal/model"
|
|
"FileRelay/internal/service"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
type ConfigHandler struct{}
|
|
|
|
func NewConfigHandler() *ConfigHandler {
|
|
return &ConfigHandler{}
|
|
}
|
|
|
|
// GetConfig 获取当前完整配置
|
|
// @Summary 获取完整配置
|
|
// @Description 获取系统的完整配置文件内容（仅管理员）
|
|
// @Tags Admin
|
|
// @Security AdminAuth
|
|
// @Produce json
|
|
// @Success 200 {object} model.Response{data=config.Config}
|
|
// @Router /api/admin/config [get]
|
|
func (h *ConfigHandler) GetConfig(c *gin.Context) {
|
|
c.JSON(http.StatusOK, model.SuccessResponse(config.GlobalConfig))
|
|
}
|
|
|
|
// UpdateConfig 更新配置
|
|
// @Summary 更新配置
|
|
// @Description 更新系统的配置文件内容（仅管理员）
|
|
// @Tags Admin
|
|
// @Security AdminAuth
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param config body config.Config true "新配置内容"
|
|
// @Success 200 {object} model.Response{data=config.Config}
|
|
// @Failure 400 {object} model.Response
|
|
// @Failure 500 {object} model.Response
|
|
// @Router /api/admin/config [put]
|
|
func (h *ConfigHandler) UpdateConfig(c *gin.Context) {
|
|
var newConfig config.Config
|
|
if err := c.ShouldBindJSON(&newConfig); err != nil {
|
|
c.JSON(http.StatusBadRequest, model.ErrorResponse(model.CodeBadRequest, err.Error()))
|
|
return
|
|
}
|
|
|
|
// 简单的校验，防止关键配置被改空
|
|
if newConfig.Database.Path == "" {
|
|
newConfig.Database.Path = config.GlobalConfig.Database.Path
|
|
}
|
|
if newConfig.Site.Port <= 0 || newConfig.Site.Port > 65535 {
|
|
newConfig.Site.Port = 8080
|
|
}
|
|
|
|
// 如果传入了明文密码，则重新生成 hash
|
|
if newConfig.Security.AdminPassword != "" {
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(newConfig.Security.AdminPassword), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "Failed to hash password: "+err.Error()))
|
|
return
|
|
}
|
|
newConfig.Security.AdminPasswordHash = string(hash)
|
|
}
|
|
|
|
// 检查取件码长度是否变化
|
|
pickupCodeLengthChanged := newConfig.Security.PickupCodeLength != config.GlobalConfig.Security.PickupCodeLength && newConfig.Security.PickupCodeLength > 0
|
|
// 检查数据库配置是否变化
|
|
dbConfigChanged := newConfig.Database != config.GlobalConfig.Database
|
|
|
|
// 如果长度变化，同步更新现有取件码 (在可能切换数据库前，先处理旧库数据)
|
|
if pickupCodeLengthChanged {
|
|
batchService := service.NewBatchService()
|
|
if err := batchService.UpdateAllPickupCodes(newConfig.Security.PickupCodeLength); err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "Failed to update existing pickup codes: "+err.Error()))
|
|
return
|
|
}
|
|
}
|
|
|
|
// 更新内存配置
|
|
config.UpdateGlobalConfig(&newConfig)
|
|
|
|
// 重新连接数据库并迁移数据（如果配置发生变化）
|
|
if dbConfigChanged {
|
|
if err := bootstrap.ReloadDB(newConfig.Database); err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "Failed to reload database: "+err.Error()))
|
|
return
|
|
}
|
|
}
|
|
|
|
// 重新初始化存储（热更新业务逻辑）
|
|
if err := bootstrap.ReloadStorage(); err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "Failed to reload storage: "+err.Error()))
|
|
return
|
|
}
|
|
|
|
// 保存到文件
|
|
if err := config.SaveConfig(); err != nil {
|
|
c.JSON(http.StatusInternalServerError, model.ErrorResponse(model.CodeInternalError, "Failed to save config: "+err.Error()))
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, model.SuccessResponse(config.GlobalConfig))
|
|
}
|