patch.sh

#!/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 w/ integrations, exiting!"
	fi
}

get_version_info() {
	## revanced cli
	resp=$(curl -sL -H 'User-Agent: revanced-creator' "https://github.com/revanced/revanced-$1/releases/latest")
	ver=$(printf '%s' "$resp" | $grep -oe "v[0-9].*[0-9]" | awk -F'/' 'NR==1 {print $1}')
	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 integrations; do
		get_version_info "$i" &
		if [ "$i" = "integrations" ] && [ "$integrations" = "enabled" ]; then
			get_version_info "$i" &
		fi
	done

	## wait for getting versions to get finished
	wait

	for i in cli patches integrations; 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"
	if [ "$integrations" = "enabled" ]; then
		log "revanced_integrations_version : $revanced_integrations_version"
	fi
}

get_stock_apk_version() {
	notset "$apk_version" && apk_version=$(curl -sL -H 'User-Agent: revanced-creator' "https://api.github.com/repos/XDream8/revanced-creator/releases" | sed 's/"name":"/\n/g; s/.apk",/\n/g' | $grep -ioe "^$what_to_patch-[0-9].*[0-9]" | $grep -oe "[0-9].*[0-9]" | awk 'END{print}')
	notset "$apk_version" && {
		err "getting $what_to_patch apk version failed, exiting!"
	}
	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 "$integrations_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 \
		$integrations_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 -b $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 "17" ]; then
		err "java 17 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 -e 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
		integrations="enabled"
		addarg "-e 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
		integrations="enabled"
		addarg "-e debug-mode"
		;;
	twitter)
		get_stock_apk_version
		apk_filename=Twitter-$apk_version.apk
		integrations="enabled"
		;;
	reddit)
		get_stock_apk_version
		apk_filename=Reddit-$apk_version.apk
		;;
	tiktok)
		get_stock_apk_version
		apk_filename=TikTok-$apk_version.apk
		integrations="enabled"
		;;
	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"
		integrations="enabled"
		[ -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=revanced-patches-$revanced_patches_version.jar
	integrations_filename=revanced-integrations-$revanced_integrations_version.apk

	## add integrations arg
	if [ "$integrations" = "enabled" ]; then
		addarg "-m $integrations_filename"
	fi

	## 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
	if [ "$integrations" = "enabled" ]; then
		[ ! -f "$integrations_filename" ] && integrations_link=https://github.com/revanced/revanced-integrations/releases/download/v$revanced_integrations_version/$integrations_filename
	fi

	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: