jostreff/wireguard-ui
1
0
Fork 0
mirror of https://github.com/ngoduykhanh/wireguard-ui.git synced 2025-07-06 17:13:13 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Config requests, TgUseridToClientID cache, fixes

Browse source
This commit is contained in:
0xCA 2023-11-17 17:11:39 +05:00
parent 4c6080b3fa
commit 659df606f6
6 changed files with 176 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

21
store/jsondb/jsondb.go
View file

@ -6,6 +6,7 @@ import (
"fmt"
"os"
"path"
"strconv"
"time"
"github.com/sdomino/scribble"
@ -161,6 +162,20 @@ func (o *JsonDB) Init() error {
}
}
// init cache
clients, err := o.GetClients(false)
if err != nil {
return nil
}
for _, cl := range clients {
client := cl.Client
if len(client.TgUserid) > 3 {
if userid, err := strconv.ParseInt(client.TgUserid, 10, 64); err == nil {
util.UpdateTgToClientID(userid, client.ID)
}
}
}
return nil
}
@ -314,6 +329,11 @@ func (o *JsonDB) GetClientByID(clientID string, qrCodeSettings model.QRCodeSetti
func (o *JsonDB) SaveClient(client model.Client) error {
clientPath := path.Join(path.Join(o.dbPath, "clients"), client.ID+".json")
output := o.conn.Write("clients", client.ID, client)
if output == nil && len(client.TgUserid) > 3 {
if userid, err := strconv.ParseInt(client.TgUserid, 10, 64); err == nil {
util.UpdateTgToClientID(userid, client.ID)
}
}
err := util.ManagePerms(clientPath)
if err != nil {
return err
@ -322,6 +342,7 @@ func (o *JsonDB) SaveClient(client model.Client) error {
}
func (o *JsonDB) DeleteClient(clientID string) error {
util.RemoveTgToClientID(clientID)
return o.conn.Delete("clients", clientID)
}