#!/bin/bash

# TODO we might want to implement all of this in Python/Rust in the future for platform-independence etc

kybra_version="$1"
rust_version="$2"

global_kybra_config_dir=~/.config/kybra
global_kybra_version_dir="$global_kybra_config_dir"/"$kybra_version"
global_kybra_rust_dir="$global_kybra_config_dir"/rust/"$rust_version"
global_kybra_rust_bin_dir="$global_kybra_rust_dir"/bin
global_kybra_logs_dir="$global_kybra_version_dir"/logs
global_kybra_cargo_bin="$global_kybra_rust_bin_dir"/cargo
global_kybra_rustup_bin="$global_kybra_rust_bin_dir"/rustup
global_kybra_wasi2ic_bin="$global_kybra_rust_bin_dir"/wasi2ic
global_kybra_rustc_bin="$global_kybra_rust_bin_dir"/rustc
global_kybra_cargo_binstall_bin="$global_kybra_rust_bin_dir"/cargo-binstall
global_kybra_candid_extractor_bin="$global_kybra_rust_bin_dir"/candid-extractor

export CARGO_TARGET_DIR="$global_kybra_config_dir"/rust/target
export CARGO_HOME="$global_kybra_rust_dir"
export RUSTUP_HOME="$global_kybra_rust_dir"

function run() {
    if ! rustup_exists || ! cargo_exists || ! rustc_exists || ! wasm32_wasip1_target_installed || ! wasi2ic_exists || ! candid_extractor_exists || ! cargo_binstall_exists; then
        echo -e "\nKybra "$kybra_version" prerequisite installation (this may take a few minutes)\n"

        mkdir -p "$global_kybra_rust_dir"
        mkdir -p "$global_kybra_logs_dir"

        install_rustup
        install_wasm32_wasip1
        install_wasi2ic
        install_candid_extractor
    else
        update_rustup
    fi
}

function install_rustup() {
    echo -e "1/4) Installing Rust"

    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- --no-modify-path -y --default-toolchain="$rust_version" &> "$global_kybra_logs_dir"/install_rustup
}

function update_rustup() {
    "$global_kybra_rustup_bin" update "$rust_version" &> "$global_kybra_logs_dir"/rustup_update
}

function install_wasm32_wasip1() {
    echo -e "2/4) Installing wasm32-wasip1"

    "$global_kybra_rustup_bin" target add wasm32-wasip1 &> "$global_kybra_logs_dir"/install_wasm32_wasip1
}

function install_wasi2ic() {
    echo -e "3/4) Installing wasi2ic"

    "$global_kybra_cargo_bin" install --git https://github.com/wasm-forge/wasi2ic --rev 7418e0bd1a7810c8e9c55cc0155c921503a793b8 &> "$global_kybra_logs_dir"/install_wasi2ic
}

function install_candid_extractor() {
    echo -e "4/4) Installing candid-extractor"

    # cargo binstall makes the installation process for candid-extractor much faster
    curl -L --proto '=https' --tlsv1.2 -sSf https://raw.githubusercontent.com/cargo-bins/cargo-binstall/main/install-from-binstall-release.sh | bash &> "$global_kybra_logs_dir"/install_cargo_binstall
    "$global_kybra_cargo_bin" binstall -y candid-extractor@0.1.2 &> "$global_kybra_logs_dir"/install_candid_extractor
}

function rustup_exists() {
    [ -e "$global_kybra_rustup_bin" ]
}

function cargo_exists() {
    [ -e "$global_kybra_cargo_bin" ]
}

function rustc_exists() {
    [ -e "$global_kybra_rustc_bin" ]
}

function wasm32_wasip1_target_installed() {
    $global_kybra_rustup_bin target list | grep -q "wasm32-wasip1 (installed)"
}

function wasi2ic_exists() {
    [ -e "$global_kybra_wasi2ic_bin" ]
}

function cargo_binstall_exists() {
    [ -e "$global_kybra_cargo_binstall_bin" ]
}

function candid_extractor_exists() {
    [ -e "$global_kybra_candid_extractor_bin" ]
}

run
