DatabaseMessageMapper.kt

package chat.rocket.android.server.infrastructure

import chat.rocket.android.db.DatabaseManager
import chat.rocket.android.db.model.*
import chat.rocket.android.util.retryDB
import chat.rocket.common.model.SimpleRoom
import chat.rocket.common.model.SimpleUser
import chat.rocket.core.model.Message
import chat.rocket.core.model.Reactions
import chat.rocket.core.model.attachment.Attachment
import chat.rocket.core.model.attachment.Color
import chat.rocket.core.model.attachment.Field
import chat.rocket.core.model.attachment.actions.Action
import chat.rocket.core.model.attachment.actions.ButtonAction
import chat.rocket.core.model.messageTypeOf
import chat.rocket.core.model.url.Meta
import chat.rocket.core.model.url.ParsedUrl
import chat.rocket.core.model.url.Url
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext
import timber.log.Timber
import java.lang.Exception

class DatabaseMessageMapper(private val dbManager: DatabaseManager) {
    suspend fun map(message: FullMessage): Message? = map(listOf(message)).firstOrNull()

    suspend fun map(messages: List<FullMessage>): List<Message> {
        val list = mutableListOf<Message>()
        messages.forEach { message ->
            val favorites = mutableListOf<SimpleUser>()
            message.favorites.forEach { user ->
                favorites.add(mapUser(user))
            }

            val mentions = mutableListOf<SimpleUser>()
            message.mentions.forEach { user ->
                mentions.add(mapUser(user))
            }

            val channels = mutableListOf<SimpleRoom>()
            message.message.channels?.forEach { channel ->
                channels.add(SimpleRoom(channel.roomId, channel.roomName))
            }

            with(message.message) {
                val sender = this.message.senderId?.let { id ->
                    SimpleUser(id, this.senderUsername, this.senderName)
                }
                val editedBy = this.message.editedBy?.let { id ->
                    SimpleUser(id, this.editUsername, this.editName)
                }
                val urls = this.urls?.let { mapUrl(it) }
                val reactions = this.reactions?.let { mapReactions(it) }
                val attachments = this.attachments?.let { mapAttachments(it).asReversed() }
                val messageType = messageTypeOf(this.message.type)

                list.add(
                    Message(
                        id = this.message.id,
                        roomId = this.message.roomId,
                        message = this.message.message,
                        timestamp = this.message.timestamp,
                        sender = sender,
                        updatedAt = this.message.updatedAt,
                        editedAt = this.message.editedAt,
                        editedBy = editedBy,
                        senderAlias = this.message.senderAlias,
                        avatar = this.message.avatar,
                        type = messageType,
                        groupable = this.message.groupable,
                        parseUrls = this.message.parseUrls,
                        urls = urls,
                        mentions = mentions,
                        channels = channels,
                        attachments = attachments,
                        pinned = this.message.pinned,
                        starred = favorites,
                        reactions = reactions,
                        role = this.message.role,
                        synced = this.message.synced,
                        unread = this.message.unread
                    )
                )
            }
        }

        return list
    }

    private fun mapReactions(reactions: List<ReactionEntity>): Reactions {
        val map = Reactions()
        reactions.forEach { reaction ->
            val usernames = reaction.usernames.split(",").map { it.trim() }
            val names = reaction.names.split(",").map { it.trim() }
            map[reaction.reaction] = Pair(usernames, names)
        }

        return map
    }

    private fun mapUrl(urls: List<UrlEntity>): List<Url> {
        val list = mutableListOf<Url>()

        urls.forEach { url ->
            val parsedUrl = url.hostname?.let {
                ParsedUrl(host = it)
            }
            val meta =
                if (!url.description.isNullOrEmpty() || !url.imageUrl.isNullOrEmpty() || !url.title.isNullOrEmpty()) {
                    val raw = HashMap<String, String>()
                    if (url.description != null) raw["ogDescription"] = url.description
                    if (url.title != null) raw["ogTitle"] = url.title
                    if (url.imageUrl != null) raw["ogImage"] = url.imageUrl
                    Meta(
                        title = url.title,
                        description = url.description,
                        imageUrl = url.imageUrl,
                        raw = raw
                    )
                } else null

            list.add(Url(url = url.url, meta = meta, parsedUrl = parsedUrl))
        }

        return list
    }

    private fun mapUser(user: UserEntity): SimpleUser {
        return with(user) {
            SimpleUser(
                id = id,
                username = username,
                name = name
            )
        }
    }

    private suspend fun mapAttachments(attachments: List<AttachmentEntity>): List<Attachment> {
        val list = mutableListOf<Attachment>()
        try {
            attachments.forEach { attachment ->
                with(attachment) {
                    val fields = if (hasFields) {
                        withContext(Dispatchers.IO) {
                            retryDB("getAttachmentFields(${attachment._id})") {
                                dbManager.messageDao().getAttachmentFields(attachment._id)
                            }
                        }.map { Field(it.title, it.value) }
                    } else {
                        null
                    }
                    val actions = if (hasActions) {
                        withContext(Dispatchers.IO) {
                            retryDB("getAttachmentActions(${attachment._id})") {
                                dbManager.messageDao().getAttachmentActions(attachment._id)
                            }
                        }.mapNotNull { mapAction(it) }
                    } else {
                        null
                    }
                    list.add(
                        Attachment(
                            title = title,
                            type = type,
                            description = description,
                            authorName = authorName,
                            text = text,
                            thumbUrl = thumbUrl,
                            color = color?.let { Color.Custom(color) },
                            titleLink = titleLink,
                            titleLinkDownload = titleLinkDownload,
                            imageUrl = imageUrl,
                            imageType = imageType,
                            imageSize = imageSize,
                            videoUrl = videoUrl,
                            videoType = videoType,
                            videoSize = videoSize,
                            audioUrl = audioUrl,
                            audioType = audioType,
                            audioSize = audioSize,
                            messageLink = messageLink,
                            attachments = null, // HOW TO MAP THIS
                            timestamp = timestamp,
                            authorIcon = authorIcon,
                            authorLink = authorLink,
                            fields = fields,
                            fallback = fallback,
                            buttonAlignment = if (actions != null && actions.isNotEmpty()) buttonAlignment
                                ?: "vertical" else null,
                            actions = actions
                        )
                    )
                }
            }
        } catch (ex: Exception) {
            Timber.e(ex)
        }
        return list
    }

    private fun mapAction(action: AttachmentActionEntity): Action? {
        return when (action.type) {
            "button" -> ButtonAction(
                action.type, action.text, action.url, action.isWebView,
                action.webViewHeightRatio, action.imageUrl, action.message,
                action.isMessageInChatWindow
            )
            else -> null
        }
    }
}