BIT-2070: Enable individual Cipher Encryption for SDK (#1148)

app/src/main/java/com/x8bit/bitwarden/data/auth/datasource/sdk/AuthSdkSourceImpl.kt

@@ -12,6 +12,10 @@ import com.bitwarden.sdk.ClientPlatform
 import com.x8bit.bitwarden.data.auth.datasource.sdk.model.PasswordStrength
 import com.x8bit.bitwarden.data.auth.datasource.sdk.util.toPasswordStrengthOrNull
 import com.x8bit.bitwarden.data.auth.datasource.sdk.util.toUByte
+import com.x8bit.bitwarden.data.platform.manager.dispatcher.DispatcherManager
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManager
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.launch
 
 /**
  * Primary implementation of [AuthSdkSource] that serves as a convenience wrapper around a
@@ -20,8 +24,18 @@ import com.x8bit.bitwarden.data.auth.datasource.sdk.util.toUByte
 class AuthSdkSourceImpl(
     private val clientAuth: ClientAuth,
     private val clientPlatform: ClientPlatform,
+    dispatcherManager: DispatcherManager,
+    featureFlagManager: BitwardenFeatureFlagManager,
 ) : AuthSdkSource {
 
+    private val ioScope = CoroutineScope(dispatcherManager.io)
+
+    init {
+        ioScope.launch {
+            clientPlatform.loadFlags(featureFlagManager.featureFlags)
+        }
+    }
+
     override suspend fun getTrustDevice(): Result<TrustDeviceResponse> = runCatching {
         clientAuth.trustDevice()
     }

app/src/main/java/com/x8bit/bitwarden/data/auth/datasource/sdk/di/AuthSdkModule.kt

@@ -3,6 +3,8 @@ package com.x8bit.bitwarden.data.auth.datasource.sdk.di
 import com.bitwarden.sdk.Client
 import com.x8bit.bitwarden.data.auth.datasource.sdk.AuthSdkSource
 import com.x8bit.bitwarden.data.auth.datasource.sdk.AuthSdkSourceImpl
+import com.x8bit.bitwarden.data.platform.manager.dispatcher.DispatcherManager
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManager
 import dagger.Module
 import dagger.Provides
 import dagger.hilt.InstallIn
@@ -20,8 +22,12 @@ object AuthSdkModule {
     @Singleton
     fun provideAuthSdkSource(
         client: Client,
+        featureFlagManager: BitwardenFeatureFlagManager,
+        dispatcherManager: DispatcherManager,
     ): AuthSdkSource = AuthSdkSourceImpl(
         clientAuth = client.auth(),
         clientPlatform = client.platform(),
+        featureFlagManager = featureFlagManager,
+        dispatcherManager = dispatcherManager,
     )
 }

app/src/main/java/com/x8bit/bitwarden/data/platform/manager/SdkClientManager.kt

@@ -11,7 +11,7 @@ interface SdkClientManager {
      * Returns the cached [Client] instance for the given [userId], otherwise creates and caches
      * a new one and returns it.
      */
-    fun getOrCreateClient(userId: String): Client
+    suspend fun getOrCreateClient(userId: String): Client
 
     /**
      * Clears any resources from the [Client] associated with the given [userId] and removes it

app/src/main/java/com/x8bit/bitwarden/data/platform/manager/SdkClientManagerImpl.kt

@@ -1,16 +1,21 @@
 package com.x8bit.bitwarden.data.platform.manager
 
 import com.bitwarden.sdk.Client
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManager
 
 /**
  * Primary implementation of [SdkClientManager].
  */
 class SdkClientManagerImpl(
-    private val clientProvider: () -> Client = { Client(null) },
+    private val featureFlagManager: BitwardenFeatureFlagManager,
+    private val clientProvider: suspend () -> Client = {
+        Client(null)
+            .apply { platform().loadFlags(featureFlagManager.featureFlags) }
+    },
 ) : SdkClientManager {
     private val userIdToClientMap = mutableMapOf<String, Client>()
 
-    override fun getOrCreateClient(
+    override suspend fun getOrCreateClient(
         userId: String,
     ): Client =
         userIdToClientMap.getOrPut(key = userId) { clientProvider() }

app/src/main/java/com/x8bit/bitwarden/data/platform/manager/di/PlatformManagerModule.kt

@@ -34,6 +34,7 @@ import com.x8bit.bitwarden.data.platform.manager.dispatcher.DispatcherManager
 import com.x8bit.bitwarden.data.platform.manager.dispatcher.DispatcherManagerImpl
 import com.x8bit.bitwarden.data.platform.repository.EnvironmentRepository
 import com.x8bit.bitwarden.data.platform.repository.SettingsRepository
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManager
 import com.x8bit.bitwarden.data.vault.repository.VaultRepository
 import dagger.Module
 import dagger.Provides
@@ -97,7 +98,11 @@ object PlatformManagerModule {
 
     @Provides
     @Singleton
-    fun provideSdkClientManager(): SdkClientManager = SdkClientManagerImpl()
+    fun provideSdkClientManager(
+        featureFlagManager: BitwardenFeatureFlagManager,
+    ): SdkClientManager = SdkClientManagerImpl(
+        featureFlagManager = featureFlagManager,
+    )
 
     @Provides
     @Singleton

app/src/main/java/com/x8bit/bitwarden/data/vault/datasource/sdk/BitwardenFeatureFlagManager.kt

@@ -0,0 +1,11 @@
+package com.x8bit.bitwarden.data.vault.datasource.sdk
+
+/**
+ * Manages the available feature flags for the Bitwarden application.
+ */
+interface BitwardenFeatureFlagManager {
+    /**
+     * Returns a map of feature flags.
+     */
+    val featureFlags: Map<String, Boolean>
+}

app/src/main/java/com/x8bit/bitwarden/data/vault/datasource/sdk/BitwardenFeatureFlagManagerImpl.kt

@@ -0,0 +1,11 @@
+package com.x8bit.bitwarden.data.vault.datasource.sdk
+
+private const val CIPHER_KEY_ENCRYPTION_KEY = "enableCipherKeyEncryption"
+
+/**
+ * Primary implementation of [BitwardenFeatureFlagManager].
+ */
+class BitwardenFeatureFlagManagerImpl : BitwardenFeatureFlagManager {
+    override val featureFlags: Map<String, Boolean>
+        get() = mapOf(CIPHER_KEY_ENCRYPTION_KEY to true)
+}

app/src/main/java/com/x8bit/bitwarden/data/vault/datasource/sdk/VaultSdkSourceImpl.kt

@@ -388,7 +388,7 @@ class VaultSdkSourceImpl(
             )
     }
 
-    private fun getClient(
+    private suspend fun getClient(
         userId: String,
     ): Client = sdkClientManager.getOrCreateClient(userId = userId)
 }

app/src/main/java/com/x8bit/bitwarden/data/vault/datasource/sdk/di/VaultSdkModule.kt

@@ -1,6 +1,8 @@
 package com.x8bit.bitwarden.data.vault.datasource.sdk.di
 
 import com.x8bit.bitwarden.data.platform.manager.SdkClientManager
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManager
+import com.x8bit.bitwarden.data.vault.datasource.sdk.BitwardenFeatureFlagManagerImpl
 import com.x8bit.bitwarden.data.vault.datasource.sdk.VaultSdkSource
 import com.x8bit.bitwarden.data.vault.datasource.sdk.VaultSdkSourceImpl
 import dagger.Module
@@ -24,4 +26,9 @@ object VaultSdkModule {
         VaultSdkSourceImpl(
             sdkClientManager = sdkClientManager,
         )
+
+    @Provides
+    @Singleton
+    fun providesBitwardenFeatureFlagManager(): BitwardenFeatureFlagManager =
+        BitwardenFeatureFlagManagerImpl()
 }