Added requirement to run script as root

1 files changed, 9 insertions, 1 deletions

diff --git a/build.sh b/build.sh
index c51e97e..741349e 100755
--- a/build.sh
+++ b/build.sh
@@ -1,9 +1,17 @@
 #!/bin/bash
+if [ "$EUID" -ne 0 ]
+	then echo "Run this script as root to continue"
+	exit	
+fi
 echo "Building Alee Boot Utility"
 if [ -f "/usr/bin/mkarchiso" ]; then
-	if [ -d "log" ]; then
+	if [ ! -d "log" ]; then
 		mkdir log
 	fi
+	if [ -d "work" ]; then
+		echo "Removing the work directory"
+		rm -rf work
+	fi
 	mkarchiso -v . | tee log/aleebootutility-$(date +%Y.%m.%d-%H.%M.%S).log
 else
 	echo "You must install the archiso package beforing continuing this process..."