Add Poll Entity to Message Entities

GetStream · Jun 13, 2024 · d02041e · d02041e
1 parent c00e3cd
commit d02041e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/...ava/io/getstream/chat/android/offline/repository/domain/message/internal/MessageEntity.kt b/...ava/io/getstream/chat/android/offline/repository/domain/message/internal/MessageEntity.kt
@@ -127,6 +127,7 @@ internal data class MessageInnerEntity(
     val moderationDetails: ModerationDetailsEntity? = null,
     /** When the message text was updated */
     val messageTextUpdatedAt: Date? = null,
+    val poll: PollEntity? = null,
 )
 
 internal const val MESSAGE_ENTITY_TABLE_NAME = "stream_chat_message"
diff --git a/...ava/io/getstream/chat/android/offline/repository/domain/message/internal/MessageMapper.kt b/...ava/io/getstream/chat/android/offline/repository/domain/message/internal/MessageMapper.kt
@@ -77,6 +77,7 @@ internal suspend fun MessageEntity.toModel(
         skipPushNotification = skipPushNotification,
         moderationDetails = moderationDetails?.toModel(),
         messageTextUpdatedAt = messageTextUpdatedAt,
+        poll = poll?.toModel(getUser),
     )
 }
 
@@ -118,6 +119,7 @@ internal fun Message.toEntity(): MessageEntity = MessageEntity(
         skipEnrichUrl = skipEnrichUrl,
         moderationDetails = moderationDetails?.toEntity(),
         messageTextUpdatedAt = messageTextUpdatedAt,
+        poll = poll?.toEntity(),
     ),
     attachments = attachments.mapIndexed { index, attachment -> attachment.toEntity(id, index) },
     latestReactions = latestReactions.map(Reaction::toEntity),
@@ -164,6 +166,7 @@ internal suspend fun ReplyMessageEntity.toModel(
             pinnedBy = pinnedByUserId?.let { getUser(it) },
             moderationDetails = moderationDetails?.toModel(),
             messageTextUpdatedAt = messageTextUpdatedAt,
+            poll = poll?.toModel(getUser),
         )
     }
 }
@@ -199,6 +202,7 @@ internal fun Message.toReplyEntity(): ReplyMessageEntity =
             pinExpires = pinExpires,
             pinnedByUserId = pinnedBy?.id,
             moderationDetails = moderationDetails?.toEntity(),
+            poll = poll?.toEntity(),
         ),
         attachments = attachments.mapIndexed { index, attachment -> attachment.toReplyEntity(id, index) },
     )

diff --git a/...o/getstream/chat/android/offline/repository/domain/message/internal/ReplyMessageEntity.kt b/...o/getstream/chat/android/offline/repository/domain/message/internal/ReplyMessageEntity.kt
@@ -100,6 +100,7 @@ internal data class ReplyMessageInnerEntity(
     val moderationDetails: ModerationDetailsEntity? = null,
     /** Date when the message text was updated **/
     val messageTextUpdatedAt: Date? = null,
+    val poll: PollEntity? = null,
 )
 
 internal const val REPLY_MESSAGE_ENTITY_TABLE_NAME = "stream_chat_reply_message"