#!/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"
ARCH="x86_64"
FILE_EXTENSION="AppImage"
APP_SOURCE="GitHub"
SCRIPT_NAME="$APP_NAME Helper Script"
SCRIPT_FILE_NAME="helium_helper.sh"
AUTO_SCRIPT_FILE_NAME="helium_launcher_update.sh"
SCRIPT_VERSION="0.5.8"
ALIAS="update-helium"

## LOCATIONS
TMP=/tmp/$APP_NAME
INSTALL_PATH=$HOME/.apps/$APP_NAME
APP_CONFIG_PATH=$HOME/.config/net.imput.helium
ASSET_PATH=$INSTALL_PATH/assets
BACKUP_PATH=$ASSET_PATH/backup
LOG_PATH=$ASSET_PATH/logs
LOG=$LOG_PATH/log
LAUNCHER_PATH=$HOME/.local/share/applications
SCRIPT_PATH=$HOME/.local/bin

## REMOTE LOCATIONS
REMOTE_REPO_BRANCH="main"
REMOTE_REPO_URL=https://lab.c95.org/fr4nz/helium-helper/raw/branch/$REMOTE_REPO_BRANCH
REMOTE_SCRIPT_URL=$REMOTE_REPO_URL/$SCRIPT_FILE_NAME
APP_SOURCE_URL=https://api.github.com/repos/imputnet/helium-linux/releases/latest

########## 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
}

### SHELL CHECK
shell_query() {
    #local SHELL_OPTION=$1

    if [ "$(echo $SHELL | grep "$SHELL_OPTION")" ] ; then
        ACTIVE_SHELL=$SHELL_OPTION
    fi
    
}

shell_check() {
    SHELLS=("bash" "fish" "ksh" "zsh")

    for SHELL_OPTION in "${SHELLS[@]}" ; do
        shell_query "$SHELL_OPTION"
    done
}

### 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.$FILE_EXTENSION" ] ; then
        echo -e "\n  ${RED}$APP_NAME_LONG directory found but no executable named ${CYN}$APP_NAME.$FILE_EXTENSION${RED}!${CRS}"
        install_helium
        exit

    fi
}

### INSTALL
install_helium() {
    if [ -f "$INSTALL_PATH/$APP_NAME.$FILE_EXTENSION" ] ; 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 $APP_SOURCE? (Y/n) : " CONFIRM_INSTALL
        echo

        if [[ "$CONFIRM_INSTALL" == "" || "$CONFIRM_INSTALL" == "y" || "$CONFIRM_INSTALL" == "Y" ]] ; then
            DOWNLOAD_URL=$(curl -s $APP_SOURCE_URL | grep "browser_download_url" | grep "$ARCH.$FILE_EXTENSION" | grep -v "zsync" | tr -d \" | awk '{print $2}')
            APP_SOURCE_FILENAME=$(curl -s $APP_SOURCE_URL | grep "browser_download_url" | grep "$ARCH.$FILE_EXTENSION" | grep -v "zsync" | tr -d \" | awk '{print $2}' | awk -F'/' '{print $9}')
            wget -P $TMP $DOWNLOAD_URL -q --show-progress
            chmod +x $TMP/$APP_SOURCE_FILENAME
            mkdir -p $INSTALL_PATH
            mv $TMP/$APP_SOURCE_FILENAME $INSTALL_PATH/$APP_NAME.$FILE_EXTENSION
            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.$FILE_EXTENSION" ] ; 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=$REMOTE_REPO_URL/assets/launcher/helium.desktop.template
            ICON_URL=$REMOTE_REPO_URL/assets/icon/helium.png
            AUTO_SCRIPT_URL=$REMOTE_REPO_URL/$AUTO_SCRIPT_FILE_NAME
            LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.$FILE_EXTENSION --version | awk '{print $2}')
            mkdir -p $LAUNCHER_PATH $SCRIPT_PATH $BACKUP_PATH $ASSET_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.$FILE_EXTENSION|" $TMP/helium.desktop.template
            sed -i "s|%ICON_LOCATION%|$ASSET_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
            sed -i "s|%OPEN_LOG_EDITOR%|$SCRIPT_PATH/$AUTO_SCRIPT_FILE_NAME -L|" $TMP/helium.desktop.template
            mv $TMP/helium.desktop.template $LAUNCHER_PATH/$APP_NAME.desktop
            mv $TMP/helium.png $ASSET_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() {
    if ! [ -f "$SCRIPT_PATH/$SCRIPT_FILE_NAME" ] ; then
        mkdir -p $SCRIPT_PATH
        wget -P $TMP $REMOTE_SCRIPT_URL -q
        chmod +x $TMP/$SCRIPT_FILE_NAME
        mv $TMP/$SCRIPT_FILE_NAME $SCRIPT_PATH
    fi
    
    if [ "$ACTIVE_SHELL" == "zsh" ] ; then
        if [ "$(cat $HOME/.zshrc | grep "$ALIAS")" ] ; then
           echo -e "\n  ${LCY}$ALIAS ${GRN}already present in${CRS} $HOME/.zshrc\n"
           exit
        fi
        echo "alias $ALIAS=\"$SCRIPT_PATH/$SCRIPT_FILE_NAME\"" >> $HOME/.zshrc
        echo -e "\n  ${LCY}$ALIAS ${GRN}added to${CRS} $HOME/.zshrc\n"

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

    elif [ "$ACTIVE_SHELL" == "fish" ] ; then
        if [ "$(cat $HOME/.config/fish/config.fish | grep "source $HOME/.config/fish/$ALIAS.fish")" ] ; then
           echo -e "\n  ${LCY}$ALIAS ${GRN}already present in${CRS} $HOME/.config/fish/config.fish\n"
           exit
        fi
        echo "### Helium Helper Alias FISH ###" > $HOME/.config/fish/$ALIAS.fish
        echo >> $HOME/.config/fish/$ALIAS.fish
        echo "function $ALIAS" >> $HOME/.config/fish/$ALIAS.fish
        echo "    $SCRIPT_PATH/$SCRIPT_FILE_NAME" >> $HOME/.config/fish/$ALIAS.fish
        echo "end" >> $HOME/.config/fish/$ALIAS.fish
        echo "source $HOME/.config/fish/$ALIAS.fish" >> $HOME/.config/fish/config.fish
        echo -e "\n  ${LCY}$ALIAS ${GRN}added to${CRS} $HOME/.config/fish/$ALIAS.fish\n"

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

    fi
}

### UPDATE
update_helium() {
    LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.$FILE_EXTENSION --version | awk '{print $2}')
    APP_SOURCE_VERSION=$(curl -s $APP_SOURCE_URL | grep "tag_name" | tr -d \", | awk '{print $2}')

    ### CHECK FOR EMPTY VERSION TAGS
    if [ -z "$APP_SOURCE_VERSION" ] ; then
        echo -e "\n  ${RED}Empty version tag from ${CYN}$APP_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 .)
    APP_SOURCE_VERSION_NUM=$(echo $APP_SOURCE_VERSION | tr -d .)

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

    elif [ "$APP_SOURCE_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}$APP_SOURCE Version :\t${CRS}${APP_SOURCE_VERSION}\n"

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

            if [[ "$CONFIRM" == "" || "$CONFIRM" == "y" || "$CONFIRM" == "Y" ]] ; then
                DOWNLOAD_URL=$(curl -s $APP_SOURCE_URL | grep "browser_download_url" | grep "$ARCH.$FILE_EXTENSION" | grep -v "zsync" | tr -d \" | awk '{print $2}')
                APP_SOURCE_FILENAME=$(curl -s $APP_SOURCE_URL | grep "browser_download_url" | grep "$ARCH.$FILE_EXTENSION" | grep -v "zsync" | tr -d \" | awk '{print $2}' | awk -F'/' '{print $9}')
                mkdir -p $TMP $BACKUP_PATH
                wget -P $TMP $DOWNLOAD_URL -q --show-progress
                chmod +x $TMP/$APP_SOURCE_FILENAME
                if [ -f "$BACKUP_PATH/$APP_NAME.old" ] ; then
                    rm $BACKUP_PATH/$APP_NAME.old
                fi
                mv $INSTALL_PATH/$APP_NAME.$FILE_EXTENSION $BACKUP_PATH/$APP_NAME.old
                mv $TMP/$APP_SOURCE_FILENAME $INSTALL_PATH/$APP_NAME.$FILE_EXTENSION
                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.$FILE_EXTENSION --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}\n"
                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}$APP_SOURCE Version :\t${CRS}${APP_SOURCE_VERSION}\n"
        echo -e "  ${RED}Update failed... Exiting...${CRS}\n"
        exit
        
    fi

}

### AUTOMATIC UPDATE --> TO DO?
auto_update() {
    if ! [ -f "$INSTALL_PATH/$APP_NAME.$FILE_EXTENSION" ] ; 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
                REMOVABLE=("$INSTALL_PATH" "$LAUNCHER_PATH/$APP_NAME.desktop" "$SCRIPT_PATH/$SCRIPT_FILE_NAME" "$SCRIPT_PATH/$AUTO_SCRIPT_FILE_NAME")
                echo
                for RM_PATH in "${REMOVABLE[@]}" ; do
                    if [[ -f "$RM_PATH" || -d "$RM_PATH" ]] ; then
                        echo -e "  ${RED}Removing $RM_PATH ...${CRS}"
                        rm -rf "$RM_PATH"
                    fi
                done
                
                shell_check

                if [ "$ACTIVE_SHELL" == "zsh" ] ; then
                    if [ "$(cat $HOME/.zshrc | grep "$ALIAS")" ] ; then
                        sed -i "/$ALIAS/d" $HOME/.zshrc
                    fi

                elif [ "$ACTIVE_SHELL" == "bash" ] ; then
                    if [ "$(cat $HOME/.bashrc | grep "$ALIAS")" ] ; then
                        sed -i "/$ALIAS/d" $HOME/.bashrc
                    fi

                fi

                echo -e "\n  ${RED}$APP_NAME_LONG removed!${CRS}\n"
                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
update_script() {
    ### 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
                mkdir -p $TMP
                wget -P $TMP $REMOTE_SCRIPT_URL -q
                chmod +x $TMP/$SCRIPT_FILE_NAME
                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}\n"
                exit

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

        done

    fi

    REMOTE_SCRIPT_VERSION=$(curl -s $REMOTE_SCRIPT_URL | grep -m 1 "SCRIPT_VERSION" | tr -d 'SCRIPT_VERSION="')

    ### 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
                mkdir -p $TMP
                wget -P $TMP $REMOTE_SCRIPT_URL -q
                chmod +x $TMP/$SCRIPT_FILE_NAME
                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.$FILE_EXTENSION --version | awk '{print $2}')
    APP_SOURCE_VERSION=$(curl -s $APP_SOURCE_URL | grep "tag_name" | tr -d \", | awk '{print $2}')

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

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

    fi

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

    if [ "$APP_SOURCE_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}$APP_SOURCE Version :\t${CRS}${APP_SOURCE_VERSION}\n"
        exit
        
    elif [ "$APP_SOURCE_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}$APP_SOURCE Version :\t${CRS}${APP_SOURCE_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}$APP_SOURCE Version :\t${CRS}${APP_SOURCE_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 $APP_SOURCE"
    echo -e "\t-l\tCreate desktop launcher"
    echo -e "\t-a\tCreate alias for $SCRIPT_FILE_NAME ${YEL}--> Working with zsh & bash${CRS}"
    echo -e "\t-u\tDownload & update $APP_NAME_LONG from $APP_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}"
            shell_check
            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