PM-27867: feat: Use SDK for prelogin API

app/src/main/kotlin/com/x8bit/bitwarden/data/auth/datasource/sdk/AuthSdkSource.kt

@@ -2,6 +2,7 @@ package com.x8bit.bitwarden.data.auth.datasource.sdk
 
 import com.bitwarden.auth.JitMasterPasswordRegistrationResponse
 import com.bitwarden.auth.KeyConnectorRegistrationResult
+import com.bitwarden.auth.PasswordPreloginResponse
 import com.bitwarden.auth.TdeRegistrationResponse
 import com.bitwarden.auth.UserMasterPasswordRegistrationResponse
 import com.bitwarden.core.AuthRequestResponse
@@ -21,6 +22,11 @@ import com.x8bit.bitwarden.data.auth.datasource.sdk.model.PasswordStrength
  */
 @Suppress("TooManyFunctions")
 interface AuthSdkSource {
+    /**
+     * Performs the pre-login request for the given email address.
+     */
+    suspend fun preLogin(email: String): Result<PasswordPreloginResponse>
+
     /**
      * Enrolls the user to master password unlock.
      */

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

@@ -3,6 +3,7 @@ package com.x8bit.bitwarden.data.auth.datasource.sdk
 import com.bitwarden.auth.JitMasterPasswordRegistrationRequest
 import com.bitwarden.auth.JitMasterPasswordRegistrationResponse
 import com.bitwarden.auth.KeyConnectorRegistrationResult
+import com.bitwarden.auth.PasswordPreloginResponse
 import com.bitwarden.auth.TdeRegistrationRequest
 import com.bitwarden.auth.TdeRegistrationResponse
 import com.bitwarden.auth.UserMasterPasswordRegistrationRequest
@@ -38,6 +39,14 @@ class AuthSdkSourceImpl(
 ) : BaseSdkSource(sdkClientManager = sdkClientManager),
     AuthSdkSource {
 
+    override suspend fun preLogin(
+        email: String,
+    ): Result<PasswordPreloginResponse> = runCatchingWithLogs {
+        withContext(context = dispatcherManager.io) {
+            useClient { auth().login().getPasswordPrelogin(email = email) }
+        }
+    }
+
     override suspend fun postKeysForJitPasswordRegistration(
         userId: String,
         organizationId: String,

app/src/main/kotlin/com/x8bit/bitwarden/data/auth/repository/AuthRepositoryImpl.kt

@@ -1,5 +1,6 @@
 package com.x8bit.bitwarden.data.auth.repository
 
+import com.bitwarden.auth.PasswordPreloginResponse
 import com.bitwarden.core.AuthRequestMethod
 import com.bitwarden.core.InitUserCryptoMethod
 import com.bitwarden.core.MasterPasswordUnlockData
@@ -529,13 +530,18 @@ internal class AuthRepositoryImpl(
     override suspend fun login(
         email: String,
         password: String,
-    ): LoginResult = identityService
-        .preLogin(email = email)
+    ): LoginResult = if (featureFlagManager.getFeatureFlag(key = FlagKey.SdkPreLogin)) {
+        authSdkSource.preLogin(email = email)
+    } else {
+        identityService
+            .preLogin(email = email)
+            .map { PasswordPreloginResponse(salt = email, kdf = it.kdfParams.toSdkParams()) }
+    }
         .flatMap {
             authSdkSource.hashPassword(
-                salt = email,
+                salt = it.salt,
                 password = password,
-                kdf = it.kdfParams.toSdkParams(),
+                kdf = it.kdf,
                 purpose = HashPurpose.SERVER_AUTHORIZATION,
             )
         }

app/src/test/kotlin/com/x8bit/bitwarden/data/auth/datasource/disk/util/FakeAuthDiskSource.kt

@@ -446,7 +446,7 @@ class FakeAuthDiskSource : AuthDiskSource {
     /**
      * Assert that the given [userState] matches the currently tracked value.
      */
-    fun assertUserState(userState: UserStateJson) {
+    fun assertUserState(userState: UserStateJson?) {
         assertEquals(userState, this.userState)
     }
 

app/src/test/kotlin/com/x8bit/bitwarden/data/auth/datasource/sdk/AuthSdkSourceTest.kt

@@ -3,6 +3,7 @@ package com.x8bit.bitwarden.data.auth.datasource.sdk
 import com.bitwarden.auth.JitMasterPasswordRegistrationRequest
 import com.bitwarden.auth.JitMasterPasswordRegistrationResponse
 import com.bitwarden.auth.KeyConnectorRegistrationResult
+import com.bitwarden.auth.PasswordPreloginResponse
 import com.bitwarden.auth.TdeRegistrationRequest
 import com.bitwarden.auth.TdeRegistrationResponse
 import com.bitwarden.auth.UserMasterPasswordRegistrationRequest
@@ -14,6 +15,7 @@ import com.bitwarden.core.MasterPasswordPolicyOptions
 import com.bitwarden.core.RegisterKeyResponse
 import com.bitwarden.core.RegisterTdeKeyResponse
 import com.bitwarden.core.data.manager.dispatcher.FakeDispatcherManager
+import com.bitwarden.core.data.util.asFailure
 import com.bitwarden.core.data.util.asSuccess
 import com.bitwarden.crypto.HashPurpose
 import com.bitwarden.crypto.Kdf
@@ -22,6 +24,7 @@ import com.bitwarden.policies.PolicyType
 import com.bitwarden.policies.PolicyView
 import com.bitwarden.sdk.AuthClient
 import com.bitwarden.sdk.Client
+import com.bitwarden.sdk.LoginClient
 import com.bitwarden.sdk.PlatformClient
 import com.bitwarden.sdk.PoliciesClient
 import com.bitwarden.sdk.RegistrationClient
@@ -40,12 +43,14 @@ import org.junit.jupiter.api.Assertions.assertEquals
 import org.junit.jupiter.api.Test
 
 class AuthSdkSourceTest {
+    private val clientLogin = mockk<LoginClient>()
     private val clientRegistration = mockk<RegistrationClient>()
     private val clientAuth = mockk<AuthClient> {
         every { registration() } returns clientRegistration
+        every { login() } returns clientLogin
     }
     private val clientPlatform = mockk<PlatformClient> {
-        coEvery { loadFlags(any()) } just runs
+        coEvery { loadFlags(flags = any()) } just runs
     }
     private val clientPolicies = mockk<PoliciesClient>()
     private val client = mockk<Client> {
@@ -63,6 +68,38 @@ class AuthSdkSourceTest {
         sdkClientManager = sdkClientManager,
     )
 
+    @Test
+    fun `preLogin should call SDK and return a Result with correct data`() = runBlocking {
+        val email = "email@example.com"
+        val expectedResult = mockk<PasswordPreloginResponse>()
+        val slot = slot<suspend Client.() -> PasswordPreloginResponse>()
+        coEvery {
+            sdkClientManager.singleUseClient(block = capture(slot))
+        } coAnswers { slot.captured(client) }
+        coEvery { clientLogin.getPasswordPrelogin(email = email) } returns expectedResult
+
+        val result = authSkdSource.preLogin(email = email)
+
+        assertEquals(expectedResult.asSuccess(), result)
+        coVerify(exactly = 1) { clientLogin.getPasswordPrelogin(email = email) }
+    }
+
+    @Test
+    fun `preLogin should return a failure when the SDK throws`() = runBlocking {
+        val email = "email@example.com"
+        val error = RuntimeException("Fail")
+        val slot = slot<suspend Client.() -> PasswordPreloginResponse>()
+        coEvery {
+            sdkClientManager.singleUseClient(block = capture(slot))
+        } coAnswers { slot.captured(client) }
+        coEvery { clientLogin.getPasswordPrelogin(email = email) } throws error
+
+        val result = authSkdSource.preLogin(email = email)
+
+        assertEquals(error.asFailure(), result)
+        coVerify(exactly = 1) { clientLogin.getPasswordPrelogin(email = email) }
+    }
+
     @Suppress("MaxLineLength")
     @Test
     fun `postKeysForJitPasswordRegistration should call SDK and return a Result with correct data`() =