feat: support dry-run mode

Author: tomeon

shutdowncheck.sh

@@ -105,6 +105,16 @@ if [ -n "${failed:-}" ]; then
   exit 1
 fi
 
+if [ -n "${ATX_RASPI_DRY_RUN:-}" ]; then
+  handle_press() {
+    echo "[dry-run] $*"
+  }
+else
+  handle_press() {
+    "$@"
+  }
+fi
+
 if { command -v gpioset && command -v gpiomon ; } 1>/dev/null 2>&1; then
   init_shutdown_pin() {
     # NOP
@@ -130,10 +140,11 @@ if { command -v gpioset && command -v gpiomon ; } 1>/dev/null 2>&1; then
 
           if [ "$pulseDuration" -gt "$REBOOTPULSEMAXIMUM" ]; then
             header 12 "SHUTDOWN request on chip ${CHIP?} from GPIO${SHUTDOWN}, halting Rpi ..."
+            handle_press poweroff
             return
           elif [ "$pulseDuration" -gt "$REBOOTPULSEMINIMUM" ]; then
             header 12 "REBOOT request on chip ${CHIP?} from GPIO${SHUTDOWN?}, recycling Rpi ..."
-            reboot
+            handle_press reboot
             return
           else
             unset pulseStart pulseEnd
@@ -171,15 +182,15 @@ elif [ -e /sys/class/gpio/export ]; then
           sleep 0.02
           if [ "$(( "$(date +%s%N)" - pulseStart ))" -gt "$REBOOTPULSEMAXIMUM" ]; then
             header 12 "SHUTDOWN request from GPIO${SHUTDOWN}, halting Rpi ..."
-            poweroff
+            handle_press poweroff
             return
           fi
           shutdownSignal=$(cat /sys/class/gpio/gpio$SHUTDOWN/value)
         done
         #pulse went LOW, check if it was long enough, and trigger reboot
         if [ "$(( "$(date +%s%N)" - pulseStart ))" -gt "$REBOOTPULSEMINIMUM" ]; then
           header 12 "REBOOT request from GPIO${SHUTDOWN}, recycling Rpi ..."
-          reboot
+          handle_press reboot
           return
         fi
       fi

shutdownirq.py

@@ -39,6 +39,15 @@ def handle_press(*command):
 		print("[dry-run] ", " ".join(command))
 		sys.exit()
 
+if os.environ.get("ATX_RASPI_DRY_RUN", "") == "":
+	def handle_press(*command):
+		os.system(*command)
+		sys.exit()
+else:
+	def handle_press(*command):
+		print("[dry-run] ", " ".join(command))
+		sys.exit()
+
 def diag(*msgs):
 	linelen = max([len(msg) for msg in msgs]) + 2
 	wrapper = "=" * linelen
@@ -92,12 +101,12 @@ def announce():
 						if pulse_duration >= REBOOTPULSEMAXIMUM:
 							print()
 							diag("SHUTDOWN request on chip {0} from GPIO{1}, halting Rpi ...".format(CHIP, SHUTDOWN))
-							os.system("poweroff")
+							handle_press("poweroff")
 							sys.exit()
 						elif pulse_duration >= REBOOTPULSEMINIMUM:
 							print()
 							diag("REBOOT request on chip {0} from GPIO{1}, recycling Rpi ...".format(CHIP, SHUTDOWN))
-							os.system("reboot")
+							handle_press("reboot")
 							sys.exit()
 						else:
 							pulse_start = None
@@ -125,13 +134,13 @@ def announce():
 				if(time.time() - pulse_start >= REBOOTPULSEMAXIMUM):
 					print()
 					diag("SHUTDOWN request from GPIO{0}, halting Rpi ...".format(SHUTDOWN))
-					os.system("poweroff")
+					handle_press("poweroff")
 					sys.exit()
 				shutdown_signal = GPIO.input(SHUTDOWN)
 			if time.time() - pulse_start >= REBOOTPULSEMINIMUM:
 				print()
 				diag("REBOOT request from GPIO{0}, recycling Rpi ...".format(SHUTDOWN))
-				os.system("reboot")
+				handle_press("reboot")
 				sys.exit()
 			if GPIO.input(SHUTDOWN): # before looping we must make sure the shutdown signal went low
 				GPIO.wait_for_edge(SHUTDOWN, GPIO.FALLING)