refactor(youtube/hide-cast-button): quality of life changes

This commit is contained in:
oSumAtrIX 2022-11-24 23:14:02 +01:00
parent f8686b7d1a
commit 8c8233a010
No known key found for this signature in database
GPG Key ID: A9B3094ACDB604B4

View File

@ -7,6 +7,7 @@ import app.revanced.patcher.data.BytecodeContext
import app.revanced.patcher.extensions.addInstructions
import app.revanced.patcher.patch.BytecodePatch
import app.revanced.patcher.patch.PatchResult
import app.revanced.patcher.patch.PatchResultError
import app.revanced.patcher.patch.PatchResultSuccess
import app.revanced.patcher.patch.annotations.DependsOn
import app.revanced.patcher.patch.annotations.Patch
@ -34,13 +35,15 @@ class HideCastButtonPatch : BytecodePatch() {
)
)
context.classes.forEach { classDef ->
classDef.methods.forEach { method ->
if (classDef.type.endsWith("MediaRouteButton;") && method.name == "setVisibility") {
val setVisibilityMethod =
context.proxy(classDef).mutableClass.methods.first { it.name == "setVisibility" }
setVisibilityMethod.addInstructions(
with(
context.findClass("MediaRouteButton")
?: return PatchResultError("MediaRouteButton class not found.")
) {
with(
mutableClass.methods.find { it.name == "setVisibility" }
?: return PatchResultError("setVisibility method not found.")
) {
addInstructions(
0, """
invoke-static {p1}, Lapp/revanced/integrations/patches/HideCastButtonPatch;->getCastButtonOverrideV2(I)I
move-result p1
@ -48,7 +51,6 @@ class HideCastButtonPatch : BytecodePatch() {
)
}
}
}
return PatchResultSuccess()
}