From 9e63db154d97b6d5c3a26b97d6a1f68a28511f8a Mon Sep 17 00:00:00 2001
From: liberodark <liberodark@gmail.com>
Date: Wed, 18 Dec 2019 13:18:16 +0100
Subject: [PATCH] Update wireguard-install.sh

First Rework
---
 wireguard-install.sh | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/wireguard-install.sh b/wireguard-install.sh
index 69f6299..574de4b 100644
--- a/wireguard-install.sh
+++ b/wireguard-install.sh
@@ -21,6 +21,44 @@ if [[ $(id -u) -ne 0 ]] ; then echo "Please run as root" ; exit 1 ; fi
 
 distribution=$(cat /etc/*release | grep "PRETTY_NAME" | sed 's/PRETTY_NAME=//g' | sed 's/["]//g' | awk '{print $1}')
 
+usage ()
+{
+     echo "usage: -install or -remove"
+     echo "options:"
+     echo "-install: Install WireGuard"
+     echo "-remove: Remove WireGuard"
+     echo "-h: Show help"
+}
+
+parse_args ()
+{
+    while [ $# -ne 0 ]
+    do
+        case "${1}" in
+            -install)
+                shift
+                install_wg >&2
+                ;;
+            -remove)
+                shift
+                remove_wg >&2
+                ;;
+            -h|--help)
+                usage
+                exit 0
+                ;;
+            *)
+                echo "Invalid argument : ${1}" >&2
+                usage >&2
+                exit 1
+                ;;
+        esac
+        shift
+    done
+
+}
+
+
 if [ "$(systemd-detect-virt)" == "openvz" ]; then
     echo "OpenVZ is not supported"
     exit