session-android/src/org/thoughtcrime/securesms/loki/api/BackgroundPollWorker.kt

package org.thoughtcrime.securesms.loki.api

import android.content.Context
import android.content.Intent
import nl.komponents.kovenant.functional.map
import org.thoughtcrime.securesms.ApplicationContext
import org.thoughtcrime.securesms.database.DatabaseFactory
import org.thoughtcrime.securesms.jobs.PushContentReceiveJob
import org.thoughtcrime.securesms.service.PersistentAlarmManagerListener
import org.thoughtcrime.securesms.util.TextSecurePreferences
import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope
import org.whispersystems.signalservice.loki.api.SnodeAPI
import java.util.concurrent.TimeUnit

class BackgroundPollWorker : PersistentAlarmManagerListener() {

    companion object {
        private val pollInterval = TimeUnit.MINUTES.toMillis(15)

        @JvmStatic
        fun schedule(context: Context) {
            BackgroundPollWorker().onReceive(context, Intent())
        }
    }

    override fun getNextScheduledExecutionTime(context: Context): Long {
        return TextSecurePreferences.getBackgroundPollTime(context)
    }

    override fun onAlarm(context: Context, scheduledTime: Long): Long {
        if (scheduledTime != 0L) {
            if (!TextSecurePreferences.isUsingFCM(context)) {
                val userPublicKey = TextSecurePreferences.getLocalNumber(context)
                val lokiAPIDatabase = DatabaseFactory.getLokiAPIDatabase(context)
                try {
                    val applicationContext = context.applicationContext as ApplicationContext
                    val broadcaster = applicationContext.broadcaster
                    SnodeAPI.configureIfNeeded(userPublicKey, lokiAPIDatabase, broadcaster)
                    SnodeAPI.shared.getMessages().map { messages ->
                        messages.forEach {
                            PushContentReceiveJob(context).processEnvelope(SignalServiceEnvelope(it), false)
                        }
                    }
                } catch (exception: Throwable) {
                    // Do nothing
                }
            }
            val openGroups = DatabaseFactory.getLokiThreadDatabase(context).getAllPublicChats().map { it.value }
            for (openGroup in openGroups) {
                val poller = LokiPublicChatPoller(context, openGroup)
                poller.stop()
                poller.pollForNewMessages()
            }
        }
        val nextTime = System.currentTimeMillis() + pollInterval
        TextSecurePreferences.setBackgroundPollTime(context, nextTime)
        return nextTime
    }
}