tristan/nixinate
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Packages Wiki Activity

Local part of magic rollbacks.

Browse source
This commit is contained in:
Chloe Kever 2022-01-28 16:43:36 -08:00
parent 8f960401c4
commit 14124ca6f8
1 changed files with 30 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
flake.nix
View file

@ -17,7 +17,29 @@
let
machines = builtins.attrNames flake.nixosConfigurations;
validMachines = final.lib.remove "" (final.lib.forEach machines (x: final.lib.optionalString (flake.nixosConfigurations."${x}"._module.args ? nixinate) "${x}" ));
mkDeployScript = machine: final.writeScript "deploy-${machine}.sh" ''
mkDeployScript = machine: let
rollback = flake.nixosConfigurations.${machine}._module.args.nixinate.rollback or {};
rollbackEnabled = rollback.enable or true;
rollbackInit = rollback.init or 500;
rollbackLimit = rollback.limit or 8;
rollbackTimeout = rollback.timeout or 10;
rollbackScript = if rollbackEnabled then ''
racc=${toString rollbackLimit}
exp=0
until ${final.openssh}/bin/ssh -o ConnectTimeout=${toString rollbackTimeout} -t $SSH_USER@$SSH_HOST 'sudo rm /tmp/.nixinate-deploy-succeeded' &> /dev/null; do
rbp=$((${toString rollbackInit} * (2 ** $exp)))
echo "Could not access ${machine}, trying again in $rbp milliseconds."
sleep $(($rbp / 1000)
exp=$(($exp + 1))
racc=$(($racc - 1))
if [[ $racc == 0 ]]; then
echo "Cannot access ${machine}. Rollback will happen." >&2
exit 1
fi
done
echo "${machine} has finished deploying."
'' else "";
in final.writeScript "deploy-${machine}.sh" ''
set -e
SSH_USER=${flake.nixosConfigurations.${machine}._module.args.nixinate.sshUser}
SSH_HOST=${flake.nixosConfigurations.${machine}._module.args.nixinate.host}
@ -32,12 +54,14 @@
( set -x; ${final.rsync}/bin/rsync -q -vz --recursive --zc=zstd ${flake}/* $SSH_USER@$SSH_HOST:/tmp/nixcfg/ )
echo "🤞 Activating configuration on ${machine} via ssh:"
( set -x; ${final.openssh}/bin/ssh -t $SSH_USER@$SSH_HOST 'sudo nixos-rebuild switch --flake /tmp/nixcfg#${machine}' )
${rollbackScript}
elif [ $BUILD_ON = "local" ]; then
echo "🔨 Building system closure locally and copying it to remote store:"
( set -x; ${final.nixFlakes}/bin/nix copy --to ssh://$SSH_USER@$SSH_HOST $SYSTEM_CLOSURE )
echo "🤞 Activating configuration on ${machine} via ssh:"
SYSTEM_CLOSURE_PATH=$(${final.nixFlakes}/bin/nix path-info $SYSTEM_CLOSURE)
( set -x; ${final.openssh}/bin/ssh -t $SSH_USER@$SSH_HOST "sudo $SYSTEM_CLOSURE_PATH/bin/switch-to-configuration switch" )
${rollbackScript}
else
echo "_module.args.nixinate.buildOn is not set to a valid value of 'local' or 'remote'"
fi