#!/bin/bash

# Helium BOWSER INSTALL & UPDATE SCRIPT ---> INTERACTIVE

########## PREREQUISITES ###########
set -e

## SOME COLOR
CYN="\e[0;36m"
YEL="\e[0;33m"
RED="\e[0;31m"
GRN="\e[0;32m"
LCY="\e[1;36m"
CRS="\e[0m"

## GLOBAL VARIABLES
APP_NAME="Helium"
APP_NAME_LONG="Helium Web Browser"
SOURCE="GitHub"
SCRIPT_NAME="$APP_NAME Helper Script"
SCRIPT_FILE_NAME="helium_helper.sh"
AUTO_SCRIPT_FILE_NAME="helium_auto_update.sh"
SCRIPT_VERSION="0.5.5"
ALIAS="update-helium"

## LOCATIONS
TMP=/tmp/$APP_NAME
INSTALL_PATH=$HOME/.apps/$APP_NAME
CONFIG_PATH=$INSTALL_PATH/assets
BACKUP_PATH=$INSTALL_PATH/assets/backup
LOG_PATH=$INSTALL_PATH/assets/logs
LOG=$LOG_PATH/log
LAUNCHER_PATH=$HOME/.local/share/applications
SCRIPT_PATH=$HOME/.local/bin
#PROFILE=$HOME/.profile     ### for AUTO UPDATE

########## FUNCTIONS ###########

### DEPENDENCIES
cmd_query() { 
    local PACKAGE="$1"

    if ! command -v "$PACKAGE" >/dev/null 2>&1 ; then
        echo "    - $PACKAGE" >> $TMP/missingDeps

    fi
}

dependency_check () {
    DEPENDENCIES=("curl" "wget")
    mkdir -p $TMP
    
    for PACKAGE in "${DEPENDENCIES[@]}" ; do
        cmd_query "$PACKAGE"

    done

    if [ -f "$TMP/missingDeps" ] ; then
        MISSING=$(cat $TMP/missingDeps)
        echo -e "\n  ${RED}Following dependencies are missing :\n\n${CRS}$MISSING\n\n  ${CYN}Please install and re-run script !${CRS}"
        exit_clean
        
    fi
}

### CHECK LOCAL INSTALL
local_check() {
    if ! [ -d "$INSTALL_PATH" ] ; then
        echo -e "\n  ${RED}$APP_NAME_LONG directory not found!${CRS}"
        install_helium
        exit
        
    elif ! [ -f "$INSTALL_PATH/$APP_NAME.AppImage" ] ; then
        echo -e "\n  ${RED}$APP_NAME_LONG directory found but no executable named ${CYN}$APP_NAME.AppImage${RED}!${CRS}"
        install_helium
        exit

    fi
}

### INSTALL
install_helium() {
    if [ -f "$INSTALL_PATH/$APP_NAME.AppImage" ] ; then
        echo -e "\n  ${YEL}$APP_NAME_LONG AppImage found! Trying to update...${CRS}"
        update_helium
        exit

    fi

    while true ; do
        read -p "  Download and install $APP_NAME_LONG from $SOURCE? (Y/n) : " CONFIRM_INSTALL
        echo

        if [[ "$CONFIRM_INSTALL" == "" || "$CONFIRM_INSTALL" == "y" || "$CONFIRM_INSTALL" == "Y" ]] ; then
            GIT_URL=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "browser_download_url" | grep "x86_64.AppImage" | grep -v "zsync" | tr -d \" | awk '{print $2}')
            GIT_FILENAME=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "browser_download_url" | grep "x86_64.AppImage" | grep -v "zsync" | tr -d \" | awk '{print $2}' | awk -F'/' '{print $9}')
            wget -P $TMP $GIT_URL -q --show-progress
            chmod +x $TMP/$GIT_FILENAME
            mkdir -p $INSTALL_PATH
            mv $TMP/$GIT_FILENAME $INSTALL_PATH/$APP_NAME.AppImage
            add_launcher
            echo -e "\n  ${GRN}$APP_NAME_LONG is installed!${CRS}\n"
            break

        elif [[ "$CONFIRM_INSTALL" == "n" || "$CONFIRM_INSTALL" == "N" ]] ; then
            echo -e "  ${RED}Cancelled!${CRS}"
            break

        else
            echo -e "  ${YEL}Invalid response!${CRS}\n"
            
        fi

    done
}

### LAUNCHER
add_launcher() {
    if [ -f "$LAUNCHER_PATH/$APP_NAME.desktop" ] ; then
        echo -e "\n  ${YEL}${APP_NAME}.desktop already exists!${CRS}"
    
    elif [ -f "$INSTALL_PATH/$APP_NAME.AppImage" ] ; then
        while true ; do
        echo
        read -p "  Add $APP_NAME_LONG launcher to $LAUNCHER_PATH? (Y/n) : " CONFIRM_LAUNCHER
        echo

        if [[ "$CONFIRM_LAUNCHER" == "" || "$CONFIRM_LAUNCHER" == "y" || "$CONFIRM_LAUNCHER" == "Y" ]] ; then
            LAUNCHER_URL=https://lab.c95.org/fr4nz/helium-helper/raw/branch/main/assets/launcher/helium.desktop.template
            ICON_URL=https://lab.c95.org/fr4nz/helium-helper/raw/branch/main/assets/icon/helium.png
            AUTO_SCRIPT_URL=https://lab.c95.org/fr4nz/helium-helper/raw/branch/main/$AUTO_SCRIPT_FILE_NAME
            LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}')
            mkdir -p $LAUNCHER_PATH $SCRIPT_PATH $BACKUP_PATH $CONFIG_PATH/icon
            if ! [ -f "$SCRIPT_PATH/$AUTO_SCRIPT_FILE_NAME" ] ; then
                wget -P $TMP $AUTO_SCRIPT_URL -q --show-progress
                chmod +x $TMP/$AUTO_SCRIPT_FILE_NAME
                mv $TMP/$AUTO_SCRIPT_FILE_NAME $SCRIPT_PATH
            fi

            wget -P $TMP $LAUNCHER_URL -q --show-progress
            wget -P $TMP $ICON_URL -q --show-progress
            sed -i "s|%APP_LOCATION%|$INSTALL_PATH/$APP_NAME.AppImage|" $TMP/helium.desktop.template
            sed -i "s|%ICON_LOCATION%|$CONFIG_PATH/icon/helium.png|" $TMP/helium.desktop.template
            sed -i "s|%APP_VERSION%|$LOCAL_VERSION|" $TMP/helium.desktop.template
            sed -i "s|%QUIET_UPDATE_SCRIPT%|$SCRIPT_PATH/$AUTO_SCRIPT_FILE_NAME|" $TMP/helium.desktop.template
            sed -i "s|%SCRIPT_UPDATE%|$SCRIPT_PATH/$AUTO_SCRIPT_FILE_NAME -S|" $TMP/helium.desktop.template
            mv $TMP/helium.desktop.template $LAUNCHER_PATH/$APP_NAME.desktop
            mv $TMP/helium.png $CONFIG_PATH/icon
            echo -e "\n  ${GRN}$APP_NAME_LONG launcher is set up!${CRS}"
            break

        elif [[ "$CONFIRM_LAUNCHER" == "n" || "$CONFIRM_LAUNCHER" == "N" ]] ; then
            echo -e "  ${RED}Launcher not added!${CRS}"
            break

        else
            echo -e "  ${YEL}Invalid response!${CRS}\n"
            
        fi

    done

    else
        echo -e "\n  ${RED}$APP_NAME_LONG AppImage not found! Can't add launcher...${CRS}"
        install_helium
        exit
    
    fi
}

### ALIAS
add_alias() {   ### WORKS BUT NEEDS WORK
    if ! [ -f "$SCRIPT_PATH/$SCRIPT_FILE_NAME" ] ; then
        mkdir -p $SCRIPT_PATH
        wget -P $TMP $SCRIPT_URL -q
        chmod +x $TMP/$SCRIPT_FILE_NAME
        mv $TMP/$SCRIPT_FILE_NAME $SCRIPT_PATH
    fi
        
    if [ -f "$HOME/.zshrc" ] ; then
        if [ "$(cat $HOME/.zshrc | grep "$ALIAS")" ] ; then
           echo -e "\n  ${LCY}$ALIAS ${GRN}already present in${CRS} $HOME/.zshrc"
           exit
        fi
        echo "alias $ALIAS=\"$SCRIPT_PATH/$SCRIPT_FILE_NAME\"" >> $HOME/.zshrc
        echo -e "\n  ${LCY}$ALIAS ${GRN}added to${CRS} $HOME/.zshrc"

    elif [ -f "$HOME/.bashrc" ] ; then
        if [ "$(cat $HOME/.bashrc | grep "$ALIAS")" ] ; then
           echo -e "\n  ${LCY}$ALIAS ${GRN}already present in${CRS} $HOME/.bashrc"
           exit
        fi
        echo "alias $ALIAS=\"$SCRIPT_PATH/$SCRIPT_FILE_NAME\"" >> $HOME/.zshrc
        echo -e "\n  ${LCY}$ALIAS ${GRN}added to${CRS} $HOME/.bashrc"

    else
        echo -e "\n  ${RED}No .shellrc file found!${CRS}"

    fi
}

### UPDATE
update_helium() {
    LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}')
    GIT_VERSION=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "tag_name" | tr -d \", | awk '{print $2}')

    ### CHECK FOR EMPTY VERSION TAGS
    if [ -z "$GIT_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}$SOURCE${RED}! Can't compare, exiting...${CRS}"
        exit

    elif [ -z "$LOCAL_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}local AppImage${RED}! Can't compare, exiting...${CRS}"
        exit

    fi

    ### COMPARE LOCAL & REMOTE VERSION
    LOCAL_VERSION_NUM=$(echo $LOCAL_VERSION | tr -d .)
    GIT_VERSION_NUM=$(echo $GIT_VERSION | tr -d .)

    if [ "$GIT_VERSION" == "$LOCAL_VERSION" ] ; then
        echo -e "\n  ${GRN}$APP_NAME_LONG is already up-to-date!${CRS}\n"

    elif [ "$GIT_VERSION_NUM" -gt "$LOCAL_VERSION_NUM" ] ; then
        echo -e "\n  ${LCY}$APP_NAME_LONG is updatable!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}$SOURCE Version :\t${CRS}${GIT_VERSION}\n"

        while true ; do
            read -p "  Download and update $APP_NAME_LONG from $SOURCE? (Y/n) : " CONFIRM
            echo

            if [[ "$CONFIRM" == "" || "$CONFIRM" == "y" || "$CONFIRM" == "Y" ]] ; then
                GIT_URL=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "browser_download_url" | grep "x86_64.AppImage" | grep -v "zsync" | tr -d \" | awk '{print $2}')
                GIT_FILENAME=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "browser_download_url" | grep "x86_64.AppImage" | grep -v "zsync" | tr -d \" | awk '{print $2}' | awk -F'/' '{print $9}')
                mkdir -p $TMP $BACKUP_PATH
                wget -P $TMP $GIT_URL -q --show-progress
                chmod +x $TMP/$GIT_FILENAME
                if [ -f "$BACKUP_PATH/$APP_NAME.old" ] ; then
                    rm $BACKUP_PATH/$APP_NAME.old
                fi
                mv $INSTALL_PATH/$APP_NAME.AppImage $BACKUP_PATH/$APP_NAME.old
                mv $TMP/$GIT_FILENAME $INSTALL_PATH/$APP_NAME.AppImage
                cp $LAUNCHER_PATH/$APP_NAME.desktop $BACKUP_PATH/$APP_NAME.desktop.old
                sed -i "s|X-AppImage-Version=.*|X-AppImage-Version=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}')|" $LAUNCHER_PATH/$APP_NAME.desktop
                echo -e "\n  ${GRN}$APP_NAME_LONG is up-to-date !${CRS}\n"
                break

            elif [[ "$CONFIRM" == "n" || "$CONFIRM" == "N" ]] ; then
                echo -e "  ${RED}Cancelled!${CRS}"
                exit

            else
                echo -e "  ${YEL}Invalid response!${CRS}\n"
                
            fi

        done

    else
        echo -e "\n  ${RED}Version tags malformed or you are using a Beta version!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}$SOURCE Version :\t${CRS}${GIT_VERSION}\n"
        echo -e "  ${RED}Update failed... Exiting...${CRS}\n"
        exit
        
    fi

}

### AUTOMATIC UPDATE --> TO DO
auto_update() {
    if ! [ -f "$INSTALL_PATH/$APP_NAME.AppImage" ] ; then
        echo -e "\n  ${RED}$APP_NAME AppImage not found! Can't add auto update script...${CRS}"
        exit
        
    else
        echo -e "\n  ${RED}Not implemented yet...${CRS}"

    fi
}

### REMOVE
remove_helium() {
    if [ -d "$INSTALL_PATH" ] ; then

        while true ; do
            read -p "  Remove $APP_NAME_LONG incl. backups and launcher? (Y/n) : " CONFIRM_REMOVE

            if [[ "$CONFIRM_REMOVE" == "" || "$CONFIRM_REMOVE" == "y" || "$CONFIRM_REMOVE" == "Y" ]] ; then
                rm -rf $INSTALL_PATH
                if [ -f "$LAUNCHER_PATH/$APP_NAME.desktop" ] ; then
                    rm $LAUNCHER_PATH/$APP_NAME.desktop
                
                fi

                if [ -f "$HOME/.zshrc" ] ; then
                    sed -i "s|$ALIAS|d|" $HOME/.zshrc

                elif [ -f "$HOME/.bashrc" ] ; then
                    sed -i "s|$ALIAS|d|" $HOME/.bashrc

                fi

                echo -e "\n  ${RED}$APP_NAME_LONG removed!${CRS}"
                exit

            elif [[ "$CONFIRM_REMOVE" == "n" || "$CONFIRM_REMOVE" == "N" ]] ; then
                echo -e "  ${RED}Cancelled!${CRS}"
                exit

            else
                echo -e "  ${YEL}Invalid response!${CRS}\n"
                
            fi

        done

    else
        echo -e "\n  ${RED}Can't remove $APP_NAME_LONG! ${YEL}$INSTALL_PATH ${RED}directory not found...${CRS}"
        exit

    fi
}

### SCRIPT VERSION
script_version() {
    echo -e "\n  ${CYN}$APP_NAME_LONG Helper Script Version :${CRS}   $SCRIPT_VERSION\n"
    exit
    
}

### UPDATE SCRIPT
remote_script_check() {
    REMOTE_SCRIPT_URL=https://lab.c95.org/fr4nz/helium-helper/raw/branch/main/$SCRIPT_FILE_NAME
    mkdir -p $TMP
    wget -P $TMP $REMOTE_SCRIPT_URL -q
    chmod +x $TMP/$SCRIPT_FILE_NAME
}

update_script() {
    remote_script_check

    ### CHECK FOR LOCAL SCRIPT
    if ! [ -f "$SCRIPT_PATH/$SCRIPT_FILE_NAME" ] ; then
        echo -e "\n  ${YEL}$SCRIPT_FILE_NAME not found...${CRS}"
        
        while true ; do
            echo
            read -p "  Copy script to $SCRIPT_PATH? (Y/n) : " CONFIRM_SCRIPT_INSTALL

            if [[ "$CONFIRM_SCRIPT_INSTALL" == "" || "$CONFIRM_SCRIPT_INSTALL" == "y" || "$CONFIRM_SCRIPT_INSTALL" == "Y" ]] ; then
                if ! [ -d "$SCRIPT_PATH" ] ; then
                    mkdir -p $SCRIPT_PATH
                fi
                
                mv $TMP/$SCRIPT_FILE_NAME $SCRIPT_PATH
                echo -e "\n  ${GRN}$SCRIPT_FILE_NAME is up-to-date!${CRS}"
                exit_clean

            elif [[ "$CONFIRM_SCRIPT_INSTALL" == "n" || "$CONFIRM_SCRIPT_INSTALL" == "N" ]] ; then
                echo -e "  ${RED}Cancelled!${CRS}"
                exit

            else
                echo -e "  ${YEL}Invalid response!${CRS}\n"
                
            fi

        done

    fi

    REMOTE_SCRIPT_VERSION=$($TMP/$SCRIPT_FILE_NAME -s | awk '{print $8}' | tr -d '\n' )

    ### CHECK FOR EMPTY VERSION TAGS
    if [ -z "$SCRIPT_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}local script${RED}! Can't compare, exiting...${CRS}"
        exit

    elif [ -z "$REMOTE_SCRIPT_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}remote script${RED}! Can't compare, exiting...${CRS}"
        exit

    fi
        
    ### COMPARE LOCAL & REMOTE VERSION
    SCRIPT_VERSION_NUM=$(echo $SCRIPT_VERSION | tr -d .)
    REMOTE_SCRIPT_VERSION_NUM=$(echo $REMOTE_SCRIPT_VERSION | tr -d .)

    if [ "$REMOTE_SCRIPT_VERSION" == "$SCRIPT_VERSION" ] ; then
        echo -e "\n  ${GRN}$SCRIPT_FILE_NAME is already up-to-date!${CRS}\n"
        
    elif [ "$REMOTE_SCRIPT_VERSION_NUM" -gt "$SCRIPT_VERSION_NUM" ] ; then
        echo -e "\n  ${YEL}$SCRIPT_FILE_NAME is updatable!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${SCRIPT_VERSION}"
        echo -e "  ${CYN}Remote Version :\t${CRS}${REMOTE_SCRIPT_VERSION}\n"
        
        while true ; do
            read -p "  Update $SCRIPT_FILE_NAME? (Y/n) : " CONFIRM_SCRIPT_UPDATE
            echo

            if [[ "$CONFIRM_SCRIPT_UPDATE" == "" || "$CONFIRM_SCRIPT_UPDATE" == "y" || "$CONFIRM_SCRIPT_UPDATE" == "Y" ]] ; then
                if ! [ -d "$BACKUP_PATH" ] ; then
                    mkdir -p $BACKUP_PATH
                fi
                mv $SCRIPT_PATH/$SCRIPT_FILE_NAME $BACKUP_PATH/$SCRIPT_FILE_NAME.old
                mv $TMP/$SCRIPT_FILE_NAME $SCRIPT_PATH
                echo -e "  ${GRN}$SCRIPT_FILE_NAME is up-to-date!${CRS}"
                break

            elif [[ "$CONFIRM_SCRIPT_UPDATE" == "n" || "$CONFIRM_SCRIPT_UPDATE" == "N" ]] ; then
                echo -e "  ${RED}Cancelled!${CRS}"
                break

            else
                echo -e "  ${YEL}Invalid response!${CRS}\n"
                
            fi

        done

    else
        echo -e "\n  ${RED}Version tags malformed or you are using a Beta version!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${SCRIPT_VERSION}"
        echo -e "  ${CYN}Remote Version :\t${CRS}${REMOTE_SCRIPT_VERSION}\n"
        echo -e "  ${RED}Updating script failed...${CRS}\n"
        
    fi
}

### COMPARE APPIMAGE VERSION
compare_version() {
    LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}')
    GIT_VERSION=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "tag_name" | tr -d \", | awk '{print $2}')

    ### CHECK FOR EMPTY VERSION TAGS
    if [ -z "$GIT_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}$SOURCE${RED}! Can't compare, exiting...${CRS}"
        exit

    elif [ -z "$LOCAL_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}local AppImage${RED}! Can't compare, exiting...${CRS}"
        exit

    fi

    ### COMPARE LOCAL & REMOTE VERSION
    LOCAL_VERSION_NUM=$(echo $LOCAL_VERSION | tr -d .)
    GIT_VERSION_NUM=$(echo $GIT_VERSION | tr -d .)

    if [ "$GIT_VERSION" == "$LOCAL_VERSION" ] ; then
        echo -e "\n  ${GRN}$APP_NAME_LONG is up-to-date!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}$SOURCE Version :\t${CRS}${GIT_VERSION}\n"
        exit
        
    elif [ "$GIT_VERSION_NUM" -gt "$LOCAL_VERSION_NUM" ] ; then
        echo -e "\n  ${YEL}$APP_NAME_LONG is updatable!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}$SOURCE Version :\t${CRS}${GIT_VERSION}\n"
        exit
        
    else
        echo -e "\n  ${RED}Version tags malformed or you are using a Beta version!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}$SOURCE Version :\t${CRS}${GIT_VERSION}\n"
        echo -e "  ${RED}Update failed ... Exiting ...${CRS}\n"
        exit
        
    fi
}

### HELP
help() {
    echo -e "\n${LCY}${SCRIPT_NAME}${CRS}\n"
    echo -e "$APP_NAME_LONG installer & updater script\n"
    echo -e "Syntax: $SCRIPT_FILE_NAME [ -i | -l | -a | -u | -r | -c | -s | -S | -h ]\n" # | -U |
    echo -e "Options:\n"
    echo -e "\t-i\tDownload & install $APP_NAME_LONG from $SOURCE"
    echo -e "\t-l\tCreate desktop launcher"
    echo -e "\t-a\tCreate alias for $SCRIPT_FILE_NAME ${YEL}--> Partially working with zsh & bash${CRS}"
    echo -e "\t-u\tDownload & update $APP_NAME_LONG from $SOURCE"
    #echo -e "\t-U\tEnable automatic updates on login ${RED}--> Not yet working${CRS}"
    echo -e "\t-r\tRemove $APP_NAME_LONG"
    echo -e "\t-c\tCompare local to remote version"
    echo -e "\t-s\tShow script version"
    echo -e "\t-S\tUpdate $SCRIPT_FILE_NAME"
    echo -e "\t-h\tPrint this help page.\n"
    echo -e "Configured locations:\n"
    echo -e "\t${CYN}${INSTALL_PATH}${CRS}\t\tInstall location"
    echo -e "\t${CYN}${LAUNCHER_PATH}${CRS}\tLauncher location"
    echo -e "\t${CYN}${SCRIPT_PATH}${CRS}\t\t\tScript location\n"
}

### EXIT CLEAN
exit_clean() {
    if [ -d "$TMP" ] ; then
        rm -rf $TMP

    fi

    exit
}

########## MAIN ##########

## GET OPTIONS
while getopts ":i,:l,:a,:u,:U,:r,:c,:s,:S,:h" OPTION ; do
    case $OPTION in
        i)
            echo -e "\n  ${CYN}Installing $APP_NAME_LONG...${CRS}"
            dependency_check
            install_helium
            exit_clean ;;

        l)
            echo -e "\n  ${CYN}Adding $APP_NAME_LONG launcher...${CRS}"
            dependency_check
            local_check
            add_launcher
            exit_clean ;;

        a)
            echo -e "\n  ${CYN}Adding alias...${CRS}"
            add_alias
            exit ;;
        
        u)
            echo -e "\n  ${CYN}Updating $APP_NAME_LONG...${CRS}"
            dependency_check
            local_check
            update_helium
            exit_clean ;;

        U)
            echo -e "\n  ${CYN}Enabling automatic updates...${CRS}"
            dependency_check
            local_check
            auto_update
            exit_clean ;;

        r)
            echo -e "\n  ${CYN}Removing $APP_NAME_LONG...${CRS}"
            remove_helium
            exit ;;

        c)
            echo -e "\n  ${CYN}Comparing remote to local version...${CRS}"
            dependency_check
            local_check
            compare_version
            exit_clean ;;

        s)
            script_version
            exit ;;

        S)
            echo -e "\n  ${CYN}Updating script...${CRS}"
            dependency_check
            update_script
            exit_clean ;;

        h)
            help
            exit ;;

        *)
            echo -e "\n${YEL}Option ${RED}$1${YEL} not recognized... Refer to help below!${CRS}"
            help
            exit ;;

    esac

done

## DEPENDENCY CHECK
dependency_check

## CHECK FOR LOCAL INSTALL
local_check

## UPDATE 
update_helium

## CLEAN
exit_clean