majjo/PurrfectSnap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: capture story_lookup/batch_story_lookup endpoints for story media URLs

Browse Source
This commit is contained in:
Majjo
2026-04-20 00:29:57 +02:00
parent 947bb399c8
commit 98e04a81cb
1 changed files with 52 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
core/src/main/kotlin/me/eternal/purrfectsnap/core/features/impl/global/CompanionServer.kt
View File

@@ -185,49 +185,72 @@ class CompanionServer : Feature("CompanionServer") {
// ---- Story capture ----
private fun storeStory(userId: String, url: String, postedAt: Long, createdAt: Long, keyB64: String, ivB64: String): Boolean {
val seen = capturedStoryUrls.getOrPut(userId) { java.util.concurrent.ConcurrentHashMap() }
if (seen.putIfAbsent(url, true) != null) return false
capturedStories.getOrPut(userId) { java.util.concurrent.CopyOnWriteArrayList() }.add(mapOf(
"userId" to userId,
"url" to url,
"postedAt" to postedAt,
"createdAt" to createdAt,
"key" to keyB64,
"iv" to ivB64,
"capturedAt" to System.currentTimeMillis()
))
return true
}
private fun subscribeToStoryEvents() {
context.event.subscribe(NetworkApiRequestEvent::class) { event ->
if (!event.url.contains("df-mixer-prod") ||
(!event.url.endsWith("stories") && !event.url.endsWith("batch_stories"))) return@subscribe
val urlPath = event.url.substringBefore("?")
if (!urlPath.contains("df-mixer-prod")) return@subscribe
val endpoint = urlPath.substringAfterLast("/")
if (endpoint !in setOf("stories", "batch_stories", "story_lookup", "batch_story_lookup")) return@subscribe
event.onSuccess { buffer ->
buffer ?: return@onSuccess
context.log.verbose("CompanionServer: [$endpoint] response ${buffer.size} bytes", "CompanionServer")
runCatching {
ProtoReader(buffer).eachBuffer(3) {
var parsed = 0
// story_lookup / batch_story_lookup: field 1 → repeated story entries
// Each entry: field 4 = userId, field 3 → media block → field 36 → field 1 → key/iv/url
ProtoReader(buffer).eachBuffer(1) {
val userId = getString(4, 1) ?: return@eachBuffer
eachBuffer(3) {
followPath(36) {
eachBuffer(1) {
val url = getString(2, 2)?.substringBefore("?") ?: return@eachBuffer
val postedAt = getVarInt(3) ?: return@eachBuffer
val createdAt = getVarInt(27) ?: -1L
val keyB64 = getString(2, 5) ?: return@eachBuffer
val ivB64 = getString(2, 4) ?: return@eachBuffer
if (storeStory(userId, url, postedAt, createdAt, keyB64, ivB64)) parsed++
}
}
}
}
// batch_stories fallback: field 3 → field 3 → field 3 → field 36 → field 1
if (parsed == 0) {
ProtoReader(buffer).eachBuffer(3) {
eachBuffer(3) {
followPath(36) {
eachBuffer(1) {
val userId = getString(8, 1) ?: return@eachBuffer
val url = getString(2, 2)?.substringBefore("?") ?: return@eachBuffer
val postedAt = getVarInt(3) ?: return@eachBuffer
val createdAt = getVarInt(27) ?: -1L
val keyB64 = getString(2, 5) ?: return@eachBuffer
val ivB64 = getString(2, 4) ?: return@eachBuffer
// Deduplicate by URL
val seen = capturedStoryUrls.getOrPut(userId) {
java.util.concurrent.ConcurrentHashMap()
}
if (seen.putIfAbsent(url, true) == null) {
capturedStories.getOrPut(userId) {
java.util.concurrent.CopyOnWriteArrayList()
}.add(mapOf(
"userId" to userId,
"url" to url,
"postedAt" to postedAt,
"createdAt" to createdAt,
"key" to keyB64,
"iv" to ivB64,
"capturedAt" to System.currentTimeMillis()
))
eachBuffer(3) {
followPath(36) {
eachBuffer(1) {
val userId = getString(8, 1) ?: return@eachBuffer
val url = getString(2, 2)?.substringBefore("?") ?: return@eachBuffer
val postedAt = getVarInt(3) ?: return@eachBuffer
val createdAt = getVarInt(27) ?: -1L
val keyB64 = getString(2, 5) ?: return@eachBuffer
val ivB64 = getString(2, 4) ?: return@eachBuffer
if (storeStory(userId, url, postedAt, createdAt, keyB64, ivB64)) parsed++
}
}
}
}
}
}
context.log.verbose("CompanionServer: [$endpoint] parsed $parsed new stories", "CompanionServer")
}.onFailure {
context.log.warn("CompanionServer: story proto parse failed: ${it.message}", "CompanionServer")
context.log.warn("CompanionServer: [$endpoint] proto parse failed: ${it.message}", "CompanionServer")
}
}
}