#!/usr/bin/env bash

#【R.I.P】非公式YouTubeアプリ総合スレ11【Vanced】
#https://egg.5ch.net/test/read.cgi/android/1669198502/

AUTHOR="revanced"
WORKSPACE="$HOME/.$AUTHOR"

echo "(first arg) any options: $1"

TIMESTAMP=`date "+%Y%m%d%H%M"`

PATCHES_ARGS=" \
    -i settings -i theme \
    -i hide-create-button -i hide-shorts-button \
    -i minimized-playback -i remember-video-quality \
    -i video-ads -i general-ads \
    -i comments -i hide-info-cards -i hide-endscreen-cards \
    -i remove-player-button-background \
    -i block-audio-ads -i block-video-ads \
    -i show-deleted-messages \
    "

PACKAGE_NAME="com.google.android.youtube"
#CLI_VERSION=""
#INTEGRATIONS_VERSION=""
#PATCHES_VERSION=""

AAPT2_PATH=`which aapt2`
AAPT2_ARGS="--custom-aapt2-binary $AAPT2_PATH"
ANY_ARGS=$1
CACHE_PATH="$WORKSPACE/revanced-cache"
OPTIONS_PATH="$WORKSPACE/options.toml"
#PACKAGE_TINY_NAME=$(echo ${PACKAGE_NAME} | sed 's/^.*\.\([^\.]*\)$/\1/')

if [ ! -e "$WORKSPACE" ]; then
    mkdir -p $WORKSPACE
fi

UMOUNT_SCRIPT_NAME="umount_revanced_${PACKAGE_NAME}.sh"
UMOUNT_SCRIPT_PATH="${WORKSPACE}/${UMOUNT_SCRIPT_NAME}"
if [ -f "$UMOUNT_SCRIPT_PATH" ]; then
    echo "Removing $UMOUNT_SCRIPT_PATH"
    rm -f $UMOUNT_SCRIPT_PATH
fi
cat <<EOF > $UMOUNT_SCRIPT_PATH
#!/usr/bin/env bash
echo "Unmounting ..."
su -mm -c 'am force-stop ${PACKAGE_NAME} && \\
    grep ${PACKAGE_NAME} /proc/mounts | \\
    while read -r line; do echo \$line | \\
    cut -d " " -f 2 | \\
    sed "s/apk.*/apk/" | \\
    xargs -r umount -l > /dev/null 2>&1; done'
EOF
chmod 755 $UMOUNT_SCRIPT_PATH
$UMOUNT_SCRIPT_PATH

if [ -f "$OPTIONS_PATH" ]; then
    echo "Removing $OPTIONS_PATH"
    rm -f $OPTIONS_PATH
fi
cat << EOF > $OPTIONS_PATH
[theme]
darkThemeBackgroundColor = "@android:color/system_neutral1_900"
lightThemeBackgroundColor = "@android:color/system_neutral1_50"
EOF

# Check if curl is installed before continuing
if ! command -v "curl" &> "/dev/null"; then
    echo -e "\e[1;31mError: curl not found\e[0m"; exit 1
fi
CURL_HEADER="User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:102.0) Gecko/20100101 Firefox/102.0"

CLI_AUTHOR=${AUTHOR}
if [ -z "$CLI_VERSION" ]; then
    CLI_VERSION="$(curl -s https://api.github.com/repos/${CLI_AUTHOR}/revanced-cli/releases/latest | grep "tag_name")"
    CLI_VERSION="${CLI_VERSION:16:-2}"
fi

echo "revanced-cli --version: $CLI_VERSION"
CLI_PATH="$WORKSPACE/revanced-cli-$CLI_VERSION-all.jar"
if [ ! -f "$CLI_PATH" ]; then
    curl "https://github.com/${CLI_AUTHOR}/revanced-cli/releases/download/v$CLI_VERSION/revanced-cli-$CLI_VERSION-all.jar" -sSL -o "$CLI_PATH"
fi

PATCHES_AUTHOR=${AUTHOR}
if [ -z "$INTEGRATIONS_VERSION" ]; then
    INTEGRATIONS_VERSION="$(curl -s https://api.github.com/repos/${PATCHES_AUTHOR}/revanced-integrations/releases/latest | grep "tag_name")"
    INTEGRATIONS_VERSION="${INTEGRATIONS_VERSION:16:-2}"
fi

echo "revanced-integrations --version: $INTEGRATIONS_VERSION"
INTEGRATIONS_PATH="$WORKSPACE/revanced-integrations-$INTEGRATIONS_VERSION.apk"
if [ ! -f "$INTEGRATIONS_PATH" ]; then
    curl "https://github.com/${PATCHES_AUTHOR}/revanced-integrations/releases/download/v$INTEGRATIONS_VERSION/app-release-unsigned.apk" -sSL -o "$INTEGRATIONS_PATH"
fi

if [ -z "$PATCHES_VERSION" ]; then
    PATCHES_VERSION="$(curl -s https://api.github.com/repos/${PATCHES_AUTHOR}/revanced-patches/releases/latest | grep "tag_name")"
    PATCHES_VERSION="${PATCHES_VERSION:16:-2}"
fi

echo "revanced-patches --version: $PATCHES_VERSION"
PATCHES_PATH="$WORKSPACE/revanced-patches-$PATCHES_VERSION.jar"
PATCHES_JSON_PATH="$WORKSPACE/patches.json"
if [ ! -f "$PATCHES_PATH" ]; then
    curl "https://github.com/${PATCHES_AUTHOR}/revanced-patches/releases/download/v$PATCHES_VERSION/revanced-patches-$PATCHES_VERSION.jar" -sSL -o "$PATCHES_PATH"
    curl "https://github.com/${PATCHES_AUTHOR}/revanced-patches/releases/download/v$PATCHES_VERSION/patches.json" -sSL -o "$PATCHES_JSON_PATH"
fi

APK_PATH="$WORKSPACE/$PACKAGE_NAME.apk"
if [ ! -f "$APK_PATH" ]; then
    su -mm -c "cp \$(pm path ${PACKAGE_NAME} | grep base | sed 's/package://g') ${APK_PATH} && \
    chown u0_a330 ${APK_PATH} && \
    chmod 644 ${APK_PATH}"
fi

OUTPUT_PATH="$APK_PATH.rv-$TIMESTAMP.apk"

CLI_COMMAND_ARGS=" \
    -jar $CLI_PATH \
    -a $APK_PATH \
    -o $OUTPUT_PATH \
    -b $PATCHES_PATH \
    -m $INTEGRATIONS_PATH \
    $PATCHES_ARGS \
    $AAPT2_ARGS \
    --mount --exclusive --experimental \
    -t $CACHE_PATH --options $OPTIONS_PATH \
    "

#echo "Command:"
echo "java $CLI_COMMAND_ARGS $ANY_ARGS" | \
    sed -e 's/\ \+/\ /g' \
    > "$WORKSPACE/command.txt"

PATCH_LIST_PATH="${WORKSPACE}/patch.list.${TIMESTAMP}.txt"
if [ -f "$PATCH_LIST_PATH" ]; then
    echo "Removing $PATCH_LIST_PATH"
    rm -f $PATCH_LIST_PATH
fi

echo "Java:"
java $CLI_COMMAND_ARGS $ANY_ARGS

rm -rf $CACHE_PATH

MOUNT_SCRIPT_NAME="mount_revanced_${PACKAGE_NAME}.sh"
MOUNT_SCRIPT_PATH="${WORKSPACE}/${MOUNT_SCRIPT_NAME}"
MOUNT_SCRIPT_ROOT_PATH="/data/adb/service.d/${MOUNT_SCRIPT_NAME}"
if [ -f "$MOUNT_SCRIPT_PATH" ]; then
    echo "Removing $MOUNT_SCRIPT_PATH"
    rm -f $MOUNT_SCRIPT_PATH
fi
cat <<EOF > $MOUNT_SCRIPT_PATH
#!/system/bin/sh
MAGISKTMP="\$(magisk --path)" || MAGISKTMP=/sbin
MIRROR="\$MAGISKTMP/.magisk/mirror"
while [ "\$(getprop sys.boot_completed | tr -d '\r')" != "1" ]; do sleep 1; done

base_path="/data/adb/${PACKAGE_NAME}.apk"
stock_path=\$( pm path ${PACKAGE_NAME} | grep base | sed 's/package://g' )

chcon u:object_r:apk_data_file:s0  \$base_path
mount -o bind \$MIRROR\$base_path \$stock_path
EOF

chmod 755 $MOUNT_SCRIPT_PATH
chmod 644 $OUTPUT_PATH

$UMOUNT_SCRIPT_PATH
echo "Copying to root ..."

su -mm -c "cp ${MOUNT_SCRIPT_PATH} ${MOUNT_SCRIPT_ROOT_PATH} && \
    cp ${OUTPUT_PATH} /data/adb/${PACKAGE_NAME}.apk; \
    ${MOUNT_SCRIPT_ROOT_PATH}"

echo "Finished!"

exit 0