🐛修复bug

Reviewed-on: #3

Dockerfile

@@ -1,9 +1,9 @@
 # 打包前端
-FROM node:18-alpine as build-front
+FROM node:18-alpine AS build-front
 
-WORKDIR front
+WORKDIR /front
 COPY . .
-WORKDIR web
+WORKDIR ./web
 
 RUN corepack enable
 RUN corepack prepare pnpm@8.6.10 --activate
@@ -13,7 +13,7 @@ RUN pnpm build
 RUN ls -lh && pwd
 
 # 前后端集成打包
-FROM golang:alpine as build-backend
+FROM golang:alpine AS build-backend
 
 RUN apk add upx
 WORKDIR /build
@@ -23,7 +23,7 @@ COPY --from=build-front /front/web/dist/ /build/dist
 ENV GO111MODULE=on
 ENV GOPROXY=https://goproxy.cn,direct
 
-RUN  go build -ldflags="-s -w" -o wgui && upx -9 wgui
+RUN go version && go build -ldflags="-s -w" -o wgui && upx -9 wgui
 
 RUN ls -lh && chmod +x ./wgui
 
@@ -43,4 +43,4 @@ COPY --from=build-backend /build/template/* /app/template/
 
 RUN chmod +x wgui
 
-ENTRYPOINT ["./wgui"]
+ENTRYPOINT ["./wgui","http:serve"]

README.md

@@ -34,14 +34,6 @@ file:
   accessSecret:                  # oss必填
   bucketName:                    # oss必填
 
-# 邮件设置
-mail:
-  host: 
-  port: 
-  user: 
-  password: 
-  skipTls: 
-
 # 一些系统配置
 wireguard:
   restartMode: DELAY
@@ -78,14 +70,25 @@ services:
 账户: admin
 密码: admin123
 ```
+## 配置示例
+```text
+  1. 邮箱配置如下：
+     code: EMAIL_SMTP
+     配置项：
+        1. host: "xxxx.xxx"
+        2. port: "123"
+        3. user: "haha"
+        4. password: "haha123"
+        5. skipTls: "false"
+```
 
-## 示例
-![img.png](img.png)
-![img_1.png](img_1.png)
-![img_7.png](img_7.png)
-![img_8.png](img_8.png)
-![img_2.png](img_2.png)
-![img_3.png](img_3.png)
-![img_4.png](img_4.png)
-![img_5.png](img_5.png)
-![img_6.png](img_6.png)
+## 页面展示
+![img.png](document/img.png)
+![img_1.png](document/img_1.png)
+![img_7.png](document/img_7.png)
+![img_8.png](document/img_8.png)
+![img_2.png](document/img_2.png)
+![img_3.png](document/img_3.png)
+![img_4.png](document/img_4.png)
+![img_5.png](document/img_5.png)
+![img_6.png](document/img_6.png)

cli/kernal.go

@@ -0,0 +1,10 @@
+package cli
+
+import (
+	"wireguard-ui/cli/tui"
+)
+
+func Kernel() error {
+	tui.NewApp().Run()
+	return nil
+}

cli/tui/app.go

@@ -0,0 +1,165 @@
+package tui
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/google/uuid"
+	"time"
+	"wireguard-ui/component"
+	"wireguard-ui/global/client"
+	"wireguard-ui/global/constant"
+	"wireguard-ui/http/vo"
+	"wireguard-ui/service"
+	"wireguard-ui/utils"
+)
+
+type App struct {
+	TokenSecret string            // token密钥
+	User        *vo.User          // 登陆用户
+	Client      *ClientComponent  // 客户端组件
+	Server      *ServerComponent  // 服务端组件
+	Setting     *SettingComponent // 设置组件
+}
+
+func NewApp() *App {
+	app := &App{}
+	if _, err := app.Login(); err != nil {
+		fmt.Println("登陆失败: ", err)
+		return nil
+	}
+	err := app.AuthLogin()
+	if err != nil {
+		fmt.Println("登陆失败: ", err)
+		return nil
+	}
+	fmt.Println("\n=============== 登陆成功 ==============================================================================")
+	app.Client = NewClientComponent(app.User)
+	app.Server = NewServerComponent(app.User)
+	app.Setting = NewSettingComponent(app.User)
+	fmt.Println("=============== 欢迎使用wireguard-tui =================================================================")
+	fmt.Println("=============== 当前用户: ", app.User.Nickname, " ================================================================")
+	fmt.Println("=============== 当前时间: ", time.Now().Format("2006-01-02 15:04:05"), " =======================================================")
+	fmt.Println("=============== 注意事项如下: =========================================================================")
+	fmt.Println("=============== 1. 请确保服务端已经安装wireguard ======================================================")
+	fmt.Println("=============== 2. 请确保服务端和客户端配置文件路径正确 ===============================================")
+	fmt.Println("=============== 3. 请确保服务端和客户端配置文件权限正确 ===============================================")
+	fmt.Println("=============== 4. 请确保服务端和客户端配置文件内容正确 ===============================================")
+	fmt.Println("=============== 5. 请勿泄露配置文件内容 ===============================================================")
+	fmt.Println("=============== 6. 每次修改客户端、服务端配置或者全局配置过后，请使用重启功能重启服务端，以保证生效 ===")
+	fmt.Println("=============== 7. 当使用重启无效时，请手动执行对应命令 ===============================================")
+	fmt.Println("=============== 8. 请勿随意删除客户端，删除后无法恢复 =================================================")
+	fmt.Println("=============== 9. 手动命令 ===========================================================================")
+	fmt.Println("=============== 10. 启动wireguard服务端: wg-quick up wg0 ==============================================")
+	fmt.Println("=============== 11. 停止wireguard服务端: wg-quick down wg0 ============================================")
+
+	return app
+}
+
+func (a *App) Run() {
+	if a == nil {
+		return
+	}
+	for {
+
+		PrintMenu()
+		chooseMenu := readInput("请选择菜单: ")
+		switch chooseMenu {
+		case "1":
+			a.Client.ConnectList()
+		case "2":
+			a.Client.Menus()
+		case "3":
+			a.Server.Menus()
+		case "4":
+			a.Setting.Menus()
+		case "q":
+			return
+		}
+	}
+}
+
+// AuthLogin
+// @description: 登陆认证
+// @receiver a
+// @return string
+func (a *App) AuthLogin() error {
+	// 先判断token是否存在
+	tokenStr, err := client.Redis.Get(context.Background(), fmt.Sprintf("%s:%s", constant.TUIUserToken, a.User.Id)).Result()
+	if err != nil {
+		// 不存在，去登陆
+		tokenStr, err = a.Login()
+		if err != nil {
+			return err
+		}
+	}
+
+	// 存在，不必要再次登陆,解析token
+	claims, err := component.JWT().ParseToken("Bearer "+tokenStr, a.TokenSecret, "tui")
+	if err != nil {
+		return err
+	}
+
+	user, err := service.User().GetUserById(claims.ID)
+	if err != nil {
+		return err
+	}
+
+	if user.Status != constant.Enabled {
+		return errors.New("用户状态异常，请联系管理员处理")
+	}
+
+	a.User = &vo.User{
+		Id:       user.Id,
+		Account:  user.Account,
+		Nickname: user.Nickname,
+		Avatar:   user.Avatar,
+		Contact:  user.Contact,
+		IsAdmin:  user.IsAdmin,
+		Status:   user.Status,
+	}
+
+	return nil
+}
+
+// Login
+// @description: 登陆
+// @receiver a
+// @return string
+func (a *App) Login() (tokenStr string, err error) {
+	fmt.Println("============== 登陆 ==============")
+
+	username := readInput("请输入用户名: ")
+	password := readInput("请输入密码: ")
+	// 验证码正确，查询用户信息
+	user, err := service.User().GetUserByAccount(username)
+	if err != nil {
+		return "", fmt.Errorf("用户不存在: %v", err)
+	}
+
+	// 对比密码
+	if !utils.Password().ComparePassword(user.Password, password) {
+		return "", errors.New("密码错误")
+	}
+
+	secret := component.JWT().GenerateSecret(password, uuid.NewString(), time.Now().Local().String())
+	// 生成token
+	token, _, err := component.JWT().GenerateToken(user.Id, secret, "tui")
+	if err != nil {
+		return "", fmt.Errorf("登陆失败: %v", err)
+	}
+
+	a.User = &vo.User{
+		Id:       user.Id,
+		Account:  user.Account,
+		Nickname: user.Nickname,
+		Avatar:   user.Avatar,
+		Contact:  user.Contact,
+		IsAdmin:  user.IsAdmin,
+		Status:   user.Status,
+	}
+
+	a.TokenSecret = secret
+
+	return "Bearer " + token, nil
+}

cli/tui/client.go

@@ -0,0 +1,520 @@
+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 = &param.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
+}

cli/tui/server.go

@@ -0,0 +1,62 @@
+package tui
+
+import (
+	"fmt"
+	"wireguard-ui/command"
+	"wireguard-ui/http/vo"
+)
+
+type ServerComponent struct {
+	LoginUser *vo.User
+	Menu      []string
+}
+
+func NewServerComponent(loginUser *vo.User) *ServerComponent {
+	return &ServerComponent{
+		LoginUser: loginUser,
+		Menu:      []string{"[1] 启动服务", "[2] 关闭服务", "[3] 重启服务", "[q] 返回上一级菜单"},
+	}
+}
+
+// Menus
+// @description: 服务端菜单
+// @receiver s
+func (s *ServerComponent) Menus() {
+	fmt.Println("")
+	for _, r := range s.Menu {
+		fmt.Println(" -> " + r)
+	}
+
+	chooseMenu := readInput("\n请选择: ")
+	switch chooseMenu {
+	case "1":
+		s.Start()
+	case "2":
+		s.Stop()
+	case "3":
+		s.Restart()
+	case "q":
+		return
+	}
+}
+
+// Start
+// @description: 启动服务
+// @receiver s
+func (s *ServerComponent) Start() {
+	command.StartWireguard("")
+}
+
+// Stop
+// @description: 停止服务
+// @receiver s
+func (s *ServerComponent) Stop() {
+	command.StopWireguard("")
+}
+
+// Restart
+// @description: 重启服务
+// @receiver s
+func (s *ServerComponent) Restart() {
+	command.RestartWireguard(false, "")
+}

cli/tui/setting.go

@@ -0,0 +1,440 @@
+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)
+
+}

cli/tui/utils.go

@@ -0,0 +1,136 @@
+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()
+}

component/jwt.go

@@ -36,7 +36,7 @@ func JWT() JwtComponent {
 // @return token
 // @return expireTime
 // @return err
-func (JwtComponent) GenerateToken(userId, secret string, times ...time.Time) (token string, expireTime *jwt.NumericDate, err error) {
+func (JwtComponent) GenerateToken(userId, secret, source string, times ...time.Time) (token string, expireTime *jwt.NumericDate, err error) {
 	var notBefore, issuedAt *jwt.NumericDate
 	if len(times) != 0 {
 		expireTime = jwt.NewNumericDate(times[0])
@@ -68,10 +68,19 @@ func (JwtComponent) GenerateToken(userId, secret string, times ...time.Time) (to
 		return "", nil, errors.New("生成token失败")
 	}
 
+	switch source {
+	case "http":
 		client.Redis.Set(context.Background(),
 			fmt.Sprintf("%s:%s", constant.UserToken, userId),
 			token,
 			time.Duration(expireTime.Sub(time.Now()).Abs().Seconds())*time.Second)
+	case "tui":
+		client.Redis.Set(context.Background(),
+			fmt.Sprintf("%s:%s", constant.TUIUserToken, userId),
+			token,
+			time.Duration(expireTime.Sub(time.Now()).Abs().Seconds())*time.Second)
+	}
+
 	return
 }
 
@@ -81,7 +90,7 @@ func (JwtComponent) GenerateToken(userId, secret string, times ...time.Time) (to
 // @param token
 // @return *JwtComponent
 // @return error
-func (JwtComponent) ParseToken(token, secret string) (*JwtComponent, error) {
+func (JwtComponent) ParseToken(token, secret, source string) (*JwtComponent, error) {
 	tokenStr := strings.Split(token, "Bearer ")[1]
 
 	t, err := jwt.ParseWithClaims(tokenStr, &JwtComponent{}, func(token *jwt.Token) (any, error) {
@@ -89,11 +98,21 @@ func (JwtComponent) ParseToken(token, secret string) (*JwtComponent, error) {
 	})
 
 	if claims, ok := t.Claims.(*JwtComponent); ok && t.Valid {
-		userToken, err := client.Redis.Get(context.Background(), fmt.Sprintf("%s:%s", constant.UserToken, claims.ID)).Result()
+		var userToken string
+		switch source {
+		case "http":
+			userToken, err = client.Redis.Get(context.Background(), fmt.Sprintf("%s:%s", constant.UserToken, claims.ID)).Result()
 			if err != nil {
 				log.Errorf("缓存中用户[%s]的token查找失败: %v", claims.ID, err.Error())
 				return nil, errors.New("token不存在")
 			}
+		case "tui":
+			userToken, err = client.Redis.Get(context.Background(), fmt.Sprintf("%s:%s", constant.TUIUserToken, claims.ID)).Result()
+			if err != nil {
+				log.Errorf("缓存中用户[%s]的token查找失败: %v", claims.ID, err.Error())
+				return nil, errors.New("token不存在")
+			}
+		}
 
 		if userToken != tokenStr {
 			log.Errorf("token不一致")

config/config.go

@@ -7,6 +7,5 @@ type config struct {
 	Database  *database  `yaml:"database"`
 	Cache     *cache     `yaml:"redis"`
 	File      *file      `yaml:"file"`
-	Mail      *mail      `yaml:"email"`
 	Wireguard *wireguard `yaml:"wireguard"`
 }

config/mail.go

@@ -1,9 +0,0 @@
-package config
-
-type mail struct {
-	Host     string `json:"host" yaml:"host"`
-	Port     int    `json:"port" yaml:"port"`
-	User     string `json:"user" yaml:"user"`
-	Password string `json:"password" yaml:"password"`
-	SkipTls  bool   `json:"skipTls" yaml:"skipTls"`
-}

cron/cron.go

@@ -1 +1,24 @@
 package cron
+
+import (
+	"gitee.ltd/lxh/logger/log"
+	"github.com/go-co-op/gocron/v2"
+	"github.com/spf13/cast"
+	"os"
+	"time"
+	"wireguard-ui/cron/task"
+)
+
+func Task() {
+	if !cast.ToBool(os.Getenv("ENABLED_CRON")) {
+		return
+	}
+	sch, err := gocron.NewScheduler(gocron.WithLocation(time.Local))
+	if err != nil {
+		log.Errorf("初始化定时任务失败")
+		return
+	}
+
+	_, _ = sch.NewJob(gocron.DurationJob(1*time.Minute), gocron.NewTask(task.NetworkClient().ClientOfflineNotify)) // 每分钟执行一次
+	sch.Start()
+}

cron/task/client.go

@@ -0,0 +1,119 @@
+package task
+
+import (
+	"context"
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	jsoniter "github.com/json-iterator/go"
+	"golang.org/x/exp/slices"
+	"strings"
+	"time"
+	"wireguard-ui/component"
+	"wireguard-ui/global/client"
+	"wireguard-ui/global/constant"
+	"wireguard-ui/model"
+	"wireguard-ui/service"
+	"wireguard-ui/utils"
+)
+
+type NetworkClientImpl interface {
+	ClientOfflineNotify() // 客户端离线通知
+}
+
+type networkClient struct{}
+
+func NetworkClient() NetworkClientImpl {
+	return networkClient{}
+}
+
+// ClientOfflineNotify
+// @description: 客户端离线通知
+// @receiver c
+// @return error
+func (c networkClient) ClientOfflineNotify() {
+	log.Debugf("开始执行离线通知任务")
+	// 开始扫描已经链接过的客户端
+	connectedPeers, err := component.Wireguard().GetClients()
+	if err != nil {
+		log.Errorf("获取已连接客户端失败: %v", err.Error())
+		return
+	}
+
+	// 查询一下通知配置
+	code, err := service.Setting().GetByCode("WECHAT_NOTIFY")
+	if err != nil {
+		log.Errorf("获取微信通知配置失败: %v", err.Error())
+		return
+	}
+
+	// 查询出所有配置了离线通知的客户端
+	var clients []model.Client
+	if err := client.DB.Where("offline_monitoring = ?", 1).Find(&clients).Error; err != nil {
+		return
+	}
+
+	if len(clients) <= 0 {
+		return
+	}
+
+	for _, peer := range connectedPeers {
+		var clientName string
+		if !slices.ContainsFunc(clients, func(cli model.Client) bool {
+			isExist := peer.PublicKey.String() == jsoniter.Get([]byte(cli.Keys), "publicKey").ToString()
+			if isExist {
+				clientName = cli.Name
+			}
+			return isExist
+		}) {
+			continue
+		}
+
+		online := time.Since(peer.LastHandshakeTime).Minutes() < 3
+		log.Debugf("客户端[%v]在线状态: %v，离线时间: %v", clientName, online, time.Since(peer.LastHandshakeTime).Minutes())
+
+		var ipAllocation string
+		for _, iaip := range peer.AllowedIPs {
+			ipAllocation += iaip.String() + ","
+		}
+		// 去除一下最右边的逗号
+		if len(ipAllocation) > 0 {
+			ipAllocation = strings.TrimRight(ipAllocation, ",")
+		}
+
+		// 如果存在,判断离线时间
+		if !online {
+			// 已经离线,发送通知
+			msg := fmt.Sprintf(`[离线通知]
+客户端名称 : %v
+客户端IP : %v
+最后在线时间 : %v`, clientName, ipAllocation, peer.LastHandshakeTime.Format("2006-01-02 15:04:05"))
+			err := utils.WechatNotify(code).SendTextMessage(msg)
+			if err != nil {
+				log.Errorf("微信消息[%v]通知失败: %v", msg, err.Error())
+				continue
+			}
+
+			// 离线了,设置离线标识
+			client.Redis.Set(context.Background(), fmt.Sprintf("%s%s", constant.ClientOffline, utils.Hash().MD5(ipAllocation)), true, 0)
+		} else {
+			// 判断是否存在缓存
+			if client.Redis.Exists(context.Background(), fmt.Sprintf("%s%s", constant.ClientOffline, utils.Hash().MD5(ipAllocation))).Val() > 0 {
+				// 存在,删除离线标识
+				client.Redis.Del(context.Background(), fmt.Sprintf("%s%s", constant.ClientOffline, utils.Hash().MD5(ipAllocation)))
+				// 微信通知该客户端已经上线
+				msg := fmt.Sprintf(`[上线通知]
+客户端名称 : %v
+客户端IP : %v
+最后在线时间 : %v`, clientName, ipAllocation, peer.LastHandshakeTime.Format("2006-01-02 15:04:05"))
+				err := utils.WechatNotify(code).SendTextMessage(msg)
+				if err != nil {
+					log.Errorf("微信消息[%v]通知失败: %v", msg, err.Error())
+					continue
+				}
+			}
+		}
+
+	}
+
+	return
+}

global/constant/redis_key.go

@@ -3,4 +3,6 @@ package constant
 const (
 	Captcha       = "captcha"
 	UserToken     = "token"
+	TUIUserToken  = "tui:token"
+	ClientOffline = "client:offline:"
 )

go.mod

@@ -1,53 +1,76 @@
 module wireguard-ui
 
-go 1.21
+go 1.22.7
+
+toolchain go1.23.0
 
 require (
-	gitee.ltd/lxh/logger v1.0.15
+	gitee.ltd/lxh/logger v1.0.19
+	github.com/charmbracelet/bubbles v0.20.0
+	github.com/charmbracelet/lipgloss v0.13.0
 	github.com/cowardmrx/go_aliyun_oss v1.0.7
-	github.com/fsnotify/fsnotify v1.7.0
-	github.com/gin-contrib/pprof v1.5.0
+	github.com/dustin/go-humanize v1.0.1
+	github.com/eiannone/keyboard v0.0.0-20220611211555-0d226195f203
+	github.com/fsnotify/fsnotify v1.8.0
+	github.com/gin-contrib/pprof v1.5.2
 	github.com/gin-gonic/gin v1.10.0
 	github.com/glebarez/sqlite v1.11.0
-	github.com/go-resty/resty/v2 v2.13.1
-	github.com/redis/go-redis/v9 v9.5.3
+	github.com/go-co-op/gocron/v2 v2.12.4
+	github.com/go-playground/locales v0.14.1
+	github.com/go-playground/universal-translator v0.18.1
+	github.com/go-playground/validator/v10 v10.23.0
+	github.com/go-resty/resty/v2 v2.15.3
+	github.com/golang-jwt/jwt/v5 v5.2.1
+	github.com/google/uuid v1.6.0
+	github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible
+	github.com/json-iterator/go v1.1.12
+	github.com/mojocn/base64Captcha v1.3.6
+	github.com/redis/go-redis/v9 v9.7.0
+	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e
+	github.com/spf13/cast v1.6.0
 	github.com/spf13/viper v1.19.0
+	github.com/urfave/cli/v2 v2.27.5
+	golang.org/x/crypto v0.31.0
+	golang.org/x/exp v0.0.0-20240613232115-7f521ea00fb8
 	golang.zx2c4.com/wireguard/wgctrl v0.0.0-20230429144221-925a1e7659e6
+	gopkg.in/fsnotify/fsnotify.v1 v1.4.7
 	gorm.io/driver/mysql v1.5.7
-	gorm.io/driver/postgres v1.5.9
+	gorm.io/driver/postgres v1.5.11
 	gorm.io/gorm v1.25.10
 )
 
 require (
 	github.com/aliyun/aliyun-oss-go-sdk v2.2.5+incompatible // indirect
+	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
-	github.com/bytedance/sonic v1.11.6 // indirect
-	github.com/bytedance/sonic/loader v0.1.1 // indirect
+	github.com/bytedance/sonic v1.12.9 // indirect
+	github.com/bytedance/sonic/loader v0.2.2 // indirect
 	github.com/caarlos0/env/v6 v6.10.1 // indirect
-	github.com/cespare/xxhash/v2 v2.2.0 // indirect
-	github.com/cloudwego/base64x v0.1.4 // indirect
-	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/cespare/xxhash/v2 v2.3.0 // indirect
+	github.com/charmbracelet/bubbletea v1.1.0 // indirect
+	github.com/charmbracelet/x/ansi v0.2.3 // indirect
+	github.com/charmbracelet/x/term v0.2.0 // indirect
+	github.com/cloudwego/base64x v0.1.5 // indirect
+	github.com/cpuguy83/go-md2man/v2 v2.0.5 // indirect
+	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
+	github.com/dennwc/varint v1.0.0 // indirect
 	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
-	github.com/dustin/go-humanize v1.0.1 // indirect
-	github.com/gabriel-vasile/mimetype v1.4.3 // indirect
-	github.com/gin-contrib/cors v1.7.2 // indirect
+	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
+	github.com/gabriel-vasile/mimetype v1.4.7 // indirect
 	github.com/gin-contrib/sse v0.1.0 // indirect
 	github.com/glebarez/go-sqlite v1.21.2 // indirect
 	github.com/go-kit/kit v0.12.0 // indirect
 	github.com/go-kit/log v0.2.1 // indirect
 	github.com/go-logfmt/logfmt v0.5.1 // indirect
-	github.com/go-playground/locales v0.14.1 // indirect
-	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.22.0 // indirect
 	github.com/go-sql-driver/mysql v1.7.0 // indirect
-	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/goccy/go-json v0.10.4 // indirect
 	github.com/gogo/protobuf v1.3.2 // indirect
-	github.com/golang-jwt/jwt/v5 v5.2.1 // indirect
 	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
-	github.com/golang/protobuf v1.5.3 // indirect
 	github.com/golang/snappy v0.0.4 // indirect
 	github.com/google/go-cmp v0.6.0 // indirect
-	github.com/google/uuid v1.6.0 // indirect
+	github.com/grafana/loki-client-go v0.0.0-20240913122146-e119d400c3a5 // indirect
+	github.com/grafana/loki/pkg/push v0.0.0-20240912152814-63e84b476a9a // indirect
+	github.com/grafana/regexp v0.0.0-20220304095617-2e8d9baf4ac2 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
@@ -55,63 +78,71 @@ require (
 	github.com/jackc/puddle/v2 v2.2.1 // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
-	github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
+	github.com/jonboulle/clockwork v0.4.0 // indirect
 	github.com/josharian/native v1.1.0 // indirect
 	github.com/jpillora/backoff v1.0.0 // indirect
-	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/klauspost/cpuid/v2 v2.2.7 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.9 // indirect
+	github.com/kylelemons/godebug v1.1.0 // indirect
 	github.com/leodido/go-urn v1.4.0 // indirect
-	github.com/lixh00/loki-client-go v1.0.1 // indirect
+	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/magiconair/properties v1.8.7 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
-	github.com/matttproud/golang_protobuf_extensions v1.0.2 // indirect
+	github.com/mattn/go-localereader v0.0.1 // indirect
+	github.com/mattn/go-runewidth v0.0.16 // indirect
 	github.com/mdlayher/genetlink v1.3.2 // indirect
 	github.com/mdlayher/netlink v1.7.2 // indirect
 	github.com/mdlayher/socket v0.4.1 // indirect
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/mojocn/base64Captcha v1.3.6 // indirect
+	github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
+	github.com/muesli/cancelreader v0.2.2 // indirect
+	github.com/muesli/termenv v0.15.2 // indirect
+	github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
 	github.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f // indirect
 	github.com/natefinch/lumberjack v2.0.0+incompatible // indirect
-	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/panjf2000/ants/v2 v2.10.0 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
-	github.com/prometheus/client_golang v1.13.0 // indirect
-	github.com/prometheus/client_model v0.3.0 // indirect
-	github.com/prometheus/common v0.37.0 // indirect
-	github.com/prometheus/procfs v0.8.0 // indirect
-	github.com/prometheus/prometheus v1.8.2-0.20201028100903-3245b3267b24 // indirect
+	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
+	github.com/prometheus/client_golang v1.20.5 // indirect
+	github.com/prometheus/client_model v0.6.1 // indirect
+	github.com/prometheus/common v0.61.0 // indirect
+	github.com/prometheus/procfs v0.15.1 // indirect
+	github.com/prometheus/prometheus v0.35.0 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+	github.com/rivo/uniseg v0.4.7 // indirect
+	github.com/robfig/cron/v3 v3.0.1 // indirect
+	github.com/russross/blackfriday/v2 v2.1.0 // indirect
 	github.com/sagikazarmark/locafero v0.4.0 // indirect
 	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
-	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e // indirect
 	github.com/sourcegraph/conc v0.3.0 // indirect
 	github.com/spf13/afero v1.11.0 // indirect
-	github.com/spf13/cast v1.6.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/stretchr/testify v1.10.0 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.2.12 // indirect
-	go.uber.org/atomic v1.10.0 // indirect
-	go.uber.org/multierr v1.9.0 // indirect
+	github.com/xrash/smetrics v0.0.0-20240521201337-686a1a2994c1 // indirect
+	go.uber.org/atomic v1.11.0 // indirect
+	go.uber.org/goleak v1.3.0 // indirect
+	go.uber.org/multierr v1.11.0 // indirect
 	go.uber.org/zap v1.23.0 // indirect
-	golang.org/x/arch v0.8.0 // indirect
-	golang.org/x/crypto v0.23.0 // indirect
-	golang.org/x/exp v0.0.0-20230905200255-921286631fa9 // indirect
-	golang.org/x/image v0.13.0 // indirect
-	golang.org/x/net v0.25.0 // indirect
-	golang.org/x/oauth2 v0.18.0 // indirect
-	golang.org/x/sync v0.6.0 // indirect
-	golang.org/x/sys v0.20.0 // indirect
-	golang.org/x/text v0.15.0 // indirect
-	golang.org/x/time v0.5.0 // indirect
+	golang.org/x/arch v0.12.0 // indirect
+	golang.org/x/image v0.18.0 // indirect
+	golang.org/x/net v0.33.0 // indirect
+	golang.org/x/oauth2 v0.24.0 // indirect
+	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/sys v0.28.0 // indirect
+	golang.org/x/text v0.21.0 // indirect
+	golang.org/x/time v0.8.0 // indirect
 	golang.zx2c4.com/wireguard v0.0.0-20230325221338-052af4a8072b // indirect
-	google.golang.org/appengine v1.6.8 // indirect
-	google.golang.org/genproto/googleapis/rpc v0.0.0-20240314234333-6e1732d8331c // indirect
-	google.golang.org/grpc v1.62.1 // indirect
-	google.golang.org/protobuf v1.34.1 // indirect
-	gopkg.in/fsnotify/fsnotify.v1 v1.4.7 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20241209162323-e6fa225c2576 // indirect
+	google.golang.org/grpc v1.69.0 // indirect
+	google.golang.org/protobuf v1.36.5 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
+	gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 	modernc.org/libc v1.22.5 // indirect

http/api/client.go

@@ -243,7 +243,14 @@ func (ClientApi) Download(c *gin.Context) {
 			return
 		}
 
-		err = utils.Mail().SendMail(data.Email, fmt.Sprintf("客户端: %s", data.Name), "请查收附件", outPath)
+		// 获取邮箱配置
+		emailConf, err := service.Setting().GetByCode("EMAIL_SMTP")
+		if err != nil {
+			response.R(c).FailedWithError("获取邮箱配置失败，请先到设置页面的【其他】里面添加code为【EMAIL_SMTP】的具体配置")
+			return
+		}
+
+		err = utils.Mail(emailConf).SendMail(data.Email, fmt.Sprintf("客户端: %s", data.Name), "请查收附件", outPath)
 		if err != nil {
 			response.R(c).FailedWithError("发送邮件失败")
 			return

http/api/login.go

@@ -76,7 +76,7 @@ func (LoginApi) Login(c *gin.Context) {
 
 	secret := component.JWT().GenerateSecret(p.Password, uuid.NewString(), time.Now().Local().String())
 	// 生成token
-	token, expireAt, err := component.JWT().GenerateToken(user.Id, secret)
+	token, expireAt, err := component.JWT().GenerateToken(user.Id, secret, "http")
 	if err != nil {
 		log.Errorf("用户[%s]生成token失败: %v", user.Account, err.Error())
 		response.R(c).FailedWithError("登陆失败！")

http/api/setting.go

@@ -67,7 +67,7 @@ func (setting) Delete(c *gin.Context) {
 		return
 	}
 
-	if err := service.Setting().Model(&model.Setting{}).Where("code = ?", code).Delete(&model.Setting{}).Error; err != nil {
+	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 {
 		response.R(c).FailedWithError("删除失败")
 		return
 	}

http/kernel.go

@@ -0,0 +1,33 @@
+package http
+
+import (
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"github.com/gin-contrib/pprof"
+	"github.com/spf13/cast"
+	"net/http"
+	"os"
+	"wireguard-ui/config"
+	"wireguard-ui/http/router"
+)
+
+// Kernel
+// @description: http启动
+// @return error
+func Kernel() error {
+	router.Rooters()
+	handler := router.InitRouter()
+	addr := fmt.Sprintf(":%d", config.Config.Http.Port)
+
+	if cast.ToBool(os.Getenv("ENABLED_PPROF")) {
+		pprof.Register(handler, "/monitoring")
+	}
+
+	httpServer := http.Server{
+		Addr:    addr,
+		Handler: handler,
+	}
+
+	log.Infof("[HTTP] server runing in %s", addr)
+	return httpServer.ListenAndServe()
+}

http/middleware/authorization.go

@@ -33,7 +33,7 @@ func Authorization() gin.HandlerFunc {
 			return
 		}
 
-		userClaims, err := component.JWT().ParseToken(token, hashPassword)
+		userClaims, err := component.JWT().ParseToken(token, hashPassword, "http")
 		if err != nil {
 			response.R(c).AuthorizationFailed("未登陆")
 			c.Abort()
@@ -70,7 +70,7 @@ func Authorization() gin.HandlerFunc {
 
 		// 生成一个新token
 		secret := component.JWT().GenerateSecret(user.Password, uuid.NewString(), time.Now().Local().String())
-		tokenStr, _, err := component.JWT().GenerateToken(user.Id, secret, userClaims.ExpiresAt.Time, time.Now().Local())
+		tokenStr, _, err := component.JWT().GenerateToken(user.Id, secret, "http", userClaims.ExpiresAt.Time, time.Now().Local())
 		if err != nil {
 			response.R(c).AuthorizationFailed("校验失败")
 			c.Abort()

http/router/client.go

@@ -10,7 +10,7 @@ import (
 // @description: 登陆相关API
 // @param r
 func ClientApi(r *gin.RouterGroup) {
-	client := r.Group("client", middleware.Authorization(), middleware.RequestLog())
+	client := r.Group("client", middleware.RequestLog(), middleware.Authorization())
 	{
 		client.POST("", api.Client().Save)                       // 新增/编辑客户端
 		client.DELETE("/:id", api.Client().Delete)               // 删除客户端

http/router/dashboard.go

@@ -10,7 +10,7 @@ import (
 // @description: 控制台相关接口
 // @param r
 func DashboardApi(r *gin.RouterGroup) {
-	dashboard := r.Group("dashboard", middleware.Authorization(), middleware.RequestLog())
+	dashboard := r.Group("dashboard", middleware.RequestLog(), middleware.Authorization())
 	{
 		dashboard.GET("/request/list", api.Dashboard().List)          // 请求日志
 		dashboard.GET("/daily-poetry", api.Dashboard().DailyPoetry)   // 每日诗词

http/router/setting.go

@@ -10,7 +10,7 @@ import (
 // @description: 设置相关API
 // @param r
 func SettingApi(r *gin.RouterGroup) {
-	setting := r.Group("setting", middleware.Authorization(), middleware.RequestLog())
+	setting := r.Group("setting", middleware.RequestLog(), middleware.Authorization())
 	{
 		setting.POST("", api.Setting().Set)                      // 新增/编辑设置
 		setting.DELETE("/:code", api.Setting().Delete)           // 删除配置

http/router/user.go

@@ -10,7 +10,7 @@ import (
 // @description: 用户相关API
 // @param r
 func UserApi(r *gin.RouterGroup) {
-	userApi := r.Group("user", middleware.Authorization(), middleware.RequestLog())
+	userApi := r.Group("user", middleware.RequestLog(), middleware.Authorization())
 	{
 		userApi.GET("/info", api.User().GetLoginUser)                // 获取当前登陆用户信息
 		userApi.POST("", api.User().SaveUser)                        // 新增/编辑用户

http/vo/setting.go

@@ -1,6 +1,9 @@
 package vo
 
-import "wireguard-ui/global/constant"
+import (
+	"wireguard-ui/global/constant"
+	"wireguard-ui/model"
+)
 
 // SettingItem
 // @description: 设置单项
@@ -8,14 +11,19 @@ type SettingItem struct {
 	Code      string         `json:"code"`
 	Data      string         `json:"data"`
 	Describe  string         `json:"describe"`
+	CreatedAt model.JsonTime `json:"createdAt"`
+	UpdatedAt model.JsonTime `json:"updatedAt"`
 }
 
 type Export struct {
 	Global  *Global  `json:"global" label:"全局配置" binding:"required"`
 	Server  *Server  `json:"server" label:"服务端配置" binding:"required"`
 	Clients []Client `json:"clients" label:"客户端" binding:"omitempty"`
+	Other   []Other  `json:"other" label:"其他" binding:"omitempty"`
 }
 
+// Global
+// @description: 全局配置
 type Global struct {
 	MTU                 int      `json:"MTU" label:"MTU" binding:"required"`
 	ConfigFilePath      string   `json:"configFilePath" label:"配置文件路径" binding:"required"`
@@ -26,6 +34,8 @@ type Global struct {
 	Table               string   `json:"table" label:"table" binding:"omitempty"`
 }
 
+// Server
+// @description: 服务端信息
 type Server struct {
 	IpScope        []string `json:"ipScope" label:"ipScope" binding:"min=1,dive,required"`
 	ListenPort     int      `json:"listenPort" label:"listenPort" binding:"required"`
@@ -35,6 +45,8 @@ type Server struct {
 	PostDownScript string   `json:"postDownScript" label:"postDownScript" binding:"omitempty"`
 }
 
+// Client
+// @description: 客户端信息
 type Client struct {
 	Name            string           `json:"name" label:"name" binding:"required"`
 	Email           string           `json:"email" label:"email" binding:"omitempty"`
@@ -52,3 +64,11 @@ type Client struct {
 	Enabled           *constant.Status `json:"enabled" label:"enabled" binding:"required"`
 	OfflineMonitoring *constant.Status `json:"offlineMonitoring" label:"offlineMonitoring" binding:"required"`
 }
+
+// Other
+// @description: 其他配置
+type Other struct {
+	Code     string `json:"code" label:"code" binding:"required"`
+	Data     string `json:"data" label:"data" binding:"required"`
+	Describe string `json:"describe" label:"describe" binding:"omitempty"`
+}

initialize/init.go

@@ -6,8 +6,10 @@ import (
 	"gitee.ltd/lxh/logger/log"
 	"github.com/cowardmrx/go_aliyun_oss"
 	"github.com/fsnotify/fsnotify"
+	"github.com/gin-gonic/gin"
 	"github.com/glebarez/sqlite"
 	"github.com/go-resty/resty/v2"
+	"github.com/joho/godotenv"
 	"github.com/redis/go-redis/v9"
 	"github.com/spf13/viper"
 	"golang.zx2c4.com/wireguard/wgctrl"
@@ -15,6 +17,7 @@ import (
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 	gl "gorm.io/gorm/logger"
+	"os"
 	"time"
 	"wireguard-ui/config"
 	"wireguard-ui/global/client"
@@ -25,6 +28,7 @@ import (
 func Init() {
 	initLogger()     // 初始化日志
 	initConfig()     // 读取配置文件
+	initEnv()        // 加载环境变量文件
 	initWireguard()  // 初始化wireguard客户端
 	initDatabase()   // 初始化数据库
 	initRedis()      // 初始化redis
@@ -140,8 +144,22 @@ func initOSS() {
 // initLogger
 // @description: 初始化日志
 func initLogger() {
+
+	mode := logger.Dev
+	if os.Getenv("GIN_MODE") == gin.ReleaseMode {
+		mode = logger.Prod
+	}
+
 	logger.InitLogger(logger.LogConfig{
-		Mode:       logger.Dev,
+		Mode:       mode,
 		FileEnable: true,
 	})
 }
+
+// initEnv
+// @description: 初始化环境变量
+func initEnv() {
+	if err := godotenv.Load(".env"); err != nil {
+		log.Errorf("加载.env文件失败: %v", err.Error())
+	}
+}

main.go

@@ -1,16 +1,15 @@
 package main
 
 import (
-	"fmt"
 	"gitee.ltd/lxh/logger/log"
-	"github.com/gin-contrib/pprof"
-	"github.com/spf13/cast"
+	"github.com/urfave/cli/v2"
 	"math/rand"
-	"net/http"
 	"os"
+	"sort"
 	"time"
-	"wireguard-ui/config"
-	"wireguard-ui/http/router"
+	tui "wireguard-ui/cli"
+	"wireguard-ui/cron"
+	"wireguard-ui/http"
 	"wireguard-ui/initialize"
 	"wireguard-ui/script"
 )
@@ -20,23 +19,38 @@ func init() {
 	if err := script.New().Do(); err != nil {
 		log.Errorf("执行脚本失败: %v", err.Error())
 	}
+	cron.Task()
 }
 
 func main() {
 	rand.New(rand.NewSource(time.Now().Local().UnixNano()))
-	router.Rooters()
-	handler := router.InitRouter()
 
-	if cast.ToBool(os.Getenv("ENABLED_PPROF")) {
-		pprof.Register(handler, "/monitoring")
+	app := &cli.App{
+		Name:  "wireguard-ui",
+		Usage: "wireguard-manager-ui",
 	}
 
-	httpServe := http.Server{
-		Addr:    fmt.Sprintf(":%d", config.Config.Http.Port),
-		Handler: handler,
+	app.Commands = []*cli.Command{
+		{
+			Name:    "http:serve",
+			Aliases: []string{"app:serve"},
+			Usage:   "",
+			Action: func(ctx *cli.Context) error {
+				return http.Kernel()
+			},
+		},
+		{
+			Name:    "cmd:serve",
+			Aliases: []string{"command:serve"},
+			Usage:   "use command exec",
+			Action: func(ctx *cli.Context) error {
+				return tui.Kernel()
+			},
+		},
 	}
 
-	if err := httpServe.ListenAndServe(); err != nil {
-		log.Panicf("启动http服务端失败: %v", err.Error())
+	sort.Sort(cli.CommandsByName(app.Commands))
+	if err := app.Run(os.Args); err != nil {
+		log.Fatalf("服务启动失败: %v", err.Error())
 	}
 }

model/oauth_client.go

@@ -1,16 +0,0 @@
-package model
-
-// AuthClient
-// @description: 认证客户端
-type AuthClient struct {
-	Base
-	Name      string `json:"name" gorm:"type:varchar(255);not null;comment: '客户端名称'"`
-	ClientID  string `json:"clientID" gorm:"type:varchar(255);not null;comment: '客户端ID'"`
-	ClientKey string `json:"clientKey" gorm:"type:varchar(255);not null;comment: '客户端key'"`
-	ExpireAt  string `json:"expireAt" gorm:"type:varchar(255);not null;comment: '过期时间'"`
-	IsEnabled int    `json:"isEnabled" gorm:"type:int(1);not null;comment: '是否启用'"`
-}
-
-func (AuthClient) TableName() string {
-	return "t_oauth_client"
-}

model/wireguard.go

@@ -22,16 +22,3 @@ type Client struct {
 func (Client) TableName() string {
 	return "t_client"
 }
-
-// Watcher
-// @description: 监听日志
-type Watcher struct {
-	Base
-	ClientId     string `json:"clientId" gorm:"type:char(36);not null;comment:'客户端id'"`
-	NotifyResult string `json:"notifyResult" gorm:"type:text;default null;comment:'通知结果'"`
-	IsSend       int    `json:"isSend" gorm:"type:tinyint(1);default 0;comment:'是否已通知'"`
-}
-
-func (Watcher) TableName() string {
-	return "t_watcher"
-}

service/base.go

@@ -3,7 +3,7 @@ package service
 import "gorm.io/gorm"
 
 func Paginate(current, size int64) func(db *gorm.DB) *gorm.DB {
-	// 如果页码是-1，就不分页
+	// 如果页码是-1，就不分页👋
 	if current == -1 {
 		return func(db *gorm.DB) *gorm.DB {
 			return db

service/setting.go

@@ -86,7 +86,18 @@ func (s setting) GetWGServerForConfig() (data *render_data.Server, err error) {
 // @return data
 // @return err
 func (s setting) GetAllSetting(blackList []string) (data []vo.SettingItem, err error) {
-	err = s.Model(&model.Setting{}).Select("code, data, describe").Where("code not in ?", blackList).Find(&data).Error
+	err = s.Model(&model.Setting{}).Select("code, data, describe,created_at,updated_at").Where("code not in ?", blackList).Find(&data).Error
+	return
+}
+
+// GetByCode
+// @description: 获取指定code的配置
+// @receiver s
+// @param code
+// @return data
+// @return err
+func (s setting) GetByCode(code string) (data *model.Setting, err error) {
+	err = s.Model(&model.Setting{}).Where("code = ?", code).Take(&data).Error
 	return
 }
 
@@ -141,6 +152,13 @@ func (s setting) Export() (data vo.Export, err error) {
 		}
 	}
 
+	// 查询其他配置
+	var others []vo.Other
+	if err = s.Model(&model.Setting{}).Where("code not in ?", []string{"WG_SETTING", "WG_SERVER"}).Find(&others).Error; err != nil {
+		return
+	}
+
+	data.Other = others
 	cj, _ := json.Marshal(clients)
 	_ = json.Unmarshal(cj, &data.Clients)
 
@@ -208,5 +226,17 @@ func (s setting) Import(data *vo.Export, loginUser *vo.User) (err error) {
 		}
 	}
 
+	// 其他配置写入
+	for _, v := range data.Other {
+		if err = s.SetData(&model.Setting{
+			Code:     v.Code,
+			Data:     v.Data,
+			Describe: v.Describe,
+		}); err != nil {
+			slog.Errorf("其他配置[%s]导入失败: %v", v.Code, err)
+			continue
+		}
+	}
+
 	return nil
 }

utils/mail.go

@@ -2,28 +2,35 @@ package utils
 
 import (
 	"crypto/tls"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/jordan-wright/email"
+	"github.com/spf13/cast"
 	"mime"
 	"net/smtp"
 	"net/textproto"
 	"path/filepath"
-	"wireguard-ui/config"
+	"wireguard-ui/model"
 )
 
 type mail struct {
 	*email.Email
 	addr string
 	auth smtp.Auth
+	conf map[string]string
 }
 
-func Mail() *mail {
+func Mail(conf *model.Setting) *mail {
+	// 解析配置文件
+	var mailConf = make(map[string]string)
+	_ = json.Unmarshal([]byte(conf.Data), &mailConf)
 	var m mail
 	em := email.NewEmail()
 	m.Email = em
-	m.auth = smtp.PlainAuth("", config.Config.Mail.User, config.Config.Mail.Password, config.Config.Mail.Host)
-	m.addr = fmt.Sprintf("%s:%d", config.Config.Mail.Host, config.Config.Mail.Port)
+	m.auth = smtp.PlainAuth("", mailConf["user"], mailConf["password"], mailConf["host"])
+	m.addr = fmt.Sprintf("%s:%s", mailConf["host"], mailConf["port"])
+	m.conf = mailConf
 	return &m
 }
 
@@ -47,7 +54,7 @@ func (m *mail) VerifyConfig() (err error) {
 // @param attacheFilePath
 // @return err
 func (m *mail) SendMail(to, subject, content, attacheFilePath string) (err error) {
-	m.From = fmt.Sprintf("wg-dashboard <%s>", config.Config.Mail.User)
+	m.From = fmt.Sprintf("wg-dashboard <%s>", m.conf["user"])
 	m.To = []string{to}
 	m.Subject = subject
 	m.Text = []byte(content)
@@ -61,13 +68,13 @@ func (m *mail) SendMail(to, subject, content, attacheFilePath string) (err error
 		atch.Header = emHeader
 	}
 
-	if config.Config.Mail.SkipTls {
+	if cast.ToBool(m.conf["skipTls"]) {
 		return m.Send(m.addr, m.auth)
 	}
 
 	tlsConfig := &tls.Config{}
-	tlsConfig.InsecureSkipVerify = config.Config.Mail.SkipTls
-	tlsConfig.ServerName = config.Config.Mail.Host
+	tlsConfig.InsecureSkipVerify = cast.ToBool(m.conf["skipTls"])
+	tlsConfig.ServerName = m.conf["host"]
 
 	return m.SendWithTLS(m.addr, m.auth, tlsConfig)
 }

utils/wechat_notify.go

@@ -0,0 +1,65 @@
+package utils
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	jsoniter "github.com/json-iterator/go"
+	"strings"
+	"wireguard-ui/global/client"
+	"wireguard-ui/model"
+)
+
+// wxid_472vas3av5ug22  /api/sendTextMsg
+type wechatNotify struct {
+	Addr     string
+	Path     string
+	Method   string
+	toUserId string
+}
+
+func WechatNotify(setting *model.Setting) wechatNotify {
+	var sm = make(map[string]string)
+	_ = json.Unmarshal([]byte(setting.Data), &sm)
+
+	return wechatNotify{
+		Addr:     sm["addr"],
+		Path:     sm["path"],
+		Method:   sm["method"],
+		toUserId: sm["toUserWxId"],
+	}
+}
+
+// SendTextMessage
+// @description: 发送文字通知
+// @receiver website
+// @param msg
+// @return error
+func (w wechatNotify) SendTextMessage(msg string) error {
+	log.Debugf("发送通知到微信: %v", msg)
+	req := client.HttpClient.R()
+
+	req.SetHeader("Content-Type", "application/json")
+	req.SetBody(map[string]string{
+		"wxid": w.toUserId,
+		"msg":  msg,
+	})
+
+	req.URL = fmt.Sprintf("%s:%s", w.Addr, w.Path)
+	switch strings.ToUpper(w.Method) {
+	case "POST":
+		req.Method = "POST"
+	}
+	result, err := req.SetDebug(true).Send()
+	if err != nil {
+		return err
+	}
+
+	if jsoniter.Get(result.Body(), "code").ToInt() != 1 {
+		log.Errorf("发送通知到微信失败: %v", jsoniter.Get(result.Body(), "msg").ToString())
+		return errors.New("发送通知到微信失败")
+	}
+
+	return nil
+}

web/src/views/client/index.vue

@@ -85,14 +85,14 @@
                 {{ cip }}
               </n-button>
             </n-form-item>
-            <n-form-item label="可访问IP:">
-              <n-button v-if="row.allowedIps.length <= 0" dashed size="small">
-                -
-              </n-button>
-              <n-button v-else dashed mr-2 type="warning" v-for="aip in row.allowedIps" size="small">
-                {{ aip }}
-              </n-button>
-            </n-form-item>
+<!--            <n-form-item label="可访问IP:">-->
+<!--              <n-button v-if="row.allowedIps.length <= 0" dashed size="small">-->
+<!--                - -->
+<!--              </n-button>-->
+<!--              <n-button v-else dashed mr-2 type="warning" v-for="aip in row.allowedIps" size="small">-->
+<!--                {{ aip }}-->
+<!--              </n-button>-->
+<!--            </n-form-item>-->
             <n-form-item label="创建人:">
               <n-button color="#54150F" dashed size="small">
                 {{ row.createUser }}
@@ -106,7 +106,7 @@
                 禁用
               </n-button>
             </n-form-item>
-            <n-form-item label="离线监听:">
+            <n-form-item label="离线通知:">
               <n-button v-if="row.offlineMonitoring === 1" color="#067748" round :bordered="false" size="small">
                 启用
               </n-button>
@@ -114,12 +114,12 @@
                 禁用
               </n-button>
             </n-form-item>
-            <n-form-item class="timeItem" label="时间:">
-              <n-space vertical>
-                <span> 创建时间: {{ row.createdAt }}</span>
-                <span> 更新时间: {{ row.updatedAt }}</span>
-              </n-space>
-            </n-form-item>
+<!--            <n-form-item class="timeItem" label="时间:">-->
+<!--              <n-space vertical>-->
+<!--                <span> 创建时间: {{ row.createdAt }}</span>-->
+<!--                <span> 更新时间: {{ row.updatedAt }}</span>-->
+<!--              </n-space>-->
+<!--            </n-form-item>-->
           </n-form>
         </div>
       </n-card>
@@ -228,12 +228,18 @@
           <n-radio :value="0" :checked="editModalForm.enabled === 0" @change="editModalForm.enabled = 0">禁用</n-radio>
         </n-radio-group>
       </n-form-item>
-      <n-form-item label="离线监听" path="offlineMonitoring">
+      <n-form-item label="离线通知" path="offlineMonitoring">
         <n-radio-group :value="editModalForm.offlineMonitoring">
           <n-radio :value="1" :checked="editModalForm.offlineMonitoring === 1" @change="editModalForm.offlineMonitoring = 1">启用</n-radio>
           <n-radio :value="0" :checked="editModalForm.offlineMonitoring === 0" @change="editModalForm.offlineMonitoring = 0">禁用</n-radio>
         </n-radio-group>
       </n-form-item>
+      <n-form-item class="timeItem" label="时间:">
+        <n-space vertical>
+          <span> 创建时间: {{ editModalForm.createdAt }}</span>
+          <span> 更新时间: {{ editModalForm.updatedAt }}</span>
+        </n-space>
+      </n-form-item>
       <n-button type="info" style="margin-left: 40%" @click="updateClient()">确认</n-button>
     </n-form>
   </n-modal>
@@ -319,7 +325,7 @@
           <n-radio :value="0" :checked="addModalForm.enabled === 0" @change="addModalForm.enabled = 0">禁用</n-radio>
         </n-radio-group>
       </n-form-item>
-      <n-form-item label="离线监听" path="offlineMonitoring">
+      <n-form-item label="离线通知" path="offlineMonitoring">
         <n-radio-group :value="addModalForm.offlineMonitoring">
           <n-radio :value="1" :checked="addModalForm.offlineMonitoring === 1" @change="addModalForm.offlineMonitoring = 1">启用</n-radio>
           <n-radio :value="0" :checked="addModalForm.offlineMonitoring === 0" @change="addModalForm.offlineMonitoring = 0">禁用</n-radio>
@@ -440,7 +446,9 @@ const editModalForm = ref({
     presharedKey: ''
   },
   enabled: 1,
-  offlineMonitoring: 1
+  offlineMonitoring: 1,
+  createdAt: '',
+  updatedAt: ''
 })
 
 // 添加模态框的数据集
@@ -576,6 +584,8 @@ function openEditModal(row) {
   editModalForm.value.keys.presharedKey = row.keys.presharedKey
   editModalForm.value.enabled = row.enabled
   editModalForm.value.offlineMonitoring = row.offlineMonitoring
+  editModalForm.value.createdAt = row.createdAt
+  editModalForm.value.updatedAt = row.updatedAt
 }
 
 // 更改客户端信息

web/src/views/setting/index.vue

@@ -174,6 +174,9 @@
             <n-radio :value="false" :checked="editFormModel.data[index] === false" @change="editFormModel.data[index] = false">否</n-radio>
           </n-radio-group>
         </n-form-item>
+        <n-form-item label="配置描述">
+          <n-input v-model:value="editFormModel.describe" />
+        </n-form-item>
         <n-form-item>
           <n-button type="info" @click="updateSetting">确认</n-button>
         </n-form-item>

web/src/views/workbench/index.vue

@@ -27,6 +27,18 @@
         remote
         :columns="connectionsColumns"
         :data="connectionsData.data"
+        :row-props="rowProps"
+      />
+      <n-dropdown
+        placement="bottom-start"
+        trigger="manual"
+        size="small"
+        :x="xRef"
+        :y="yRef"
+        :options="rightMenuOpts"
+        @select="rowSelect"
+        @clickoutside="rowClick"
+        :show="showDropdownRef"
       />
     </n-card>
   </AppPage>
@@ -35,7 +47,7 @@
 <script setup>
 import { useUserStore } from '@/store'
 import api from '@/views/workbench/api'
-import { debounce } from '@/utils'
+import { debounce, renderIcon } from '@/utils'
 import { NTag } from 'naive-ui'
 const userStore = useUserStore()
 
@@ -152,6 +164,49 @@ const connectionsColumns = [
   }
 ]
 
+// 链接列表邮件刷新菜单
+const rightMenuOpts = [
+  {
+    label: () => h('span',{ style: { color: 'green' }}, '刷新'),
+    key: "refresh",
+    icon: renderIcon('tabler:refresh',{ size: 14 })
+  }
+]
+
+// 右键菜单的设置
+const showDropdownRef = ref(false);
+const xRef = ref(0);
+const yRef = ref(0);
+
+// 右键菜单的基本位置逻辑
+function rowProps(row) {
+  return {
+    onContextmenu: (e) => {
+      // $message.info(JSON.stringify(row, null, 2));
+      e.preventDefault();
+      showDropdownRef.value = false;
+      nextTick().then(() => {
+        showDropdownRef.value = true;
+        xRef.value = e.clientX;
+        yRef.value = e.clientY;
+      });
+    }
+  };
+}
+
+// 右键菜单的逻辑
+function rowSelect(row) {
+  switch (row) {
+    case "refresh":
+      getClientConnections()
+      showDropdownRef.value = false
+  }
+}
+
+function rowClick() {
+  showDropdownRef.value = false
+}
+
 // 表格数据
 const tableData = ref({
   data: []