From 9a4b66627484aec403d72f27d66fac27d817e9da Mon Sep 17 00:00:00 2001
From: Jaewook Lee <11328376+jaewooklee93@users.noreply.github.com>
Date: Sun, 4 Aug 2024 02:59:06 +0900
Subject: [PATCH] fix wg code

---
 wireguard/wireguard.py | 30 ++++++++++++++++--------------
 1 file changed, 16 insertions(+), 14 deletions(-)

diff --git a/wireguard/wireguard.py b/wireguard/wireguard.py
index a634f13..48913d9 100644
--- a/wireguard/wireguard.py
+++ b/wireguard/wireguard.py
@@ -57,7 +57,7 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     sudo wg-quick down ~/.wg/server.conf >/dev/null 2>&1
                     echo "interface: server deleted"
-                fi
+                else
                     echo
                     sudo wg
                     echo
@@ -66,12 +66,13 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     echo "no interface: server"
                 else
-                sudo wg-quick up ~/.wg/server.conf >/dev/null 2>&1
-                echo
-                echo "interface: server created"
-                echo
-                sudo wg
-                echo
+                    sudo wg-quick up ~/.wg/server.conf >/dev/null 2>&1
+                    echo
+                    echo "interface: server created"
+                    echo
+                    sudo wg
+                    echo
+                fi
             fi
         '''))
         
@@ -105,7 +106,7 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     sudo wg-quick down ~/.wg/client.conf >/dev/null 2>&1
                     echo "interface: client deleted"
-                fi
+                else
                     echo
                     sudo wg
                     echo
@@ -114,12 +115,13 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     echo "no interface: client"
                 else
-                sudo wg-quick up ~/.wg/client.conf >/dev/null 2>&1
-                echo
-                echo "interface: client created"
-                echo
-                sudo wg
-                echo
+                    sudo wg-quick up ~/.wg/client.conf >/dev/null 2>&1
+                    echo
+                    echo "interface: client created"
+                    echo
+                    sudo wg
+                    echo
+                fi
             fi
         '''))