From f8fd5acbed66f9afeb34cf38e24a74599050ed10 Mon Sep 17 00:00:00 2001
From: Richard Goodwin <github@richardgoodwin.com>
Date: Fri, 26 Oct 2018 10:40:08 -0500
Subject: [PATCH] Optionally use LED for setup progress in either setup mode

---
 setup/pi/setup-teslausb | 40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/setup/pi/setup-teslausb b/setup/pi/setup-teslausb
index 4563c72..9b800f1 100644
--- a/setup/pi/setup-teslausb
+++ b/setup/pi/setup-teslausb
@@ -5,6 +5,8 @@ SETUP_LOGFILE=/boot/teslausb-headless-setup.log
 REPO=${REPO:-cimryan}
 BRANCH=${BRANCH:-master}
 HEADLESS_SETUP=${HEADLESS_SETUP:-false}
+USE_LED_FOR_SETUP_PROGRESS=true
+
 
 
 if ! [ $(id -u) = 0 ]
@@ -29,6 +31,7 @@ function headless_setup_populate_variables () {
   fi
 }
 
+
 function headless_setup_mark_setup_failed () {
   if [ $HEADLESS_SETUP = "true" ]
   then
@@ -61,6 +64,22 @@ function headless_setup_progress_flash () {
   fi
 }
 
+function setup_led_off () {
+
+  if [ $USE_LED_FOR_SETUP_PROGRESS = "true" ]
+  then
+    echo 1 | sudo tee /sys/class/leds/led1/brightness
+  fi
+}
+
+function setup_led_on () {
+
+  if [ $USE_LED_FOR_SETUP_PROGRESS = "true" ]
+  then
+    echo 0 | sudo tee /sys/class/leds/led1/brightness
+  fi
+}
+
 function check_variable () {
   local var_name="$1"
   if [ -z "${!var_name+x}" ]
@@ -232,6 +251,12 @@ function make_root_fs_readonly () {
 
 headless_setup_populate_variables
 
+# If USE_LED_FOR_SETUP_PROGRESS = true. 
+setup_led_off
+
+# Flash for stage 2 headless (verify variables)
+headless_setup_progress_flash 2
+
 setup_progress "Verifying environment variables..."
 
 RSYNC_ENABLE="${RSYNC_ENABLE:-false}"
@@ -255,6 +280,9 @@ check_variable "campercent"
 
 check_pushover_enabled
 
+# Flash for Stage 3 headless (grab scripts)
+headless_setup_progress_flash 3
+
 if [ ! -e /root/bin ]
 then
   mkdir /root/bin
@@ -280,6 +308,7 @@ get_ancillary_setup_scripts
 
 pushd ~
 
+
 configure_archive_scripts
 
 configure_pushover_scripts
@@ -288,6 +317,9 @@ fix_cmdline_txt_modules_load
 
 echo "" >> /etc/fstab
 
+# Flash for stage 4 headless (Create backing files)
+headless_setup_progress_flash 4
+
 create_usb_drive_backing_files
 
 /root/bin/configure-archive.sh
@@ -296,8 +328,14 @@ configure_rc_local
 
 configure_hostname
 
+# Flash for stage 5 headless (Mark success, FS readonly)
+headless_setup_progress_flash 5
+
 headless_setup_mark_setup_success
 
 make_root_fs_readonly
 
-echo "All done."
+# If USE_LED_FOR_SETUP_PROGRESS = true. 
+setup_led_on
+
+setup_progress "All done."