#!/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"

## GLOBAL VARIABLES
APP_NAME="Helium"
APP_NAME_LONG="Helium Web Browser"
SOURCE="GitHub"
SCRIPT_NAME="$APP_NAME Helper Script"
SCRIPT_VERSION="0.5.1"
DATE=$(date)

## LOCATIONS
TMP=/tmp/$APP_NAME
INSTALL_PATH=$HOME/.apps/$APP_NAME
CONFIG_PATH=$INSTALL_PATH/assets
BACKUP_PATH=$INSTALL_PATH/assets/backups
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
        exit
        
    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  ${CYN}$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

        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
            echo -e "\n  ${GRN}$APP_NAME_LONG is installed!${CRS}\n"
            exit

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

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

    done
}

### LAUNCHER --> TO DO
add_launcher() {
    if [ -f "$INSTALL_PATH/$APP_NAME.AppImage" ] ; then
        while true ; do
        read -p "  Add $APP_NAME_LONG launcher to $LAUNCHER_PATH? (Y/n) : " CONFIRM_LAUNCHER

        if [[ "$CONFIRM_LAUNCHER" == "" || "$CONFIRM_LAUNCHER" == "y" || "$CONFIRM_LAUNCHER" == "Y" ]] ; then
            ASSET_URL=https://lab.c95.org/fr4nz/helium-helper/src/branch/main/assets
            LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}')
            wget -P $TMP $ASSET_URL -q --show-progress
            sed -i "s|%APP_LOCATION%|$INSTALL_PATH/$APP_NAME.AppImage|" $TMP/assets/launcher/helium.desktop.template
            sed -i "s|%ICON_LOCATION%|$INSTALL_PATH/$CONFIG_PATH/icon/helium.png|" $TMP/assets/launcher/helium.desktop.template
            sed -i "s|%APP_VERSION%|$LOCAL_VERSION|" $TMP/assets/launcher/helium.desktop.template
            mkdir -p $LAUNCHER_PATH
            mv $TMP/assets/launcher/helium.desktop.template $LAUNCHER_PATH/$APP_NAME.desktop
            rm $TMP/assets/launcher
            mv $TMP/assets $CONFIG_PATH
            echo -e "\n  ${GRN}$APP_NAME_LONG launcher is set up!${CRS}\n"
            exit

        elif [[ "$CONFIRM_LAUNCHER" == "n" || "$CONFIRM_LAUNCHER" == "N" ]] ; then
            echo -e "\n  ${RED}Launcher creation cancelled!${CRS}\n"
            exit

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

    done

    elif [ -f "$LAUNCHER_PATH/$APP_NAME.desktop" ] ; then
        echo -e "\n  ${RED}${APP_NAME}.desktop already exists!${CRS}"
        exit

    else
        echo -e "\n  ${RED}$APP_NAME_LONG AppImage not found! Can't add launcher...${CRS}"
        install_helium
        exit
    
    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  ${RED}$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

            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
                if [ -f "$INSTALL_PATH/$APP_NAME.old" ] ; then
                    rm $INSTALL_PATH/$APP_NAME.old
                fi
                mv $INSTALL_PATH/$APP_NAME.AppImage $INSTALL_PATH/$APP_NAME.old
                mv $TMP/$GIT_FILENAME $INSTALL_PATH/$APP_NAME.AppImage
                echo -e "\n  ${GRN}$APP_NAME_LONG is up-to-date !${CRS}\n"
                break

            elif [[ "$CONFIRM" == "n" || "$CONFIRM" == "N" ]] ; then
                echo -e "\n  ${RED}Update 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}$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}"
        exit

    fi
}

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

        while true ; do
        read -p "  Remove $APP_NAME_LONG 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}$APP_NAME_LONG removed!${CRS}\n"
            exit

        elif [[ "$CONFIRM_REMOVE" == "n" || "$CONFIRM_REMOVE" == "N" ]] ; then
            echo -e "\n  ${RED}Removal cancelled!${CRS}\n"
            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}\n"
        exit

    fi
}

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

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

    fi
}

### COMPARE APPIMAGE VERSION
compare_version() {
    LOCAL_VERSION=$($INSTALL_PATH/$APP_NAME.AppImage --version | awk '{print $2}' >/dev/null 2>&1)
    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${CYN}${SCRIPT_NAME}${CRS}\n"
    echo -e "$APP_NAME_LONG installer & updater script\n"
    echo -e "Syntax: ${APP_NAME}_helper.sh [ -i | -L | -u | -U | -r | -c | -s | -S | -h ]\n"
    echo -e "Options:\n"
    echo -e "\t-i\tDownload & install $APP_NAME_LONG from $SOURCE"
    echo -e "\t-L\tCreate desktop launcher ${RED}--> Not yet working${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 ${RED}--> Not yet working${CRS}"
    echo -e "\t-h\tPrint this help page.\n"
    echo -e "Configured locations:\n"
    echo -e "\t${YEL}${INSTALL_PATH}${CRS}\t\tInstall location"
    echo -e "\t${YEL}${LAUNCHER_PATH}${CRS}\tLauncher location"
    echo -e "\t${YEL}${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,:u,:U,:r,:c,:s,:S,:h" OPTION ; do
    case $OPTION in
        i)
            echo -e "\n  ${YEL}Installing $APP_NAME_LONG...${CRS}"
            dependency_check
            local_check
            install_helium
            exit_clean
            exit ;;

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

        u)
            echo -e "\n  ${YEL}Updating $APP_NAME_LONG...${CRS}"
            dependency_check
            local_check
            update_helium
            exit_clean
            exit ;;

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

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

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

        s)
            script_version
            exit ;;

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

        h)
            help
            exit;;

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