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 index b87662d86a..e1ba723910 100644 --- 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/src/main/kotlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsResourcePatch.kt b/src/main/kotlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsResourcePatch.kt index 714fa62d2e..a3c58a30b1 100644 --- a/src/main/kotlin/app/revanced/patches/youtube/layout/hide/shorts/HideShortsComponentsResourcePatch.kt +++ b/src/main/kotlin/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) }