118 lines
4.6 KiB
Kotlin
118 lines
4.6 KiB
Kotlin
package org.thoughtcrime.securesms.loki
|
|
|
|
import android.content.Context
|
|
import android.database.ContentObserver
|
|
import android.text.TextUtils
|
|
import nl.komponents.kovenant.Promise
|
|
import nl.komponents.kovenant.functional.bind
|
|
import nl.komponents.kovenant.functional.map
|
|
import org.thoughtcrime.securesms.ApplicationContext
|
|
import org.thoughtcrime.securesms.database.DatabaseContentProviders
|
|
import org.thoughtcrime.securesms.database.DatabaseFactory
|
|
import org.thoughtcrime.securesms.groups.GroupManager
|
|
import org.thoughtcrime.securesms.loki.redesign.messaging.LokiPublicChatPoller
|
|
import org.thoughtcrime.securesms.util.TextSecurePreferences
|
|
import org.thoughtcrime.securesms.util.Util
|
|
import org.whispersystems.signalservice.loki.api.publicchats.LokiPublicChat
|
|
|
|
class LokiPublicChatManager(private val context: Context) {
|
|
private var chats = mutableMapOf<Long, LokiPublicChat>()
|
|
private val pollers = mutableMapOf<Long, LokiPublicChatPoller>()
|
|
private val observers = mutableMapOf<Long, ContentObserver>()
|
|
private var isPolling = false
|
|
|
|
public fun startPollersIfNeeded() {
|
|
refreshChatsAndPollers()
|
|
|
|
for ((threadId, chat) in chats) {
|
|
val poller = pollers[threadId] ?: LokiPublicChatPoller(context, chat)
|
|
poller.startIfNeeded()
|
|
listenToThreadDeletion(threadId)
|
|
if (!pollers.containsKey(threadId)) { pollers[threadId] = poller }
|
|
}
|
|
isPolling = true
|
|
}
|
|
|
|
public fun stopPollers() {
|
|
pollers.values.forEach { it.stop() }
|
|
isPolling = false
|
|
}
|
|
|
|
public fun addChat(server: String, channel: Long): Promise<LokiPublicChat, Exception> {
|
|
val groupChatAPI = ApplicationContext.getInstance(context).lokiPublicChatAPI ?: return Promise.ofFail(IllegalStateException("LokiPublicChatAPI is not set!"))
|
|
return groupChatAPI.getAuthToken(server).bind {
|
|
groupChatAPI.getChannelInfo(channel, server)
|
|
}.map {
|
|
addChat(server, channel, it)
|
|
}
|
|
}
|
|
|
|
public fun addChat(server: String, channel: Long, name: String): LokiPublicChat {
|
|
val chat = LokiPublicChat(channel, server, name, true)
|
|
var threadID = GroupManager.getPublicChatThreadId(chat.id, context)
|
|
// Create the group if we don't have one
|
|
if (threadID < 0) {
|
|
val result = GroupManager.createPublicChatGroup(chat.id, context, null, chat.displayName)
|
|
threadID = result.threadId
|
|
}
|
|
DatabaseFactory.getLokiThreadDatabase(context).setPublicChat(chat, threadID)
|
|
// Set our name on the server
|
|
val displayName = TextSecurePreferences.getProfileName(context)
|
|
if (!TextUtils.isEmpty(displayName)) {
|
|
ApplicationContext.getInstance(context).lokiPublicChatAPI?.setDisplayName(displayName, server)
|
|
}
|
|
// Start polling
|
|
Util.runOnMain{ startPollersIfNeeded() }
|
|
|
|
return chat
|
|
}
|
|
|
|
private fun refreshChatsAndPollers() {
|
|
val chatsInDB = DatabaseFactory.getLokiThreadDatabase(context).getAllPublicChats()
|
|
val removedChatThreadIds = chats.keys.filter { !chatsInDB.keys.contains(it) }
|
|
removedChatThreadIds.forEach { pollers.remove(it)?.stop() }
|
|
|
|
// Only append to chats if we have a thread for the chat
|
|
chats = chatsInDB.filter { GroupManager.getPublicChatThreadId(it.value.id, context) > -1 }.toMutableMap()
|
|
}
|
|
|
|
private fun listenToThreadDeletion(threadID: Long) {
|
|
if (threadID < 0 || observers[threadID] != null) { return }
|
|
val observer = createDeletionObserver(threadID, Runnable {
|
|
val chat = chats[threadID]
|
|
|
|
// Reset last message cache
|
|
if (chat != null) {
|
|
val apiDatabase = DatabaseFactory.getLokiAPIDatabase(context)
|
|
apiDatabase.removeLastDeletionServerID(chat.channel, chat.server)
|
|
apiDatabase.removeLastMessageServerID(chat.channel, chat.server)
|
|
}
|
|
|
|
DatabaseFactory.getLokiThreadDatabase(context).removePublicChat(threadID)
|
|
pollers.remove(threadID)?.stop()
|
|
observers.remove(threadID)
|
|
startPollersIfNeeded()
|
|
})
|
|
observers[threadID] = observer
|
|
|
|
context.applicationContext.contentResolver.registerContentObserver(DatabaseContentProviders.Conversation.getUriForThread(threadID), true, observer)
|
|
}
|
|
|
|
private fun createDeletionObserver(threadID: Long, onDelete: Runnable): ContentObserver {
|
|
return object : ContentObserver(null) {
|
|
|
|
override fun onChange(selfChange: Boolean) {
|
|
super.onChange(selfChange)
|
|
// Stop the poller if thread is deleted
|
|
try {
|
|
if (!DatabaseFactory.getThreadDatabase(context).hasThread(threadID)) {
|
|
onDelete.run()
|
|
context.applicationContext.contentResolver.unregisterContentObserver(this)
|
|
}
|
|
} catch (e: Exception) {
|
|
// TODO: Handle
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |