Support networking relink flows

financial-connections/api/financial-connections.api

@@ -438,14 +438,6 @@ public final class com/stripe/android/financialconnections/features/attachpaymen
 	public final fun getLambda-1$financial_connections_release ()Lkotlin/jvm/functions/Function2;
 }
 
-public final class com/stripe/android/financialconnections/features/bankauthrepair/BankAuthRepairViewModel$Args$Creator : android/os/Parcelable$Creator {
-	public fun <init> ()V
-	public final fun createFromParcel (Landroid/os/Parcel;)Lcom/stripe/android/financialconnections/features/bankauthrepair/BankAuthRepairViewModel$Args;
-	public synthetic fun createFromParcel (Landroid/os/Parcel;)Ljava/lang/Object;
-	public final fun newArray (I)[Lcom/stripe/android/financialconnections/features/bankauthrepair/BankAuthRepairViewModel$Args;
-	public synthetic fun newArray (I)[Ljava/lang/Object;
-}
-
 public final class com/stripe/android/financialconnections/features/common/ComposableSingletons$AccountItemKt {
 	public static final field INSTANCE Lcom/stripe/android/financialconnections/features/common/ComposableSingletons$AccountItemKt;
 	public fun <init> ()V

financial-connections/src/main/java/com/stripe/android/financialconnections/di/FinancialConnectionsSheetNativeComponent.kt

@@ -6,7 +6,6 @@ import com.stripe.android.financialconnections.FinancialConnectionsSheet
 import com.stripe.android.financialconnections.features.accountpicker.AccountPickerViewModel
 import com.stripe.android.financialconnections.features.accountupdate.AccountUpdateRequiredViewModel
 import com.stripe.android.financialconnections.features.attachpayment.AttachPaymentViewModel
-import com.stripe.android.financialconnections.features.bankauthrepair.BankAuthRepairViewModel
 import com.stripe.android.financialconnections.features.consent.ConsentViewModel
 import com.stripe.android.financialconnections.features.error.ErrorViewModel
 import com.stripe.android.financialconnections.features.exit.ExitViewModel
@@ -50,7 +49,6 @@ internal interface FinancialConnectionsSheetNativeComponent {
     val manualEntryViewModelFactory: ManualEntryViewModel.Factory
     val manualEntrySuccessViewModelFactory: ManualEntrySuccessViewModel.Factory
     val partnerAuthViewModelFactory: PartnerAuthViewModel.Factory
-    val bankAuthRepairViewModelFactory: BankAuthRepairViewModel.Factory
     val successViewModelFactory: SuccessViewModel.Factory
     val attachPaymentViewModelFactory: AttachPaymentViewModel.Factory
     val resetViewModelFactory: ResetViewModel.Factory

financial-connections/src/main/java/com/stripe/android/financialconnections/di/FinancialConnectionsSheetSharedModule.kt

@@ -30,7 +30,9 @@ import com.stripe.android.financialconnections.analytics.FinancialConnectionsAna
 import com.stripe.android.financialconnections.analytics.FinancialConnectionsEventReporter
 import com.stripe.android.financialconnections.domain.GetOrFetchSync
 import com.stripe.android.financialconnections.domain.IsLinkWithStripe
+import com.stripe.android.financialconnections.domain.IsNetworkingRelinkSession
 import com.stripe.android.financialconnections.domain.RealIsLinkWithStripe
+import com.stripe.android.financialconnections.domain.RealIsNetworkingRelinkSession
 import com.stripe.android.financialconnections.features.common.enableWorkManager
 import com.stripe.android.financialconnections.repository.ConsumerSessionProvider
 import com.stripe.android.financialconnections.repository.ConsumerSessionRepository
@@ -86,6 +88,9 @@ internal interface FinancialConnectionsSheetSharedModule {
     @Binds
     fun bindsIsLinkWithStripe(impl: RealIsLinkWithStripe): IsLinkWithStripe
 
+    @Binds
+    fun bindsIsNetworkingRelinkSession(impl: RealIsNetworkingRelinkSession): IsNetworkingRelinkSession
+
     companion object {
 
         @Provides

financial-connections/src/main/java/com/stripe/android/financialconnections/domain/IsNetworkingRelinkSession.kt

@@ -0,0 +1,17 @@
+package com.stripe.android.financialconnections.domain
+
+import com.stripe.android.financialconnections.repository.CoreAuthorizationPendingNetworkingRepairRepository
+import javax.inject.Inject
+
+internal fun interface IsNetworkingRelinkSession {
+    operator fun invoke(): Boolean
+}
+
+internal class RealIsNetworkingRelinkSession @Inject constructor(
+    private val pendingRepairRepository: CoreAuthorizationPendingNetworkingRepairRepository,
+) : IsNetworkingRelinkSession {
+
+    override fun invoke(): Boolean {
+        return pendingRepairRepository.get() != null
+    }
+}

financial-connections/src/main/java/com/stripe/android/financialconnections/domain/RepairAuthorizationSession.kt

@@ -0,0 +1,25 @@
+package com.stripe.android.financialconnections.domain
+
+import com.stripe.android.financialconnections.FinancialConnectionsSheet
+import com.stripe.android.financialconnections.di.APPLICATION_ID
+import com.stripe.android.financialconnections.model.FinancialConnectionsAuthorizationSession
+import com.stripe.android.financialconnections.repository.FinancialConnectionsManifestRepository
+import javax.inject.Inject
+import javax.inject.Named
+
+internal class RepairAuthorizationSession @Inject constructor(
+    private val repository: FinancialConnectionsManifestRepository,
+    private val configuration: FinancialConnectionsSheet.Configuration,
+    @Named(APPLICATION_ID) private val applicationId: String,
+) {
+
+    suspend operator fun invoke(
+        coreAuthorization: String
+    ): FinancialConnectionsAuthorizationSession {
+        return repository.repairAuthorizationSession(
+            clientSecret = configuration.financialConnectionsSessionClientSecret,
+            coreAuthorization = coreAuthorization,
+            applicationId = applicationId,
+        )
+    }
+}

financial-connections/src/main/java/com/stripe/android/financialconnections/domain/SaveAccountToLink.kt

@@ -24,6 +24,7 @@ internal class SaveAccountToLink @Inject constructor(
     private val successContentRepository: SuccessContentRepository,
     private val repository: FinancialConnectionsManifestRepository,
     private val accountsRepository: FinancialConnectionsAccountsRepository,
+    private val isNetworkingRelinkSession: IsNetworkingRelinkSession,
 ) {
 
     suspend fun new(
@@ -90,9 +91,13 @@ internal class SaveAccountToLink @Inject constructor(
         }.mapCatching {
             action(selectedAccountIds)
         }.onSuccess { manifest ->
-            storeSavedToLinkMessage(manifest, selectedAccountIds.size)
+            if (!isNetworkingRelinkSession()) {
+                storeSavedToLinkMessage(manifest, selectedAccountIds.size)
+            }
         }.onFailure {
-            storeFailedToSaveToLinkMessage(selectedAccountIds.size)
+            if (!isNetworkingRelinkSession()) {
+                storeFailedToSaveToLinkMessage(selectedAccountIds.size)
+            }
         }.getOrThrow()
     }
 

financial-connections/src/main/java/com/stripe/android/financialconnections/features/accountupdate/AccountUpdateRequiredViewModel.kt

@@ -24,6 +24,7 @@ import com.stripe.android.financialconnections.presentation.Async
 import com.stripe.android.financialconnections.presentation.Async.Uninitialized
 import com.stripe.android.financialconnections.presentation.FinancialConnectionsViewModel
 import com.stripe.android.financialconnections.repository.AccountUpdateRequiredContentRepository
+import com.stripe.android.financialconnections.repository.CoreAuthorizationPendingNetworkingRepairRepository
 import dagger.assisted.Assisted
 import dagger.assisted.AssistedFactory
 import dagger.assisted.AssistedInject
@@ -33,6 +34,7 @@ internal class AccountUpdateRequiredViewModel @AssistedInject constructor(
     @Assisted initialState: AccountUpdateRequiredState,
     nativeAuthFlowCoordinator: NativeAuthFlowCoordinator,
     private val updateRequiredContentRepository: AccountUpdateRequiredContentRepository,
+    private val pendingRepairRepository: CoreAuthorizationPendingNetworkingRepairRepository,
     private val navigationManager: NavigationManager,
     private val eventTracker: FinancialConnectionsAnalyticsTracker,
     private val updateLocalManifest: UpdateLocalManifest,
@@ -60,7 +62,7 @@ internal class AccountUpdateRequiredViewModel @AssistedInject constructor(
             val referrer = state.referrer
             when (val type = requireNotNull(state.payload()?.type)) {
                 is Type.Repair -> {
-                    handleUnsupportedRepairAction(referrer)
+                    openBankAuthRepair(type.institution, type.authorization, referrer)
                 }
                 is Type.Supportability -> {
                     openPartnerAuth(type.institution, referrer)
@@ -69,15 +71,31 @@ internal class AccountUpdateRequiredViewModel @AssistedInject constructor(
         }
     }
 
-    private fun handleUnsupportedRepairAction(referrer: Pane) {
-        eventTracker.logError(
-            extraMessage = "Updating a repair account, but repairs are not supported in Mobile.",
-            error = UnclassifiedError("UpdateRepairAccountError"),
-            logger = logger,
-            pane = PANE,
-        )
-        // Fall back to the institution picker for now
-        navigationManager.tryNavigateTo(InstitutionPicker(referrer))
+    private fun openBankAuthRepair(
+        institution: FinancialConnectionsInstitution?,
+        authorization: String?,
+        referrer: Pane,
+    ) {
+        if (institution != null && authorization != null) {
+            updateLocalManifest {
+                it.copy(activeInstitution = institution)
+            }
+
+            pendingRepairRepository.set(authorization)
+            navigationManager.tryNavigateTo(Destination.BankAuthRepair(referrer))
+        } else {
+            val missingAuth = authorization == null
+            val missingInstitution = institution == null
+            eventTracker.logError(
+                extraMessage = "Unable to open repair flow " +
+                    "(missing auth: $missingAuth, missing institution: $missingInstitution).",
+                error = UnclassifiedError("UpdateRepairAccountError"),
+                logger = logger,
+                pane = PANE,
+            )
+            // Fall back to the institution picker
+            navigationManager.tryNavigateTo(InstitutionPicker(referrer))
+        }
     }
 
     private fun openPartnerAuth(

financial-connections/src/main/java/com/stripe/android/financialconnections/features/attachpayment/AttachPaymentViewModel.kt

@@ -9,10 +9,13 @@ import com.stripe.android.financialconnections.analytics.FinancialConnectionsAna
 import com.stripe.android.financialconnections.analytics.FinancialConnectionsAnalyticsTracker
 import com.stripe.android.financialconnections.analytics.logError
 import com.stripe.android.financialconnections.di.FinancialConnectionsSheetNativeComponent
+import com.stripe.android.financialconnections.domain.CachedPartnerAccount
 import com.stripe.android.financialconnections.domain.GetCachedAccounts
 import com.stripe.android.financialconnections.domain.GetOrFetchSync
+import com.stripe.android.financialconnections.domain.IsNetworkingRelinkSession
 import com.stripe.android.financialconnections.domain.NativeAuthFlowCoordinator
 import com.stripe.android.financialconnections.domain.PollAttachPaymentAccount
+import com.stripe.android.financialconnections.model.FinancialConnectionsSessionManifest
 import com.stripe.android.financialconnections.model.FinancialConnectionsSessionManifest.Pane
 import com.stripe.android.financialconnections.model.LinkAccountSessionPaymentAccount
 import com.stripe.android.financialconnections.model.PaymentAccountParams
@@ -41,7 +44,8 @@ internal class AttachPaymentViewModel @AssistedInject constructor(
     private val getCachedAccounts: GetCachedAccounts,
     private val navigationManager: NavigationManager,
     private val getOrFetchSync: GetOrFetchSync,
-    private val logger: Logger
+    private val logger: Logger,
+    private val isNetworkingRelinkSession: IsNetworkingRelinkSession,
 ) : FinancialConnectionsViewModel<AttachPaymentState>(initialState, nativeAuthFlowCoordinator) {
 
     init {
@@ -60,15 +64,8 @@ internal class AttachPaymentViewModel @AssistedInject constructor(
                     params = PaymentAccountParams.LinkedAccount(requireNotNull(id))
                 )
             }
-            if (manifest.isNetworkingUserFlow == true && manifest.accountholderIsLinkConsumer == true) {
-                result.networkingSuccessful?.let {
-                    successContentRepository.set(
-                        message = PluralId(
-                            value = R.plurals.stripe_success_pane_desc_link_success,
-                            count = accounts.size
-                        )
-                    )
-                }
+            if (result.networkingSuccessful == true) {
+                setSuccessMessageIfNecessary(manifest, accounts)
             }
             eventTracker.track(
                 PollAttachPaymentsSucceeded(
@@ -91,6 +88,20 @@ internal class AttachPaymentViewModel @AssistedInject constructor(
         )
     }
 
+    private fun setSuccessMessageIfNecessary(
+        manifest: FinancialConnectionsSessionManifest,
+        accounts: List<CachedPartnerAccount>,
+    ) {
+        if (manifest.canSetCustomLinkSuccessMessage && !isNetworkingRelinkSession()) {
+            successContentRepository.set(
+                message = PluralId(
+                    value = R.plurals.stripe_success_pane_desc_link_success,
+                    count = accounts.size
+                )
+            )
+        }
+    }
+
     private fun logErrors() {
         onAsync(
             AttachPaymentState::linkPaymentAccount,
@@ -132,3 +143,6 @@ internal class AttachPaymentViewModel @AssistedInject constructor(
 internal data class AttachPaymentState(
     val linkPaymentAccount: Async<LinkAccountSessionPaymentAccount> = Uninitialized
 )
+
+private val FinancialConnectionsSessionManifest.canSetCustomLinkSuccessMessage: Boolean
+    get() = isNetworkingUserFlow == true && accountholderIsLinkConsumer == true