116 lines
2.4 KiB
Go
116 lines
2.4 KiB
Go
package api
|
|
|
|
import (
|
|
"gitee.ltd/lxh/logger/log"
|
|
"github.com/gin-gonic/gin"
|
|
"slices"
|
|
"wireguard-ui/http/param"
|
|
"wireguard-ui/http/response"
|
|
"wireguard-ui/model"
|
|
"wireguard-ui/script"
|
|
"wireguard-ui/service"
|
|
"wireguard-ui/utils"
|
|
)
|
|
|
|
type setting struct{}
|
|
|
|
func Setting() setting {
|
|
return setting{}
|
|
}
|
|
|
|
// Set
|
|
// @description: 设置配置
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) Set(c *gin.Context) {
|
|
var p param.SetSetting
|
|
if err := c.ShouldBind(&p); err != nil {
|
|
response.R(c).Validator(err)
|
|
return
|
|
}
|
|
|
|
if err := service.Setting().SetData(&model.Setting{
|
|
Code: p.Code,
|
|
Data: p.Data,
|
|
Describe: p.Describe,
|
|
}); err != nil {
|
|
response.R(c).FailedWithError(err)
|
|
return
|
|
}
|
|
|
|
var whiteCodes = []string{"WG_SETTING", "WG_SERVER"}
|
|
if slices.Contains(whiteCodes, p.Code) {
|
|
go func() {
|
|
if err := script.New().GenerateConfig(); err != nil {
|
|
log.Errorf("执行脚本失败")
|
|
}
|
|
}()
|
|
}
|
|
|
|
response.R(c).OK()
|
|
}
|
|
|
|
// Delete
|
|
// @description: 删除配置
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) Delete(c *gin.Context) {
|
|
code := c.Param("code")
|
|
if code == "" || code == "undefined" {
|
|
response.R(c).FailedWithError("code不能为空")
|
|
return
|
|
}
|
|
|
|
if err := service.Setting().Model(&model.Setting{}).Where("code = ?", code).Delete(&model.Setting{}).Error; err != nil {
|
|
response.R(c).FailedWithError("删除失败")
|
|
return
|
|
}
|
|
|
|
response.R(c).OK()
|
|
}
|
|
|
|
// GetSetting
|
|
// @description: 获取指定配置
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) GetSetting(c *gin.Context) {
|
|
code := c.Query("code")
|
|
if code == "" {
|
|
response.R(c).FailedWithError("code不能为空")
|
|
return
|
|
}
|
|
|
|
var data *model.Setting
|
|
if err := service.Setting().Model(&model.Setting{}).Where("code = ?", code).Take(&data).Error; err != nil {
|
|
response.R(c).FailedWithError("获取指定配置失败")
|
|
return
|
|
}
|
|
|
|
response.R(c).OkWithData(data.Data)
|
|
}
|
|
|
|
// GetAllSetting
|
|
// @description: 获取全部配置
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) GetAllSetting(c *gin.Context) {
|
|
// 不查询的配置
|
|
var blackList = []string{"WG_SETTING", "WG_SERVER"}
|
|
|
|
data, err := service.Setting().GetAllSetting(blackList)
|
|
if err != nil {
|
|
response.R(c).FailedWithError("获取配置失败")
|
|
return
|
|
}
|
|
|
|
response.R(c).OkWithData(data)
|
|
}
|
|
|
|
// GetPublicAddr
|
|
// @description: 获取公网地址
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) GetPublicAddr(c *gin.Context) {
|
|
response.R(c).OkWithData(utils.Network().GetHostPublicIP())
|
|
}
|