initialize sae

Author: gbirch-stripe

StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/Checkout/Checkout.swift

@@ -59,6 +59,9 @@ public final class Checkout: ObservableObject {
     /// The CurrencySelectorElement for this Checkout instance, when Adaptive Pricing is available.
     private var currencySelectorElement: CurrencySelectorElement?
 
+    /// The ShippingAddressElement for this Checkout instance.
+    private var shippingAddressElement: ShippingAddressElement!
+
     // TODO(gbirch) TODO(porter) remove this nonisolatedSession
     //  once MPE is properly MainActor isolated
     /// A snapshot of the current ``session`` accessible from non-MainActor contexts.
@@ -133,6 +136,7 @@ public final class Checkout: ObservableObject {
             try await applyDefaults()
 
             // Load elements
+            self.shippingAddressElement = ShippingAddressElement(checkout: self)
             self.paymentElement = try await PaymentElement(checkout: self)
             let sessionSource = CheckoutSessionSource(initialSession: session, sessionPublisher: $session)
             self.expressCheckoutElement = ExpressCheckoutElement(
@@ -287,6 +291,11 @@ public final class Checkout: ObservableObject {
         return currencySelectorElement
     }
 
+    /// Returns the ShippingAddressElement for this Checkout instance.
+    public func getShippingAddressElement() -> ShippingAddressElement {
+        return shippingAddressElement
+    }
+
     // MARK: - Confirm
 
     /// Use this method to confirm the Checkout Session.

StripePaymentSheet/StripePaymentSheet/Source/PaymentSheet/Checkout/Shipping Address Element/ShippingAddressElement.swift

@@ -8,4 +8,27 @@
 @MainActor
 @_spi(STP)
 @_spi(ReactNativeSDK)
-public final class ShippingAddressElement {}
+public final class ShippingAddressElement {
+
+    private(set) var addressViewController: AddressViewController!
+
+    init(checkout: Checkout) {
+        let configuration = checkout.configuration.shippingAddressElement.makeAddressViewControllerConfiguration(
+            shippingAddress: checkout.session.shippingAddress,
+            allowedCountries: checkout.session.allowedShippingCountries,
+            apiClient: checkout.apiClient,
+            useAutocompleteEndpoints: checkout.session.elementsSession.shouldUseAutocompleteProxyEndpoints
+        )
+        addressViewController = AddressViewController(
+            configuration: configuration,
+            delegate: self
+        )
+    }
+}
+
+extension ShippingAddressElement: AddressViewControllerDelegate {
+    public func addressViewControllerDidFinish(
+        _ addressViewController: AddressViewController,
+        with address: AddressViewController.AddressDetails?
+    ) {}
+}

StripePaymentSheet/StripePaymentSheetTests/PaymentSheet/Checkout/ShippingAddressElementConfigurationTests.swift

@@ -11,6 +11,55 @@ import XCTest
 
 @MainActor
 final class ShippingAddressElementConfigurationTests: XCTestCase {
+    func testCheckoutInitializesShippingAddressElement() async throws {
+        // Given
+        var sessionJSON = CheckoutTestHelpers.openSessionJSON
+        sessionJSON["shipping_address_collection"] = ["allowed_countries": ["US", "CA"]]
+        var elementsSessionJSON = CheckoutTestHelpers.minimalElementsSessionJSON
+        elementsSessionJSON["flags"] = ["ocs_mobile_should_use_autocomplete_proxy_endpoints": true]
+        sessionJSON["elements_session"] = elementsSessionJSON
+        let apiResponse = try XCTUnwrap(PaymentPagesAPIResponse.decodedObject(fromAPIResponse: sessionJSON))
+
+        var configuration = Checkout.Configuration(
+            clientSecret: "cs_test_123_secret_abc",
+            returnURL: "stripe-ios-test://checkout-return"
+        )
+        configuration.shippingAddressElement.title = "Delivery address"
+        configuration.shippingAddressElement.buttonTitle = "Save delivery address"
+        configuration.shippingAddressElement.appearance.colors.primary = .purple
+        var shippingDetails = Checkout.Configuration.Defaults.ShippingDetails()
+        shippingDetails.name = "Jenny Rosen"
+        shippingDetails.address = .init(
+            country: "US",
+            line1: "510 Townsend St.",
+            city: "San Francisco",
+            state: "CA",
+            postalCode: "94103"
+        )
+        configuration.defaults.shippingDetails = shippingDetails
+        configuration = CheckoutTestHelpers.makeConfiguration(
+            apiResponse: apiResponse,
+            configuration: configuration
+        )
+
+        // When
+        let checkout = try await Checkout(configuration: configuration)
+        let shippingAddressElement = checkout.getShippingAddressElement()
+        let addressConfiguration = shippingAddressElement.addressViewController.configuration
+
+        // Then
+        XCTAssertTrue(shippingAddressElement === checkout.getShippingAddressElement())
+        XCTAssertEqual(addressConfiguration.title, "Delivery address")
+        XCTAssertEqual(addressConfiguration.buttonTitle, "Save delivery address")
+        XCTAssertEqual(addressConfiguration.appearance.colors.primary, .purple)
+        XCTAssertEqual(addressConfiguration.allowedCountries, ["US", "CA"])
+        XCTAssertEqual(addressConfiguration.defaultValues.name, "Jenny Rosen")
+        XCTAssertEqual(addressConfiguration.defaultValues.address.country, "US")
+        XCTAssertEqual(addressConfiguration.defaultValues.address.line1, "510 Townsend St.")
+        XCTAssertTrue(addressConfiguration.apiClient === configuration.apiClient)
+        XCTAssertTrue(addressConfiguration.useAutocompleteEndpoints)
+    }
+
     func testMakeAddressViewControllerConfiguration() {
         // Given
         let apiClient = STPAPIClient(publishableKey: "pk_test_123")