From aed4ad0210532f406c0f2afac9093367445c59ba Mon Sep 17 00:00:00 2001
From: Navratan Gupta <navilg0409@gmail.com>
Date: Tue, 30 Jun 2020 16:44:10 +0530
Subject: [PATCH] Added checkOS function

Check OS operations have been encapsulated in a function named checkOS
---
 wireguard-install.sh | 35 +++++++++++++++++++----------------
 1 file changed, 19 insertions(+), 16 deletions(-)

diff --git a/wireguard-install.sh b/wireguard-install.sh
index f6768fb..1901bcb 100644
--- a/wireguard-install.sh
+++ b/wireguard-install.sh
@@ -1,5 +1,23 @@
 #!/bin/bash
 
+function checkOS() {
+    # Check OS version
+    if [[ -e /etc/debian_version ]]; then
+        source /etc/os-release
+        OS=$ID # debian or ubuntu
+    elif [[ -e /etc/fedora-release ]]; then
+        source /etc/os-release
+        OS=$ID
+    elif [[ -e /etc/centos-release ]]; then
+        OS=centos
+    elif [[ -e /etc/arch-release ]]; then
+        OS=arch
+    else
+        echo "Looks like you aren't running this installer on a Debian, Ubuntu, Fedora, CentOS or Arch Linux system"
+        exit 1
+    fi
+}
+
 function addClient() {
 	# Load params
 	source /etc/wireguard/params
@@ -94,22 +112,7 @@ elif [[ -e /etc/wireguard ]]; then
 	exit 1
 fi
 
-# Check OS version
-if [[ -e /etc/debian_version ]]; then
-	source /etc/os-release
-	OS=$ID # debian or ubuntu
-elif [[ -e /etc/fedora-release ]]; then
-	source /etc/os-release
-	OS=$ID
-elif [[ -e /etc/centos-release ]]; then
-	OS=centos
-elif [[ -e /etc/arch-release ]]; then
-	OS=arch
-else
-	echo "Looks like you aren't running this installer on a Debian, Ubuntu, Fedora, CentOS or Arch Linux system"
-	exit 1
-fi
-
+checkOS
 # Detect public IPv4 address and pre-fill for the user
 SERVER_PUB_IPV4=$(ip addr | grep 'inet' | grep -v inet6 | grep -vE '127\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' | grep -oE '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' | head -1)
 read -rp "IPv4 or IPv6 public address: " -e -i "$SERVER_PUB_IPV4" SERVER_PUB_IP