Mirrors/WGDashboard
1
0
Fork 0
mirror of https://github.com/donaldzou/WGDashboard.git synced 2025-10-03 07:46:18 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update Peer.py

Browse Source
This commit is contained in:
Donald Zou
2025-09-16 07:57:30 +08:00
parent 0b054ae668
commit 77112675ae
1 changed files with 29 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/modules/Peer.py
View File

@@ -1,7 +1,9 @@
""" """
Peer Peer
""" """
import base64
import datetime import datetime
import json
import os, subprocess, uuid, random, re import os, subprocess, uuid, random, re
from datetime import timedelta from datetime import timedelta
@@ -122,6 +124,10 @@ class Peer:
return False, exc.output.decode("UTF-8").strip() return False, exc.output.decode("UTF-8").strip()
def downloadPeer(self) -> dict[str, str]: def downloadPeer(self) -> dict[str, str]:
final = {
"fileName": "",
"file": ""
}
filename = self.name filename = self.name
if len(filename) == 0: if len(filename) == 0:
filename = "UntitledPeer" filename = "UntitledPeer"
@@ -133,10 +139,9 @@ class Peer:
for i in illegal_filename: for i in illegal_filename:
filename = filename.replace(i, "") filename = filename.replace(i, "")
finalFilename = ""
for i in filename: for i in filename:
if re.match("^[a-zA-Z0-9_=+.-]$", i): if re.match("^[a-zA-Z0-9_=+.-]$", i):
finalFilename += i final["fileName"] += i
interfaceSection = { interfaceSection = {
"PrivateKey": self.private_key, "PrivateKey": self.private_key,
@@ -179,19 +184,33 @@ class Peer:
"PresharedKey": self.preshared_key "PresharedKey": self.preshared_key
} }
combine = [interfaceSection.items(), peerSection.items()] combine = [interfaceSection.items(), peerSection.items()]
peerConfiguration = ""
for s in range(len(combine)): for s in range(len(combine)):
if s == 0: if s == 0:
peerConfiguration += "[Interface]\n" final["file"] += "[Interface]\n"
else: else:
peerConfiguration += "\n[Peer]\n" final["file"] += "\n[Peer]\n"
for (key, val) in combine[s]: for (key, val) in combine[s]:
if val is not None and ((type(val) is str and len(val) > 0) or (type(val) is int and val > 0)): if val is not None and ((type(val) is str and len(val) > 0) or (type(val) is int and val > 0)):
peerConfiguration += f"{key} = {val}\n" final["file"] += f"{key} = {val}\n"
return { if self.configuration.Protocol == "awg":
"fileName": finalFilename, final["amneziaVPN"] = json.dumps({
"file": jinja2.Template(peerConfiguration).render(configuration=self.configuration) "containers": [{
} "awg": {
"isThirdPartyConfig": True,
"last_config": final['file'],
"port": self.configuration.ListenPort,
"transport_proto": "udp"
},
"container": "amnezia-awg"
}],
"defaultContainer": "amnezia-awg",
"description": self.name,
"hostName": (
self.configuration.configurationInfo.OverridePeerSettings.PeerRemoteEndpoint
if self.configuration.configurationInfo.OverridePeerSettings.PeerRemoteEndpoint
else self.configuration.DashboardConfig.GetConfig("Peers", "remote_endpoint")[1])
})
return final
def getJobs(self): def getJobs(self):
self.jobs = self.configuration.AllPeerJobs.searchJob(self.configuration.Name, self.id) self.jobs = self.configuration.AllPeerJobs.searchJob(self.configuration.Name, self.id)