521 lines
12 KiB
Go
521 lines
12 KiB
Go
package tui
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"gitee.ltd/lxh/logger/log"
|
||
"github.com/charmbracelet/bubbles/table"
|
||
"github.com/spf13/cast"
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
"wireguard-ui/component"
|
||
"wireguard-ui/global/constant"
|
||
"wireguard-ui/http/param"
|
||
"wireguard-ui/http/vo"
|
||
"wireguard-ui/service"
|
||
"wireguard-ui/utils"
|
||
)
|
||
|
||
type ClientComponent struct {
|
||
LoginUser *vo.User
|
||
Menu []string
|
||
Clients [][]string
|
||
}
|
||
|
||
func NewClientComponent(loginUser *vo.User) *ClientComponent {
|
||
ccp := &ClientComponent{
|
||
LoginUser: loginUser,
|
||
Menu: []string{"[1] 客户端列表", "[2] 查看客户端配置", "[3] 添加客户端", "[4] 编辑客户端", "[5] 删除客户端", "[q] 返回上一级菜单"},
|
||
}
|
||
|
||
return ccp
|
||
}
|
||
|
||
// Menus
|
||
// @description: 客户端菜单
|
||
// @receiver c
|
||
func (c *ClientComponent) Menus() {
|
||
fmt.Println("")
|
||
for _, r := range c.Menu {
|
||
fmt.Println(" -> " + r)
|
||
}
|
||
|
||
chooseMenu := readInput("\n请选择: ")
|
||
switch chooseMenu {
|
||
case "1":
|
||
c.List(true)
|
||
case "2":
|
||
c.ShowConfig()
|
||
case "3":
|
||
c.Add()
|
||
case "4":
|
||
c.Edit()
|
||
case "5":
|
||
c.Delete()
|
||
case "q":
|
||
return
|
||
}
|
||
}
|
||
|
||
// ConnectList
|
||
// @description: 客户端链接列表
|
||
// @receiver c
|
||
// @return string
|
||
func (c *ClientComponent) ConnectList() {
|
||
fmt.Println("\n客户端链接列表")
|
||
connectList, err := component.Wireguard().GetClients()
|
||
if err != nil {
|
||
fmt.Println("获取客户端链接列表失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
var data [][]string
|
||
for _, peer := range connectList {
|
||
// 获取客户端链接信息
|
||
clientInfo, err := service.Client().GetByPublicKey(peer.PublicKey.String())
|
||
if err != nil {
|
||
continue
|
||
}
|
||
var ipAllocation string
|
||
for _, iaip := range peer.AllowedIPs {
|
||
ipAllocation += iaip.String() + ","
|
||
}
|
||
// 去除一下最右边的逗号
|
||
if len(ipAllocation) > 0 {
|
||
ipAllocation = strings.TrimRight(ipAllocation, ",")
|
||
}
|
||
var isOnline = "否"
|
||
if time.Since(peer.LastHandshakeTime).Minutes() < 3 {
|
||
isOnline = "是"
|
||
}
|
||
data = append(data, []string{clientInfo.Name,
|
||
clientInfo.Email,
|
||
ipAllocation,
|
||
isOnline,
|
||
utils.FlowCalculation().Parse(peer.TransmitBytes),
|
||
utils.FlowCalculation().Parse(peer.ReceiveBytes),
|
||
peer.Endpoint.String(),
|
||
peer.LastHandshakeTime.Format("2006-01-02 15:04:05")})
|
||
}
|
||
|
||
//if len(data) <= 0 {
|
||
// //data = append(data, []string{"暂无数据"})
|
||
// // data = append(data, []string{"名称1", "12345678910@qq.com", "192.168.100.1", "是", "10G", "20G", "1.14.30.133:51280", "2024-12-20 15:07:36"}, []string{"名称2", "12345678910@qq.com", "192.168.100.2", "否", "20G", "40G", "1.14.30.133:51280", "2024-12-22 15:07:36"})
|
||
//}
|
||
|
||
title := []table.Column{
|
||
{
|
||
Title: "客户端名称",
|
||
Width: 20,
|
||
}, {
|
||
Title: "联系邮箱",
|
||
Width: 20,
|
||
}, {
|
||
Title: "分配的IP",
|
||
Width: 30,
|
||
}, {
|
||
Title: "是否在线",
|
||
Width: 10,
|
||
}, {
|
||
Title: "接收流量",
|
||
Width: 10,
|
||
}, {
|
||
Title: "传输流量",
|
||
Width: 10,
|
||
}, {
|
||
Title: "链接端点",
|
||
Width: 30,
|
||
}, {
|
||
Title: "最后握手时间",
|
||
Width: 30,
|
||
}}
|
||
|
||
Show(GenerateTable(title, data))
|
||
return
|
||
}
|
||
|
||
// List
|
||
// @description: 客户端列表
|
||
// @receiver c
|
||
func (c *ClientComponent) List(showMenu bool) {
|
||
|
||
title := []table.Column{
|
||
{
|
||
Title: "序号",
|
||
Width: 5,
|
||
},
|
||
{
|
||
Title: "ID",
|
||
Width: 35,
|
||
},
|
||
{
|
||
Title: "名称",
|
||
Width: 25,
|
||
},
|
||
{
|
||
Title: "联系邮箱",
|
||
Width: 30,
|
||
},
|
||
{
|
||
Title: "客户端IP",
|
||
Width: 20,
|
||
},
|
||
{
|
||
Title: "状态",
|
||
Width: 10,
|
||
},
|
||
{
|
||
Title: "离线通知",
|
||
Width: 10,
|
||
},
|
||
}
|
||
|
||
records, _, err := service.Client().List(param.ClientList{
|
||
Page: param.Page{
|
||
Current: -1,
|
||
},
|
||
})
|
||
if err != nil {
|
||
fmt.Println("获取客户端列表失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
fmt.Println("\n客户端列表")
|
||
var data [][]string
|
||
for i, client := range records {
|
||
var status, offlineNotify string
|
||
if client.Enabled == 1 {
|
||
status = "启用"
|
||
} else {
|
||
status = "禁用"
|
||
}
|
||
|
||
if client.OfflineMonitoring == 1 {
|
||
offlineNotify = "开启"
|
||
} else {
|
||
offlineNotify = "关闭"
|
||
}
|
||
|
||
data = append(data, []string{
|
||
cast.ToString(i + 1),
|
||
client.Id,
|
||
client.Name,
|
||
client.Email,
|
||
client.IpAllocationStr,
|
||
status,
|
||
offlineNotify,
|
||
})
|
||
}
|
||
|
||
Show(GenerateTable(title, data))
|
||
c.Clients = data
|
||
if showMenu {
|
||
c.Menus()
|
||
}
|
||
return
|
||
}
|
||
|
||
// ShowConfig
|
||
// @description: 显示配置
|
||
// @receiver c
|
||
func (c *ClientComponent) ShowConfig() {
|
||
c.List(false)
|
||
|
||
clientIdx := readInput("请输入客户端序号: ")
|
||
downloadType := readInput("请输入下载类型(FILE - 文件 | EMAIL - 邮件): ")
|
||
idx, err := strconv.Atoi(clientIdx)
|
||
if err != nil {
|
||
fmt.Println("输入有误: ", err.Error())
|
||
return
|
||
}
|
||
|
||
if idx < 1 || idx > len(c.Clients) {
|
||
fmt.Println("输入有误")
|
||
return
|
||
}
|
||
|
||
client := c.Clients[idx-1]
|
||
// 取到id
|
||
clientID := client[1]
|
||
// 查询客户端信息
|
||
clientInfo, err := service.Client().GetByID(clientID)
|
||
if err != nil {
|
||
fmt.Println("获取客户端信息失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
// 渲染配置
|
||
var keys vo.Keys
|
||
_ = json.Unmarshal([]byte(clientInfo.Keys), &keys)
|
||
|
||
globalSet, err := service.Setting().GetWGSetForConfig()
|
||
if err != nil {
|
||
fmt.Println("获取全局配置失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
serverConf, err := service.Setting().GetWGServerForConfig()
|
||
if err != nil {
|
||
fmt.Println("获取服务器配置失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
outPath, err := component.Wireguard().GenerateClientFile(clientInfo, serverConf, globalSet)
|
||
if err != nil {
|
||
fmt.Println("生成客户端配置失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
// 根据不同下载类型执行不同逻辑
|
||
switch downloadType {
|
||
case "FILE": // 二维码
|
||
// 读取文件内容
|
||
fileContent, err := os.ReadFile(outPath)
|
||
if err != nil {
|
||
fmt.Println("读取文件失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
fmt.Println("\n#请将以下内容复制到客户端配置文件中【不包含本行提示语】")
|
||
fmt.Println("\n" + string(fileContent))
|
||
|
||
if err = os.Remove(outPath); err != nil {
|
||
log.Errorf("删除临时文件失败: %s", err.Error())
|
||
}
|
||
|
||
case "EMAIL": // 邮件
|
||
if clientInfo.Email == "" {
|
||
fmt.Println("当前客户端并未配置通知邮箱")
|
||
return
|
||
}
|
||
|
||
// 获取邮箱配置
|
||
emailConf, err := service.Setting().GetByCode("EMAIL_SMTP")
|
||
if err != nil {
|
||
fmt.Println("获取邮箱配置失败，请先到设置页面的【其他】里面添加code为【EMAIL_SMTP】的具体配置")
|
||
return
|
||
}
|
||
|
||
err = utils.Mail(emailConf).SendMail(clientInfo.Email, fmt.Sprintf("客户端: %s", clientInfo.Name), "请查收附件", outPath)
|
||
if err != nil {
|
||
fmt.Println("发送邮件失败")
|
||
return
|
||
}
|
||
|
||
if err = os.Remove(outPath); err != nil {
|
||
log.Errorf("删除临时文件失败: %s", err.Error())
|
||
}
|
||
|
||
fmt.Println("发送邮件成功，请注意查收！")
|
||
}
|
||
|
||
c.Menus()
|
||
}
|
||
|
||
// Add
|
||
// @description: 添加客户端
|
||
// @receiver c
|
||
func (c *ClientComponent) Add() {
|
||
fmt.Println("\n添加客户端")
|
||
clientIP, serverIP, err := GenerateIP()
|
||
if err != nil {
|
||
fmt.Println("生成客户端IP失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
keys, err := GenerateKeys()
|
||
if err != nil {
|
||
fmt.Println("生成密钥对失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
var p param.SaveClient
|
||
p.Name = readInput("请输入客户端名称: ")
|
||
p.Email = readInput("请输入联系邮箱: ")
|
||
clientIPIn := readInput("请输入客户端IP(默认自动生成，多个采用 ',' 分割,例如 10.10.0.1/32,10.10.0.2/32): ")
|
||
if clientIPIn == "" {
|
||
p.IpAllocation = clientIP
|
||
} else {
|
||
p.IpAllocation = strings.Split(clientIPIn, ",")
|
||
}
|
||
|
||
p.AllowedIps = serverIP
|
||
p.Keys = ¶m.Keys{
|
||
PrivateKey: keys.PrivateKey,
|
||
PublicKey: keys.PublicKey,
|
||
PresharedKey: keys.PresharedKey,
|
||
}
|
||
|
||
var useServerDNS, enabled, offlineNotify constant.Status
|
||
|
||
useServerDNSIn := readInput("是否使用服务器DNS(默认不使用 1 - 是 | 0 - 否 ): ")
|
||
switch useServerDNSIn {
|
||
case "1":
|
||
useServerDNS = constant.Enabled
|
||
case "0":
|
||
useServerDNS = constant.Disabled
|
||
default:
|
||
useServerDNS = constant.Disabled
|
||
}
|
||
|
||
enabledIn := readInput("是否启用(默认启用 1 - 是 | 0 - 否 ): ")
|
||
switch enabledIn {
|
||
case "1":
|
||
enabled = constant.Enabled
|
||
case "0":
|
||
enabled = constant.Disabled
|
||
default:
|
||
enabled = constant.Enabled
|
||
}
|
||
|
||
offlineNotifyIn := readInput("是否开启离线通知(默认关闭 1 - 是 | 0 - 否 ): ")
|
||
switch offlineNotifyIn {
|
||
case "1":
|
||
offlineNotify = constant.Enabled
|
||
case "0":
|
||
offlineNotify = constant.Disabled
|
||
default:
|
||
offlineNotify = constant.Disabled
|
||
}
|
||
p.UseServerDns = &useServerDNS
|
||
p.Enabled = &enabled
|
||
p.OfflineMonitoring = &offlineNotify
|
||
|
||
err = service.Client().SaveClient(p, c.LoginUser)
|
||
if err != nil {
|
||
fmt.Println("添加客户端失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
fmt.Println("添加客户端成功")
|
||
c.List(true)
|
||
return
|
||
}
|
||
|
||
// Edit
|
||
// @description: 编辑客户端
|
||
// @receiver c
|
||
func (c *ClientComponent) Edit() {
|
||
fmt.Println("\n编辑客户端")
|
||
c.List(false)
|
||
|
||
clientIdx := readInput("请输入客户端序号: ")
|
||
idx, err := strconv.Atoi(clientIdx)
|
||
if err != nil {
|
||
fmt.Println("输入有误: ", err.Error())
|
||
return
|
||
}
|
||
|
||
if idx < 1 || idx > len(c.Clients) {
|
||
fmt.Println("输入有误")
|
||
return
|
||
}
|
||
|
||
client := c.Clients[idx-1]
|
||
// 取到id
|
||
clientID := client[1]
|
||
// 查询客户端信息
|
||
clientInfo, err := service.Client().GetByID(clientID)
|
||
if err != nil {
|
||
fmt.Println("获取客户端信息失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
var p param.SaveClient
|
||
p.Id = clientID
|
||
p.Name = readInput("请输入客户端名称[无需改变请回车跳过，下同]: ")
|
||
p.Email = readInput("请输入联系邮箱: ")
|
||
clientIPIn := readInput("请输入客户端IP(默认自动生成，多个采用 ',' 分割,例如 10.10.0.1/32,10.10.0.2/32): ")
|
||
if clientIPIn == "" {
|
||
p.IpAllocation = strings.Split(clientInfo.IpAllocation, ",")
|
||
} else {
|
||
p.IpAllocation = strings.Split(clientIPIn, ",")
|
||
}
|
||
|
||
p.AllowedIps = strings.Split(clientInfo.AllowedIps, ",")
|
||
var keys *param.Keys
|
||
_ = json.Unmarshal([]byte(clientInfo.Keys), &keys)
|
||
|
||
p.Keys = keys
|
||
|
||
var useServerDNS, enabled, offlineNotify constant.Status
|
||
|
||
useServerDNSIn := readInput("是否使用服务器DNS(默认不使用 1 - 是 | 0 - 否 ): ")
|
||
switch useServerDNSIn {
|
||
case "1":
|
||
useServerDNS = constant.Enabled
|
||
case "0":
|
||
useServerDNS = constant.Disabled
|
||
default:
|
||
useServerDNS = clientInfo.UseServerDns
|
||
}
|
||
|
||
enabledIn := readInput("是否启用(默认启用 1 - 是 | 0 - 否 ): ")
|
||
switch enabledIn {
|
||
case "1":
|
||
enabled = constant.Enabled
|
||
case "0":
|
||
enabled = constant.Disabled
|
||
default:
|
||
enabled = clientInfo.Enabled
|
||
}
|
||
|
||
offlineNotifyIn := readInput("是否开启离线通知(默认关闭 1 - 是 | 0 - 否 ): ")
|
||
switch offlineNotifyIn {
|
||
case "1":
|
||
offlineNotify = constant.Enabled
|
||
case "0":
|
||
offlineNotify = constant.Disabled
|
||
default:
|
||
offlineNotify = clientInfo.OfflineMonitoring
|
||
}
|
||
p.UseServerDns = &useServerDNS
|
||
p.Enabled = &enabled
|
||
p.OfflineMonitoring = &offlineNotify
|
||
|
||
err = service.Client().SaveClient(p, c.LoginUser)
|
||
if err != nil {
|
||
fmt.Println("编辑客户端失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
fmt.Println("编辑客户端成功")
|
||
c.List(true)
|
||
return
|
||
}
|
||
|
||
// Delete
|
||
// @description: 删除客户端
|
||
// @receiver c
|
||
func (c *ClientComponent) Delete() {
|
||
fmt.Println("\n删除客户端")
|
||
|
||
c.List(false)
|
||
|
||
clientIdx := readInput("请输入客户端序号: ")
|
||
idx, err := strconv.Atoi(clientIdx)
|
||
if err != nil {
|
||
fmt.Println("输入有误: ", err.Error())
|
||
return
|
||
}
|
||
|
||
if idx < 1 || idx > len(c.Clients) {
|
||
fmt.Println("输入有误")
|
||
return
|
||
}
|
||
|
||
client := c.Clients[idx-1]
|
||
// 取到id
|
||
clientID := client[1]
|
||
if err := service.Client().Delete(clientID); err != nil {
|
||
fmt.Println("删除客户端失败: ", err.Error())
|
||
return
|
||
}
|
||
|
||
c.List(true)
|
||
fmt.Println("删除客户端成功")
|
||
return
|
||
}
|