From e0bc6a10d026315df8906bea0da1efe365883b3b Mon Sep 17 00:00:00 2001
From: Donald Zou <donaldzou@live.hk>
Date: Sun, 29 Dec 2024 16:37:19 +0800
Subject: [PATCH 1/2] Edited configuration not updating to dashboard

---
 src/dashboard.py | 51 +++++++++++++++++++++++++++---------------------
 1 file changed, 29 insertions(+), 22 deletions(-)

diff --git a/src/dashboard.py b/src/dashboard.py
index dd61827..b86afab 100644
--- a/src/dashboard.py
+++ b/src/dashboard.py
@@ -1098,35 +1098,42 @@ class WireguardConfiguration:
         original = []
         dataChanged = False
         with open(os.path.join(DashboardConfig.GetConfig("Server", "wg_conf_path")[1], f'{self.Name}.conf'), 'r') as f:
-            original = f.readlines()
-            original = [l.rstrip("\n") for l in original]
-            allowEdit = ["Address", "PreUp", "PostUp", "PreDown", "PostDown", "ListenPost"]
+            original = [l.rstrip("\n") for l in f.readlines()]
+            allowEdit = ["Address", "PreUp", "PostUp", "PreDown", "PostDown", "ListenPort"]
             start = original.index("[Interface]")
-            for line in range(start+1, len(original)):
-                if original[line] == "[Peer]":
-                    break
+            try:
+                end = original.index("[Peer]")
+            except ValueError as e:
+                end = len(original)
+
+            new = ["[Interface]"]
+            peerFound = False
+            for line in range(start, end):
                 split = re.split(r'\s*=\s*', original[line], 1)
                 if len(split) == 2:
-                    key = split[0]
-                    value = split[1]
-                    if key in allowEdit and key in newData.keys() and value != newData[key]:
-                        split[1] = newData[key]
-                        original[line] = " = ".join(split)
-                        if isinstance(getattr(self, key), bool):
-                            setattr(self, key, _strToBool(newData[key]))
-                        else:
-                            setattr(self, key, str(newData[key]))
-                        dataChanged = True
-                    print(original[line])
-        if dataChanged:
-            with open(os.path.join(DashboardConfig.GetConfig("Server", "wg_conf_path")[1], f'{self.Name}.conf'), 'w') as f:
-                f.write("\n".join(original))
+                    if split[0] not in allowEdit:
+                        new.append(original[line])
+            for key in allowEdit:
+                new.insert(1, f"{key} = {str(newData[key]).strip()}")
+            new.append("")
+            for line in range(end, len(original)):
+                new.append(original[line])
             self.backupConfigurationFile()
-        
-        
+            print(f"[WGDashboard] Edited Configuration -- {self.Name}.conf")
+            print("\n".join(new))
+            with open(os.path.join(DashboardConfig.GetConfig("Server", "wg_conf_path")[1], f'{self.Name}.conf'), 'w') as f:
+                f.write("\n".join(new))
+            
         status, msg = self.toggleConfiguration()        
         if not status:
             return False, msg
+        
+        for i in allowEdit:
+            if isinstance(getattr(self, i), bool):
+                setattr(self, i, _strToBool(newData[i]))
+            else:
+                setattr(self, i, str(newData[i]))
+            
         return True, ""
     
     def deleteConfiguration(self):

From bf6f58eb5e759186ad13fa68a3ffddefe2cbb76c Mon Sep 17 00:00:00 2001
From: Donald Zou <donaldzou@live.hk>
Date: Mon, 30 Dec 2024 20:31:24 +0800
Subject: [PATCH 2/2] Remove `with sqldb` in SQL Select

---
 src/dashboard.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/dashboard.py b/src/dashboard.py
index b86afab..5f1851e 100644
--- a/src/dashboard.py
+++ b/src/dashboard.py
@@ -1638,13 +1638,13 @@ sqldb.isolation_level = None
 cursor = sqldb.cursor()
 
 def sqlSelect(statement: str, paramters: tuple = ()) -> sqlite3.Cursor:
-    with sqldb:
-        try:
-            cursor = sqldb.cursor()
-            return cursor.execute(statement, paramters)
-        except Exception as e:
-            print("[WGDashboard] SQLite Error:" + str(error) + " | Statement: " + statement)
-            return []
+    try:
+        cursor = sqldb.cursor()
+        return cursor.execute(statement, paramters)
+    except Exception as e:
+        print("[WGDashboard] SQLite Error:" + str(error) + " | Statement: " + statement)
+        return []
+       
 
 
 def sqlUpdate(statement: str, paramters: tuple = ()) -> sqlite3.Cursor: