#!/bin/bash

STEP_COUNT=0

# Helpers to pretty-print step titles and count.
pre_step() {
    STEP_COUNT=$(($STEP_COUNT + 1))
    echo -e "\e[1m[$STEP_COUNT/$TOTAL_STEPS] <====== $1 =======>\e[0m"
}

# Ensures that all existing Apt packages are up-to-date.
ensure_apt_up_to_date() {
    pre_step "Ensuring apt packages are up-to-date" 
    
    if [ -z "$(apt --version 2> /dev/null)" ]; then
        echo -e "\e[33mApt not installed, skipping updates.\e[0m"
    else
        echo -e "\e[1mEnsuring system packages are up-to-date...\e[0m"
        
        sudo apt update && sudo apt upgrade -y --autoremove

        echo -e "\e[1;32mSystem packages up-to-date.\e[0m"
    fi
}

# Installs packages as specified in apt.txt
ensure_apt_dependencies() {
    pre_step "Installing extra packages"

    if [ -z "$(apt --version 2> /dev/null)" ]; then
        echo -e "\e[33mApt not installed, skipping packages.\e[0m"
    else
        echo -e "\e[1mInstalling packages...\e[0m"
        
        sudo apt install $(cat ./apt.txt) -y --autoremove

        echo -e "\e[1;32mSystem packages installed.\e[0m"
    fi
}

ensure_nvm() {
    pre_step "Installing nvm to manage node version"
    curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.7/install.sh | bash

    # Ensuring that nvm is usable right away without restarting the shell
    export NVM_DIR="$HOME/.nvm"
    [ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"
    [ -s "$NVM_DIR/bash_completion" ] && \. "$NVM_DIR/bash_completion" 
}

# Injects a managed block in the shell configuration.
inject_shell_configuration() {
    pre_step "Injecting managed block in shell configuration"

    BLOCK_DELIMITER_PATTERN="mcataford/env"
    WORKING_PATH=$(git rev-parse --show-toplevel)
    SHELL_CONFIG_PATH="$HOME/.zshrc"

    echo "Setting up shell configuration extras..."
    if [[ -z $(cat $SHELL_CONFIG_PATH | grep $BLOCK_DELIMITER_PATTERN) ]]; then
        echo "# $BLOCK_DELIMITER_PATTERN\:start
    source $WORKING_PATH/shell_extras
    # $BLOCK_DELIMITER_PATTERN\:end" >> $SHELL_CONFIG_PATH
        echo "✅ Added managed block to $SHELL_CONFIG_PATH"
    else
        echo "No changes to apply!"
    fi

    EDITOR_CONFIG=$HOME/.config/nvim
    EDITOR_CONFIG_FILE=$EDITOR_CONFIG/init.vim
}

# Injects a managed block in the vim configuration
inject_vim_configuration() {
    pre_step "Inject managed block in vim configuration"

    if [[ -f $EDITOR_CONFIG_FILE ]]; then
        echo "Setting up NVIM configuration extras..."
        if [[ -z $(cat $EDITOR_CONFIG_FILE | grep $BLOCK_DELIMITER_PATTERN) ]]; then
            echo "\" $BLOCK_DELIMITER_PATTERN\:start
    source $WORKING_PATH/extras.vim
    \" $BLOCK_DELIMITER_PATTERN\:end\n\n" >> $EDITOR_CONFIG_FILE.new
            cat $EDITOR_CONFIG_FILE >> $EDITOR_CONFIG_FILE.new

            mv $EDITOR_CONFIG_FILE $EDITOR_CONFIG_FILE.old
            mv $EDITOR_CONFIG_FILE.new $EDITOR_CONFIG_FILE
            echo "✅ Added managed block to $EDITOR_CONFIG_FILE"
        else
            echo "No changes to apply!"
        fi
    fi

    echo "Setting up git configuration..."
    source $WORKING_PATH/git_config
    echo "✅ Set up git configuration, see $WORKING_PATH/git_config for details!"
}