From a306bfe87267f3325282bc5f0dfcca8be7469d38 Mon Sep 17 00:00:00 2001
From: Kaj Forney <kaj.forney@gmail.com>
Date: Mon, 1 Oct 2018 22:42:40 -0600
Subject: [PATCH] Fix unary operators.

Former-commit-id: f727ba95af39612968bc2a80c3787d023659f050
Former-commit-id: 3e3fb956b1d16717d9344131fe312bb9297ec5a4
---
 runme.sh | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/runme.sh b/runme.sh
index c0845eec..88c9e7a3 100755
--- a/runme.sh
+++ b/runme.sh
@@ -13,19 +13,20 @@ fi
 
 #Arguments
 
+DISTRO=$(lsb_release -i -s)
 
-while getopts ud:r: option
+while getopts ul:d:r: option
 do
     case "${option}" in
         u) UNATTEND="true" && echo "UNATTEND";;
+        l) DISTRO=${OPTARG} && echo $DISTRO;;
         d) DESKTOP=${OPTARG} && echo $DESKTOP;;
         r) REPO=${OPTARG} && echo $REPO;;
     esac
 done
 
-DISTRO=$(lsb_release -i -s)
 
-if [ $UNATTEND != "true" ];
+if [[ $UNATTEND != "true" ]];
 then
     whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70
 
@@ -104,7 +105,7 @@ Parrot)
 ;;
 esac
 
-if [ $UNATTEND != "true" ];
+if [[ $UNATTEND != "true" ]];
 then
 whiptail --backtitle "Pentest Build Script" --title "Finished" --msgbox "Fingers crossed there were no errors... Your image should be done!" 7 70
 fi