Add threadInfo field to message like timeline events (#5930)

* Add `threadInfo` field to message like timeline events:
- Polls
- Stickers
- UTDs

* Add missing cases for `EventTimeline.threadInfo()`

libraries/matrix/api/src/main/kotlin/io/element/android/libraries/matrix/api/timeline/item/event/EventContent.kt

@@ -26,7 +26,7 @@ data class MessageContent(
     val inReplyTo: InReplyTo?,
     val isEdited: Boolean,
     val threadInfo: EventThreadInfo?,
-    val type: MessageType
+    val type: MessageType,
 ) : EventContent
 
 data object RedactedContent : EventContent
@@ -36,6 +36,7 @@ data class StickerContent(
     val body: String?,
     val info: ImageInfo,
     val source: MediaSource,
+    val threadInfo: EventThreadInfo?,
 ) : EventContent {
     val bestDescription: String
         get() = body ?: filename
@@ -49,10 +50,12 @@ data class PollContent(
     val votes: ImmutableMap<String, ImmutableList<UserId>>,
     val endTime: ULong?,
     val isEdited: Boolean,
+    val threadInfo: EventThreadInfo?,
 ) : EventContent
 
 data class UnableToDecryptContent(
-    val data: Data
+    val data: Data,
+    val threadInfo: EventThreadInfo?,
 ) : EventContent {
     @Immutable
     sealed interface Data {

libraries/matrix/api/src/main/kotlin/io/element/android/libraries/matrix/api/timeline/item/event/EventTimelineItem.kt

@@ -39,7 +39,13 @@ data class EventTimelineItem(
         return (content as? MessageContent)?.inReplyTo
     }
 
-    fun threadInfo(): EventThreadInfo? = (content as? MessageContent)?.threadInfo
+    fun threadInfo(): EventThreadInfo? = when (content) {
+        is MessageContent -> content.threadInfo
+        is PollContent -> content.threadInfo
+        is StickerContent -> content.threadInfo
+        is UnableToDecryptContent -> content.threadInfo
+        else -> null
+    }
 
     fun hasNotLoadedInReplyTo(): Boolean {
         val details = inReplyTo()

libraries/matrix/impl/src/main/kotlin/io/element/android/libraries/matrix/impl/timeline/item/event/TimelineEventContentMapper.kt

@@ -36,6 +36,7 @@ import io.element.android.libraries.matrix.impl.room.join.map
 import kotlinx.collections.immutable.toImmutableList
 import kotlinx.collections.immutable.toImmutableMap
 import org.matrix.rustcomponents.sdk.EmbeddedEventDetails
+import org.matrix.rustcomponents.sdk.MsgLikeContent
 import org.matrix.rustcomponents.sdk.MsgLikeKind
 import org.matrix.rustcomponents.sdk.TimelineItemContent
 import org.matrix.rustcomponents.sdk.use
@@ -68,37 +69,11 @@ class TimelineEventContentMapper(
                     when (val kind = it.content.kind) {
                         is MsgLikeKind.Message -> {
                             val inReplyTo = it.content.inReplyTo
-                            val threadSummary = it.content.threadSummary?.use { summary ->
-                                val numberOfReplies = summary.numReplies().toLong()
-                                val latestEvent = summary.latestEvent()
-                                val details = when (latestEvent) {
-                                    is EmbeddedEventDetails.Unavailable -> AsyncData.Uninitialized
-                                    is EmbeddedEventDetails.Pending -> AsyncData.Loading()
-                                    is EmbeddedEventDetails.Error -> AsyncData.Failure(IllegalStateException(latestEvent.message))
-                                    is EmbeddedEventDetails.Ready -> {
-                                        AsyncData.Success(
-                                            EmbeddedEventInfo(
-                                                eventOrTransactionId = latestEvent.eventOrTransactionId.map(),
-                                                content = map(latestEvent.content),
-                                                senderId = UserId(latestEvent.sender),
-                                                senderProfile = latestEvent.senderProfile.map(),
-                                                timestamp = latestEvent.timestamp.toLong(),
-                                            )
-                                        )
-                                    }
-                                }
-                                ThreadSummary(
-                                    latestEvent = details,
-                                    numberOfReplies = numberOfReplies,
-                                )
-                            }
-                            val threadRootId = it.content.threadRoot?.let(::ThreadId)
-                            val threadInfo = when {
-                                threadSummary != null -> EventThreadInfo.ThreadRoot(threadSummary)
-                                threadRootId != null -> EventThreadInfo.ThreadResponse(threadRootId)
-                                else -> null
-                            }
-                            eventMessageMapper.map(kind, inReplyTo, threadInfo)
+                            eventMessageMapper.map(
+                                message = kind,
+                                inReplyTo = inReplyTo,
+                                threadInfo = extractThreadInfo(it.content)
+                            )
                         }
                         is MsgLikeKind.Redacted -> {
                             RedactedContent
@@ -114,11 +89,13 @@ class TimelineEventContentMapper(
                                 }.toImmutableMap(),
                                 endTime = kind.endTime,
                                 isEdited = kind.hasBeenEdited,
+                                threadInfo = extractThreadInfo(it.content),
                             )
                         }
                         is MsgLikeKind.UnableToDecrypt -> {
                             UnableToDecryptContent(
-                                data = kind.msg.map()
+                                data = kind.msg.map(),
+                                threadInfo = extractThreadInfo(it.content),
                             )
                         }
                         is MsgLikeKind.Sticker -> {
@@ -127,6 +104,7 @@ class TimelineEventContentMapper(
                                 body = null,
                                 info = kind.info.map(),
                                 source = kind.source.map(),
+                                threadInfo = extractThreadInfo(it.content),
                             )
                         }
                         is MsgLikeKind.Other -> UnknownContent
@@ -159,6 +137,43 @@ class TimelineEventContentMapper(
             }
         }
     }
+
+    private fun extractThreadInfo(content: MsgLikeContent): EventThreadInfo? {
+        val threadSummary = extractThreadSummary(content.threadSummary)
+        val threadRootId = content.threadRoot?.let(::ThreadId)
+        return when {
+            threadSummary != null -> EventThreadInfo.ThreadRoot(threadSummary)
+            threadRootId != null -> EventThreadInfo.ThreadResponse(threadRootId)
+            else -> null
+        }
+    }
+
+    private fun extractThreadSummary(threadSummary: org.matrix.rustcomponents.sdk.ThreadSummary?): ThreadSummary? {
+        return threadSummary?.use { summary ->
+            val numberOfReplies = summary.numReplies().toLong()
+            val latestEvent = summary.latestEvent()
+            val details = when (latestEvent) {
+                is EmbeddedEventDetails.Unavailable -> AsyncData.Uninitialized
+                is EmbeddedEventDetails.Pending -> AsyncData.Loading()
+                is EmbeddedEventDetails.Error -> AsyncData.Failure(IllegalStateException(latestEvent.message))
+                is EmbeddedEventDetails.Ready -> {
+                    AsyncData.Success(
+                        EmbeddedEventInfo(
+                            eventOrTransactionId = latestEvent.eventOrTransactionId.map(),
+                            content = map(latestEvent.content),
+                            senderId = UserId(latestEvent.sender),
+                            senderProfile = latestEvent.senderProfile.map(),
+                            timestamp = latestEvent.timestamp.toLong(),
+                        )
+                    )
+                }
+            }
+            ThreadSummary(
+                latestEvent = details,
+                numberOfReplies = numberOfReplies,
+            )
+        }
+    }
 }
 
 private fun RustMembershipChange.map(): MembershipChange {

libraries/matrix/test/src/main/kotlin/io/element/android/libraries/matrix/test/timeline/TimelineFixture.kt

@@ -123,11 +123,13 @@ fun aStickerContent(
     info: ImageInfo,
     mediaSource: MediaSource,
     body: String? = null,
+    threadInfo: EventThreadInfo? = null,
 ) = StickerContent(
     filename = filename,
     body = body,
     info = info,
     source = mediaSource,
+    threadInfo = threadInfo,
 )
 
 fun aTimelineItemDebugInfo(
@@ -148,6 +150,7 @@ fun aPollContent(
     votes: ImmutableMap<String, ImmutableList<UserId>> = persistentMapOf(),
     endTime: ULong? = null,
     isEdited: Boolean = false,
+    threadInfo: EventThreadInfo? = null,
 ) = PollContent(
     question = question,
     kind = kind,
@@ -156,4 +159,5 @@ fun aPollContent(
     votes = votes,
     endTime = endTime,
     isEdited = isEdited,
+    threadInfo = threadInfo,
 )