Ryan Hughes
d6e4060d56
Graceful guest poweroff over SSH first, escalating to TERM then KILL; --kill skips straight to force. Finds the newest live run by its pidfile and confirms before acting when run from a terminal, since a harness still driving the VM dies with it.
102 lines
2.5 KiB
Bash
Executable File
102 lines
2.5 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Stop the VM a `omarchy-iso-test --keep-running` run left behind: graceful
|
|
# guest poweroff over SSH first, escalating to TERM then KILL. The base image
|
|
# is untouched either way; only the throwaway run overlay is affected.
|
|
#
|
|
# Usage: omarchy-iso-test-stop [--kill] [--port PORT]
|
|
#
|
|
# --kill Skip the graceful poweroff and kill the VM immediately
|
|
# --port PORT Host port forwarded to guest SSH (default 2222)
|
|
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
REPO_DIR="$(dirname "$SCRIPT_DIR")"
|
|
|
|
GUEST_USER="omarchy"
|
|
GUEST_PASSWORD="omarchy"
|
|
SSH_PORT=2222
|
|
GRACEFUL=true
|
|
|
|
while (($#)); do
|
|
case "$1" in
|
|
--kill) GRACEFUL=false ;;
|
|
--port) SSH_PORT="$2"; shift ;;
|
|
-h | --help)
|
|
sed -n '3,10p' "${BASH_SOURCE[0]}" | sed 's/^# \{0,1\}//'
|
|
exit 0
|
|
;;
|
|
*) echo "Unknown option: $1" >&2; exit 1 ;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
# The newest run dir whose qemu is still alive is the VM to stop.
|
|
PID=""
|
|
PIDFILE=""
|
|
while IFS= read -r candidate; do
|
|
pid=$(<"$candidate")
|
|
if kill -0 "$pid" 2>/dev/null; then
|
|
PID="$pid"
|
|
PIDFILE="$candidate"
|
|
break
|
|
fi
|
|
done < <(ls -t "$REPO_DIR"/test-runs/*/runs/*/qemu.pid 2>/dev/null)
|
|
|
|
if [[ -z $PID ]]; then
|
|
echo "No running omarchy-iso-test VM found." >&2
|
|
exit 1
|
|
fi
|
|
|
|
BASE_DIR="$(dirname "$(dirname "$(dirname "$PIDFILE")")")"
|
|
SSH_KEY="$BASE_DIR/id_ed25519"
|
|
|
|
# A harness that is still driving this VM dies with it; make sure that is
|
|
# what the caller wants before pulling anything down.
|
|
if [[ -t 0 ]]; then
|
|
if pgrep -f "bin/omarchy-iso-test .*\.iso" >/dev/null; then
|
|
echo "Warning: an omarchy-iso-test run appears to be in progress."
|
|
fi
|
|
if ! gum confirm "Stop test VM (pid $PID, ${PIDFILE#$REPO_DIR/})?"; then
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
echo "Stopping test VM (pid $PID, ${PIDFILE#$REPO_DIR/})"
|
|
|
|
if $GRACEFUL && [[ -f $SSH_KEY ]]; then
|
|
ssh -i "$SSH_KEY" -p "$SSH_PORT" \
|
|
-o BatchMode=yes \
|
|
-o IdentitiesOnly=yes \
|
|
-o StrictHostKeyChecking=no \
|
|
-o UserKnownHostsFile=/dev/null \
|
|
-o ConnectTimeout=5 \
|
|
-o LogLevel=ERROR \
|
|
"$GUEST_USER@127.0.0.1" \
|
|
"echo $GUEST_PASSWORD | sudo -S systemctl poweroff" >/dev/null 2>&1 || true
|
|
|
|
waited=0
|
|
while kill -0 "$PID" 2>/dev/null && ((waited < 15)); do
|
|
sleep 1
|
|
((waited += 1))
|
|
done
|
|
fi
|
|
|
|
if kill -0 "$PID" 2>/dev/null; then
|
|
kill "$PID" 2>/dev/null || true
|
|
sleep 1
|
|
fi
|
|
|
|
if kill -0 "$PID" 2>/dev/null; then
|
|
kill -9 "$PID" 2>/dev/null || true
|
|
sleep 1
|
|
fi
|
|
|
|
if kill -0 "$PID" 2>/dev/null; then
|
|
echo "VM (pid $PID) refused to die." >&2
|
|
exit 1
|
|
fi
|
|
|
|
echo "VM stopped; port $SSH_PORT is free."
|