#!/bin/env fish
function runSetup
    set -l dir (status dirname)
    source "$dir/../Scripts/config.fish"
    source "$dir/../Scripts/hooks.fish"
    set -l mountDir (getConfig valhalla.partition.rootDir)
    set -l projectRoot (realpath "$dir/../../..")
    set -l projectName (basename "$projectRoot")
    set -l PROJECT_CLONE_ROOT "/opt/$(basename "$projectName")"
    set -l script (mktemp)

    if not type -q runChroot
        function runChroot -S
            command chroot $argv
        end
    end

    if not type -q installValhallaDeps
        function installValhallaDeps
            echo 'Please set up the hook `installValhallaDeps` for installing `fish`, `nix` and `tmux`.'
            exit 1
        end
    end

    function getCloneFile -S -a path
        set -l relativeDir (realpath --relative-to "$projectRoot" "$dir")
        set -l relativePath (realpath --relative-to "$dir" "$path")
        echo "$PROJECT_CLONE_ROOT/$relativeDir/$relativePath"
    end

    function runInOS -S
        set -l script "/root/run_once"
        wrapScript $argv | chroot "$mountDir" tee "$script" > /dev/null
        runChroot "$mountDir" chmod +x "$script"
        runChroot "$mountDir" "$script"
        runChroot "$mountDir" rm "$script"
    end

    function wrapScript -S
        printf %s\n \
            "cd $PROJECT_CLONE_ROOT" \
            "export CONFIG_MODULE=$(string escape (getCloneFile "$CONFIG_MODULE"))" \
            "$argv"
    end

    getConfig valhalla.partition.script > "$script"
    bash "$script"

    # Copy `nixpkgs` channel
    and begin
        set -l channelDir /nix/var/nix/profiles/per-user/root/channels/nixpkgs
        mkdir -p (dirname "$mountDir/$channelDir")
        cp -r "$channelDir" "$mountDir/$channelDir"
    end

    and runHook installValhallaDeps
    and source "$dir/../../copy-repo.fish" "$mountDir$PROJECT_CLONE_ROOT"
    and runHook setupOS || true
    and runHook autologin || true

    and begin
        set -l script (string escape (getCloneFile (getInstallerScript)))
        wrapScript "INSTALLER_SCRIPT=$(string escape "$script") $script"
    end | runChroot "$mountDir" tee /root/.bash_profile > /dev/null

    and echo "Setup finished!"
    and echo "This machine will reboot in 5 seconds..."
    and echo "Press CTRL-C to abort..."
    and sleep 5
    and systemctl reboot
end