Fix power action terminal scope race

Co-authored-by: Jon Kinney <jonkinney@gmail.com>

bin/omarchy-system-reboot

@@ -4,10 +4,11 @@
 # omarchy:examples=omarchy reboot | omarchy system reboot
 # omarchy:aliases=omarchy reboot
 
-omarchy-state clear re*-required
+# Schedule the reboot in the user manager so closing the terminal's systemd
+# scope cannot terminate it before it runs.
+systemd-run --user --collect --quiet --on-active="2s" systemctl reboot --no-wall || exit 1
 
-# Schedule the reboot to happen after closing windows (detached from terminal)
-nohup bash -c "sleep 2 && systemctl reboot --no-wall" >/dev/null 2>&1 &
+omarchy-state clear re*-required
 
 # Now close all windows
 omarchy-hyprland-window-close-all

bin/omarchy-system-shutdown

@@ -4,10 +4,11 @@
 # omarchy:examples=omarchy shutdown | omarchy system shutdown
 # omarchy:aliases=omarchy shutdown
 
-omarchy-state clear re*-required
+# Schedule the shutdown in the user manager so closing the terminal's systemd
+# scope cannot terminate it before it runs.
+systemd-run --user --collect --quiet --on-active="2s" systemctl poweroff --no-wall || exit 1
 
-# Schedule the shutdown to happen after closing windows (detached from terminal)
-nohup bash -c "sleep 2 && systemctl poweroff --no-wall" >/dev/null 2>&1 &
+omarchy-state clear re*-required
 
 # Now close all windows
 omarchy-hyprland-window-close-all

test/shell.d/system-power-test.sh

@@ -0,0 +1,68 @@
+#!/bin/bash
+
+source "$(dirname "$0")/base-test.sh"
+
+test_tmp=$(mktemp -d)
+trap 'rm -rf "$test_tmp"' EXIT
+
+mock_bin="$test_tmp/bin"
+call_log="$test_tmp/calls.log"
+mkdir -p "$mock_bin"
+
+cat >"$mock_bin/systemd-run" <<'SH'
+#!/bin/bash
+
+printf 'systemd-run %s\n' "$*" >>"$CALL_LOG"
+[[ ${FAIL_SYSTEMD_RUN:-false} == "true" ]] && exit 1
+exit 0
+SH
+
+for command in omarchy-state omarchy-hyprland-window-close-all sleep; do
+  cat >"$mock_bin/$command" <<'SH'
+#!/bin/bash
+
+printf '%s %s\n' "$(basename "$0")" "$*" >>"$CALL_LOG"
+SH
+done
+chmod +x "$mock_bin"/*
+
+run_power_command() {
+  local action="$1"
+
+  : >"$call_log"
+  PATH="$mock_bin:$PATH" CALL_LOG="$call_log" "$ROOT/bin/omarchy-system-$action"
+}
+
+assert_power_calls() {
+  local action="$1"
+  local systemctl_action="$2"
+  local expected_log="$test_tmp/$action-expected.log"
+
+  cat >"$expected_log" <<EOF
+systemd-run --user --collect --quiet --on-active=2s systemctl $systemctl_action --no-wall
+omarchy-state clear re*-required
+omarchy-hyprland-window-close-all 
+sleep 1
+EOF
+
+  diff -u "$expected_log" "$call_log" || fail "$action runs after being scheduled outside the terminal scope"
+  pass "$action runs after being scheduled outside the terminal scope"
+}
+
+run_power_command reboot
+assert_power_calls reboot reboot
+
+run_power_command shutdown
+assert_power_calls shutdown poweroff
+
+for action in reboot shutdown; do
+  : >"$call_log"
+  if PATH="$mock_bin:$PATH" CALL_LOG="$call_log" FAIL_SYSTEMD_RUN=true "$ROOT/bin/omarchy-system-$action"; then
+    fail "$action aborts when scheduling fails"
+  fi
+
+  if (( $(wc -l <"$call_log") != 1 )); then
+    fail "$action leaves state and windows alone when scheduling fails"
+  fi
+  pass "$action leaves state and windows alone when scheduling fails"
+done