session-android/app/src/main/java/org/thoughtcrime/securesms/conversation/v2/utilities/MentionManagerUtilities.kt

package org.thoughtcrime.securesms.conversation.v2.utilities

import android.content.Context
import org.session.libsession.messaging.mentions.MentionsManager
import org.session.libsession.utilities.TextSecurePreferences
import org.thoughtcrime.securesms.database.model.MessageRecord
import org.thoughtcrime.securesms.dependencies.DatabaseComponent

object MentionManagerUtilities {

    fun populateUserPublicKeyCacheIfNeeded(threadID: Long, context: Context) {
        val result = mutableSetOf<String>()
        val recipient = DatabaseComponent.get(context).threadDatabase().getRecipientForThreadId(threadID) ?: return
        val storage = DatabaseComponent.get(context).storage()
        when {
            recipient.address.isLegacyClosedGroup -> {
                val members = DatabaseComponent.get(context).groupDatabase().getGroupMembers(recipient.address.toGroupString(), false).map { it.address.serialize() }
                result.addAll(members)
            }
            recipient.address.isClosedGroup -> {
                val members = storage.getMembers(recipient.address.serialize())
                 result.addAll(members.map { it.sessionId })
            }
            recipient.address.isOpenGroup -> {
                val messageDatabase = DatabaseComponent.get(context).mmsSmsDatabase()
                val reader = messageDatabase.readerFor(messageDatabase.getConversation(threadID, true, 0, 200))
                var record: MessageRecord? = reader.next
                while (record != null) {
                    result.add(record.individualRecipient.address.serialize())
                    try {
                        record = reader.next
                    } catch (exception: Exception) {
                        record = null
                    }
                }
                reader.close()
                result.add(TextSecurePreferences.getLocalNumber(context)!!)
            }
        }
        MentionsManager.userPublicKeyCache[threadID] = result
    }
}