Merge branch 'v2-develop' into v2
All checks were successful
continuous-integration/drone/tag Build is passing
All checks were successful
continuous-integration/drone/tag Build is passing
This commit is contained in:
coward
commit
904907f796
@ -1,5 +1,10 @@
|
||||
package cli
|
||||
|
||||
import (
|
||||
"wireguard-ui/cli/tui"
|
||||
)
|
||||
|
||||
func Kernel() error {
|
||||
tui.NewApp().Run()
|
||||
return nil
|
||||
}
|
||||
|
||||
165
cli/tui/app.go
Normal file
165
cli/tui/app.go
Normal file
@ -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
|
||||
}
|
||||
520
cli/tui/client.go
Normal file
520
cli/tui/client.go
Normal file
@ -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 = ¶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
|
||||
}
|
||||
62
cli/tui/server.go
Normal file
62
cli/tui/server.go
Normal file
@ -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, "")
|
||||
}
|
||||
440
cli/tui/setting.go
Normal file
440
cli/tui/setting.go
Normal file
@ -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)
|
||||
|
||||
}
|
||||
136
cli/tui/utils.go
Normal file
136
cli/tui/utils.go
Normal file
@ -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()
|
||||
}
|
||||
@ -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失败")
|
||||
}
|
||||
|
||||
client.Redis.Set(context.Background(),
|
||||
fmt.Sprintf("%s:%s", constant.UserToken, userId),
|
||||
token,
|
||||
time.Duration(expireTime.Sub(time.Now()).Abs().Seconds())*time.Second)
|
||||
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,10 +98,20 @@ 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()
|
||||
if err != nil {
|
||||
log.Errorf("缓存中用户[%s]的token查找失败: %v", claims.ID, err.Error())
|
||||
return nil, errors.New("token不存在")
|
||||
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 {
|
||||
|
||||
@ -3,5 +3,6 @@ package constant
|
||||
const (
|
||||
Captcha = "captcha"
|
||||
UserToken = "token"
|
||||
TUIUserToken = "tui:token"
|
||||
ClientOffline = "client:offline:"
|
||||
)
|
||||
|
||||
15
go.mod
15
go.mod
@ -4,8 +4,11 @@ go 1.21
|
||||
|
||||
require (
|
||||
gitee.ltd/lxh/logger v1.0.18
|
||||
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/dustin/go-humanize v1.0.1
|
||||
github.com/eiannone/keyboard v0.0.0-20220611211555-0d226195f203
|
||||
github.com/fsnotify/fsnotify v1.7.0
|
||||
github.com/gin-gonic/gin v1.10.0
|
||||
github.com/glebarez/sqlite v1.11.0
|
||||
@ -35,15 +38,20 @@ require (
|
||||
|
||||
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.12.5 // indirect
|
||||
github.com/bytedance/sonic/loader v0.2.0 // indirect
|
||||
github.com/caarlos0/env/v6 v6.10.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.2.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.4 // indirect
|
||||
github.com/cloudwego/iasm v0.2.0 // indirect
|
||||
github.com/cpuguy83/go-md2man/v2 v2.0.5 // indirect
|
||||
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.3 // indirect
|
||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||
github.com/glebarez/go-sqlite v1.21.2 // indirect
|
||||
@ -70,8 +78,11 @@ require (
|
||||
github.com/klauspost/cpuid/v2 v2.2.7 // 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/mattn/go-localereader v0.0.1 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.2 // indirect
|
||||
github.com/mdlayher/genetlink v1.3.2 // indirect
|
||||
github.com/mdlayher/netlink v1.7.2 // indirect
|
||||
@ -79,6 +90,9 @@ require (
|
||||
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/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/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f // indirect
|
||||
github.com/natefinch/lumberjack v2.0.0+incompatible // indirect
|
||||
github.com/panjf2000/ants/v2 v2.10.0 // indirect
|
||||
@ -90,6 +104,7 @@ require (
|
||||
github.com/prometheus/procfs v0.8.0 // indirect
|
||||
github.com/prometheus/prometheus v1.8.2-0.20201028100903-3245b3267b24 // 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
|
||||
|
||||
36
go.sum
36
go.sum
@ -106,6 +106,10 @@ github.com/aws/aws-lambda-go v1.13.3/go.mod h1:4UKl9IzQMoD+QF79YdCuzCwp8VbmG4VAQ
|
||||
github.com/aws/aws-sdk-go v1.27.0/go.mod h1:KmX6BPdI08NWTb3/sm4ZGu5ShLoqVDhKgpiN924inxo=
|
||||
github.com/aws/aws-sdk-go v1.35.5/go.mod h1:tlPOdRjfxPBpNIwqDj61rmsnA85v9jc0Ps9+muhnW+k=
|
||||
github.com/aws/aws-sdk-go-v2 v0.18.0/go.mod h1:JWVYvqSMppoMJC0x5wdwiImzgXTI9FuZwxzkQq9wy+g=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k=
|
||||
github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8=
|
||||
github.com/aymanbagabas/go-udiff v0.2.0 h1:TK0fH4MteXUDspT88n8CKzvK0X9O2xu9yQjWpi6yML8=
|
||||
github.com/aymanbagabas/go-udiff v0.2.0/go.mod h1:RE4Ex0qsGkTAJoQdQQCA0uG+nAzJO/pI/QwceO5fgrA=
|
||||
github.com/baiyubin/aliyun-sts-go-sdk v0.0.0-20180326062324-cfa1a18b161f/go.mod h1:AuiFmCCPBSrqvVMvuqFuk0qogytodnVFVSN5CeJB8Gc=
|
||||
github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=
|
||||
github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=
|
||||
@ -139,6 +143,18 @@ github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XL
|
||||
github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
||||
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/charmbracelet/bubbles v0.20.0 h1:jSZu6qD8cRQ6k9OMfR1WlM+ruM8fkPWkHvQWD9LIutE=
|
||||
github.com/charmbracelet/bubbles v0.20.0/go.mod h1:39slydyswPy+uVOHZ5x/GjwVAFkCsV8IIVy+4MhzwwU=
|
||||
github.com/charmbracelet/bubbletea v1.1.0 h1:FjAl9eAL3HBCHenhz/ZPjkKdScmaS5SK69JAK2YJK9c=
|
||||
github.com/charmbracelet/bubbletea v1.1.0/go.mod h1:9Ogk0HrdbHolIKHdjfFpyXJmiCzGwy+FesYkZr7hYU4=
|
||||
github.com/charmbracelet/lipgloss v0.13.0 h1:4X3PPeoWEDCMvzDvGmTajSyYPcZM4+y8sCA/SsA3cjw=
|
||||
github.com/charmbracelet/lipgloss v0.13.0/go.mod h1:nw4zy0SBX/F/eAO1cWdcvy6qnkDUxr8Lw7dvFrAIbbY=
|
||||
github.com/charmbracelet/x/ansi v0.2.3 h1:VfFN0NUpcjBRd4DnKfRaIRo53KRgey/nhOoEqosGDEY=
|
||||
github.com/charmbracelet/x/ansi v0.2.3/go.mod h1:dk73KoMTT5AX5BsX0KrqhsTqAnhZZoCBjs7dGWp4Ktw=
|
||||
github.com/charmbracelet/x/exp/golden v0.0.0-20240815200342-61de596daa2b h1:MnAMdlwSltxJyULnrYbkZpp4k58Co7Tah3ciKhSNo0Q=
|
||||
github.com/charmbracelet/x/exp/golden v0.0.0-20240815200342-61de596daa2b/go.mod h1:wDlXFlCrmJ8J+swcL/MnGUuYnqgQdW9rhSD61oNMb6U=
|
||||
github.com/charmbracelet/x/term v0.2.0 h1:cNB9Ot9q8I711MyZ7myUR5HFWL/lc3OpU8jZ4hwm0x0=
|
||||
github.com/charmbracelet/x/term v0.2.0/go.mod h1:GVxgxAbjUrmpvIINHIQnJJKpMlHiZ4cktEQCN6GWyF0=
|
||||
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
|
||||
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
|
||||
github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
|
||||
@ -189,6 +205,8 @@ github.com/eapache/go-xerial-snappy v0.0.0-20180814174437-776d5712da21/go.mod h1
|
||||
github.com/eapache/queue v1.1.0/go.mod h1:6eCeP0CKFpHLu8blIFXhExK/dRa7WDZfr6jVFPTqq+I=
|
||||
github.com/eclipse/paho.mqtt.golang v1.2.0/go.mod h1:H9keYFcgq3Qr5OUJm/JZI/i6U7joQ8SYLhZwfeOo6Ts=
|
||||
github.com/edsrzf/mmap-go v1.0.0/go.mod h1:YO35OhQPt3KJa3ryjFM5Bs14WD66h8eGKpfaBNrHW5M=
|
||||
github.com/eiannone/keyboard v0.0.0-20220611211555-0d226195f203 h1:XBBHcIb256gUJtLmY22n99HaZTz+r2Z51xUPi01m3wg=
|
||||
github.com/eiannone/keyboard v0.0.0-20220611211555-0d226195f203/go.mod h1:E1jcSv8FaEny+OP/5k9UxZVw9YFWGj7eI4KR/iOBqCg=
|
||||
github.com/elazarl/goproxy v0.0.0-20180725130230-947c36da3153/go.mod h1:/Zj4wYkgs4iZTTu3o/KG3Itv/qCCa8VVMlb3i9OVuzc=
|
||||
github.com/emicklei/go-restful v0.0.0-20170410110728-ff4f55a20633/go.mod h1:otzb+WCGbkyDHkqmQmT5YD2WR4BBwUdeQoFo8l/7tVs=
|
||||
github.com/envoyproxy/go-control-plane v0.6.9/go.mod h1:SBwIajubJHhxtWwsL9s8ss4safvEdbitLhGGK48rN6g=
|
||||
@ -196,6 +214,8 @@ github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymF
|
||||
github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
|
||||
github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=
|
||||
github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6baUTXGLOoWe4PQhGxaX0KpnayAqC48p4=
|
||||
github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM=
|
||||
github.com/evanphx/json-patch v4.9.0+incompatible/go.mod h1:50XU6AFN0ol/bzJsmQLiYLvXMP4fmwYFNcr97nuDLSk=
|
||||
github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
|
||||
github.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU=
|
||||
@ -572,6 +592,8 @@ github.com/lightstep/lightstep-tracer-common/golang/gogo v0.0.0-20190605223551-b
|
||||
github.com/lightstep/lightstep-tracer-go v0.18.1/go.mod h1:jlF1pusYV4pidLvZ+XD0UBX0ZE6WURAspgAczcDHrL4=
|
||||
github.com/lixh00/loki-client-go v1.0.1 h1:y/ePf/s66N77eikIujRS/QQAKvbMmPmesMxAuMuP8lM=
|
||||
github.com/lixh00/loki-client-go v1.0.1/go.mod h1:JSeu3fIBPjnmf5bBq6I8hvJlhYum2eLQEzwU149vyfQ=
|
||||
github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
|
||||
github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
|
||||
github.com/lyft/protoc-gen-validate v0.0.13/go.mod h1:XbGvPuh87YZc5TdIa2/I4pLk0QoUACkjt2znoq26NVQ=
|
||||
github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=
|
||||
github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
|
||||
@ -595,8 +617,12 @@ github.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOA
|
||||
github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4=
|
||||
github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
|
||||
github.com/mattn/go-runewidth v0.0.2/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
|
||||
github.com/mattn/go-runewidth v0.0.3/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
|
||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/mattn/go-sqlite3 v1.11.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=
|
||||
github.com/mattn/go-tty v0.0.0-20180907095812-13ff1204f104/go.mod h1:XPvLUNfbS4fJH25nqRHfWLMa1ONC8Amw+mIA639KxkE=
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
|
||||
@ -638,6 +664,12 @@ github.com/mojocn/base64Captcha v1.3.6/go.mod h1:i5CtHvm+oMbj1UzEPXaA8IH/xHFZ3DG
|
||||
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
|
||||
github.com/morikuni/aec v1.0.0/go.mod h1:BbKIizmSmc5MMPqRYbxO4ZU0S0+P200+tUnFx7PXmsc=
|
||||
github.com/mschoch/smat v0.0.0-20160514031455-90eadee771ae/go.mod h1:qAyveg+e4CE+eKJXWVjKXM4ck2QobLqTDytGJbLLhJg=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
|
||||
github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
|
||||
github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
|
||||
github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
|
||||
github.com/muesli/termenv v0.15.2 h1:GohcuySI0QmI3wN8Ok9PtKGkgkFIk7y6Vpb5PvrY+Wo=
|
||||
github.com/muesli/termenv v0.15.2/go.mod h1:Epx+iuz8sNs7mNKhxzH4fWXGNpZwUaJKRS1noLXviQ8=
|
||||
github.com/munnerz/goautoneg v0.0.0-20120707110453-a547fc61f48d/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
|
||||
github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
|
||||
github.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f h1:KUppIJq7/+SVif2QVs3tOP0zanoHgBEVAwHxUSIzRqU=
|
||||
@ -758,6 +790,9 @@ github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0/go.mod h1:qq
|
||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
|
||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
|
||||
github.com/retailnext/hllpp v1.0.1-0.20180308014038-101a6d2f8b52/go.mod h1:RDpi1RftBQPUCDRw6SmxeaREsAaRKnOclghuzp/WRzc=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
|
||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
||||
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
||||
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
||||
github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
|
||||
@ -1108,6 +1143,7 @@ golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7w
|
||||
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210603081109-ebe580a85c40/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220114195835-da31bd327af9/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
|
||||
@ -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("登陆失败！")
|
||||
|
||||
@ -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
|
||||
}
|
||||
|
||||
@ -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()
|
||||
|
||||
@ -1,13 +1,18 @@
|
||||
package vo
|
||||
|
||||
import "wireguard-ui/global/constant"
|
||||
import (
|
||||
"wireguard-ui/global/constant"
|
||||
"wireguard-ui/model"
|
||||
)
|
||||
|
||||
// SettingItem
|
||||
// @description: 设置单项
|
||||
type SettingItem struct {
|
||||
Code string `json:"code"`
|
||||
Data string `json:"data"`
|
||||
Describe string `json:"describe"`
|
||||
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 {
|
||||
|
||||
@ -6,6 +6,7 @@ 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/redis/go-redis/v9"
|
||||
@ -15,6 +16,7 @@ import (
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
gl "gorm.io/gorm/logger"
|
||||
"os"
|
||||
"time"
|
||||
"wireguard-ui/config"
|
||||
"wireguard-ui/global/client"
|
||||
@ -140,8 +142,14 @@ 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,
|
||||
})
|
||||
}
|
||||
|
||||
@ -86,7 +86,7 @@ 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
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user