feat(core): snap score changes

Signed-off-by: rhunk <[email protected]>

Author: rhunk

app/src/main/kotlin/me/rhunk/snapenhance/RemoteTracker.kt

@@ -8,6 +8,7 @@ import me.rhunk.snapenhance.common.data.TrackerRuleEvent
 import me.rhunk.snapenhance.common.util.toSerialized
 import me.rhunk.snapenhance.storage.getRuleTrackerScopes
 import me.rhunk.snapenhance.storage.getTrackerEvents
+import me.rhunk.snapenhance.storage.updateFriendScore
 
 
 class RemoteTracker(
@@ -26,4 +27,8 @@ class RemoteTracker(
             ScopedTrackerRule(it.key, context.database.getRuleTrackerScopes(it.key.id))
         }).toSerialized()
     }
+
+    override fun updateFriendScore(userId: String, score: Long): Long {
+        return context.database.updateFriendScore(userId, score)
+    }
 }

app/src/main/kotlin/me/rhunk/snapenhance/storage/AppDatabase.kt

@@ -78,6 +78,10 @@ class AppDatabase(
                 "params TEXT",
                 "actions TEXT"
             ),
+            "friend_scores" to listOf(
+                "userId CHAR(36) PRIMARY KEY",
+                "score BIGINT"
+            ),
             "quick_tiles" to listOf(
                 "key VARCHAR PRIMARY KEY",
                 "position INTEGER",

app/src/main/kotlin/me/rhunk/snapenhance/storage/Tracker.kt

@@ -9,6 +9,7 @@ import me.rhunk.snapenhance.common.data.TrackerRuleActionParams
 import me.rhunk.snapenhance.common.data.TrackerRuleEvent
 import me.rhunk.snapenhance.common.data.TrackerScopeType
 import me.rhunk.snapenhance.common.util.ktx.getInteger
+import me.rhunk.snapenhance.common.util.ktx.getLongOrNull
 import me.rhunk.snapenhance.common.util.ktx.getStringOrNull
 import kotlin.coroutines.suspendCoroutine
 
@@ -195,3 +196,24 @@ fun AppDatabase.getRuleTrackerScopes(ruleId: Int, limit: Int = Int.MAX_VALUE): M
     }
     return scopes
 }
+
+fun AppDatabase.updateFriendScore(userId: String, score: Long): Long {
+    return runBlocking {
+        suspendCoroutine { continuation ->
+            executeAsync {
+                val currentScore = database.rawQuery("SELECT score FROM friend_scores WHERE userId = ?", arrayOf(userId)).use { cursor ->
+                    if (!cursor.moveToFirst()) return@use null
+                    cursor.getLongOrNull("score")
+                }
+
+                if (currentScore != null) {
+                    database.execSQL("UPDATE friend_scores SET score = ? WHERE userId = ?", arrayOf(score, userId))
+                } else {
+                    database.execSQL("INSERT INTO friend_scores (userId, score) VALUES (?, ?)", arrayOf(userId, score))
+                }
+
+                continuation.resumeWith(Result.success(currentScore ?: -1))
+            }
+        }
+    }
+}

common/src/main/aidl/me/rhunk/snapenhance/bridge/logger/TrackerInterface.aidl

@@ -2,4 +2,6 @@ package me.rhunk.snapenhance.bridge.logger;
 
 interface TrackerInterface {
     String getTrackedEvents(String eventType); // returns serialized TrackerEventsResult
+
+    long updateFriendScore(String userId, long score); // returns old score (-1 if not found)
 }

common/src/main/assets/lang/en_US.json

@@ -1152,6 +1152,10 @@
                     "prevent_forced_logout": {
                         "name": "Prevent Forced Logout",
                         "description": "Prevents Snapchat from logging you out when you login on another device"
+                    },
+                    "snapscore_changes": {
+                        "name": "Snapscore Changes",
+                        "description": "Tracks changes in friends Snapscore\nUse this feature in newer versions of Snapchat only"
                     }
                 }
             },

common/src/main/kotlin/me/rhunk/snapenhance/common/config/impl/Experimental.kt

@@ -94,4 +94,5 @@ class Experimental : ConfigContainer() {
         "added_by_quick_add",
     ) { addNotices(FeatureNotice.BAN_RISK) }
     val preventForcedLogout = boolean("prevent_forced_logout") { requireRestart(); addNotices(FeatureNotice.BAN_RISK, FeatureNotice.INTERNAL_BEHAVIOR); }
+    val snapScoreChanges = boolean("snapscore_changes") { requireRestart() }
 }

core/src/main/kotlin/me/rhunk/snapenhance/core/features/FeatureManager.kt

@@ -142,6 +142,7 @@ class FeatureManager(
             VoiceNoteOverride(),
             FriendNotes(),
             DoubleTapChatAction(),
+            SnapScoreChanges(),
         )
 
         features.values.toList().forEach { feature ->

core/src/main/kotlin/me/rhunk/snapenhance/core/features/impl/experiments/SnapScoreChanges.kt

@@ -0,0 +1,68 @@
+package me.rhunk.snapenhance.core.features.impl.experiments
+
+import android.view.ViewGroup
+import me.rhunk.snapenhance.common.util.protobuf.ProtoReader
+import me.rhunk.snapenhance.core.event.events.impl.AddViewEvent
+import me.rhunk.snapenhance.core.event.events.impl.UnaryCallEvent
+import me.rhunk.snapenhance.core.features.Feature
+import me.rhunk.snapenhance.core.ui.getComposerContext
+import me.rhunk.snapenhance.core.ui.getComposerViewNode
+import me.rhunk.snapenhance.core.util.ktx.getObjectField
+import me.rhunk.snapenhance.core.wrapper.impl.SnapUUID
+
+class SnapScoreChanges: Feature("Snap Score Changes") {
+    private val scores = mutableMapOf<String, Long>()
+    private var lastViewedUserId: String? = null
+
+    override fun init() {
+        if (!context.config.experimental.snapScoreChanges.get()) return
+
+        context.event.subscribe(UnaryCallEvent::class) { event ->
+            if (event.uri != "/com.snapchat.atlas.gw.AtlasGw/GetFriendsUserScore") return@subscribe
+
+            event.addResponseCallback {
+                synchronized(scores) {
+                    ProtoReader(buffer).eachBuffer(1) {
+                        val friendUUID = getByteArray(1) ?: return@eachBuffer
+                        val score = getVarInt(2) ?: return@eachBuffer
+
+                        scores[SnapUUID(friendUUID).toString()] = score
+                    }
+                }
+            }
+        }
+
+        context.event.subscribe(AddViewEvent::class) { event ->
+            if (event.viewClassName.endsWith("UnifiedProfileFlatlandProfileViewTopViewFrameLayout")) {
+                val composerView = (event.view as ViewGroup).getChildAt(0) ?: return@subscribe
+                val composerContext = composerView.getComposerContext() ?: return@subscribe
+
+                lastViewedUserId = composerContext.viewModel?.getObjectField("_userId")?.toString()
+            }
+
+            if (event.viewClassName.endsWith("ProfileFlatlandFriendSnapScoreIdentityPillDialogView")) {
+                event.view.post {
+                    val composerViewNode = event.view.getComposerViewNode() ?: return@post
+                    val surface = composerViewNode.getChildren().getOrNull(1) ?: return@post
+
+                    val snapTextView = surface.getChildren().lastOrNull {
+                        it.getClassName() == "com.snap.composer.views.ComposerSnapTextView"
+                    } ?: return@post
+
+
+                    val currentFriendScore = scores[lastViewedUserId] ?: (event.view.getComposerContext()?.viewModel?.getObjectField("_friendSnapScore") as? Double)?.toLong() ?: return@post
+
+                    val oldSnapScore = context.bridgeClient.getTracker().updateFriendScore(
+                        lastViewedUserId ?: return@post,
+                        currentFriendScore
+                    )
+
+                    val diff = currentFriendScore - oldSnapScore
+
+                    snapTextView.setAttribute("value", "${if (oldSnapScore != -1L && diff > 0) "\uD83D\uDCC8 +$diff !\n\n" else ""}Last Checked Score: ${oldSnapScore.takeIf { it != -1L } ?: "N/A"}")
+                    event.view.invalidate()
+                }
+            }
+        }
+    }
+}