install.sh

set -e

#########################################################################################
# Functions
#########################################################################################

command_exists() {
    command -v "$1" &> /dev/null
}

log_info() {
    if command_exists printf
    then
        printf "🤖 \e[0;33m${1}\e[0;0m\n"
    else
        echo "🤖 ${1}"
    fi
}

confirm() {
    # Prompt the user for confirmation
    log_info "$1 (y/N)"
    read choice
    case "$choice" in
        y|Y ) return 0;;
        n|N ) return 1;;
        * ) echo "Invalid choice"; return 1;;
    esac
}

try_sudo() {
    if command_exists sudo
    then
        sudo $@
    else
        $@
    fi
}

register_pyenv() {
    log_info "Registering Pyenv to $1"
    echo 'if [ -d "${HOME}/.pyenv" ]' >> $1
    echo 'then' >> $1
    echo '    export PYENV_ROOT="$HOME/.pyenv"' >> $1
    echo '    export PATH="$PYENV_ROOT/bin:$PATH"' >> $1
    echo '    eval "$(pyenv init --path)"' >> $1
    echo 'fi' >> $1
}

register_local_venv() {
    log_info "Registering .local-venv to $1"
    echo 'if [ -f "${HOME}/.local-venv/bin/activate" ]' >> $1
    echo 'then' >> $1
    echo '    source "${HOME}/.local-venv/bin/activate"' >> $1
    echo 'fi' >> $1
}

create_and_register_local_venv() {
    log_info "Creating local venv"
    python -m venv $HOME/.local-venv
    . $HOME/.local-venv/bin/activate
    
    # register local venv to .zshrc
    if command_exists zsh
    then
        register_local_venv "$HOME/.zshrc"
    fi
    # register local venv to .bashrc
    if command_exists bash
    then
        register_local_venv "$HOME/.bashrc"
    fi
}

install_pyenv() {
    log_info "Installing pyenv"
    curl https://pyenv.run | bash

    # register .pyenv to .zshrc
    if [ -f "$HOME/.zshrc" ]
    then
        register_pyenv "$HOME/.zshrc"
    fi
    # register .pyenv to .bashrc
    if [ -f "$HOME/.bashrc" ]
    then
        register_pyenv "$HOME/.bashrc"
    fi
    # activate pyenv
    log_info "Activating pyenv"
    export PYENV_ROOT="$HOME/.pyenv"
    export PATH="$PYENV_ROOT/bin:$PATH"
    eval "$(pyenv init -)"
    eval "$(pyenv virtualenv-init -)"
}

install_python_on_pyenv() {
    # install python 3.10.0
    log_info "Installing python 3.10.0"
    pyenv install 3.10.0
    # set global python to 3.10.0
    log_info "Setting python 3.10.0 as global"
    pyenv global 3.10.0
}

install_poetry() {
    log_info "Installing Poetry"
    pip install --upgrade pip setuptools
    pip install poetry
}

install_zrb() {
    log_info "Installing Zrb"
    eval pip install --pre zrb
}


#########################################################################################
# Getting variables
#########################################################################################
OS_TYPE=$(uname)

if [ -n "$PREFIX" ] && [ "$PREFIX" = "/data/data/com.termux/files/usr" ]
then
    IS_TERMUX=1
else
    IS_TERMUX=0
fi

IS_PYENV_INSTALLED=0
IS_LOCAL_VENV_INSTALLED=0

#########################################################################################
# Installation
#########################################################################################

if [ "$IS_TERMUX" = "1" ] && [ ! -d "$HOME/.local-venv" ] && confirm "Do you want to setup termux?"
then
    log_info "Setting environment variables"
    export CFLAGS="-Wno-incompatible-function-pointer-types" # ruamel.yaml need this.

    log_info "Change repo"
    termux-change-repo

    log_info "Updating packages"
    pkg update
    pkg upgrade -y

    if [ ! -d "${HOME}/storage" ]
    then
        log_info "Setup storage"
        termux-setup-storage
    fi

    log_info "Installing prerequisites"
    pkg install termux-api openssh curl wget git which \
        python rust clang cmake build-essential golang swig \
        binutils ninja patchelf libxml2 libxslt \
        postgresql sqlite

elif [ "$IS_TERMUX" = "0" ] && confirm "Do you want to install pyenv?"
then
    if [ ! -d "$HOME/.pyenv" ]
    then
        # Install prerequisites

        if [ "$OS_TYPE" = "Darwin" ]
        then
            if command_exists brew
            then
                log_info "Using brew to install pyenv prerequisites"
                brew install openssl readline sqlite3 xz zlib tcl-tk
            else
                log_info "Brew not found, continuing anyway"
            fi
        elif [ "$OS_TYPE" = "Linux" ]
        then
            if command_exists pkg
            then
                log_info "Using pkg to install pyenv prerequisites"
                try_sudo pkg update
                try_sudo pkg install -y build-essential
            elif command_exists apt
            then
                log_info "Using apt to install pyenv prerequisites"
                try_sudo apt update
                try_sudo apt install -y build-essential libssl-dev zlib1g-dev \
                    libbz2-dev libreadline-dev libsqlite3-dev curl \
                    libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
            elif command_exists yum
            then
                log_info "Using yum to install pyenv prerequisites"
                try_sudo yum install -y gcc make patch zlib-devel bzip2 bzip2-devel readline-devel sqlite sqlite-devel openssl-devel tk-devel libffi-devel xz-devel
            elif command_exists dnf
            then
                log_info "Using dnf to install pyenv prerequisites"
                try_sudo dnf install -y make gcc patch zlib-devel bzip2 bzip2-devel readline-devel sqlite sqlite-devel openssl-devel tk-devel libffi-devel xz-devel libuuid-devel gdbm-libs libnsl2
            elif command_exists pacman
            then
                log_info "Using pacman to install pyenv prerequisites"
                try_sudo pacman -syu --noconfirm base-devel openssl zlib xz tk
            elif command_exists apk
            then
                log_info "Using apk to install pyenv prerequisites"
                try_sudo apk add --no-cache git bash build-base libffi-dev openssl-dev bzip2-dev zlib-dev xz-dev readline-dev sqlite-dev tk-dev
            else
                log_info "No known package manager found, continuing anyway"
            fi
        else
            log_info "Unsupported OS, cannot install pyenv pre-requisites, continuing anyway"
        fi
    fi

    if ! command_exists pyenv
    then
        install_pyenv
        IS_PYENV_INSTALLED=1
    fi

    if ! command_exists python
    then
        install_python_on_pyenv
    fi
elif ! command_exists python
then
    log_info "You need to install Python first."
    log_info "Exiting"
    exit 1
fi

if ! command_exists poetry && confirm "Do you want to install poetry?"
then
    install_poetry
fi

if [ ! -d "${HOME}/.local-venv" ] && confirm "Do you want to create virtual environment?"
then
    create_and_register_local_venv
    IS_LOCAL_VENV_INSTALLED=1
fi

if ! command_exists zrb
then
    install_zrb
fi

if [ "$IS_PYENV_INSTALLED" = 1 ] || [ "$IS_LOCAL_VENV_INSTALLED" = 1 ]
then
    log_info "You need to restart your terminal session!!!"
fi