🆕为实现数据迁移以及备份新增配置导出

service/setting.go

@@ -3,6 +3,7 @@ package service
 import (
 	"encoding/json"
 	"gorm.io/gorm"
+	"strings"
 	gdb "wireguard-ui/global/client"
 	"wireguard-ui/http/vo"
 	"wireguard-ui/model"
@@ -84,3 +85,60 @@ func (s setting) GetAllSetting(blackList []string) (data []vo.SettingItem, err e
 	err = s.Model(&model.Setting{}).Select("code, data, describe").Where("code not in ?", blackList).Find(&data).Error
 	return
 }
+
+// Export
+// @description: 导出
+// @receiver s
+// @return data
+// @return err
+func (s setting) Export() (data vo.Export, err error) {
+	// 先查询global配置
+	var gs, ss *model.Setting
+	if err = s.Model(&model.Setting{}).Where("code = ?", "WG_SETTING").Take(&gs).Error; err != nil {
+		return
+	}
+	if err = json.Unmarshal([]byte(gs.Data), &data.Global); err != nil {
+		return
+	}
+
+	// 查询server配置
+	if err = s.Model(&model.Setting{}).Where("code = ?", "WG_SERVER").Take(&ss).Error; err != nil {
+		return
+	}
+	if err = json.Unmarshal([]byte(ss.Data), &data.Server); err != nil {
+		return
+	}
+
+	// 查询client配置
+	var clients []vo.ClientItem
+	if err = s.Model(&model.Client{}).Select("id,name,email,ip_allocation as ip_allocation_str," +
+		"allowed_ips as allowed_ips_str,extra_allowed_ips as extra_allowed_ips_str," +
+		"endpoint,use_server_dns,keys as keys_str," +
+		"enabled,offline_monitoring").Order("created_at DESC").Find(&clients).Error; err != nil {
+		return
+	}
+
+	for i, v := range clients {
+		if v.KeysStr != "" {
+			_ = json.Unmarshal([]byte(v.KeysStr), &clients[i].Keys)
+		}
+		if v.IpAllocationStr != "" {
+			clients[i].IpAllocation = strings.Split(v.IpAllocationStr, ",")
+		}
+		if v.AllowedIpsStr != "" {
+			clients[i].AllowedIps = strings.Split(v.AllowedIpsStr, ",")
+		} else {
+			clients[i].AllowedIps = []string{}
+		}
+		if v.ExtraAllowedIpsStr != "" {
+			clients[i].ExtraAllowedIps = strings.Split(v.ExtraAllowedIpsStr, ",")
+		} else {
+			clients[i].ExtraAllowedIps = []string{}
+		}
+	}
+
+	cj, _ := json.Marshal(clients)
+	_ = json.Unmarshal(cj, &data.Clients)
+
+	return
+}