137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package tui
|
||
|
||
import (
|
||
"bufio"
|
||
"fmt"
|
||
"gitee.ltd/lxh/logger/log"
|
||
"github.com/charmbracelet/bubbles/table"
|
||
"github.com/charmbracelet/lipgloss"
|
||
"github.com/spf13/cast"
|
||
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||
"os"
|
||
"strings"
|
||
"wireguard-ui/http/vo"
|
||
"wireguard-ui/model"
|
||
"wireguard-ui/service"
|
||
"wireguard-ui/utils"
|
||
)
|
||
|
||
var BaseStyle = lipgloss.NewStyle().
|
||
BorderStyle(lipgloss.NormalBorder()).
|
||
BorderForeground(lipgloss.Color("240"))
|
||
|
||
var menus = []string{"[1] 客户端链接列表", "[2] 客户端", "[3] 服务端", "[4] 设置", "[q] 退出"}
|
||
|
||
// PrintMenu
|
||
// @description: 打印一下菜单
|
||
func PrintMenu() {
|
||
fmt.Println("\n菜单:")
|
||
for _, menu := range menus {
|
||
fmt.Println(menu)
|
||
}
|
||
}
|
||
|
||
// GenerateTable
|
||
// @description: 生成数据表
|
||
// @param column
|
||
// @param rows
|
||
// @return string
|
||
func GenerateTable(column []table.Column, rows [][]string) string {
|
||
var data []table.Row
|
||
for _, v := range rows {
|
||
data = append(data, v)
|
||
}
|
||
|
||
tl := table.New(
|
||
table.WithColumns(column),
|
||
table.WithRows(data),
|
||
table.WithHeight(len(data)+1),
|
||
)
|
||
|
||
s := table.DefaultStyles()
|
||
s.Header = s.Header.
|
||
BorderStyle(lipgloss.NormalBorder()).
|
||
BorderForeground(lipgloss.Color("240")).
|
||
BorderBottom(true).
|
||
Bold(false)
|
||
s.Selected = lipgloss.NewStyle()
|
||
|
||
tl.SetStyles(s)
|
||
|
||
return BaseStyle.Render(tl.View()+"\n") + "\n一共有 【" + fmt.Sprintf("%d", len(data)) + "】 条数据"
|
||
}
|
||
|
||
// GenerateIP
|
||
// @description: 生成IP
|
||
// @return clientIPS
|
||
// @return serverIPS
|
||
// @return err
|
||
func GenerateIP() (clientIPS, serverIPS []string, err error) {
|
||
// 获取一下服务端信息，因为IP分配需要根据服务端的IP制定
|
||
serverInfo, err := service.Setting().GetWGServerForConfig()
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
|
||
var assignIPS []string
|
||
// 只获取最新的一个
|
||
var clientInfo *model.Client
|
||
if err = service.Client().Order("created_at DESC").Take(&clientInfo).Error; err == nil {
|
||
// 遍历每一个ip是否可允许再分配
|
||
for _, ip := range strings.Split(clientInfo.IpAllocation, ",") {
|
||
if cast.ToInt64(utils.Network().GetIPSuffix(ip)) >= 255 {
|
||
log.Errorf("IP：[%s]已无法分配新IP", ip)
|
||
continue
|
||
} else {
|
||
assignIPS = append(assignIPS, ip)
|
||
}
|
||
}
|
||
}
|
||
|
||
ips := utils.Network().GenerateIPByIPS(serverInfo.Address, assignIPS...)
|
||
|
||
return ips, serverInfo.Address, nil
|
||
}
|
||
|
||
// GenerateKeys
|
||
// @description: 生成密钥对
|
||
// @return keys
|
||
// @return err
|
||
func GenerateKeys() (keys *vo.Keys, err error) {
|
||
// 为空，新增
|
||
privateKey, err := wgtypes.GeneratePrivateKey()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
publicKey := privateKey.PublicKey().String()
|
||
presharedKey, err := wgtypes.GenerateKey()
|
||
if err != nil {
|
||
return nil, fmt.Errorf("生成预共享密钥失败: %s", err.Error())
|
||
}
|
||
keys = &vo.Keys{
|
||
PrivateKey: privateKey.String(),
|
||
PublicKey: publicKey,
|
||
PresharedKey: presharedKey.String(),
|
||
}
|
||
|
||
return keys, nil
|
||
}
|
||
|
||
// Show
|
||
// @description: 展示
|
||
// @param data
|
||
func Show(data any) {
|
||
fmt.Println(data)
|
||
}
|
||
|
||
// readInput
|
||
// @description: 读取输入
|
||
// @param prompt
|
||
// @return string
|
||
func readInput(prompt string) string {
|
||
fmt.Print(prompt)
|
||
scanner := bufio.NewScanner(os.Stdin)
|
||
scanner.Scan()
|
||
return scanner.Text()
|
||
}
|