main.com.sceyt.chatuikit.data.repositories.AttachmentsRepositoryImpl.kt Maven / Gradle / Ivy

Go to download
package com.sceyt.chatuikit.data.repositories

import com.sceyt.chat.models.SceytException
import com.sceyt.chat.models.attachment.Attachment
import com.sceyt.chat.models.attachment.AttachmentListQuery
import com.sceyt.chat.models.link.LinkDetails
import com.sceyt.chat.models.link.LoadLinkDetailsRequest
import com.sceyt.chat.models.user.User
import com.sceyt.chat.sceyt_callbacks.AttachmentsCallback
import com.sceyt.chat.sceyt_callbacks.LinkDetailsCallback
import com.sceyt.chatuikit.SceytChatUIKit
import com.sceyt.chatuikit.data.models.SceytResponse
import com.sceyt.chatuikit.data.models.messages.SceytUser
import com.sceyt.chatuikit.extensions.TAG
import com.sceyt.chatuikit.logger.SceytLog
import com.sceyt.chatuikit.persistence.extensions.safeResume
import com.sceyt.chatuikit.persistence.mappers.toSceytUser
import com.sceyt.chatuikit.persistence.repositories.AttachmentsRepository
import kotlinx.coroutines.suspendCancellableCoroutine

class AttachmentsRepositoryImpl : AttachmentsRepository {

    private fun getQuery(conversationId: Long, types: List) = AttachmentListQuery.Builder(conversationId)
        .setLimit(SceytChatUIKit.config.queryLimits.attachmentListQueryLimit)
        .withTypes(types)
        .build()


    override suspend fun getPrevAttachments(
            conversationId: Long,
            lastAttachmentId: Long,
            types: List
    ): SceytResponse, Map>> {
        return suspendCancellableCoroutine { continuation ->
            getQuery(conversationId, types).loadPrev(lastAttachmentId, object : AttachmentsCallback {
                override fun onResult(attachments: MutableList?, userMutableMap: MutableMap?) {
                    continuation.safeResume(SceytResponse.Success(Pair(attachments ?: listOf(),
                        userMutableMap?.mapValues { it.value.toSceytUser() } ?: mapOf())))
                }

                override fun onError(e: SceytException?) {
                    continuation.safeResume(SceytResponse.Error(e))
                    SceytLog.e(TAG, "getPrevAttachments error: ${e?.message}")
                }
            })
        }
    }

    override suspend fun getNextAttachments(
            conversationId: Long,
            lastAttachmentId: Long,
            types: List
    ): SceytResponse, Map>> {
        return suspendCancellableCoroutine { continuation ->
            getQuery(conversationId, types).loadNext(lastAttachmentId, object : AttachmentsCallback {
                override fun onResult(attachments: MutableList?, userMutableMap: MutableMap?) {
                    continuation.safeResume(SceytResponse.Success(Pair(attachments ?: listOf(),
                        userMutableMap?.mapValues { it.value.toSceytUser() } ?: mapOf())))
                }

                override fun onError(e: SceytException?) {
                    continuation.safeResume(SceytResponse.Error(e))
                    SceytLog.e(TAG, "getNextAttachments error: ${e?.message}")
                }
            })
        }
    }

    override suspend fun getNearAttachments(
            conversationId: Long,
            attachmentId: Long,
            types: List
    ): SceytResponse, Map>> {
        return suspendCancellableCoroutine { continuation ->
            getQuery(conversationId, types).loadNear(attachmentId, object : AttachmentsCallback {
                override fun onResult(attachments: MutableList?, userMutableMap: MutableMap?) {
                    continuation.safeResume(SceytResponse.Success(Pair(attachments ?: listOf(),
                        userMutableMap?.mapValues { it.value.toSceytUser() } ?: mapOf())))
                }

                override fun onError(e: SceytException?) {
                    continuation.safeResume(SceytResponse.Error(e))
                    SceytLog.e(TAG, "getNearAttachments error: ${e?.message}")
                }
            })
        }
    }

    override suspend fun getLinkPreviewData(link: String): SceytResponse {
        return suspendCancellableCoroutine { continuation ->
            LoadLinkDetailsRequest(link).execute(object : LinkDetailsCallback {
                override fun onResult(linkDetails: LinkDetails?) {
                    continuation.safeResume(SceytResponse.Success(linkDetails))
                }

                override fun onError(e: SceytException?) {
                    continuation.safeResume(SceytResponse.Error(e))
                    SceytLog.e(TAG, "getLinkPreviewData error: ${e?.message}, for link: $link")
                }
            })
        }
    }
}