wireguard update

wireguard/wireguard.py

@@ -28,7 +28,7 @@ def keygen():
 def generate_config(server_ip, server_port=51820):
     server_key, client_key = keygen(), keygen()
 
-    with open('server', 'w') as f:
+    with open('server.sh', 'w') as f:
         f.write(dedent(f'''\
             #!/bin/sh
 
@@ -76,7 +76,7 @@ def generate_config(server_ip, server_port=51820):
             fi
         '''))
         
-    with open('client', 'w') as f:
+    with open('client.sh', 'w') as f:
         f.write(dedent(f'''\
             #!/bin/sh
             
@@ -125,10 +125,10 @@ def generate_config(server_ip, server_port=51820):
             fi
         '''))
         
-    os.system('chmod +x server client')
+    os.system('chmod +x server.sh client.sh')
 
     return dedent(f'''
-        Shell scripts generated: ./server ./client\n
+        Shell scripts generated: ./server.sh ./client.sh\n
         {YELLOW}[Server]
         - External IP: {server_ip}
         - Internal IP: 10.0.0.1
@@ -137,14 +137,14 @@ def generate_config(server_ip, server_port=51820):
         - Internal IP: 10.0.0.2\n{WHITE}
         🚨  {WHITE}{server_ip}{YELLOW}:{server_port}/udp{WHITE} should be reachable from {CYAN}client\n
         {WHITE}Handshake:
-        {YELLOW}server $ ./server
-        {CYAN}client $ ./client\n
+        {YELLOW}(server) $ ./server.sh
+        {CYAN}(client) $ ./client.sh\n
         {WHITE}cURL test:
-        {CYAN}client $ python -m http.server
-        {YELLOW}server $ curl 10.0.0.2:8000\n
+        {CYAN}(client) $ python -m http.server
+        {YELLOW}(server) $ curl 10.0.0.2:8000\n
         {WHITE}Disconnect:
-        {CYAN}server $ ./server down
-        {YELLOW}client $ ./client down{WHITE}
+        {CYAN}(server) $ ./server.sh down
+        {YELLOW}(client) $ ./client.sh down{WHITE}
     ''')
 
 if __name__ == '__main__':