version="${1:-1.0.1}"
bashrc=~/.bashrc
zshrc=~/.zshrc
originaldir=$PWD
environment=$(ps -o args= -p $$ | grep -Em 1 -o '\w{0,5}sh' | head -1)
if [ "$environment" = "bash" ] || [ "$environment" = "zsh" ];
then
    if ! grep -qs 'bash ~/.welcome/welcome.sh' $bashrc && ! grep -qs 'zsh ~/.welcome/welcome.sh' $zshrc && ! grep -qs 'bash /home/$USER/.welcome/welcome.sh' $bashrc && ! grep -qs 'zsh /home/$USER/.welcome/welcome.sh' $zshrc;
    then
        echo "Welcome! Installing v$version in $environment..."
        tput sc
        cd ~/
        mkdir -p ~/.welcome
        if which curl >/dev/null ;
        then
            curl -SL https://github.com/G2-Games/welcome.sh/releases/download/v${version}/welcome.sh --output ~/.welcome/welcome.sh
            if [[ $(echo $version | sed 's/[.][.]*//g' ) -ge 100 ]]; then
                curl -SL https://github.com/G2-Games/welcome.sh/releases/download/v${version}/config.cfg --output ~/.welcome/config.cfg
            fi
        elif which wget >/dev/null ;
        then
            wget https://github.com/G2-Games/welcome.sh/releases/download/v${version}/welcome.sh --P ~/.welcome/
            if [[ $(echo $version | sed 's/[.][.]*//g' ) -ge 100 ]]; then
                wget https://github.com/G2-Games/welcome.sh/releases/download/v${version}/config.cfg --P ~/.welcome/
            fi
        else
            echo "Cannot download, neither wget nor curl is available."
            exit 1
        fi
        chmod +x ~/.welcome/welcome.sh
        if [[ "$environment" = "bash" ]];
        then
            echo "Installing to bashrc"
            echo 'bash ~/.welcome/welcome.sh' >> $bashrc
        elif [[ "$environment" = "zsh" ]];
        then
            echo "Installing to zshrc"
            echo 'zsh ~/.welcome/welcome.sh' >> $zshrc
        fi
        cd "$originaldir"
        tput rc el ed
        echo -e "\e[36mInstalled! \e[0m"
    else
        tput sc
        echo -e "\e[35mwelcome.sh\e[0m already installed!"
        if [[ $(echo $version | sed 's/[.][.]*//g' ) -gt $(grep version ~/.welcome/welcome.sh | sed 's/.*=//' | sed 's/[.][.]*//g') ]]; then
            echo -en "Do you want to \e[36mupdate \e[35mwelcome.sh\e[0m? (v$(grep version ~/.welcome/welcome.sh | sed 's/.*=//') => v$version) \n\e[36mY/n\e[0m"
            if [[ "$environment" = "bash" ]]; then read -p " " -n 1 -r;
            elif [[ "$environment" = "zsh" ]]; then read -q "REPLY? " -n 1 -r; fi
            echo
            if [[ $REPLY =~ ^[Yy]$ ]]
            then
                tput rc el ed
                echo "Updating..."
                tput sc
                rm ~/.welcome/welcome.sh
                if which curl >/dev/null ;
                then
                    curl -SL https://github.com/G2-Games/welcome.sh/releases/download/v${version}/welcome.sh --output ~/.welcome/welcome.sh
                    if ! [[ -a "~/.welcome/config.cfg" ]]; then
                        curl -SL https://github.com/G2-Games/welcome.sh/releases/download/v${version}/config.cfg --output ~/.welcome/config.cfg
                    fi
                elif which wget >/dev/null ;
                then
                    wget https://github.com/G2-Games/welcome.sh/releases/download/v${version}/welcome.sh --P ~/.welcome/
                    if ! [[ -a "~/.welcome/config.cfg" ]]; then
                        wget https://github.com/G2-Games/welcome.sh/releases/download/v${version}/config.cfg --P ~/.welcome/
                    fi
                else
                    echo "Cannot download, neither Wget nor cURL is available."
                    exit 1
                fi
                # Check for older versions and replace bashrc lines #
                if grep -sn 'bash /home/$USER/.welcome/welcome.sh' $bashrc ;
                then
                    line=$(grep -sn 'bash /home/$USER/.welcome/welcome.sh' $bashrc)
                    line=${line%:*}
                    sed "${line}d" $bashrc > file.tmp && mv file.tmp $bashrc
                    echo 'bash ~/.welcome/welcome.sh' >> $bashrc
                fi
                if grep -sn 'zsh /home/$USER/.welcome/welcome.sh' $zshrc ;
                then
                    line=$(grep -sn 'zsh /home/$USER/.welcome/welcome.sh' $zshrc)
                    line=${line%:*}
                    sed "${line}d" $zshrc > file.tmp && mv file.tmp $zshrc
                    echo 'zsh ~/.welcome/welcome.sh' >> $zshrc
                fi
                # End older version check #

                tput rc el ed
                echo -e "\e[32mUpdated to v$version! \e[0m"
                exit 0
            else
                tput rc el ed
                echo -e "\e[35mwelcome.sh\e[0m already installed!"
            fi
        fi
        echo -en "Do you want to \e[31muninstall \e[35mwelcome.sh\e[0m?\n\e[36mY/n\e[0m"
        if [[ "$environment" = "bash" ]]; then read -p " " -n 1 -r;
        elif [[ "$environment" = "zsh" ]]; then read -q "REPLY? " -n 1 -r; fi
        echo
        if [[ $REPLY =~ ^[Yy]$ ]]
        then
            tput rc el ed
            echo "Goodbye. Uninstalling..."
            tput sc
            rm ~/.welcome/welcome.sh
            rm ~/.welcome/config.cfg
            rmdir ~/.welcome
            if grep -sn 'bash ~/.welcome/welcome.sh' $bashrc ;
            then
                line=$(grep -sn 'bash ~/.welcome/welcome.sh' $bashrc)
                line=${line%:*}
                sed -i "${line}d" $bashrc
                sed "${line}d" $bashrc > file.tmp && mv file.tmp $bashrc
            fi
            if grep -sn 'zsh ~/.welcome/welcome.sh' $zshrc ;
            then
                line=$(grep -sn 'zsh ~/.welcome/welcome.sh' $zshrc)
                line=${line%:*}
                sed "${line}d" $zshrc > file.tmp && mv file.tmp $zshrc
            fi

            # Check for older versions #
            if grep -sn 'bash /home/$USER/.welcome/welcome.sh' $bashrc ;
            then
                line=$(grep -sn 'bash /home/$USER/.welcome/welcome.sh' $bashrc)
                line=${line%:*}
                sed "${line}d" $bashrc > file.tmp && mv file.tmp $bashrc
            fi
            if grep -sn 'zsh /home/$USER/.welcome/welcome.sh' $zshrc ;
            then
                line=$(grep -sn 'zsh /home/$USER/.welcome/welcome.sh' $zshrc)
                line=${line%:*}
                sed "${line}d" $zshrc > file.tmp && mv file.tmp $zshrc
            fi
            # End older version check #

            tput rc el ed
            echo -e "\e[36mUninstalled! \e[0m"
        else
            tput rc el ed
            echo -e "\e[31mCancelled. \e[0m"
            exit 0
        fi
    fi
else
    printf "\e[31;5mERROR:\e[0m \e[31;3mThis script can only be installed in Bash or Zsh.\e[0m\n"
fi