TaskPollWorker queries /workflow (general + downloaded categories) every 15 minutes and posts a notification when a task transitions to a terminal state. A persisted id→status map drives the diff; the first run only seeds it, so enabling doesn't replay history. Toggle lives in the settings dialog and requests POST_NOTIFICATIONS on API 33+. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>pull/3589/head
Tomas Dvorak
2 weeks ago
parent
575707bd43
commit
0c7119b2a6
@ -0,0 +1,71 @@
|
||||
package org.cloudreve.android.data
|
||||
|
||||
import android.content.Context
|
||||
import androidx.datastore.preferences.core.booleanPreferencesKey
|
||||
import androidx.datastore.preferences.core.edit
|
||||
import androidx.datastore.preferences.core.stringPreferencesKey
|
||||
import androidx.datastore.preferences.preferencesDataStore
|
||||
import kotlinx.coroutines.flow.Flow
|
||||
import kotlinx.coroutines.flow.first
|
||||
import kotlinx.coroutines.flow.map
|
||||
import kotlinx.serialization.Serializable
|
||||
import kotlinx.serialization.encodeToString
|
||||
import kotlinx.serialization.json.Json
|
||||
|
||||
private val Context.taskNotifyStore by preferencesDataStore(name = "task_notify")
|
||||
|
||||
/**
|
||||
* Task-notification preferences: the on/off toggle plus a map of
|
||||
* task id → last seen status, so TaskPollWorker only notifies on
|
||||
* transitions to a terminal state. The map is pruned to the newest
|
||||
* entries to stay small.
|
||||
*/
|
||||
class TaskNotifySettings(private val context: Context) {
|
||||
|
||||
private val json = Json { ignoreUnknownKeys = true }
|
||||
|
||||
@Serializable
|
||||
private data class Seen(val map: Map<String, String> = emptyMap())
|
||||
|
||||
private object Keys {
|
||||
val enabled = booleanPreferencesKey("enabled")
|
||||
val seen = stringPreferencesKey("seen")
|
||||
}
|
||||
|
||||
val enabled: Flow<Boolean> = context.taskNotifyStore.data.map {
|
||||
it[Keys.enabled] ?: false
|
||||
}
|
||||
|
||||
suspend fun enabledNow(): Boolean = enabled.first()
|
||||
|
||||
suspend fun setEnabled(enabled: Boolean) {
|
||||
context.taskNotifyStore.edit { it[Keys.enabled] = enabled }
|
||||
}
|
||||
|
||||
suspend fun seenNow(): Map<String, String> =
|
||||
runCatching {
|
||||
json.decodeFromString<Seen>(
|
||||
context.taskNotifyStore.data.first()[Keys.seen] ?: "{}"
|
||||
).map
|
||||
}.getOrDefault(emptyMap())
|
||||
|
||||
suspend fun markSeen(id: String, status: String) {
|
||||
context.taskNotifyStore.edit { prefs ->
|
||||
val map = runCatching {
|
||||
json.decodeFromString<Seen>(prefs[Keys.seen] ?: "{}").map
|
||||
}.getOrDefault(emptyMap()).toMutableMap()
|
||||
map.remove(id)
|
||||
map[id] = status
|
||||
if (map.size > MAX_SEEN) {
|
||||
// Keep the newest tail; insertion order preserved.
|
||||
val trimmed = map.entries.toList().takeLast(MAX_SEEN).associate { it.toPair() }
|
||||
map.clear(); map.putAll(trimmed)
|
||||
}
|
||||
prefs[Keys.seen] = json.encodeToString(Seen(map))
|
||||
}
|
||||
}
|
||||
|
||||
companion object {
|
||||
private const val MAX_SEEN = 200
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,111 @@
|
||||
package org.cloudreve.android.work
|
||||
|
||||
import android.app.NotificationChannel
|
||||
import android.app.NotificationManager
|
||||
import android.content.Context
|
||||
import androidx.core.app.NotificationCompat
|
||||
import androidx.core.app.NotificationManagerCompat
|
||||
import androidx.work.Constraints
|
||||
import androidx.work.CoroutineWorker
|
||||
import androidx.work.ExistingPeriodicWorkPolicy
|
||||
import androidx.work.ExistingWorkPolicy
|
||||
import androidx.work.NetworkType
|
||||
import androidx.work.OneTimeWorkRequestBuilder
|
||||
import androidx.work.PeriodicWorkRequestBuilder
|
||||
import androidx.work.WorkManager
|
||||
import androidx.work.WorkerParameters
|
||||
import org.cloudreve.android.CloudreveApp
|
||||
import java.util.concurrent.TimeUnit
|
||||
|
||||
/**
|
||||
* Periodically polls /workflow for the user's tasks and posts a
|
||||
* notification when a task reaches a terminal state (completed, error,
|
||||
* canceled). The first run after enabling only seeds the seen-map so
|
||||
* history doesn't flood the shade.
|
||||
*/
|
||||
class TaskPollWorker(context: Context, params: WorkerParameters) :
|
||||
CoroutineWorker(context, params) {
|
||||
|
||||
override suspend fun doWork(): Result {
|
||||
val app = applicationContext as CloudreveApp
|
||||
val settings = app.taskNotifySettings
|
||||
if (!settings.enabledNow()) return Result.success()
|
||||
|
||||
createChannel()
|
||||
|
||||
val seen = settings.seenNow()
|
||||
val tasks = runCatching {
|
||||
app.apiClient.service().listTasks(pageSize = 20, category = "general")
|
||||
.body()?.data?.tasks.orEmpty() +
|
||||
app.apiClient.service().listTasks(pageSize = 20, category = "downloaded")
|
||||
.body()?.data?.tasks.orEmpty()
|
||||
}.getOrDefault(emptyList())
|
||||
|
||||
val fresh = seen.isNotEmpty()
|
||||
tasks.forEach { t ->
|
||||
val previous = seen[t.id]
|
||||
if (previous == t.status) return@forEach
|
||||
settings.markSeen(t.id, t.status)
|
||||
if (fresh && t.status in TERMINAL) {
|
||||
notify(t.id, t.type, t.status, t.error)
|
||||
}
|
||||
}
|
||||
return Result.success()
|
||||
}
|
||||
|
||||
private fun notify(id: String, type: String, status: String, error: String?) {
|
||||
if (!NotificationManagerCompat.from(applicationContext).areNotificationsEnabled()) return
|
||||
val label = type.substringAfterLast('.').replace('_', ' ')
|
||||
val text = when (status) {
|
||||
"completed" -> "$label completed"
|
||||
"canceled" -> "$label canceled"
|
||||
else -> "$label failed" + (error?.let { ": $it" } ?: "")
|
||||
}
|
||||
val notification = NotificationCompat.Builder(applicationContext, CHANNEL_ID)
|
||||
.setSmallIcon(android.R.drawable.stat_sys_download_done)
|
||||
.setContentTitle("Cloudreve task")
|
||||
.setContentText(text)
|
||||
.setAutoCancel(true)
|
||||
.build()
|
||||
NotificationManagerCompat.from(applicationContext)
|
||||
.notify(id.hashCode(), notification)
|
||||
}
|
||||
|
||||
private fun createChannel() {
|
||||
val manager =
|
||||
applicationContext.getSystemService(Context.NOTIFICATION_SERVICE) as NotificationManager
|
||||
manager.createNotificationChannel(
|
||||
NotificationChannel(CHANNEL_ID, "Task updates", NotificationManager.IMPORTANCE_DEFAULT)
|
||||
)
|
||||
}
|
||||
|
||||
companion object {
|
||||
private const val WORK_NAME = "task_poll_periodic"
|
||||
private const val CHANNEL_ID = "task_updates"
|
||||
private val TERMINAL = setOf("completed", "error", "canceled")
|
||||
|
||||
private val constraints = Constraints.Builder()
|
||||
.setRequiredNetworkType(NetworkType.CONNECTED)
|
||||
.build()
|
||||
|
||||
fun apply(context: Context) {
|
||||
val wm = WorkManager.getInstance(context)
|
||||
val periodic = PeriodicWorkRequestBuilder<TaskPollWorker>(15, TimeUnit.MINUTES)
|
||||
.setConstraints(constraints)
|
||||
.build()
|
||||
wm.enqueueUniquePeriodicWork(WORK_NAME, ExistingPeriodicWorkPolicy.UPDATE, periodic)
|
||||
// Prime the seen-map immediately so enabling doesn't notify on history.
|
||||
wm.enqueueUniqueWork(
|
||||
"$WORK_NAME.once",
|
||||
ExistingWorkPolicy.REPLACE,
|
||||
OneTimeWorkRequestBuilder<TaskPollWorker>().setConstraints(constraints).build(),
|
||||
)
|
||||
}
|
||||
|
||||
fun cancel(context: Context) {
|
||||
val wm = WorkManager.getInstance(context)
|
||||
wm.cancelUniqueWork(WORK_NAME)
|
||||
wm.cancelUniqueWork("$WORK_NAME.once")
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue