From 2e752ad783ffb817f6d627a999d51ac6656411f9 Mon Sep 17 00:00:00 2001
From: "Ycarus (Yannick Chabanois)" <ycarus@zugaina.org>
Date: Fri, 5 Mar 2021 19:56:56 +0100
Subject: [PATCH] Fix wireguard peer

---
 omr-admin.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/omr-admin.py b/omr-admin.py
index 072302d..d678031 100755
--- a/omr-admin.py
+++ b/omr-admin.py
@@ -2094,6 +2094,8 @@ class WireGuard(BaseModel):
 
 @app.post('/wireguard', summary="Modify Wireguard configuration")
 def wireguard(*, params: WireGuard, current_user: User = Depends(get_current_user)):
+    if not os.path.isfile('/etc/wireguard/wg0.conf'):
+        return {'result': 'error', 'reason': 'Wireguard config not found', 'route': 'wireguard'}
     wg_config = configparser.ConfigParser(strict=False)
     wg_config.read_file(open(r'/etc/wireguard/wg0.conf'))
     wg_port = wg_config.get('Interface', 'ListenPort')
@@ -2107,7 +2109,7 @@ def wireguard(*, params: WireGuard, current_user: User = Depends(get_current_use
         n.write('PrivateKey = ' + wg_key + '\n')
         for peer in params.peers:
             n.write('\n')
-            n.write('[Peer]')
+            n.write('[Peer]\n')
             n.write('PublicKey  = ' + peer.key + '\n')
             n.write('AllowedIPs = ' + peer.ip + '\n')
     move(tmpfile, '/etc/wireguard/wg0.conf')