installer.sh

#!/bin/bash
# Copyright 2021-2023 nunopenim @github
# Copyright 2021-2023 prototype74 @github
#
# Licensed under the PEL (Penim Enterprises License), v1.0
#
# You may not use this file or any of the content within it, unless in
# compliance with the PE License

RED="\033[91m"
GREEN="\033[92m"
YELLOW="\033[93m"
END="\033[0m"

function setColor() {
    echo "$1$2${END}"
}

if ([ -f ./userbot/__init__.py ] && \
    [ -f ./userbot/__main__.py ]) || \
   ([ -d ./HyperUBot ] && \
    [ -f ./HyperUBot/userbot/__init__.py ] && \
    [ -f ./HyperUBot/userbot/__main__.py ]); then
    printf "$(setColor $GREEN 'HyperUBot is installed already')\n"
    exit 0
fi

os_name=$(setColor $RED "Unsupported")

# Check OS
if [[ "$OSTYPE" == "linux-android" || "$OSTYPE" == "linux-androideabi" ]]; then
    os_name="Android"
elif [[ "$OSTYPE" == "linux-gnu"* ]]; then
    if [ -f "/etc/debian_version" ]; then
        os_name="Debian"
    elif [ -f "/etc/arch-release" ]; then
        os_name="Arch Linux"
    elif [ -f "/etc/redhat-release" ]; then
        os_name="Red Hat"
    fi
elif [[ "$OSTYPE" =~ "darwin"* ]]; then
    os_name="macOS"
elif [ "$OSTYPE" == "linux-musl" ]; then
    if [ -f "/etc/alpine-release" ]; then
        os_name="Alpine Linux"
    fi
fi

# Generated with ASCII Art Generator: http://patorjk.com/software/taag/
printf " _   _                       _   _ ____        _   \n"
printf "| | | |_   _ _ __   ___ _ __| | | | __ )  ___ | |_ \n"
printf "| |_| | | | | '_ \ / _ \ '__| | | |  _ \ / _ \| __|\n"
printf "|  _  | |_| | |_) |  __/ |  | |_| | |_) | (_) | |_ \n"
printf "|_| |_|\__, | .__/ \___|_|   \___/|____/ \___/ \__|\n"
printf "       |___/|_|                                    \n"
printf "A customizable, modular Telegram userbot, with innovative components.\n\n"

printf "Operating System: $os_name\n"

function printRelease() {
    source /etc/os-release
    echo "Release: $PRETTY_NAME"
    printf "\n"
}

function ver_lt() {
    printf "$1\n$2\n" | sort -V | tail -1
}

function getLatestPyBin() {
    usr_bin=/usr/bin
    find_type=f
    case "$os_name" in
        "Android")
            usr_bin=$PREFIX/bin
            ;;
        "macOS")
            usr_bin=/usr/local/bin
            find_type=l
            ;;
    esac
    latest_py_bin=$(find $usr_bin -type $find_type -name "python*" -exec basename {} \; | grep -E "^python[0-9].[0-9]{1,4}$" | sort -V | tail -1)
    echo "$latest_py_bin"
}

# Install pre-requisites packages
case "$os_name" in
    "Android")
        release=$(getprop ro.build.version.release)
        release_sdk=$(getprop ro.build.version.sdk)
        printf "Release: %s (API: %s)\n\n" $release $release_sdk
        printf "%s\n" "-- Installing pre-requisites packages..."
        pkg update
        pkg install python python-cryptography git neofetch ffmpeg flac libffi
        #pkg install rust # reference to line no. 225 below
        ;;
    "Debian")
        printRelease
        printf "%s\n" "-- Installing pre-requisites packages (sudo needed)..."
        sudo apt-get update
        curr_py_bin=$(getLatestPyBin)
        if [[ -z $curr_py_bin || $(ver_lt $($curr_py_bin --version | cut -d " " -f2) "3.7."*) == "3.7."* ]]; then
            # check if (outdated) python is (not) installed.
            # If so, install the latest python version the distro does offer
            py_apt=$(apt-cache pkgnames | grep -E "^python3.[0-9]{1,4}$" | sort -V | tail -1)
            if [ -z $py_apt ]; then
                py_apt=python3
            fi
            sudo apt-get install $py_apt python3-pip $py_apt-dev libffi-dev git neofetch ffmpeg flac net-tools
        else
            sudo apt-get install python3-pip $curr_py_bin-dev libffi-dev git neofetch ffmpeg flac net-tools
        fi
        ;;
    "Arch Linux")
        printf "\n"
        printf "%s\n" "-- Installing pre-requisites packages (sudo needed)..."
        sudo pacman -Sy python3 python-pip git neofetch ffmpeg flac libffi
        ;;
    "Red Hat")
        printRelease
        printf "%s\n" "-- Installing pre-requisites packages (sudo needed)..."
        sudo dnf update
        is_fedora=$(cat /etc/redhat-release | grep "Fedora")
        if [ ! -z "$is_fedora" ]; then
            sudo dnf -y install https://download1.rpmfusion.org/free/fedora/rpmfusion-free-release-$(rpm -E %fedora).noarch.rpm https://download1.rpmfusion.org/nonfree/fedora/rpmfusion-nonfree-release-$(rpm -E %fedora).noarch.rpm
            sudo dnf install python3 python3-pip git neofetch flac ffmpeg libffi
        else  # other RHEL 8 systems
            sudo dnf install epel-release
            sudo dnf config-manager --enable epel
            sudo dnf config-manager --set-enabled powertools
            sudo dnf install --nogpgcheck https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm
            sudo dnf install --nogpgcheck https://mirrors.rpmfusion.org/free/el/rpmfusion-free-release-8.noarch.rpm https://mirrors.rpmfusion.org/nonfree/el/rpmfusion-nonfree-release-8.noarch.rpm
            curr_py_bin=$(getLatestPyBin)
            if [[ -z $curr_py_bin || $(ver_lt $($curr_py_bin --version | cut -d " " -f2) "3.7."*) == "3.7."* ]]; then
                py_dnf=$(dnf search python | grep -E "^python3[0-9]{1,4}" | sort -V | tail -1 | cut -d "." -f1)
                if [ -z $py_dnf ]; then
                    py_dnf=$(dnf search python | grep -E "^python3.[0-9]{1,4}" | sort -V | tail -1 | cut -d "." -f1-2)
                    if [ -z $py_dnf ]; then
                        py_dnf=python3
                    fi
                fi
                sudo dnf install $py_dnf python3-pip git neofetch flac ffmpeg libffi
            else
                sudo dnf install python3-pip git neofetch flac ffmpeg libffi
            fi
        fi
        ;;
    "macOS")
        release=$(sw_vers -productVersion)
        printf "Release: %s\n\n" $release
        printf "%s\n" "-- Checking for Homebrew Package Manager..."
        if [ -z $(command -v brew) ]; then
            printf "%s\n" "-- Installing Homebrew Package Manager (sudo needed)..."
            # From https://brew.sh/
            /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
        fi
        printf "%s\n" "-- Installing pre-requisites packages..."
        brew install git python3 ffmpeg flac neofetch libffi
        ;;
    "Alpine Linux")
        printRelease
        printf "%s\n" "-- Installing pre-requisites packages (sudo needed)..."
        sudo apk update
        sudo apk add git gcc python3 py3-pip python3-dev libffi-dev musl-dev openssl-dev cargo libressl-dev ffmpeg flac neofetch
        ;;
    *)
        printf "Current operating system is not supported...\n"
        exit 1
        ;;
esac

# Python --version >=3.8
printf "%s\n" "-- Checking for Python..."
py_exec=$(getLatestPyBin)

if [ -z $py_exec ]; then
    printf "$(setColor $RED '%s')\n" "-- Python is not installed"
    exit 1
fi

py_ver_str=$($py_exec --version | cut -d " " -f2)

if [ $(ver_lt $py_ver_str "3.7."*) == "3.7."* ]; then
    printf "$(setColor $YELLOW '%s %s')\n" "-- HyperUBot requires at least Python v3.8! Latest installed version is" "$py_ver_str"
    exit 1
fi

printf "%s\n" "-- Python $py_ver_str is installed!"

printf "%s\n" "-- Fetching latest release from HyperUBot's Repository..."
get_release=$(curl --silent -H "Accept: application/json" "https://api.github.com/repos/prototype74/HyperUBot/releases/latest")

tar_pkg=$(grep '"tarball_url"' <<< "$get_release" | cut -d '"' -f4)
if [ -z $tar_pkg ]; then
    message=$(grep '"message"' <<< "$get_release" | cut -d '"' -f4)
    printf "$(setColor $RED '%s: %s')\n" "-- Failed to download HyperUBot" "$message"
    exit 1
fi

release_ver=$(grep '"tag_name"' <<< "$get_release" | cut -d '"' -f4)
curr_path=$(pwd)
dir_name="HyperUBot"

printf "%s\n" "-- Downloading HyperUBot ($release_ver)..."
curl -L --silent "$tar_pkg" --output HyperUBot.tar.gz || { printf "$(setColor $RED '%s')\n" "-- Failed to download HyperUBot" && exit 1; }

printf "%s\n" "-- Creating HyperUBot's directory in '$curr_path'..."
mkdir $dir_name || { printf "$(setColor $RED '%s')\n" "-- Failed to create directory" && exit 1; }

printf "%s\n" "-- Installing HyperUBot..."
tar -xf ./HyperUBot.tar.gz --directory ./$dir_name --strip-components=1 || { printf "$(setColor $RED '%s')\n" "-- Installation failed!" && exit 1; }
rm -f ./HyperUBot.tar.gz

if [ -d $dir_name ] && \
   [ -f $dir_name/userbot/__init__.py ] && \
   [ -f $dir_name/userbot/__main__.py ]; then
    printf "%s\n" "-- HyperUBot has been installed successfully!"
else
    printf "$(setColor $RED '%s')\n" "-- Installation was not successful!"
    exit 1
fi

cd $dir_name

# Uncomment if you want to build cryptography yourself using pip and rust
# As supported by Android ABIs: https://developer.android.com/ndk/guides/abis#sa
#if [ "$os_name" == "Android" ]; then
#    arch_type=""
#    case $(uname -m) in
#        "aarch64") arch_type="aarch64-linux-android" ;;
#        "armv7l") arch_type="armv7-linux-androideabi" ;;
#        "arm")  arch_type="arm-linux-androideabi" ;;
#        "i686" | "x86") arch_type="i686-linux-android" ;;
#        "x86_64") arch_type="x86_64-linux-android" ;;
#        *)
#            printf "$(setColor $YELLOW '%s')\n\n" "-- Could not detect hardware ABI automatically!"
#            printf "Please select the hardware ABI of your device:\n"
#            select choice in "arm64-v8a (AArch64)" "armeabi-v7a" "armeabi" "x86 (i686)" "x86_64";
#            do
#                case $choice in
#                    "arm64-v8a (AArch64)") arch_type="aarch64-linux-android" ; break ;;
#                    "armeabi-v7a") arch_type="armv7-linux-androideabi" ; break ;;
#                    "armeabi")  arch_type="arm-linux-androideabi" ; break ;;
#                    "x86 (i686)") arch_type="i686-linux-android" ; break ;;
#                    "x86_64") arch_type="x86_64-linux-android" ; break ;;
#                esac
#            done
#            printf "\n"
#    esac
#    # Required for rust compiler
#    export CARGO_BUILD_TARGET=$arch_type
#fi

printf "%s\n" "-- Upgrading pip and setuptools..."
$py_exec -m pip install --upgrade pip setuptools

printf "%s\n" "-- Installing required pip packages..."
while true; do
    $py_exec -m pip install -r requirements.txt
    if [ $? -ne 0 ]; then
        printf "\n"
        printf "$(setColor $YELLOW '%s')\n" "-- pip installation was not successful. If pip is not installed, install it manually. For all other cases it may be possible that a pre-requisites package is missing. Install the package/lib/app the pip package does require. Finally, try the pip installation again..."
        printf "\n"
        while true; do
            read -p "Re-try pip installation? (y/n): " user_input
            if [[ $user_input =~ ^[yY]$ ]]; then
                break
            elif [[ $user_input =~ ^[nN]$ ]]; then
                printf "$(setColor $RED '%s')\n" "-- Installer cancelled..."
                cd ..
                rm -rf $dir_name
                exit 1
            else
                printf "$(setColor $YELLOW 'Invalid input. Try again...')\n"
            fi
        done
    else
        break
    fi
done

printf "$(setColor $GREEN '%s')\n" "-- Installer finished successfully!"
printf "\n"

while true; do
    read -p "Do you wish to start the Setup Assistant now? (y/n): " user_input
    if [[ $user_input =~ ^[yY]$ ]]; then
        printf "Starting Setup Assistant...\n"
        printf "\n"
        $py_exec setup.py -nopip
        break
    elif [[ $user_input =~ ^[nN]$ ]]; then
        break
    else
        printf "$(setColor $YELLOW 'Invalid input. Try again...')\n"
    fi
done

exit 0