fix: reconcile push registration on startup

Author: Priveetee

app/src/main/java/dev/typetype/android/MainViewModel.kt

@@ -27,6 +27,7 @@ import dev.typetype.android.domain.subscriptions.SubscriptionsRepository
 import dev.typetype.android.domain.usersettings.UserSettings
 import dev.typetype.android.domain.usersettings.UserSettingsRepository
 import dev.typetype.android.feature.player.host.PlayerHostController
+import dev.typetype.android.services.push.PushRegistrationManager
 import javax.inject.Inject
 import kotlinx.coroutines.channels.Channel
 import kotlinx.coroutines.flow.MutableStateFlow
@@ -63,6 +64,7 @@ class MainViewModel @Inject constructor(
     private val subscriptionsRepository: SubscriptionsRepository,
     private val libraryRepository: LibraryRepository,
     private val activeAccountScope: ActiveAccountScope,
+    private val pushRegistrationManager: PushRegistrationManager,
     private val startupLandingStore: StartupLandingStore,
     private val playbackResumeRepository: PlaybackResumeRepository,
     private val playbackQueueRepository: PlaybackQueueRepository,
@@ -182,6 +184,7 @@ class MainViewModel @Inject constructor(
             launch { profileRepository.refresh() }
             launch { subscriptionsRepository.refresh() }
             launch { libraryRepository.resumePendingWrites() }
+            launch { pushRegistrationManager.reconcileRegistration() }
         }
     }
 
@@ -221,6 +224,7 @@ class MainViewModel @Inject constructor(
             launch { profileRepository.refresh() }
             launch { subscriptionsRepository.refresh() }
             launch { libraryRepository.resumePendingWrites() }
+            launch { pushRegistrationManager.reconcileRegistration() }
             launch {
                 restorePlaybackUnlessExternalRequestArrives()
             }

app/src/main/java/dev/typetype/android/services/push/PushRegistrationManager.kt

@@ -4,6 +4,7 @@ import android.content.Context
 import dagger.hilt.android.qualifiers.ApplicationContext
 import dev.typetype.android.R
 import dev.typetype.android.data.account.ActiveAccountScope
+import dev.typetype.android.data.account.AccountScope
 import dev.typetype.android.data.push.PushRegistrationStore
 import dev.typetype.android.data.push.pushInstanceName
 import dev.typetype.android.data.push.scopeFromInstanceName
@@ -37,6 +38,30 @@ class PushRegistrationManager @Inject constructor(
         if (!repository.currentCapability().enabled) {
             return update(PushRegistrationStatus.Unavailable)
         }
+        return startRegistration(scope)
+    }
+
+    suspend fun reconcileRegistration(): PushRegistrationStatus {
+        val scope = activeAccountScope.require()
+        val registration = registrationStore.registrationOnce(scope)
+            ?: return update(PushRegistrationStatus.Disabled)
+        if (!repository.currentCapability().enabled) {
+            return update(PushRegistrationStatus.Unavailable)
+        }
+        val endpoint = registration.endpoint ?: return startRegistration(scope)
+        val devices = repository.devices().getOrElse {
+            return update(PushRegistrationStatus.Failed)
+        }
+        if (devices.any { device -> device.deviceId == registration.deviceId }) {
+            return update(PushRegistrationStatus.Registered)
+        }
+        return repository.registerDevice(registration.deviceId, endpoint).fold(
+            onSuccess = { update(PushRegistrationStatus.Registered) },
+            onFailure = { update(PushRegistrationStatus.Failed) },
+        )
+    }
+
+    private suspend fun startRegistration(scope: AccountScope): PushRegistrationStatus {
         val instance = pushInstanceName(scope)
         if (UnifiedPush.getDistributors(context).isEmpty()) {
             return update(PushRegistrationStatus.MissingDistributor)