From ff5c7aef41b4a7c998e89003f8abd407ebedca7b Mon Sep 17 00:00:00 2001
From: Angristan <angristan@pm.me>
Date: Sun, 19 Aug 2018 00:52:24 +0200
Subject: [PATCH] Add a functions and main section, add initialCheck ()

---
 openvpn-install.sh | 43 +++++++++++++++++++++++++++++++++----------
 1 file changed, 33 insertions(+), 10 deletions(-)

diff --git a/openvpn-install.sh b/openvpn-install.sh
index e4b10c9..7790bba 100644
--- a/openvpn-install.sh
+++ b/openvpn-install.sh
@@ -3,18 +3,41 @@
 # Secure OpenVPN server installer for Debian, Ubuntu, CentOS and Arch Linux
 # https://github.com/Angristan/OpenVPN-install
 
+#########################################
+############    FUNCTIONS    ############
+#########################################
 
-# Verify root
-if [[ "$EUID" -ne 0 ]]; then
-	echo "Sorry, you need to run this as root"
-	exit 1
-fi
+function tunAvailable () {
+	if [[ -e /dev/net/tun ]]; then
+		return 0
+	else
+		return 1
+	fi
+}
 
-# Verify tun
-if [[ ! -e /dev/net/tun ]]; then
-	echo "TUN is not available"
-	exit 2
-fi
+function isRoot () {
+	if [[ "$EUID" -eq 0 ]]; then
+		return 0
+	else
+		return 1
+	fi
+}
+
+function initialCheck () {
+	if ! isRoot; then
+		echo "Sorry, you need to run this as root"
+		exit 1
+	elif ! tunAvailable; then
+		echo "TUN is not available"
+		exit 2
+	fi
+}
+
+##########################################
+###############    MAIN    ###############
+##########################################
+
+initialCheck
 
 # Check if CentOS 5
 if grep -qs "CentOS release 5" "/etc/redhat-release"; then