#!/bin/sh -e
# shellcheck disable=2154

#---------------------------#
#     Made by XDream8       #
#---------------------------#
#           deps            #
#---------------------------#
# aria2, axel, curl or wget #
#          awk              #
#         java(17)          #
#     grep or ripgrep       #
#---------------------------#

RED='\033[0;31m'
GREEN='\033[1;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m'

out() {
    # print a message
    printf '%b\n' "$@"
}

log() {
    printf '%b\n' "${YELLOW}$*${NC}"
}

err() {
    printf '%b\n' "${RED}$*${NC}"
    exit 1
}

notset() {
    case $1 in '') return 0 ;; *) return 1 ;; esac
}

equals() {
    case $1 in "$2") return 0 ;; *) return 1 ;; esac
}

cleanup() {
    for tmp_file in "cache/tmp.revanced_"*; do
        rm -rf "$tmp_file"
    done
}

check_dep() {
    if ! command -v "$1" >/dev/null; then
        [ "$2" ] && out "${RED}$2${NC}"
        return 1
    else
        return 0
    fi
}

checkadb() {
    if [ ! "$(pidof adb)" ]; then
        if check_dep "rdo"; then
            sudo=rdo
        elif check_dep "doas"; then
            sudo=doas
        elif check_dep "sudo"; then
            sudo=sudo
        else
            sudo=
        fi

        log "starting adb server"
        $sudo adb start-server || {
            err "starting adb server failed, exiting!"
        }
    fi

    device_id=$(adb devices | awk 'FNR == 2 {print $1}')
    if [ "$device_id" = "" ]; then
        err "your phone is not connected to pc, exiting!"
    else
        log "adb device_id=$device_id"
    fi
}

checkyt() {
    # shellcheck disable=2143
    if [ ! "$(adb shell cmd package list packages | $grep -o 'com.google.android.youtube')" ]; then
        err "root variant: install $what_to_patch v${apk_version} on your device to mount, exiting!"
    fi
}

get_version_info() {
    ## revanced cli
    resp=$(curl -sL "https://github.com/ReVanced/revanced-$1/releases/latest")
    ver=$(printf '%s' "$resp" | $grep -oe "tag/v[0-9].*[0-9]" | sort -n | awk -F'/' 'NR==1 {print $NF}')
    ver=${ver#v}
    printf '%s\n' "$ver" >"cache/tmp.revanced_$1"
}

get_and_print_versions() {
    ## getting versions information
    out "${BLUE}getting latest versions info${NC}"
    for i in cli patches; do
        get_version_info "$i" &
    done

    ## wait for getting versions to get finished
    wait

    for i in cli patches; do
        export "revanced_${i}_version=$(cat cache/tmp.revanced_$i)"
    done

    ## print info
    log "revanced_cli_version : $revanced_cli_version"
    log "revanced_patches_version : $revanced_patches_version"
}

get_stock_apk_version() {
    notset "$apk_version" && {
        apk_version=$(curl -sL "https://api.github.com/repos/XDream8/revanced-creator/releases" | $grep -ioe "$what_to_patch-[0-9].*[0-9].apk" | $grep -oe "[0-9].*[0-9]" | sort -n | awk 'END{print}')
    }
    notset "$apk_version" && {
        if curl -sL "https://api.github.com" | grep -q "API rate limit exceeded for"; then
            err "getting $what_to_patch apk version failed: github rate limit exceeded"
        else
            err "getting $what_to_patch apk version failed, exiting!"
        fi
    }
    log "$what_to_patch version to be patched : $apk_version"
}

remove_old() {
    if check_dep "find"; then
        find . -maxdepth 1 -type f -name "revanced-*.jar" ! \( -name "*.keystore" -or -name "$cli_filename" -or -name "$patches_filename" -or -name "$apk_filename" \) -delete && out "${BLUE}removed old files${NC}"
    fi
}

download_needed() {
    # number
    n=0

    out "${BLUE}Downloading needed files${NC}"
    for i in \
        $cli_link \
        $patches_link \
        $aapt2_link \
        $apk_link; do
        n=$((n + 1))
        out "${CYAN}$n) ${YELLOW}downloading $i${NC}"
        $downloader "$i" &
    done
    wait
}

patch() {
    out "${BLUE}patching process started(${RED}$root_text${BLUE})${NC}"
    out "${BLUE}it may take a while please be patient${NC}"
    base_cmd="java -jar $cli_filename patch $apk_filename -o $output_apk -p $patches_filename"
    # shellcheck disable=2086
    $base_cmd $additional_args
}

addarg() {
    if notset "$additional_args"; then
        additional_args="$*"
    else
        additional_args="$additional_args $*"
    fi
}

main() {
    ## defaults
    : "${what_to_patch=youtube}"
    : "${root=0}"
    : "${grep=$(command -v rg || command -v grep)}"

    [ ! -d "cache" ] && mkdir -p cache

    ## downloader
    if notset "$downloader"; then
        if check_dep "aria2c"; then
            downloader="aria2c -x 16 -s 1"
        elif check_dep "axel"; then
            downloader="axel -n 16"
        elif check_dep "curl"; then
            downloader="curl -qLJO"
        elif check_dep "wget"; then
            downloader="wget"
        fi
    fi

    ## dependecy checks
    check_dep "curl" "curl is required, exiting!"
    equals "$downloader" "curl*" || check_dep "${downloader%% -*}" "${downloader%% -*} is missing, exiting!"
    equals "$root" "1" && check_dep "adb" "adb is required for root variant, exiting!"
    check_dep "java" "java 17 is required, exiting!"
    check_dep "awk" "awk is required, exiting!"
    check_dep "${grep%% -*}" "${grep%% -*} is required, exiting!"

    ## java version check
    JAVA_VERSION="$(java -version 2>&1 | $grep -o "version \".*\"" | awk 'match($0, /([0-9]+)/) {print substr($0, RSTART, RLENGTH)}')"
    if [ "$JAVA_VERSION" -lt "11" ]; then
        err "java 11(minimum) is required but you have version $JAVA_VERSION, exiting!"
    else
        log "java version $JAVA_VERSION found"
    fi

    ## check $root
    if [ "$root" -eq 0 ]; then
        root_text="non-root"
    else
        root_text="root"
        out "${RED}please be sure that your phone is connected to your pc, waiting 5 seconds${NC}"
        sleep 5s
        checkadb
        addarg "-d $device_id --disable=microg-support --mount"
    fi

    # termux support
    if command -v termux-setup-storage >/dev/null; then
        ## check arch
        case "$(uname -m)" in
        aarch64)
            aapt2_filename="termux-arm64-v8a-aapt2"
            ;;
        *)
            err "that architecture is not supported by revanced-creator at the moment, please create a issue"
            ;;
        esac

        addarg "--custom-aapt2-binary=./$aapt2_filename"
        [ ! -f "$aapt2_filename" ] && aapt2_link="https://github.com/XDream8/revanced-creator/releases/download/other/$aapt2_filename"
    fi

    ## what should we patch
    case "$what_to_patch" in
    youtube)
        get_stock_apk_version
        apk_filename=YouTube-$apk_version.apk
        addarg "--disable=enable-debugging"
        ;;
    youtube-music)
        get_stock_apk_version
        apk_filename=YouTube-Music-$apk_version.apk
        ;;
    twitch)
        get_stock_apk_version
        apk_filename=Twitch-$apk_version.apk
        addarg "--disable=debug-mode"
        ;;
    twitter)
        get_stock_apk_version
        apk_filename=Twitter-$apk_version.apk
        ;;
    reddit)
        get_stock_apk_version
        apk_filename=Reddit-$apk_version.apk
        ;;
    tiktok)
        get_stock_apk_version
        apk_filename=TikTok-$apk_version.apk
        ;;
    spotify)
        get_stock_apk_version
        apk_filename=Spotify-$apk_version.apk
        ;;
    *)
        if notset "$what_to_patch" && check_dep "find"; then
            what_to_patch=$(find . -maxdepth 1 -type f -iname "*.apk" -not -iname "app-release-unsigned.apk" -or -not -iname "revanced-*.apk" -or -not -iname "*.keystore" | sort -r | awk -F'/' 'NR==1 {print $2}')
        elif [ -z "$what_to_patch" ] && ! check_dep "find"; then
            err "please specify an apk file using '--patch/-p' flag"
        else
            [ -z "$what_to_patch" ] && {
                err "please specify an apk file using '--path/-p' flag"
            }
        fi
        if [ ! -f "$what_to_patch" ]; then
            err "apk file does not exist, please specify an existing apk file using 'apk_filename' arg"
        fi
        apk_filename="$what_to_patch"
        [ -z "$output_apk" ] && output_apk=revanced-$apk_filename
        log "custom apk : $apk_filename"
        ;;
    esac

    ## get and print versions
    get_and_print_versions

    ## set filenames
    cli_filename=revanced-cli-$revanced_cli_version-all.jar
    patches_filename=patches-$revanced_patches_version.rvp

    ## set output apk name
    notset "$output_apk" && {
        output_apk=revanced-${what_to_patch#*-}-$apk_version-$root_text.apk
    }

    ## link to download $what_to_patch
    [ ! -f "$apk_filename" ] && apk_link=https://github.com/XDream8/revanced-creator/releases/download/$what_to_patch/$apk_filename

    ## remove old files with find
    remove_old

    [ ! -f "$cli_filename" ] && cli_link=https://github.com/revanced/revanced-cli/releases/download/v$revanced_cli_version/$cli_filename
    [ ! -f "$patches_filename" ] && patches_link=https://github.com/revanced/revanced-patches/releases/download/v$revanced_patches_version/$patches_filename
    download_needed

    if [ "$root" -eq 1 ]; then
        out "${BLUE}root variant: installing stock youtube-$apk_version first${NC}"
        adb install -r "$apk_filename" || {
            err "install failed, exiting!"
        }
        checkyt
    fi

    if [ -f "$aapt2_filename" ]; then
        chmod +x "$aapt2_filename"
    fi

    patch
}

for opt in "$@"; do
    case $opt in
    --nocolors | -nc)
        # unset colors
        for color in RED GREEN YELLOW BLUE CYAN NC; do
            unset $color
        done
        ;;
    --root)
        root=1
        ;;
    --downloader=*)
        downloader="${opt#*=}"
        ;;
    --patch=* | -p=*)
        what_to_patch="${opt#*=}"
        ;;
    --additional-args=* | -args=*)
        addarg "${opt#*=}"
        ;;
    --help | -h | *)
        out "${RED}------${NC} ${CYAN}revanced-creator${NC} ${RED}------${NC}
${GREEN}Create revanced apps easily${NC}

${YELLOW}Usage:${NC} $0 ${BLUE}<flags>${NC}

${RED}--nocolors${NC},${RED} -nc\t\t\t\t\t${GREEN}Do not print colored text${NC}
${RED}--downloader=${BLUE}<string>\t\t\t\t${GREEN}Set a custom downloader
${RED}--root\t\t\t\t\t\t${GREEN}Build for rooted phones
${RED}--patch=${BLUE}<string>${NC},${RED} -p=${BLUE}<string>\t\t\t${GREEN}What should we patch
${RED}--additional-args=${BLUE}<string>${NC},${RED} -args=${BLUE}<string>\t${GREEN}Args that will be passed to java cmd
"
        exit 0
        ;;
    esac
done

trap cleanup EXIT
main

# vim: set sw=4 sts=4 ft=sh: