From da8c3a74fdd6c0e89464df3932a1d8fddfe877fa Mon Sep 17 00:00:00 2001
From: peepsnet <peepsnet@yahoo.com>
Date: Sat, 3 Jun 2017 11:27:50 -0400
Subject: [PATCH] Restart Stratux on exit

The script will now restart your stratux service if you exit the script before completing.

And some aesthetics things
---
 image/sdr-tool.sh | 38 ++++++++++++++++++++++++++++++--------
 1 file changed, 30 insertions(+), 8 deletions(-)

diff --git a/image/sdr-tool.sh b/image/sdr-tool.sh
index 3609a664..4cb8a4d3 100644
--- a/image/sdr-tool.sh
+++ b/image/sdr-tool.sh
@@ -54,6 +54,19 @@ function STOPSTRATUX {
 	sleep 3
 }
 
+function STARTSTRATUX {
+	HEAD
+	echo "Give me a few seconds to get STRATUX running again..."
+	SDRs=`systemctl start stratux.service`
+	sleep 3
+	if [ "`systemctl is-active $SERVICE`" = "active" ] 
+	then
+	    echo "$SERVICE is now running"
+	else 
+		echo "$SERVICE did not restart. Try 'reboot' to restart your RaspberryPI"
+	fi
+}
+
 #Function to set the serial function
 function SETSDRSERIAL {
 	HEAD
@@ -91,6 +104,7 @@ function SETSDRSERIAL {
                 	SDRs=`reboot`
     	  	;;
     		exit)
+			STARTSTRATUX
     	  		echo "Exiting. "
     	  	exit 0
   		esac
@@ -123,6 +137,7 @@ function SDRInfo {
     	  		echo " "
     	  	;;
     		exit)
+			STARTSTRATUX
     	  		echo "Exiting. "
     	  	exit 0
   		esac
@@ -162,6 +177,7 @@ function PICKFALLBACK {
 				echo " "
     	  	;;
     		exit)
+			STARTSTRATUX
     	  		echo "Exiting. "
     	  	exit 0
   		esac
@@ -195,7 +211,8 @@ function PICKFREQ {
     	  		SDRInfo
     	  	;;
     		exit)
-    	  		echo "Exiting. "
+    	  		STARTSTRATUX
+			echo "Exiting. "
     	  	exit 0
   		esac
     break
@@ -207,9 +224,10 @@ function MAINMENU {
 	echo "Loading SDR info..."
     	sleep 2
 	HEAD
-	echo "-----------------------------------------------------------"
+	echo "#                CONFIRM ONLY ONE SDR INSTALLED                      #"
+	echo "----------------------------------------------------------------------"
 	SDRs=`rtl_eeprom`
-    echo "-----------------------------------------------------------"
+    echo "----------------------------------------------------------------------"
 	echo " "
 	echo "${BOLD}${RED}Read the lines above.${NORM}"
 	echo "${BOLD}How many SDRs were found?${NORM}"
@@ -231,14 +249,17 @@ function MAINMENU {
 		      PICKFREQ
       		;;
 		    '2 or more')
-		      echo "#####################################################################################"
-		      echo "#      ${RED}Too Many SDRs Plugged in. Unplug all SDRs except one and try again!!${NORM}         #"
-		      echo "#####################################################################################"
+				echo "#####################################################################################"
+				echo "#      ${RED}Too Many SDRs Plugged in. Unplug all SDRs except one and try again!!${NORM}         #"
+				echo "#####################################################################################"
+				STARTSTRATUX
+				echo "Exiting. "
 		      exit 0
 		      ;;
 		    exit)
-		      echo "Exiting. "
-		      exit 0
+		    	STARTSTRATUX
+		    	echo "Exiting... "
+		    	exit 0
 		  esac
  	# Getting here means that a valid choice was made,
   	# so break out of the select statement and continue below,
@@ -284,6 +305,7 @@ function START {
     	    	MAINMENU
     	    ;;
     		exit)
+			STARTSTRATUX
     	  		echo "Exiting. "
     	  	exit 0
   		esac