mirror of
https://github.com/nezhahq/nezha.git
synced 2025-03-12 08:48:12 -04:00
f1e3613daf
* fix: config fields not generated on first startup * cleanup * fix reference * replace yaml module * remove duplicated fields * remove yaml.v3 as a direct dependency * update dependency * chore
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/nezhahq/nezha/model"
|
|
"github.com/nezhahq/nezha/service/singleton"
|
|
)
|
|
|
|
// List settings
|
|
// @Summary List settings
|
|
// @Schemes
|
|
// @Description List settings
|
|
// @Security BearerAuth
|
|
// @Tags common
|
|
// @Produce json
|
|
// @Success 200 {object} model.CommonResponse[model.SettingResponse]
|
|
// @Router /setting [get]
|
|
func listConfig(c *gin.Context) (*model.SettingResponse, error) {
|
|
u, authorized := c.Get(model.CtxKeyAuthorizedUser)
|
|
var isAdmin bool
|
|
if authorized {
|
|
user := u.(*model.User)
|
|
isAdmin = user.Role == model.RoleAdmin
|
|
}
|
|
|
|
config := *singleton.Conf
|
|
config.Language = strings.Replace(config.Language, "_", "-", -1)
|
|
|
|
conf := model.SettingResponse{
|
|
Config: model.Setting{
|
|
ConfigForGuests: config.ConfigForGuests,
|
|
ConfigDashboard: config.ConfigDashboard,
|
|
IgnoredIPNotificationServerIDs: config.IgnoredIPNotificationServerIDs,
|
|
Oauth2Providers: config.Oauth2Providers,
|
|
},
|
|
Version: singleton.Version,
|
|
FrontendTemplates: singleton.FrontendTemplates,
|
|
}
|
|
|
|
if !authorized || !isAdmin {
|
|
configForGuests := config.ConfigForGuests
|
|
if authorized {
|
|
configForGuests.AgentTLS = singleton.Conf.AgentTLS
|
|
configForGuests.InstallHost = singleton.Conf.InstallHost
|
|
}
|
|
conf = model.SettingResponse{
|
|
Config: model.Setting{
|
|
ConfigForGuests: configForGuests,
|
|
Oauth2Providers: config.Oauth2Providers,
|
|
},
|
|
}
|
|
}
|
|
|
|
return &conf, nil
|
|
}
|
|
|
|
// Edit config
|
|
// @Summary Edit config
|
|
// @Security BearerAuth
|
|
// @Schemes
|
|
// @Description Edit config
|
|
// @Tags admin required
|
|
// @Accept json
|
|
// @Param body body model.SettingForm true "SettingForm"
|
|
// @Produce json
|
|
// @Success 200 {object} model.CommonResponse[any]
|
|
// @Router /setting [patch]
|
|
func updateConfig(c *gin.Context) (any, error) {
|
|
var sf model.SettingForm
|
|
if err := c.ShouldBindJSON(&sf); err != nil {
|
|
return nil, err
|
|
}
|
|
var userTemplateValid bool
|
|
for _, v := range singleton.FrontendTemplates {
|
|
if !userTemplateValid && v.Path == sf.UserTemplate && !v.IsAdmin {
|
|
userTemplateValid = true
|
|
}
|
|
if userTemplateValid {
|
|
break
|
|
}
|
|
}
|
|
if !userTemplateValid {
|
|
return nil, errors.New("invalid user template")
|
|
}
|
|
|
|
singleton.Conf.Language = strings.Replace(sf.Language, "-", "_", -1)
|
|
|
|
singleton.Conf.EnableIPChangeNotification = sf.EnableIPChangeNotification
|
|
singleton.Conf.EnablePlainIPInNotification = sf.EnablePlainIPInNotification
|
|
singleton.Conf.Cover = sf.Cover
|
|
singleton.Conf.InstallHost = sf.InstallHost
|
|
singleton.Conf.IgnoredIPNotification = sf.IgnoredIPNotification
|
|
singleton.Conf.IPChangeNotificationGroupID = sf.IPChangeNotificationGroupID
|
|
singleton.Conf.SiteName = sf.SiteName
|
|
singleton.Conf.DNSServers = sf.DNSServers
|
|
singleton.Conf.CustomCode = sf.CustomCode
|
|
singleton.Conf.CustomCodeDashboard = sf.CustomCodeDashboard
|
|
singleton.Conf.RealIPHeader = sf.RealIPHeader
|
|
singleton.Conf.AgentTLS = sf.AgentTLS
|
|
singleton.Conf.UserTemplate = sf.UserTemplate
|
|
|
|
if err := singleton.Conf.Save(); err != nil {
|
|
return nil, newGormError("%v", err)
|
|
}
|
|
|
|
singleton.OnNameserverUpdate()
|
|
singleton.OnUpdateLang(singleton.Conf.Language)
|
|
return nil, nil
|
|
}
|