feat(YouTube - Hide Shorts components): Add patch option to hide Shor…

…ts from app launcher widget Beta (ReVanced#3707)
zainarbani · Sep 28, 2024 · 838f183 · 838f183
1 parent b794236
commit 838f183
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 7 deletions.
diff --git a/src/main/kotlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsPatch.kt b/src/main/kotlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsPatch.kt
@@ -81,7 +81,14 @@ object HideShortsComponentsPatch : BytecodePatch(
         key = "hideShortsAppShortcut",
         default = false,
         title = "Hide Shorts app shortcut",
-        description = "Permanently hides the shortcut to open Shorts from long pressing the app icon in your launcher."
+        description = "Permanently hides the shortcut to open Shorts when long pressing the app icon in your launcher."
+    )
+
+    internal val hideShortsWidget by booleanPatchOption(
+        key = "hideShortsWidget",
+        default = false,
+        title = "Hide Shorts widget",
+        description = "Permanently hides the launcher widget Shorts button."
     )
 
     override fun execute(context: BytecodeContext) {

diff --git a/...tlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsResourcePatch.kt b/...tlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsResourcePatch.kt
@@ -7,6 +7,7 @@ import app.revanced.patches.all.misc.resources.AddResourcesPatch
 import app.revanced.patches.shared.misc.mapping.ResourceMappingPatch
 import app.revanced.patches.shared.misc.settings.preference.SwitchPreference
 import app.revanced.patches.youtube.layout.hide.shorts.HideShortsComponentsPatch.hideShortsAppShortcut
+import app.revanced.patches.youtube.layout.hide.shorts.HideShortsComponentsPatch.hideShortsWidget
 import app.revanced.patches.youtube.misc.settings.SettingsPatch
 import app.revanced.util.findElementByAttributeValueOrThrow
 import org.w3c.dom.Element
@@ -57,12 +58,21 @@ object HideShortsComponentsResourcePatch : ResourcePatch() {
 
         if (hideShortsAppShortcut == true) {
             context.xmlEditor["res/xml/main_shortcuts.xml"].use { editor ->
-                val shortcuts = editor.file.getElementsByTagName("shortcuts").item(0) as Element
-                val shortsItem =
-                    shortcuts.getElementsByTagName("shortcut").findElementByAttributeValueOrThrow(
-                        "android:shortcutId",
-                        "shorts-shortcut"
-                    )
+                val shortsItem = editor.file.childNodes.findElementByAttributeValueOrThrow(
+                    "android:shortcutId",
+                    "shorts-shortcut"
+                )
+
+                shortsItem.parentNode.removeChild(shortsItem)
+            }
+        }
+
+        if (hideShortsWidget == true) {
+            context.xmlEditor["res/layout/appwidget_two_rows.xml"].use { editor ->
+                val shortsItem = editor.file.childNodes.findElementByAttributeValueOrThrow(
+                    "android:id",
+                    "@id/button_shorts_container"
+                )
 
                 shortsItem.parentNode.removeChild(shortsItem)
             }