Add Link warmup pane for deferred intents

Author: tillh-stripe

StripeFinancialConnections/StripeFinancialConnections/Source/Native/Consent/ConsentDataSource.swift

@@ -14,7 +14,12 @@ protocol ConsentDataSource: AnyObject {
     var merchantLogo: [String]? { get }
     var analyticsClient: FinancialConnectionsAnalyticsClient { get }
 
-    func markConsentAcquired() -> Promise<FinancialConnectionsSessionManifest>
+    func markConsentAcquired() -> Future<ConsentAcquiredResult>
+}
+
+struct ConsentAcquiredResult {
+    var manifest: FinancialConnectionsSessionManifest
+    var nextPane: FinancialConnectionsSessionManifest.NextPane?
 }
 
 final class ConsentDataSourceImplementation: ConsentDataSource {
@@ -25,24 +30,62 @@ final class ConsentDataSourceImplementation: ConsentDataSource {
     private let apiClient: any FinancialConnectionsAPI
     private let clientSecret: String
     let analyticsClient: FinancialConnectionsAnalyticsClient
+    private let elementsSessionContext: ElementsSessionContext?
 
     init(
         manifest: FinancialConnectionsSessionManifest,
         consent: FinancialConnectionsConsent,
         merchantLogo: [String]?,
         apiClient: any FinancialConnectionsAPI,
         clientSecret: String,
-        analyticsClient: FinancialConnectionsAnalyticsClient
+        analyticsClient: FinancialConnectionsAnalyticsClient,
+        elementsSessionContext: ElementsSessionContext?
     ) {
         self.manifest = manifest
         self.consent = consent
         self.merchantLogo = merchantLogo
         self.apiClient = apiClient
         self.clientSecret = clientSecret
         self.analyticsClient = analyticsClient
+        self.elementsSessionContext = elementsSessionContext
     }
 
-    func markConsentAcquired() -> Promise<FinancialConnectionsSessionManifest> {
-        return apiClient.markConsentAcquired(clientSecret: clientSecret)
+    func markConsentAcquired() -> Future<ConsentAcquiredResult> {
+        return apiClient.markConsentAcquired(clientSecret: clientSecret).chained { [weak self] manifest in
+            guard let self, manifest.shouldLookupConsumerSession, let email = elementsSessionContext?.prefillDetails?.email else {
+                let result = ConsentAcquiredResult(manifest: manifest, nextPane: nil)
+                return Promise(value: result)
+            }
+            
+            let promise = Promise<ConsentAcquiredResult>()
+            
+            apiClient.consumerSessionLookup(
+                emailAddress: email,
+                clientSecret: clientSecret,
+                sessionId: manifest.id,
+                emailSource: .userAction,
+                useMobileEndpoints: manifest.verified,
+                pane: .consent
+            ).observe { lookupResult in
+                switch lookupResult {
+                case .success(let response):
+                    let nextPane: FinancialConnectionsSessionManifest.NextPane? = response.exists ? .networkingLinkLoginWarmup : nil
+                    let result = ConsentAcquiredResult(manifest: manifest, nextPane: nextPane)
+                    promise.resolve(with: result)
+                case .failure:
+                    let result = ConsentAcquiredResult(manifest: manifest, nextPane: nil)
+                    promise.resolve(with: result)
+                }
+            }
+            
+            return promise
+        }
+    }
+}
+
+private extension FinancialConnectionsSessionManifest {
+    
+    var shouldLookupConsumerSession: Bool {
+        isLinkWithStripe == true && accountholderCustomerEmailAddress == nil
     }
 }

StripeFinancialConnections/StripeFinancialConnections/Source/Native/Consent/ConsentViewController.swift

@@ -19,7 +19,8 @@ protocol ConsentViewControllerDelegate: AnyObject {
     )
     func consentViewController(
         _ viewController: ConsentViewController,
-        didConsentWithManifest manifest: FinancialConnectionsSessionManifest
+        didConsentWithManifest manifest: FinancialConnectionsSessionManifest,
+        customNextPane: FinancialConnectionsSessionManifest.NextPane?
     )
 }
 
@@ -146,8 +147,8 @@ class ConsentViewController: UIViewController {
             .observe(on: .main) { [weak self] result in
                 guard let self = self else { return }
                 switch result {
-                case .success(let manifest):
-                    self.delegate?.consentViewController(self, didConsentWithManifest: manifest)
+                case .success(let result):
+                    self.delegate?.consentViewController(self, didConsentWithManifest: result.manifest, customNextPane: result.nextPane)
                 case .failure(let error):
                     // we display no errors on failure
                     self.dataSource

StripeFinancialConnections/StripeFinancialConnections/Source/Native/NativeFlowController.swift

@@ -696,7 +696,8 @@ extension NativeFlowController: ConsentViewControllerDelegate {
 
     func consentViewController(
         _ viewController: ConsentViewController,
-        didConsentWithManifest manifest: FinancialConnectionsSessionManifest
+        didConsentWithManifest manifest: FinancialConnectionsSessionManifest,
+        customNextPane: FinancialConnectionsSessionManifest.NextPane?
     ) {
         delegate?.nativeFlowController(
             self,
@@ -705,7 +706,7 @@ extension NativeFlowController: ConsentViewControllerDelegate {
 
         dataManager.manifest = manifest
 
-        let nextPane = manifest.nextPane
+        let nextPane = customNextPane ?? manifest.nextPane
         if nextPane == .networkingLinkLoginWarmup {
             presentPaneAsSheet(nextPane)
         } else {
@@ -1453,7 +1454,8 @@ private func CreatePaneViewController(
                 merchantLogo: dataManager.merchantLogo,
                 apiClient: dataManager.apiClient,
                 clientSecret: dataManager.clientSecret,
-                analyticsClient: dataManager.analyticsClient
+                analyticsClient: dataManager.analyticsClient,
+                elementsSessionContext: dataManager.elementsSessionContext
             )
             let consentViewController = ConsentViewController(dataSource: consentDataSource)
             consentViewController.delegate = nativeFlowController
@@ -1540,9 +1542,10 @@ private func CreatePaneViewController(
     case .networkingLinkVerification:
         let accountholderCustomerEmailAddress = dataManager.manifest.accountholderCustomerEmailAddress
         let consumerSessionEmailAddress = dataManager.consumerSession?.emailAddress
-        if let accountholderCustomerEmailAddress = consumerSessionEmailAddress ?? accountholderCustomerEmailAddress {
+        let prefillEmailAddress = dataManager.elementsSessionContext?.prefillDetails?.email
+        if let email = consumerSessionEmailAddress ?? accountholderCustomerEmailAddress ?? prefillEmailAddress {
             let networkingLinkVerificationDataSource = NetworkingLinkVerificationDataSourceImplementation(
-                accountholderCustomerEmailAddress: accountholderCustomerEmailAddress,
+                accountholderCustomerEmailAddress: email,
                 manifest: dataManager.manifest,
                 apiClient: dataManager.apiClient,
                 clientSecret: dataManager.clientSecret,
@@ -1667,7 +1670,8 @@ private func CreatePaneViewController(
             apiClient: dataManager.apiClient,
             clientSecret: dataManager.clientSecret,
             analyticsClient: dataManager.analyticsClient,
-            nextPaneOrDrawerOnSecondaryCta: parameters?.nextPaneOrDrawerOnSecondaryCta
+            nextPaneOrDrawerOnSecondaryCta: parameters?.nextPaneOrDrawerOnSecondaryCta,
+            elementsSessionContext: dataManager.elementsSessionContext
         )
         let networkingLinkWarmupViewController = NetworkingLinkLoginWarmupViewController(
             dataSource: networkingLinkWarmupDataSource,

StripeFinancialConnections/StripeFinancialConnections/Source/Native/NetworkingLinkLoginWarmup/NetworkingLinkLoginWarmupDataSource.swift

@@ -11,6 +11,7 @@ import Foundation
 protocol NetworkingLinkLoginWarmupDataSource: AnyObject {
     var manifest: FinancialConnectionsSessionManifest { get }
     var analyticsClient: FinancialConnectionsAnalyticsClient { get }
+    var email: String? { get }
 
     func disableNetworking() -> Future<FinancialConnectionsSessionManifest>
 }
@@ -22,19 +23,26 @@ final class NetworkingLinkLoginWarmupDataSourceImplementation: NetworkingLinkLog
     private let clientSecret: String
     let analyticsClient: FinancialConnectionsAnalyticsClient
     private let nextPaneOrDrawerOnSecondaryCta: String?
+    private let elementsSessionContext: ElementsSessionContext?
+    
+    var email: String? {
+        manifest.accountholderCustomerEmailAddress ?? elementsSessionContext?.prefillDetails?.email
+    }
 
     init(
         manifest: FinancialConnectionsSessionManifest,
         apiClient: any FinancialConnectionsAPI,
         clientSecret: String,
         analyticsClient: FinancialConnectionsAnalyticsClient,
-        nextPaneOrDrawerOnSecondaryCta: String?
+        nextPaneOrDrawerOnSecondaryCta: String?,
+        elementsSessionContext: ElementsSessionContext?
     ) {
         self.manifest = manifest
         self.apiClient = apiClient
         self.clientSecret = clientSecret
         self.analyticsClient = analyticsClient
         self.nextPaneOrDrawerOnSecondaryCta = nextPaneOrDrawerOnSecondaryCta
+        self.elementsSessionContext = elementsSessionContext
     }
 
     func disableNetworking() -> Future<FinancialConnectionsSessionManifest> {

StripeFinancialConnections/StripeFinancialConnections/Source/Native/NetworkingLinkLoginWarmup/NetworkingLinkLoginWarmupViewController.swift

@@ -70,10 +70,8 @@ final class NetworkingLinkLoginWarmupViewController: SheetViewController {
                     "The subtitle/description of a screen where users are informed that they can sign-in-to Link."
                 ),
                 contentView: NetworkingLinkLoginWarmupBodyView(
-                    // `accountholderCustomerEmailAddress` should always be non-null, and
-                    // since the email is only used as a visual, it's not worth to throw an error
-                    // if it is null
-                    email: dataSource.manifest.accountholderCustomerEmailAddress ?? "you"
+                    // `email` should always be non-null, and since the email is only used as a visual, it's not worth to throw an error if it is null
+                    email: dataSource.email ?? "you"
                 )
             ),
             footerView: PaneLayoutView.createFooterView(