441 lines
11 KiB
Go
441 lines
11 KiB
Go
|
package tui
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/charmbracelet/bubbles/table"
|
|||
|
|
"github.com/eiannone/keyboard"
|
|||
|
|
"github.com/spf13/cast"
|
|||
|
|
"strings"
|
|||
|
|
"wireguard-ui/http/vo"
|
|||
|
|
"wireguard-ui/model"
|
|||
|
|
"wireguard-ui/service"
|
|||
|
|
"wireguard-ui/utils"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type SettingComponent struct {
|
|||
|
|
LoginUser *vo.User
|
|||
|
|
Menu []string
|
|||
|
|
Other *OtherSettingComponent
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewSettingComponent(loginUser *vo.User) *SettingComponent {
|
|||
|
|
return &SettingComponent{
|
|||
|
|
LoginUser: loginUser,
|
|||
|
|
Menu: []string{"[1] 服务端配置", "[2] 全局设置", "[3] 其他配置", "[q] 返回上一级菜单"},
|
|||
|
|
Other: NewOtherSettingComponent(),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Menus
|
|||
|
|
// @description: 设置菜单
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *SettingComponent) Menus() {
|
|||
|
|
fmt.Println("")
|
|||
|
|
for _, r := range s.Menu {
|
|||
|
|
fmt.Println(" -> " + r)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
chooseMenu := readInput("\n请选择: ")
|
|||
|
|
switch chooseMenu {
|
|||
|
|
case "1":
|
|||
|
|
s.ServerSetting()
|
|||
|
|
case "2":
|
|||
|
|
s.GlobalSetting()
|
|||
|
|
case "3":
|
|||
|
|
s.Other.Menus(s)
|
|||
|
|
case "q":
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ServerSetting
|
|||
|
|
// @description: 服务端配置
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *SettingComponent) ServerSetting() {
|
|||
|
|
fmt.Println("\n服务端配置")
|
|||
|
|
// 先读取一下服务端配置
|
|||
|
|
servConf, err := service.Setting().GetByCode("WG_SERVER")
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("获取服务端配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type serverConf struct {
|
|||
|
|
IpScope []string `json:"ipScope"`
|
|||
|
|
ListenPort int `json:"listenPort"`
|
|||
|
|
PrivateKey string `json:"privateKey"`
|
|||
|
|
PublicKey string `json:"publicKey"`
|
|||
|
|
PostUpScript string `json:"postUpScript"`
|
|||
|
|
PostDownScript string `json:"postDownScript"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析出来好渲染
|
|||
|
|
var conf serverConf
|
|||
|
|
if err = json.Unmarshal([]byte(servConf.Data), &conf); err != nil {
|
|||
|
|
fmt.Println("解析服务端配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ipScopeIn := readInput(fmt.Sprintf("请输入IP段多个采用 ',[英文逗号]' 分割，不填写默认当前值，下同，当前值[%s] ：", strings.Replace(strings.Join(conf.IpScope, ","), " ", "", -1)))
|
|||
|
|
listenPortIn := readInput(fmt.Sprintf("请输入监听端口，当前值[%d]： ", conf.ListenPort))
|
|||
|
|
privateKeyIn := readInput(fmt.Sprintf("请输入私钥，当前值[%s]： ", conf.PrivateKey))
|
|||
|
|
publicKeyIn := readInput(fmt.Sprintf("请输入公钥，当前值[%s]： ", conf.PublicKey))
|
|||
|
|
postUpScriptIn := readInput(fmt.Sprintf("请输入PostUp脚本，当前值[%s]： ", conf.PostUpScript))
|
|||
|
|
postDownScriptIn := readInput(fmt.Sprintf("请输入PostDown脚本，当前值[%s]： ", conf.PostDownScript))
|
|||
|
|
|
|||
|
|
if ipScopeIn != "" {
|
|||
|
|
conf.IpScope = strings.Split(ipScopeIn, ",")
|
|||
|
|
}
|
|||
|
|
if listenPortIn != "" {
|
|||
|
|
conf.ListenPort = cast.ToInt(listenPortIn)
|
|||
|
|
}
|
|||
|
|
if privateKeyIn != "" {
|
|||
|
|
conf.PrivateKey = privateKeyIn
|
|||
|
|
}
|
|||
|
|
if publicKeyIn != "" {
|
|||
|
|
conf.PublicKey = publicKeyIn
|
|||
|
|
}
|
|||
|
|
if postUpScriptIn != "" {
|
|||
|
|
conf.PostUpScript = postUpScriptIn
|
|||
|
|
}
|
|||
|
|
if postDownScriptIn != "" {
|
|||
|
|
conf.PostDownScript = postDownScriptIn
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data, _ := json.Marshal(conf)
|
|||
|
|
|
|||
|
|
if err := service.Setting().SetData(&model.Setting{
|
|||
|
|
Code: "WG_SERVER",
|
|||
|
|
Data: string(data),
|
|||
|
|
}); err != nil {
|
|||
|
|
fmt.Println("保存服务端配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("修改服务端配置成功")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GlobalSetting
|
|||
|
|
// @description: 全局设置
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *SettingComponent) GlobalSetting() {
|
|||
|
|
fmt.Println("\n服务端配置")
|
|||
|
|
// 先读取一下服务端配置
|
|||
|
|
globalConf, err := service.Setting().GetByCode("WG_SETTING")
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("获取服务端配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type gConf struct {
|
|||
|
|
MTU int `json:"MTU"`
|
|||
|
|
ConfigFilePath string `json:"configFilePath"`
|
|||
|
|
DnsServer []string `json:"dnsServer"`
|
|||
|
|
EndpointAddress string `json:"endpointAddress"`
|
|||
|
|
FirewallMark string `json:"firewallMark"`
|
|||
|
|
PersistentKeepalive int `json:"persistentKeepalive"`
|
|||
|
|
Table string `json:"table"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析出来好渲染
|
|||
|
|
var conf gConf
|
|||
|
|
if err = json.Unmarshal([]byte(globalConf.Data), &conf); err != nil {
|
|||
|
|
fmt.Println("解析全局配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mtu := readInput(fmt.Sprintf("请输入mtu，不填写默认当前值，下同，当前值[%d] ：", conf.MTU))
|
|||
|
|
configFilePath := readInput(fmt.Sprintf("请输入配置文件地址，当前值[%s]： ", conf.ConfigFilePath))
|
|||
|
|
dnsServer := readInput(fmt.Sprintf("请输入dns，多个采用 ',[英文逗号]' 分割，当前值[%s]： ", strings.Replace(strings.Join(conf.DnsServer, ","), " ", "", -1)))
|
|||
|
|
endpointAddress := readInput(fmt.Sprintf("请输入公网IP，默认系统自动获取，当前值[%s]： ", conf.EndpointAddress))
|
|||
|
|
firewallMark := readInput(fmt.Sprintf("请输入FirewallMark，当前值[%s]： ", conf.FirewallMark))
|
|||
|
|
persistentKeepalive := readInput(fmt.Sprintf("请输入PersistentKeepalive，当前值[%d]： ", conf.PersistentKeepalive))
|
|||
|
|
tableRule := readInput(fmt.Sprintf("请输入Table，当前值[%s]： ", conf.Table))
|
|||
|
|
|
|||
|
|
if mtu != "" {
|
|||
|
|
conf.MTU = cast.ToInt(mtu)
|
|||
|
|
}
|
|||
|
|
if configFilePath != "" {
|
|||
|
|
conf.ConfigFilePath = configFilePath
|
|||
|
|
}
|
|||
|
|
if dnsServer != "" {
|
|||
|
|
conf.DnsServer = strings.Split(dnsServer, ",")
|
|||
|
|
}
|
|||
|
|
if endpointAddress != "" {
|
|||
|
|
conf.EndpointAddress = endpointAddress
|
|||
|
|
} else {
|
|||
|
|
conf.EndpointAddress = utils.Network().GetHostPublicIP()
|
|||
|
|
}
|
|||
|
|
if firewallMark != "" {
|
|||
|
|
conf.FirewallMark = firewallMark
|
|||
|
|
}
|
|||
|
|
if persistentKeepalive != "" {
|
|||
|
|
conf.PersistentKeepalive = cast.ToInt(persistentKeepalive)
|
|||
|
|
}
|
|||
|
|
if tableRule != "" {
|
|||
|
|
conf.Table = tableRule
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data, _ := json.Marshal(conf)
|
|||
|
|
|
|||
|
|
if err := service.Setting().SetData(&model.Setting{
|
|||
|
|
Code: "WG_SETTING",
|
|||
|
|
Data: string(data),
|
|||
|
|
}); err != nil {
|
|||
|
|
fmt.Println("保存全局配置失败: ", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("修改全局配置成功")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// OtherSettingComponent
|
|||
|
|
// @description: 其他配置杂项
|
|||
|
|
type OtherSettingComponent struct {
|
|||
|
|
Setting *SettingComponent
|
|||
|
|
Menu []string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewOtherSettingComponent() *OtherSettingComponent {
|
|||
|
|
return &OtherSettingComponent{
|
|||
|
|
Menu: []string{"[1] 列表", "[2] 添加", "[3] 编辑", "[4] 删除", "[q] 返回上一级菜单"},
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *OtherSettingComponent) Menus(setting *SettingComponent) {
|
|||
|
|
if s.Setting == nil {
|
|||
|
|
s.Setting = setting
|
|||
|
|
}
|
|||
|
|
fmt.Println("")
|
|||
|
|
for _, r := range s.Menu {
|
|||
|
|
fmt.Println(" -> " + r)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
chooseMenu := readInput("\n请选择: ")
|
|||
|
|
switch chooseMenu {
|
|||
|
|
case "1":
|
|||
|
|
s.List(true)
|
|||
|
|
case "2":
|
|||
|
|
s.Add()
|
|||
|
|
case "3":
|
|||
|
|
s.Edit()
|
|||
|
|
case "4":
|
|||
|
|
s.Delete()
|
|||
|
|
|
|||
|
|
case "q":
|
|||
|
|
s.Setting.Menus()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// List
|
|||
|
|
// @description: 其他配置
|
|||
|
|
// @receiver s
|
|||
|
|
// @param showMenu
|
|||
|
|
func (s *OtherSettingComponent) List(showMenu bool) {
|
|||
|
|
fmt.Println("\n其他配置列表")
|
|||
|
|
// 不查询的配置
|
|||
|
|
var blackList = []string{"WG_SETTING", "WG_SERVER"}
|
|||
|
|
|
|||
|
|
data, err := service.Setting().GetAllSetting(blackList)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("获取配置失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
title := []table.Column{
|
|||
|
|
{
|
|||
|
|
Title: "序号",
|
|||
|
|
Width: 10,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
Title: "编码",
|
|||
|
|
Width: 40,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
Title: "描述",
|
|||
|
|
Width: 50,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
Title: "创建时间",
|
|||
|
|
Width: 30,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
Title: "更新时间",
|
|||
|
|
Width: 30,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var result [][]string
|
|||
|
|
for i, v := range data {
|
|||
|
|
result = append(result, []string{
|
|||
|
|
cast.ToString(i + 1),
|
|||
|
|
v.Code,
|
|||
|
|
v.Describe,
|
|||
|
|
v.CreatedAt.Format("2006-01-02 15:04:05"),
|
|||
|
|
v.UpdatedAt.Format("2006-01-02 15:04:05"),
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Show(GenerateTable(title, result))
|
|||
|
|
if showMenu {
|
|||
|
|
s.Menus(nil)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Add
|
|||
|
|
// @description: 新增其他配置
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *OtherSettingComponent) Add() {
|
|||
|
|
fmt.Println("\n新增其他配置")
|
|||
|
|
|
|||
|
|
code := readInput("请输入配置编码，此编码是唯一编码不可重复：")
|
|||
|
|
desc := readInput("请输入配置描述：")
|
|||
|
|
|
|||
|
|
// 监听键盘事件，只监听 + 和 - 和 enter
|
|||
|
|
if err := keyboard.Open(); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
defer func() {
|
|||
|
|
_ = keyboard.Close()
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// + <=> 43 | - <=> 45 | enter <=> 0
|
|||
|
|
fmt.Println("请按下 + 或者 - 进行配置项的新增和删除")
|
|||
|
|
fmt.Println("每一项配置如此： key=val ")
|
|||
|
|
fmt.Println("确认输入完成后 enter[按一次代表当前配置项输入完成，两次代表新增完成]")
|
|||
|
|
fmt.Println("首先进入时请输入 + 进行第一个配置项填写")
|
|||
|
|
var breakCycle bool
|
|||
|
|
var keyVal []string
|
|||
|
|
for {
|
|||
|
|
char, _, err := keyboard.GetKey()
|
|||
|
|
if err != nil {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if breakCycle {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
switch char {
|
|||
|
|
case 0:
|
|||
|
|
// 收到enter事件触发，执行后面的
|
|||
|
|
var dm = make(map[string]any)
|
|||
|
|
for _, kv := range keyVal {
|
|||
|
|
kvs := strings.Split(kv, "=")
|
|||
|
|
key := kvs[0]
|
|||
|
|
val := kvs[1]
|
|||
|
|
dm[key] = val
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
dms, err := json.Marshal(dm)
|
|||
|
|
if err != nil {
|
|||
|
|
breakCycle = true
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = service.Setting().SetData(&model.Setting{
|
|||
|
|
Code: code,
|
|||
|
|
Data: string(dms),
|
|||
|
|
Describe: desc,
|
|||
|
|
}); err != nil {
|
|||
|
|
breakCycle = true
|
|||
|
|
fmt.Println("保存配置失败: ", err.Error())
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("保存配置成功")
|
|||
|
|
s.List(true)
|
|||
|
|
|
|||
|
|
breakCycle = true
|
|||
|
|
case 43:
|
|||
|
|
keyVal = append(keyVal, readInput("请输入配置项："))
|
|||
|
|
case 45:
|
|||
|
|
keyVal = keyVal[:len(keyVal)-1]
|
|||
|
|
fmt.Println("已删除最后一个配置项，当前配置项为：", keyVal)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Edit
|
|||
|
|
// @description: 编辑
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *OtherSettingComponent) Edit() {
|
|||
|
|
fmt.Println("\n编辑其他配置")
|
|||
|
|
|
|||
|
|
s.List(false)
|
|||
|
|
|
|||
|
|
code := readInput("请输入需要编辑的配置编码：")
|
|||
|
|
|
|||
|
|
// 通过编码查询配置
|
|||
|
|
setting, err := service.Setting().GetByCode(code)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("查找["+code+"]配置失败：", err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
desc := readInput("请输入需要编辑的配置描述：")
|
|||
|
|
if desc != "" {
|
|||
|
|
setting.Describe = desc
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var kvm = make(map[string]any)
|
|||
|
|
if err = json.Unmarshal([]byte(setting.Data), &kvm); err != nil {
|
|||
|
|
fmt.Println("配置解析失败: ", err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for k, v := range kvm {
|
|||
|
|
valIn := readInput(fmt.Sprintf("请输入配置项值，仅能修改值，当前键值对：%s=%v ：", k, v))
|
|||
|
|
if valIn != "" {
|
|||
|
|
kvm[k] = valIn
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理以下数据
|
|||
|
|
kvmStr, err := json.Marshal(kvm)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("序列化数据失败: ", err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
setting.Data = string(kvmStr)
|
|||
|
|
|
|||
|
|
if err = service.Setting().SetData(setting); err != nil {
|
|||
|
|
fmt.Println("修改配置失败: ", err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("修改配置成功")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Delete
|
|||
|
|
// @description: 删除
|
|||
|
|
// @receiver s
|
|||
|
|
func (s *OtherSettingComponent) Delete() {
|
|||
|
|
fmt.Println("\n 删除指定配置")
|
|||
|
|
|
|||
|
|
s.List(false)
|
|||
|
|
|
|||
|
|
code := readInput("请输入要删除的配置项编码：")
|
|||
|
|
// 查询配置是否存在
|
|||
|
|
if err := service.Setting().Model(&model.Setting{}).
|
|||
|
|
Where("code NOT IN (?)", []string{"WG_SETTING", "WG_SERVER"}).
|
|||
|
|
Where("code = ?", code).Delete(&model.Setting{}).Error; err != nil {
|
|||
|
|
fmt.Println("删除[" + code + "]配置失败")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Println("删除成功")
|
|||
|
|
|
|||
|
|
s.List(true)
|
|||
|
|
|
|||
|
|
}
|