# Check if an argument is provided
if [ $# -eq 0 ]; then
    echo "Error: Please provide a valid hostname as an argument."
    exit 1
fi

HOST=$1
FLAKE=github:alyraffauf/nixcfg#$HOST

echo "Warning: Running this script will wipe the currently installed system."
read -p "Do you want to continue? (y/n): " answer

if [ "$answer" != "y" ]; then
    echo "Aborted."
    exit 0
fi

sudo nix --experimental-features "nix-command flakes" run \
    github:nix-community/disko -- --mode disko --flake $FLAKE

# Install NixOS with the updated flake input and root settings
sudo nixos-install --no-root-password --root /mnt --flake $FLAKE