X-Git-Url: https://iankelling.org/git/?p=automated-distro-installer;a=blobdiff_plain;f=dsfull;h=c5cb230546219ffd0d9198e0fcf6892ee07f905d;hp=f141bc508a2f7a7012f96aaa06ea96f99cee8296;hb=57f8fa6a5b08bafdb6a422b6837d9b8cb3f7c396;hpb=d9b78f72ce932fff58c4b031f35f8722d8dbae33

diff --git a/dsfull b/dsfull
index f141bc5..c5cb230 100755
--- a/dsfull
+++ b/dsfull
@@ -7,6 +7,12 @@
 set -eE -o pipefail
 trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR
 
+reboot=true
+if [[ $1 == -r ]]; then
+    reboot=false
+    shift
+fi
+
 host=$1
 
 if [[ ! $host || $host == -h ]]; then
@@ -15,14 +21,15 @@ if [[ ! $host || $host == -h ]]; then
 fi
 
 set -x
-ssh $host sudo bash <<'EOF'
-set -e
-systemctl stop keyscriptoff.service
+if $reboot; then
+    ssh $host sudo bash <<'EOF' || [[ $? == 255 ]]
+touch /tmp/keyscript-off
 reboot ||:
 EOF
+fi
 
 pxe-server fai $host
 while ! ssh $host :; do
-      sleep 5
+    sleep 5
 done
 dsremote $host