#!/bin/bash

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

########## PREREQUISITES ###########

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

## LOCATIONS
TMP=/tmp/Helium
INSTALL_PATH=$HOME/.apps/Helium
CONFIG_PATH=$INSTALL_PATH/config
LAUNCHER_PATH=$HOME/.local/share/applications
SCRIPT_PATH=$HOME/.local/bin
#PROFILE=$HOME/.profile     ### for AUTO UPDATE

## GLOBAL VARIABLES
SCRIPT_NAME="Helium Helper Script"
SCRIPT_VERSION="0.5.1"
DEPENDENCIES=("curl" "wget")
MISSING=$(cat $TMP/missingDeps 2>/dev/null)
GIT_VERSION=$(curl -s https://api.github.com/repos/imputnet/helium-linux/releases/latest | grep "tag_name" | tr -d \", | awk '{print $2}')

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

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

        if command -v "$PACKAGE" >/dev/null 2>&1 ; then
            :

        else
            echo "$PACKAGE" >> $TMP/missingDeps

        fi
    }

    for PACKAGE in "${DEPENDENCIES[@]}" ; do
        cmd_query "$PACKAGE"

    done

    if ! [ "$MISSING" ] ; then
        :

    else
        echo -e "\n  ${RED}Following dependencies are missing :\n\n${YEL}$MISSING\n\n  ${CYN}Please install and re-run script!${CRS}"
        rm $TMP/missingDeps
        exit 1

    fi
}

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

    else 
        :

    fi
}

### INSTALL
install_helium() {
    if [ -f "$INSTALL_PATH/helium-browser.AppImage" ] ; then
        echo -e "\n  ${CYN}Helium AppImage found! Trying to update...${CRS}"
        update_helium
        exit 0
        
    else
        :

    fi

    while true ; do
        read -p "  Download and install Helium from GitHub? (Y/n) : " CONFIRM_INSTALL

        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}')
            mkdir -p $TMP
            wget -P $TMP $GIT_URL -q --show-progress
            chmod +x $TMP/$GIT_FILENAME
            mkdir -p $INSTALL_PATH
            mv $TMP/$GIT_FILENAME $INSTALL_PATH/helium-browser.AppImage
            echo -e "\n  ${GRN}Helium is installed!${CRS}\n"
            clean_tmp
            exit 0

        elif [[ "$CONFIRM_INSTALL" == "n" || "$CONFIRM_INSTALL" == "N" ]] ; then
            echo -e "\n  ${RED}Installation cancelled!${CRS}\n"
            exit 1

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

    done
}

### LAUNCHER
add_launcher() {
    if ! [ -f "$INSTALL_PATH/helium-browser.AppImage" ] ; then
        echo -e "\n  ${RED}Helium AppImage not found! Can't add launcher...${CRS}"
        exit 1
        
    else
        echo -e "\n  ${RED}Not implemented yet...${CRS}"
        exit 0

    fi
}

### UPDATE
update_helium() {
    LOCAL_VERSION=$($INSTALL_PATH/helium-browser.AppImage --version | awk '{print $2}')

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

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

    else
        :

    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}Helium is already up-to-date!${CRS}\n"

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

        while true ; do
            read -p "  Download and update Helium from GitHub? (Y/n) : " CONFIRM

            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
                wget -P $TMP $GIT_URL -q --show-progress
                chmod +x $TMP/$GIT_FILENAME
                rm $INSTALL_PATH/helium-browser.OldAppImage >/dev/null 2>&1
                mv $INSTALL_PATH/helium-browser.AppImage $INSTALL_PATH/helium-browser.OldAppImage >/dev/null 2>&1
                mv $TMP/$GIT_FILENAME $INSTALL_PATH/helium-browser.AppImage
                echo -e "\n  ${GRN}Helium is up-to-date !${CRS}\n"
                break

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

            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}GitHub Version :\t${CRS}${GIT_VERSION}\n"
        echo -e "  ${RED}Update failed... Exiting...${CRS}\n"
        exit 1
        
    fi

}

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

    fi
}

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

        while true ; do
        read -p "  Remove Helium folder ${INSTALL_PATH}? (Y/n) : " CONFIRM_REMOVE

        if [[ "$CONFIRM_REMOVE" == "" || "$CONFIRM_REMOVE" == "y" || "$CONFIRM_REMOVE" == "Y" ]] ; then
            rm -rf $INSTALL_PATH
            echo -e "\n  ${RED}Helium removed!${CRS}\n"
            exit 0

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

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

    done

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

    fi
}

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

### UPDATE SCRIPT
update_script() {
    if ! [ -f "$INSTALL_PATH/helium-browser.AppImage" ] ; then
        echo -e "\n  ${RED}Helium AppImage not found! Can't update script...${CRS}"
        exit 1
        
    else
        echo -e "\n  ${RED}Not implemented yet...${CRS}"
        exit 0

    fi
}

### COMPARE APPIMAGE VERSION
compare_version() {
    LOCAL_VERSION=$($INSTALL_PATH/helium-browser.AppImage --version | awk '{print $2}')

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

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

    else
        :

    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}Helium is up-to-date!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}GitHub Version :\t${CRS}${GIT_VERSION}\n"
        exit 0
        
    elif [ "$GIT_VERSION_NUM" -gt "$LOCAL_VERSION_NUM" ] ; then
        echo -e "\n  ${YEL}Helium is updatable!${CRS}\n"
        echo -e "  ${CYN}Local Version :\t${CRS}${LOCAL_VERSION}"
        echo -e "  ${CYN}GitHub Version :\t${CRS}${GIT_VERSION}\n"
        exit 0
        
    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}GitHub Version :\t${CRS}${GIT_VERSION}\n"
        echo -e "  ${RED}Update failed ... Exiting ...${CRS}\n"
        exit 1
        
    fi
}

### HELP
help() {
    echo -e "\n${CYN}${SCRIPT_NAME}${CRS}\n"
    echo -e "Helium Browser installer & updater script\n"
    echo -e "Syntax: helium_helper.sh [ -i | -L | -u | -U | -r | -c | -s | -S | -h ]\n"
    echo -e "Options:\n"
    echo -e "\t-i\tDownload & install Helium from GitHub"
    echo -e "\t-L\tCreate desktop launcher ${RED}--> Not yet working${CRS}"
    echo -e "\t-u\tDownload & update Helium from GitHub"
    echo -e "\t-U\tEnable automatic updates on login ${RED}--> Not yet working${CRS}"
    echo -e "\t-r\tRemove Helium"
    echo -e "\t-c\tCompare local to remote version"
    echo -e "\t-s\tShow script version"
    echo -e "\t-S\tUpdate script ${RED}--> Not yet working${CRS}"
    echo -e "\t-h\tPrint this help page.\n"
    echo -e "Configured locations:\n"
    echo -e "\t${GRN}${INSTALL_PATH}${CRS}\t\tInstall location"
    echo -e "\t${GRN}${LAUNCHER_PATH}${CRS}\tLauncher location"
    echo -e "\t${GRN}${SCRIPT_PATH}${CRS}\t\t\tScript location\n"
}

### CLEAN
clean_tmp() {
    rm -rf $TMP
}

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

## GET OPTIONS
while getopts ":i,:L,:u,:U,:r,:c,:s,:S,:h" OPTION ; do
    case $OPTION in
        i)
            echo -e "\n  ${YEL}Installing Helium Browser...${CRS}"
            dependency_check
            install_helium
            clean_tmp
            exit ;;

        L)
            echo -e "\n  ${YEL}Adding Helium Browser launcher...${CRS}"
            dependency_check
            add_launcher
            #clean_tmp
            exit ;;

        u)
            echo -e "\n  ${YEL}Updating Helium Browser...${CRS}"
            dependency_check
            update_helium
            clean_tmp
            exit ;;

        r)
            echo -e "\n  ${YEL}Removing Helium Browser...${CRS}"
            remove_helium
            exit ;;

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

        s)
            script_version
            exit ;;

        h)
            help
            exit;;

        *)
            echo -e "\n  ${YEL}Option 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
clean_tmp

exit 0