Split push module into several modules: Firebase, UnifiedPush, store

libraries/pushstore/impl/src/main/kotlin/io/element/android/libraries/pushstore/impl/DefaultUserPushStoreFactory.kt

@@ -0,0 +1,63 @@
+/*
+ * Copyright (c) 2023 New Vector Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.element.android.libraries.pushstore.impl
+
+import android.content.Context
+import com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.libraries.di.AppScope
+import io.element.android.libraries.di.ApplicationContext
+import io.element.android.libraries.di.SingleIn
+import io.element.android.libraries.pushstore.api.UserPushStore
+import io.element.android.libraries.pushstore.api.UserPushStoreFactory
+import io.element.android.libraries.sessionstorage.api.observer.SessionListener
+import io.element.android.libraries.sessionstorage.api.observer.SessionObserver
+import javax.inject.Inject
+
+@SingleIn(AppScope::class)
+@ContributesBinding(AppScope::class, boundType = UserPushStoreFactory::class)
+class DefaultUserPushStoreFactory @Inject constructor(
+    @ApplicationContext private val context: Context,
+    private val sessionObserver: SessionObserver,
+) : UserPushStoreFactory, SessionListener {
+    init {
+        observeSessions()
+    }
+
+    // We can have only one class accessing a single data store, so keep a cache of them.
+    private val cache = mutableMapOf<String, UserPushStore>()
+    override fun create(userId: String): UserPushStore {
+        return cache.getOrPut(userId) {
+            UserPushStoreDataStore(
+                context = context,
+                userId = userId
+            )
+        }
+    }
+
+    private fun observeSessions() {
+        sessionObserver.addListener(this)
+    }
+
+    override suspend fun onSessionCreated(userId: String) {
+        // Nothing to do
+    }
+
+    override suspend fun onSessionDeleted(userId: String) {
+        // Delete the store
+        create(userId).reset()
+    }
+}

libraries/pushstore/impl/src/main/kotlin/io/element/android/libraries/pushstore/impl/UserPushStoreDataStore.kt

@@ -0,0 +1,64 @@
+/*
+ * Copyright (c) 2023 New Vector Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.element.android.libraries.pushstore.impl
+
+import android.content.Context
+import androidx.datastore.core.DataStore
+import androidx.datastore.preferences.core.Preferences
+import androidx.datastore.preferences.core.edit
+import androidx.datastore.preferences.core.stringPreferencesKey
+import androidx.datastore.preferences.preferencesDataStore
+import io.element.android.libraries.pushstore.api.UserPushStore
+import kotlinx.coroutines.flow.first
+
+/**
+ * Store data related to push about a user.
+ */
+class UserPushStoreDataStore(
+    private val context: Context,
+    userId: String,
+) : UserPushStore {
+    private val Context.dataStore: DataStore<Preferences> by preferencesDataStore(name = "push_store_$userId")
+    private val notificationMethod = stringPreferencesKey("notificationMethod")
+    private val currentPushKey = stringPreferencesKey("currentPushKey")
+
+    override suspend fun getNotificationMethod(): String? {
+        return context.dataStore.data.first()[notificationMethod]
+    }
+
+    override suspend fun setNotificationMethod(value: String) {
+        context.dataStore.edit {
+            it[notificationMethod] = value
+        }
+    }
+
+    override suspend fun getCurrentRegisteredPushKey(): String? {
+        return context.dataStore.data.first()[currentPushKey]
+    }
+
+    override suspend fun setCurrentRegisteredPushKey(value: String) {
+        context.dataStore.edit {
+            it[currentPushKey] = value
+        }
+    }
+
+    override suspend fun reset() {
+        context.dataStore.edit {
+            it.clear()
+        }
+    }
+}