AppWidgetRefreshWorker.kt

package to.bitkit.appwidget

import android.appwidget.AppWidgetManager
import android.content.ComponentName
import android.content.Context
import androidx.glance.appwidget.GlanceAppWidgetReceiver
import androidx.glance.appwidget.updateAll
import androidx.hilt.work.HiltWorker
import androidx.work.Constraints
import androidx.work.CoroutineWorker
import androidx.work.ExistingPeriodicWorkPolicy
import androidx.work.NetworkType
import androidx.work.PeriodicWorkRequestBuilder
import androidx.work.WorkManager
import androidx.work.WorkerParameters
import dagger.assisted.Assisted
import dagger.assisted.AssistedInject
import to.bitkit.appwidget.model.AppWidgetType
import to.bitkit.appwidget.ui.headlines.HeadlinesGlanceReceiver
import to.bitkit.appwidget.ui.headlines.HeadlinesGlanceWidget
import to.bitkit.appwidget.ui.price.PriceGlanceReceiver
import to.bitkit.appwidget.ui.price.PriceGlanceWidget
import to.bitkit.utils.Logger
import kotlin.time.Duration.Companion.minutes
import kotlin.time.toJavaDuration

@HiltWorker
class AppWidgetRefreshWorker @AssistedInject constructor(
    @Assisted private val appContext: Context,
    @Assisted workerParams: WorkerParameters,
    private val dataRepository: AppWidgetDataRepository,
    private val preferencesStore: AppWidgetPreferencesStore,
) : CoroutineWorker(appContext, workerParams) {

    companion object {
        private const val TAG = "AppWidgetRefreshWorker"
        private const val WORK_NAME = "appwidget_refresh"

        fun enqueue(context: Context) {
            val constraints = Constraints.Builder()
                .setRequiredNetworkType(NetworkType.CONNECTED)
                .build()

            val request = PeriodicWorkRequestBuilder<AppWidgetRefreshWorker>(15.minutes.toJavaDuration())
                .setConstraints(constraints)
                .build()

            WorkManager.getInstance(context).enqueueUniquePeriodicWork(
                WORK_NAME,
                ExistingPeriodicWorkPolicy.KEEP,
                request,
            )
        }

        fun cancelIfNoWidgets(context: Context) {
            val manager = AppWidgetManager.getInstance(context)
            val hasAny = AppWidgetType.entries.any { type ->
                manager.getAppWidgetIds(ComponentName(context, receiverClassFor(type))).isNotEmpty()
            }
            if (!hasAny) {
                WorkManager.getInstance(context).cancelUniqueWork(WORK_NAME)
            }
        }

        private fun receiverClassFor(type: AppWidgetType): Class<out GlanceAppWidgetReceiver> = when (type) {
            AppWidgetType.PRICE -> PriceGlanceReceiver::class.java
            AppWidgetType.HEADLINES -> HeadlinesGlanceReceiver::class.java
        }
    }

    override suspend fun doWork(): Result {
        val activeTypes = preferencesStore.getActiveWidgetTypes()
        if (activeTypes.isEmpty()) return Result.success()

        Logger.debug("Refreshing data for widget types: '$activeTypes'", context = TAG)

        for (type in activeTypes) {
            when (type) {
                AppWidgetType.PRICE -> {
                    val periods = preferencesStore.getActivePricePeriods()
                    periods.forEach { period ->
                        dataRepository.fetchPriceData(period)
                            .onSuccess { preferencesStore.cachePriceData(period, it) }
                            .onFailure {
                                Logger.warn("Failed to refresh price for '$period'", it, context = TAG)
                            }
                    }
                    PriceGlanceWidget().updateAll(appContext)
                }

                AppWidgetType.HEADLINES -> {
                    dataRepository.fetchArticles()
                        .onSuccess { preferencesStore.cacheArticles(it) }
                        .onFailure {
                            Logger.warn("Failed to refresh headlines", it, context = TAG)
                        }
                    preferencesStore.bumpArticleRotationTick()
                    HeadlinesGlanceWidget().updateAll(appContext)
                }
            }
        }

        return Result.success()
    }
}