#!/usr/bin/env sh

# Author: caem - https://caem.dev

# install.sh - Installation script for my NixOS configuration
#
# This script aims to automate the deployment of my configuration
# on a new machine. 

args_ensure_extra_arg() {
    if [ -z "$2" ] || [ "$(echo "$2" | cut -c 1-1)" = "-" ]; then
        >&2 echo "Argument '$1' requires an extra argument. Run --help for more info."
        exit 1
    fi
}

args_ensure_is_set() {
    if [ -z "$2" ]; then
        >&2 echo "Argument '$1' is required to be set. Please consult the README or run again with --help."
        exit 1
    fi
}

args() {
    while [ -n "$1" ]; do
        case "$1" in
            "-h" | "--help")
                echo ""
                echo "$0 - Installation script for my NixOS configuration"
                echo ""
                echo "arguments:"
                echo "  -h|--help                           Print this and exit"
                echo "  -d|--device [device]    (required)  The device you want to install NixOS on to"
                echo "  -o|--host   [hostname]  (required)  The host from ./hosts you want to install"
                echo "  -b|--build                          Build the system without installing"
                echo ""
                echo "origin: https://github.com/c4em/dotnix"
                echo ""
                exit 0
                ;;

            "-d" | "--device")
                args_ensure_extra_arg "$@"
                if [ ! -b "$2" ]; then
                    >&2 echo "'$2' is not a valid block device. Make sure you selected the right drive"
                    exit 1
                fi

                DOTNIX_INSTALL_DEVICE="$2"
                shift 2
                ;;

            "-o" | "--host")
                args_ensure_extra_arg "$@"

                if [ ! -d "./hosts/$2" ]; then
                    >&2 echo "Invalid hostname '$2'. Make sure it exists in ./hosts"
                    exit 1
                fi

                DOTNIX_HOSTNAME="$2"
                shift 2
                ;;
            "-b" | "--build")
                DOTNIX_DO_ONLY_BUILD=1
                shift 1
                ;;

            *)
                >&2 echo "Unrecognized argument '$1'. Run with --help to view accepted arguments."
                exit 1
                ;;
        esac
    done

    args_ensure_is_set "--host" "$DOTNIX_HOSTNAME"
    if [ -z "$DOTNIX_DO_ONLY_BUILD" ]; then
        args_ensure_is_set "--device" "$DOTNIX_INSTALL_DEVICE"
    fi
}

sed_safe () {
    # I got this off of some random StackOverflow answer. Don't put too much trust in this.
    printf "%s" "$1" | sed -r 's/([\$\.\*\/\[\\^])/\\\1/g' | sed 's/[]]/\[]]/g'
}

update_managed_values() {
    sed -i 's/\( *device = \)".*"\(; #.*\)/\1"'"$(sed_safe "$DOTNIX_INSTALL_DEVICE")"'"\2/' "./hosts/$DOTNIX_HOSTNAME/default.nix"
    sed -i 's/\( *device = \)".*"\(; #.*\)/\1"'"$(sed_safe "$DOTNIX_INSTALL_DEVICE")"'"\2/' "./hosts/$DOTNIX_HOSTNAME/disko.nix"
    sed -i 's/\( *system.stateVersion = \)".*"\(; #.*\)/\1"'"$(sed_safe "$(nixos-version | cut -f1,2 -d '.')")"'"\2/' "./hosts/$DOTNIX_HOSTNAME/default.nix"
}

build() {
    nix build ".#nixosConfigurations.${DOTNIX_HOSTNAME}.config.system.build.toplevel"
}

permissions() {
    if [ "$(id -u)" = "0" ]; then
        sudo () {
            true
        }
    else
        sudo -v
    fi
}

ensure_confirmation() {
    printf "\e[1;31m=== ARE YOU SURE YOU WANT TO CONTINUE WITH THE INSTALLATION ===\e[0m\n\n"
    printf "This will \e[1;31mIRREVERSIBLY\e[0m wipe all data in '%s'\n" "$DOTNIX_INSTALL_DEVICE"
    printf "This disk contains following partitions:\n\n"
    lsblk -o NAME,SIZE,TYPE,FSTYPE "$DOTNIX_INSTALL_DEVICE"
    printf "\n"
    lsblk -no NAME "$DOTNIX_INSTALL_DEVICE" | tail -n +2 | tr -cd '[:alnum:][:space:]' | xargs -I {} -- df -h "/dev/{}"
    printf "\n"

    printf "Please write 'Yes, do as I say!' to continue with the installation\n> "
    read -r install_prompt
    if [ "$install_prompt" != "Yes, do as I say!" ]; then
        echo "Cancelling installation"
        exit 0
    else
        DOTNIX_CONFIRM_DISK_NUKE="yes"
    fi
}

partition_disk() {
    if [ "$DOTNIX_CONFIRM_DISK_NUKE" = "yes" ]; then
        sudo nix --experimental-features 'flakes nix-command' run --github:nix-community/disko/latest \
            -- --mode destroy,format,mount --yes-wipe-all-disks "./hosts/$DOTNIX_HOSTNAME/disko.nix"
    else
        >&2 echo "Aborted installation due to invalid state in the partitioning step."
        exit 1
    fi
}

generate_config() {
    nixos-generate-config --no-filesystems --root /mnt
}

main () {
    args "$@"
    permissions

    if [ -n "$DOTNIX_DO_ONLY_BUILD" ]; then
        if [ -n "$DOTNIX_INSTALL_DEVICE" ]; then
            update_managed_values
        fi

        build
        exit 0
    fi

    ensure_confirmation
    update_managed_values
    partition_disk
    generate_config

    # todo install
}

set -e
main "$@"