#! /usr/bin/env dash

# shellcheck source=/dev/null
SHELL_LIBRARY_VERSION="1.6.0" . %SHELL_LIBRARY_PATH

INSTALL=y . %SCRIPT_ENSURE_CONFIG_VARIABLES

msg "Important information:"
msg2 "Choose disk id is: $NIX_DISK"
warning "This information needs to be entered in the host configuration BEFORE rebuilding it. Otherwise you'll have to wait for a 1 h cooldown!"
info_applied=false
while [ "$info_applied" = false ]; do
    readp "Have you updated your chosen host with the choose disk id? [N/y]: " result
    case $result in
        [Yy])
            info_applied=true
            ;;
        *)
            warning "You won't be able to boot, if you don't update it!"
            ;;
    esac
done

msg "Started creating filesystem..."
nix run github:nix-community/disko --experimental-features 'nix-command flakes' -- --flake "git+https://codeberg.org/soispha/nixos-config#$NIX_HOST" --mode disko
msg "Finished mounting and generating btrfs subvolumes"

msg "Creating swapfile..."
msg2 "$(btrfs filesystem mkswapfile --size "$(free -m | awk '{if (NR==2) {printf "%sm\n", $2 + 500}}')" /mnt/swap/swapfile)"
msg "Finished creating swapfile!"

msg "Important information:"
msg2 "Swapfile UUID is: $(findmnt -no UUID -T /mnt/swap/swapfile)"
msg2 "Swapfile resume offset is: $(btrfs inspect-internal map-swapfile -r /mnt/swap/swapfile)"
warning "This information needs to be entered in the host configuration BEFORE rebuilding it. Otherwise the system won't boot!"

info_applied=false
while [ "$info_applied" = false ]; do
    readp "Have you updated your chosen host with this information? [N/y]: " result
    case $result in
        [Yy])
            info_applied=true
            ;;
        *)
            warning "You won't be able to boot, if you don't update it!"
            ;;
    esac
done

msg "Checking for incompatibilities..."
ssd_or_hdd="$(cat /sys/block/"$(basename "$(readlink -f "/dev/disk/by-id/$NIX_DISK")" | tr -d '\n')"/queue/rotational)"
case "$ssd_or_hdd" in
    0)
        msg2 "You seem to use a ssd."
        trim_support=$(lsblk --bytes --json --discard | jq --arg name "$(basename "$(readlink -f "/dev/disk/by-id/$NIX_DISK")" | tr -d '\n')" '.blockdevices | .[] | select(.name == $name) | (.["disc-gran"] + .["disc-max"]) != 0')
        case $trim_support in
            "true")
                msg2 "Yay, your ssd supports trim, go on and activate it"
                ;;
            "false")
                msg2 "Nay, your ssd doesn't support trim"
                ;;
            *)
                warning "Your ssd doesn't seem to exists, if this bothers you please open an issue."
                ;;
        esac
        ;;
    1)
        msg2 "You seem to use a hdd there is nothing you have to do"
        ;;
    *)
        warning "There is no indicator, which shows, which drive your are using.\n This means, that you have to check yourself, which optimizations you should activate."
        ;;
esac

case "$INSTALL_TO_SETUP" in
    "yes")
        nix run "git+https://codeberg.org/soispha/nixos-config#setup" --experimental-features 'nix-command flakes'
        ;;
    *) ;;
esac