commit c1d3e0736a3633760b237abed7b3c87b5a39933a
parent 342bc7c68bc0c99de11ba0029367d3bf04662697
Author: rhunk <101876869+rhunk@users.noreply.github.com>
Date:   Sun,  3 Mar 2024 19:41:40 +0100

feat: composer utils

Diffstat:
Mcore/src/main/kotlin/me/rhunk/snapenhance/core/data/SnapClassCache.kt | 4++++
Mcore/src/main/kotlin/me/rhunk/snapenhance/core/ui/ViewAppearanceHelper.kt | 12++++++++++++
Acore/src/main/kotlin/me/rhunk/snapenhance/core/wrapper/impl/composer/ComposerViewNode.kt | 51+++++++++++++++++++++++++++++++++++++++++++++++++++
3 files changed, 67 insertions(+), 0 deletions(-)

diff --git a/core/src/main/kotlin/me/rhunk/snapenhance/core/data/SnapClassCache.kt b/core/src/main/kotlin/me/rhunk/snapenhance/core/data/SnapClassCache.kt @@ -17,6 +17,10 @@ class SnapClassCache ( val feedEntry by lazy { findClass("com.snapchat.client.messaging.FeedEntry") } val conversation by lazy { findClass("com.snapchat.client.messaging.Conversation") } val feedManager by lazy { findClass("com.snapchat.client.messaging.FeedManager\$CppProxy") } + val nativeBridge by lazy { findClass("com.snapchat.client.composer.NativeBridge") } + val composerView by lazy { findClass("com.snap.composer.views.ComposerView") } + val composerAction by lazy { findClass("com.snap.composer.actions.ComposerAction") } + val composerFunctionActionAdapter by lazy { findClass("com.snap.composer.callable.ComposerFunctionActionAdapter") } private fun findClass(className: String): Class<*> { return try { diff --git a/core/src/main/kotlin/me/rhunk/snapenhance/core/ui/ViewAppearanceHelper.kt b/core/src/main/kotlin/me/rhunk/snapenhance/core/ui/ViewAppearanceHelper.kt @@ -19,9 +19,11 @@ import android.view.View import android.view.ViewGroup import android.widget.Switch import android.widget.TextView +import me.rhunk.snapenhance.core.SnapEnhance import me.rhunk.snapenhance.core.util.ktx.getDimens import me.rhunk.snapenhance.core.util.ktx.getDimensFloat import me.rhunk.snapenhance.core.util.ktx.getIdentifier +import me.rhunk.snapenhance.core.wrapper.impl.composer.ComposerViewNode import kotlin.random.Random fun View.applyTheme(componentWidth: Int? = null, hasRadius: Boolean = false, isAmoled: Boolean = true) { @@ -91,6 +93,16 @@ fun View.iterateParent(predicate: (View) -> Boolean) { } } +fun View.getComposerViewNode(): ComposerViewNode? { + if (!this::class.java.isAssignableFrom(SnapEnhance.classCache.composerView)) return null + val composerViewNode = this::class.java.methods.firstOrNull { + it.name == "getComposerViewNode" + }?.invoke(this) ?: return null + + return ComposerViewNode(composerViewNode::class.java.methods.firstOrNull { + it.name == "getNativeHandle" + }?.invoke(composerViewNode) as? Long ?: return null) +} object ViewAppearanceHelper { private fun createRoundedBackground(color: Int, radius: Float, hasRadius: Boolean): Drawable { diff --git a/core/src/main/kotlin/me/rhunk/snapenhance/core/wrapper/impl/composer/ComposerViewNode.kt b/core/src/main/kotlin/me/rhunk/snapenhance/core/wrapper/impl/composer/ComposerViewNode.kt @@ -0,0 +1,50 @@ +package me.rhunk.snapenhance.core.wrapper.impl.composer + +import me.rhunk.snapenhance.core.SnapEnhance +import me.rhunk.snapenhance.core.wrapper.AbstractWrapper +import java.lang.reflect.Proxy + +fun createComposerFunction(block: (args: Array<*>) -> Any?): Any { + return SnapEnhance.classCache.composerFunctionActionAdapter.constructors.first().newInstance( + Proxy.newProxyInstance( + SnapEnhance.classCache.composerAction.classLoader, + arrayOf(SnapEnhance.classCache.composerAction), + ) { _, _, args -> + block(args?.get(0) as Array<*>) + } + ) +} + +class ComposerViewNode(obj: Long) : AbstractWrapper(obj) { + fun getAttribute(name: String): Any? { + return SnapEnhance.classCache.nativeBridge.methods.firstOrNull { + it.name == "getValueForAttribute" + }?.invoke(null, instanceNonNull(), name) + } + + fun setAttribute(name: String, value: Any) { + SnapEnhance.classCache.nativeBridge.methods.firstOrNull { + it.name == "setValueForAttribute" + }?.invoke(null, instanceNonNull(), name, value, false) + } + + fun getChildren(): List<ComposerViewNode> { + return ((SnapEnhance.classCache.nativeBridge.methods.firstOrNull { + it.name == "getRetainedViewNodeChildren" + }?.invoke(null, instanceNonNull(), 1))!! as? LongArray)?.map { + ComposerViewNode(it) + } ?: emptyList() + } + + fun getClassName(): String { + return SnapEnhance.classCache.nativeBridge.methods.firstOrNull { + it.name == "getViewClassName" + }?.invoke(null, instanceNonNull()).toString() + } + + override fun toString(): String { + return SnapEnhance.classCache.nativeBridge.methods.firstOrNull { + it.name == "getViewNodeDebugDescription" + }?.invoke(null, instanceNonNull()).toString() + } +}+ \ No newline at end of file