2022-11-10 16:42:06 +03:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
set -eufo pipefail
|
|
|
|
set -x
|
|
|
|
|
|
|
|
showUsage() {
|
|
|
|
cat <<USAGE
|
|
|
|
Usage: $0 [options] ssh-host
|
|
|
|
|
|
|
|
Options:
|
|
|
|
|
|
|
|
* -f, --flake flake
|
|
|
|
set the flake to install the system from
|
|
|
|
* --arg name value
|
|
|
|
pass value to nix-build. can be used to set disk-names for example
|
|
|
|
* --argstr name value
|
|
|
|
pass value to nix-build as string
|
2022-11-10 17:30:50 +03:00
|
|
|
* --kexec url
|
|
|
|
use another kexec tarball to bootstrap NixOS
|
2022-11-10 16:42:06 +03:00
|
|
|
USAGE
|
|
|
|
}
|
|
|
|
|
|
|
|
abort() {
|
|
|
|
echo "aborted: $*" >&2
|
|
|
|
exit 1
|
|
|
|
}
|
|
|
|
|
|
|
|
nix_args=()
|
2022-11-10 17:30:50 +03:00
|
|
|
kexec_url=https://github.com/nix-community/nixos-images/releases/download/nixos-22.05/nixos-kexec-installer-x86_64-linux.tar.gz
|
2022-11-10 16:42:06 +03:00
|
|
|
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
|
|
case "$1" in
|
|
|
|
-f | --flake)
|
|
|
|
flake=$2
|
|
|
|
shift
|
|
|
|
;;
|
|
|
|
--argstr | --arg)
|
|
|
|
nix_args+=("$1" "$2" "$3")
|
|
|
|
shift
|
|
|
|
shift
|
|
|
|
;;
|
|
|
|
--help)
|
|
|
|
showUsage
|
|
|
|
exit 0
|
|
|
|
;;
|
2022-11-10 17:30:50 +03:00
|
|
|
--kexec)
|
|
|
|
kexec_url=$2
|
|
|
|
shift
|
|
|
|
;;
|
2022-11-10 16:42:06 +03:00
|
|
|
*)
|
|
|
|
if [ -z ${ssh_connection+x} ]; then
|
|
|
|
ssh_connection=$1
|
|
|
|
else
|
|
|
|
showUsage
|
|
|
|
exit 1
|
|
|
|
fi
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
shift
|
|
|
|
done
|
|
|
|
|
|
|
|
# ssh wrapper
|
2022-11-10 18:45:14 +03:00
|
|
|
timeout_ssh_() {
|
2022-11-10 17:31:43 +03:00
|
|
|
timeout 10 ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no "$ssh_connection" "$@"
|
2022-11-10 16:42:06 +03:00
|
|
|
}
|
2022-11-10 18:45:14 +03:00
|
|
|
ssh_() {
|
|
|
|
ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no "$ssh_connection" "$@"
|
|
|
|
}
|
2022-11-10 16:42:06 +03:00
|
|
|
|
|
|
|
# first check if the remote system is kexec booted
|
2022-11-10 17:31:43 +03:00
|
|
|
if $(ssh_ -- test -e /etc/is-kexec); then
|
2022-11-10 16:42:06 +03:00
|
|
|
is_kexec=y
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
echo ${is_kexec-n}
|
|
|
|
if [ ${is_kexec-n} != "y" ]; then
|
|
|
|
# TODO we probably need an architecture detection here
|
2022-11-10 17:31:43 +03:00
|
|
|
ssh_ << SSH
|
2022-11-10 16:42:06 +03:00
|
|
|
set -efux
|
2022-11-10 19:15:00 +03:00
|
|
|
os=\$(uname)
|
|
|
|
if [[ "\$os" != "Linux" ]]; then
|
|
|
|
echo "This script requires Linux as the operating system, but got \${os}" >&2
|
|
|
|
exit 1
|
|
|
|
fi
|
2022-11-10 16:42:06 +03:00
|
|
|
fetch(){
|
|
|
|
if command -v curl >/dev/null 2>&1; then
|
|
|
|
curl --fail -Ss -L "\$1"
|
|
|
|
elif command -v wget >/dev/null 2>&1; then
|
|
|
|
wget "\$1" -O-
|
|
|
|
else
|
|
|
|
echo "no downloader (curl or wget) found, bailing out"
|
|
|
|
exit 1
|
|
|
|
fi
|
|
|
|
}
|
2022-11-10 18:46:53 +03:00
|
|
|
if command -v tar >/dev/null 2>&1; then
|
|
|
|
echo "no tar command found, but required to unpack kexec tarball" >&2
|
|
|
|
exit 1
|
|
|
|
fi
|
2022-11-10 16:42:06 +03:00
|
|
|
rm -rf /root/kexec
|
|
|
|
mkdir -p /root/kexec
|
2022-11-10 17:30:50 +03:00
|
|
|
fetch "$kexec_url" | tar -C /root/kexec -xvzf-
|
2022-11-10 16:42:06 +03:00
|
|
|
export TMPDIR=/root/kexec
|
|
|
|
setsid /root/kexec/kexec/run
|
|
|
|
SSH
|
|
|
|
# wait for machine to become unreachable
|
2022-11-10 18:45:14 +03:00
|
|
|
while timeout_ssh_ -- exit 0; do sleep 1; done
|
2022-11-10 16:42:06 +03:00
|
|
|
|
|
|
|
# watiting for machine to become available again
|
2022-11-10 17:31:43 +03:00
|
|
|
until ssh_ -o ConnectTimeout=10 -- exit 0; do sleep 5; done
|
2022-11-10 16:42:06 +03:00
|
|
|
fi
|
|
|
|
|
|
|
|
|
2022-11-10 17:31:43 +03:00
|
|
|
ssh_ << SSH
|
2022-11-10 16:42:06 +03:00
|
|
|
set -efux
|
|
|
|
$(declare -p nix_args)
|
|
|
|
nix --extra-experimental-features nix-command --extra-experimental-features flakes \
|
|
|
|
run github:nix-community/disko \
|
|
|
|
--no-write-lock-file -- \
|
|
|
|
--debug -m create "\${nix_args[@]}" --flake "$flake"
|
|
|
|
|
|
|
|
nix --extra-experimental-features nix-command --extra-experimental-features flakes \
|
|
|
|
run github:nix-community/disko \
|
|
|
|
--no-write-lock-file -- \
|
|
|
|
--debug -m mount "\${nix_args[@]}" --flake "$flake"
|
|
|
|
|
|
|
|
nixos-install --flake "$flake"
|
|
|
|
reboot
|
|
|
|
SSH
|