163 lines
3.5 KiB
Go
163 lines
3.5 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"gitee.ltd/lxh/logger/log"
|
|
"github.com/gin-gonic/gin"
|
|
"slices"
|
|
"wireguard-ui/http/param"
|
|
"wireguard-ui/http/response"
|
|
"wireguard-ui/http/vo"
|
|
"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())
|
|
}
|
|
|
|
// Export
|
|
// @description: 导出配置
|
|
// @receiver setting
|
|
// @param c
|
|
func (setting) Export(c *gin.Context) {
|
|
// 获取当前登陆用户
|
|
var loginUser *vo.User
|
|
if loginUser = GetCurrentLoginUser(c); c.IsAborted() {
|
|
return
|
|
}
|
|
|
|
if loginUser.Account != "admin" {
|
|
response.R(c).FailedWithError("非法操作，你被捕啦！")
|
|
return
|
|
}
|
|
|
|
// 获取配置
|
|
data, err := service.Setting().Export()
|
|
if err != nil {
|
|
response.R(c).FailedWithError(err)
|
|
return
|
|
}
|
|
|
|
// 生成配置文件
|
|
dataBytes, _ := json.Marshal(data)
|
|
filepath, err := utils.FileUtils().GenerateFile("config.json", dataBytes)
|
|
if err != nil {
|
|
response.R(c).FailedWithError(err)
|
|
return
|
|
}
|
|
|
|
c.Header("Content-Type", "application/octet-stream")
|
|
c.Header("Content-Disposition", "attachment; filename="+filepath)
|
|
c.Header("Content-Transfer-Encoding", "binary")
|
|
c.Header("Connection", "keep-alive")
|
|
c.File(filepath)
|
|
//if err = os.Remove(filepath); err != nil {
|
|
// log.Errorf("删除临时文件失败: %s", err.Error())
|
|
//}
|
|
}
|
|
|
|
func (setting) Import(c *gin.Context) {
|
|
|
|
}
|