Implement Push client secret store and test it.

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/VectorPushHandler.kt

@@ -28,6 +28,7 @@ import io.element.android.libraries.core.log.logger.LoggerTag
 import io.element.android.libraries.core.meta.BuildMeta
 import io.element.android.libraries.di.ApplicationContext
 import io.element.android.libraries.push.api.store.PushDataStore
+import io.element.android.libraries.push.impl.clientsecret.PushClientSecret
 import io.element.android.libraries.push.impl.model.PushData
 import io.element.android.libraries.push.impl.notifications.NotifiableEventResolver
 import io.element.android.libraries.push.impl.notifications.NotificationActionIds
@@ -49,6 +50,7 @@ class VectorPushHandler @Inject constructor(
     // private val activeSessionHolder: ActiveSessionHolder,
     private val pushDataStore: PushDataStore,
     private val defaultPushDataStore: DefaultPushDataStore,
+    private val pushClientSecret: PushClientSecret,
     private val actionIds: NotificationActionIds,
     @ApplicationContext private val context: Context,
     private val buildMeta: BuildMeta
@@ -114,7 +116,8 @@ class VectorPushHandler @Inject constructor(
             }
 
             /* TODO EAx
-            - Open session
+            - Retrieve secret and use pushClientSecret
+            - Open matching session
             - get the event
             - display the notif
 

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecret.kt

@@ -0,0 +1,35 @@
+/*
+ * 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.push.impl.clientsecret
+
+interface PushClientSecret {
+    /**
+     * To call when registering a pusher. It will return the existing secret or create a new one.
+     */
+    suspend fun getSecretForUser(userId: String): String
+
+    /**
+     * To call when receiving a push containing a client secret.
+     * Return null if not found.
+     */
+    suspend fun getUserIdFromSecret(clientSecret: String): String?
+
+    /**
+     * To call when the user signs out.
+     */
+    suspend fun resetSecretForUser(userId: String)
+}

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecretFactory.kt

@@ -0,0 +1,21 @@
+/*
+ * 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.push.impl.clientsecret
+
+interface PushClientSecretFactory {
+    fun create(): String
+}

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecretFactoryImpl.kt

@@ -0,0 +1,28 @@
+/*
+ * 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.push.impl.clientsecret
+
+import com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.libraries.di.AppScope
+import java.util.UUID
+
+@ContributesBinding(AppScope::class)
+class PushClientSecretFactoryImpl : PushClientSecretFactory {
+    override fun create(): String {
+        return UUID.randomUUID().toString()
+    }
+}

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecretImpl.kt

@@ -0,0 +1,45 @@
+/*
+ * 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.push.impl.clientsecret
+
+import com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.libraries.di.AppScope
+import javax.inject.Inject
+
+@ContributesBinding(AppScope::class)
+class PushClientSecretImpl @Inject constructor(
+    private val pushClientSecretFactory: PushClientSecretFactory,
+    private val pushClientSecretStore: PushClientSecretStore,
+) : PushClientSecret {
+    override suspend fun getSecretForUser(userId: String): String {
+        val existingSecret = pushClientSecretStore.getSecret(userId)
+        if (existingSecret != null) {
+            return existingSecret
+        }
+        val newSecret = pushClientSecretFactory.create()
+        pushClientSecretStore.storeSecret(userId, newSecret)
+        return newSecret
+    }
+
+    override suspend fun getUserIdFromSecret(clientSecret: String): String? {
+        return pushClientSecretStore.getUserIdFromSecret(clientSecret)
+    }
+
+    override suspend fun resetSecretForUser(userId: String) {
+        pushClientSecretStore.resetSecret(userId)
+    }
+}

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecretStore.kt

@@ -0,0 +1,24 @@
+/*
+ * 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.push.impl.clientsecret
+
+interface PushClientSecretStore {
+    suspend fun storeSecret(userId: String, clientSecret: String)
+    suspend fun getSecret(userId: String): String?
+    suspend fun resetSecret(userId: String)
+    suspend fun getUserIdFromSecret(clientSecret: String): String?
+}

libraries/push/impl/src/main/kotlin/io/element/android/libraries/push/impl/clientsecret/PushClientSecretStoreDataStore.kt

@@ -0,0 +1,62 @@
+/*
+ * 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.push.impl.clientsecret
+
+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 com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.libraries.di.AppScope
+import io.element.android.libraries.di.ApplicationContext
+import kotlinx.coroutines.flow.first
+import javax.inject.Inject
+
+private val Context.dataStore: DataStore<Preferences> by preferencesDataStore(name = "push_client_secret_store")
+
+@ContributesBinding(AppScope::class)
+class PushClientSecretStoreDataStore @Inject constructor(
+    @ApplicationContext private val context: Context,
+) : PushClientSecretStore {
+    override suspend fun storeSecret(userId: String, clientSecret: String) {
+        context.dataStore.edit { settings ->
+            settings[getPreferenceKeyForUser(userId)] = clientSecret
+        }
+    }
+
+    override suspend fun getSecret(userId: String): String? {
+        return context.dataStore.data.first()[getPreferenceKeyForUser(userId)]
+    }
+
+    override suspend fun resetSecret(userId: String) {
+        context.dataStore.edit { settings ->
+            settings.remove(getPreferenceKeyForUser(userId))
+        }
+    }
+
+    override suspend fun getUserIdFromSecret(clientSecret: String): String? {
+        val keyValues = context.dataStore.data.first().asMap()
+        val matchingKey = keyValues.keys.firstOrNull {
+            keyValues[it] == clientSecret
+        }
+        return matchingKey?.name
+    }
+
+    private fun getPreferenceKeyForUser(userId: String) = stringPreferencesKey(userId)
+}