feat(sample): add full-customization example covering every auth screen

FirebaseSwiftUI/FirebaseAuthSwiftUI/Sources/Views/AuthPickerContent.swift

@@ -17,31 +17,36 @@ import FirebaseAuthUIComponents
 import SwiftUI
 
 /// An ``AuthPickerView`` with custom sheet content, returned by
-/// `AuthPickerView`'s `pickerContent` and ``AuthPickerView/pickerDestination(_:)`` modifiers.
+/// `AuthPickerView`'s `pickerContent`, ``AuthPickerView/pickerDestination(_:)`` and
+/// ``AuthPickerView/signedInContent(_:)`` modifiers.
 ///
 /// It owns the authentication sheet, its `NavigationStack`, error alerts, and MFA and
-/// account-conflict handling; only the first screen and the pushed screens come from the slots.
-/// You don't create it directly.
+/// account-conflict handling; only the first screen, the signed-in screen and the pushed screens
+/// come from the slots. You don't create it directly.
 @MainActor
 public struct AuthPickerContent<
   Content: View,
   PickerContent: View,
-  DestinationContent: View
+  DestinationContent: View,
+  SignedInContent: View
 >: View {
   init(content: @escaping () -> Content,
        pickerContent: @escaping (AuthMethodPickerState) -> PickerContent,
        destination: @escaping (AuthView) -> DestinationContent,
+       signedInContent: @escaping (SignedInContentState) -> SignedInContent,
        showsPickerTitle: Bool = true) {
     self.content = content
     self.pickerContent = pickerContent
     self.destination = destination
+    self.signedInContent = signedInContent
     self.showsPickerTitle = showsPickerTitle
   }
 
   @Environment(AuthService.self) private var authService
   private let content: () -> Content
   private let pickerContent: (AuthMethodPickerState) -> PickerContent
   private let destination: (AuthView) -> DestinationContent
+  private let signedInContent: (SignedInContentState) -> SignedInContent
   /// `false` when the whole first screen is custom, so the library's large title doesn't sit
   /// above the app's own heading. The close button stays so the sheet can still be dismissed.
   private let showsPickerTitle: Bool
@@ -89,13 +94,13 @@ public struct AuthPickerContent<
       }
   }
 
-  /// The sheet's first screen. The slot only replaces the signed-out screen, so signed-in users
-  /// always reach ``SignedInView`` and the progress overlay always shows while signing in.
+  /// The sheet's first screen: the signed-in slot for signed-in users, otherwise the picker slot.
+  /// The progress overlay always shows while signing in, whichever slot is on screen.
   @ViewBuilder
   private var root: some View {
     VStack {
       if authService.authenticationState == .authenticated {
-        SignedInView()
+        SignedInScreen(content: signedInContent)
       } else {
         AuthMethodPickerStateReader { state in
           pickerContent(state)
@@ -122,11 +127,12 @@ public struct AuthPickerContent<
   /// the no-argument `pickerContent` on ``AuthPickerView``.
   public func pickerContent<NewPickerContent: View>(
     @ViewBuilder _ pickerContent: @escaping () -> NewPickerContent
-  ) -> AuthPickerContent<Content, NewPickerContent, DestinationContent> {
-    AuthPickerContent<Content, NewPickerContent, DestinationContent>(
+  ) -> AuthPickerContent<Content, NewPickerContent, DestinationContent, SignedInContent> {
+    AuthPickerContent<Content, NewPickerContent, DestinationContent, SignedInContent>(
       content: content,
       pickerContent: { _ in pickerContent() },
       destination: destination,
+      signedInContent: signedInContent,
       showsPickerTitle: true
     )
   }
@@ -135,11 +141,12 @@ public struct AuthPickerContent<
   /// ``AuthMethodPickerState``. See the matching `pickerContent` on ``AuthPickerView``.
   public func pickerContent<NewPickerContent: View>(
     @ViewBuilder _ pickerContent: @escaping (AuthMethodPickerState) -> NewPickerContent
-  ) -> AuthPickerContent<Content, NewPickerContent, DestinationContent> {
-    AuthPickerContent<Content, NewPickerContent, DestinationContent>(
+  ) -> AuthPickerContent<Content, NewPickerContent, DestinationContent, SignedInContent> {
+    AuthPickerContent<Content, NewPickerContent, DestinationContent, SignedInContent>(
       content: content,
       pickerContent: pickerContent,
       destination: destination,
+      signedInContent: signedInContent,
       showsPickerTitle: false
     )
   }
@@ -148,11 +155,26 @@ public struct AuthPickerContent<
   /// ``AuthPickerView/pickerDestination(_:)``.
   public func pickerDestination<NewDestinationContent: View>(
     @ViewBuilder _ destination: @escaping (AuthView) -> NewDestinationContent
-  ) -> AuthPickerContent<Content, PickerContent, NewDestinationContent> {
-    AuthPickerContent<Content, PickerContent, NewDestinationContent>(
+  ) -> AuthPickerContent<Content, PickerContent, NewDestinationContent, SignedInContent> {
+    AuthPickerContent<Content, PickerContent, NewDestinationContent, SignedInContent>(
       content: content,
       pickerContent: pickerContent,
       destination: destination,
+      signedInContent: signedInContent,
+      showsPickerTitle: showsPickerTitle
+    )
+  }
+
+  /// Replaces the signed-in screen of the authentication sheet. See
+  /// ``AuthPickerView/signedInContent(_:)``.
+  public func signedInContent<NewSignedInContent: View>(
+    @ViewBuilder _ signedInContent: @escaping (SignedInContentState) -> NewSignedInContent
+  ) -> AuthPickerContent<Content, PickerContent, DestinationContent, NewSignedInContent> {
+    AuthPickerContent<Content, PickerContent, DestinationContent, NewSignedInContent>(
+      content: content,
+      pickerContent: pickerContent,
+      destination: destination,
+      signedInContent: signedInContent,
       showsPickerTitle: showsPickerTitle
     )
   }

FirebaseSwiftUI/FirebaseAuthSwiftUI/Sources/Views/AuthPickerView.swift

@@ -20,8 +20,9 @@ import SwiftUI
 /// Wraps your app content and presents the authentication sheet whenever
 /// `AuthService.isPresented` is `true`.
 ///
-/// Use the `pickerContent` modifiers and ``pickerDestination(_:)`` to replace the sheet's first screen or
-/// any pushed screen while the library keeps driving navigation, MFA and account conflicts.
+/// Use the `pickerContent` modifiers, ``pickerDestination(_:)`` and ``signedInContent(_:)`` to
+/// replace the sheet's first screen, any pushed screen or the signed-in screen while the library
+/// keeps driving navigation, MFA and account conflicts.
 ///
 /// Example:
 /// ```swift
@@ -42,12 +43,14 @@ public struct AuthPickerView<Content: View>: View {
     AuthPickerContent(
       content: content,
       pickerContent: { _ in DefaultAuthPickerContent() },
-      destination: { DefaultAuthPickerDestination(screen: $0) }
+      destination: { DefaultAuthPickerDestination(screen: $0) },
+      signedInContent: { DefaultSignedInContent(state: $0) }
     )
   }
 
-  /// Replaces the signed-out first screen of the authentication sheet. The library still shows
-  /// ``SignedInView`` to signed-in users and its progress overlay while signing in.
+  /// Replaces the signed-out first screen of the authentication sheet. Signed-in users still see
+  /// ``DefaultSignedInContent`` unless you replace it with ``signedInContent(_:)``, and the
+  /// progress overlay still shows while signing in.
   ///
   /// ```swift
   /// AuthPickerView { authenticatedApp }
@@ -58,11 +61,17 @@ public struct AuthPickerView<Content: View>: View {
   /// ```
   public func pickerContent<NewPickerContent: View>(
     @ViewBuilder _ pickerContent: @escaping () -> NewPickerContent
-  ) -> AuthPickerContent<Content, NewPickerContent, DefaultAuthPickerDestination> {
+  ) -> AuthPickerContent<
+    Content,
+    NewPickerContent,
+    DefaultAuthPickerDestination,
+    DefaultSignedInContent
+  > {
     AuthPickerContent(
       content: content,
       pickerContent: { _ in pickerContent() },
-      destination: { DefaultAuthPickerDestination(screen: $0) }
+      destination: { DefaultAuthPickerDestination(screen: $0) },
+      signedInContent: { DefaultSignedInContent(state: $0) }
     )
   }
 
@@ -71,9 +80,9 @@ public struct AuthPickerView<Content: View>: View {
   /// account conflicts and errors are handled the same as the default buttons.
   ///
   /// Your screen replaces the logo, email sign-in, terms and the large "Sign in" title (the close
-  /// button stays), so show your own heading, email entry (for example one built around
-  /// ``EmailAuthView``) and any legal text you need. To keep the stock screen and only lay out
-  /// the provider buttons, use ``DefaultAuthPickerContent/init(authMethodPicker:)``.
+  /// button stays), so show your own heading, email entry (for example ``EmailAuthScreen``) and
+  /// any legal text you need. To keep the stock screen and only lay out the provider buttons, use
+  /// ``DefaultAuthPickerContent/init(authMethodPicker:)``.
   ///
   /// Example:
   /// ```swift
@@ -84,11 +93,17 @@ public struct AuthPickerView<Content: View>: View {
   /// ```
   public func pickerContent<NewPickerContent: View>(
     @ViewBuilder _ pickerContent: @escaping (AuthMethodPickerState) -> NewPickerContent
-  ) -> AuthPickerContent<Content, NewPickerContent, DefaultAuthPickerDestination> {
+  ) -> AuthPickerContent<
+    Content,
+    NewPickerContent,
+    DefaultAuthPickerDestination,
+    DefaultSignedInContent
+  > {
     AuthPickerContent(
       content: content,
       pickerContent: pickerContent,
       destination: { DefaultAuthPickerDestination(screen: $0) },
+      signedInContent: { DefaultSignedInContent(state: $0) },
       showsPickerTitle: false
     )
   }
@@ -115,12 +130,45 @@ public struct AuthPickerView<Content: View>: View {
   ) -> AuthPickerContent<
     Content,
     DefaultAuthPickerContent<DefaultAuthMethodPicker>,
-    NewDestinationContent
+    NewDestinationContent,
+    DefaultSignedInContent
+  > {
+    AuthPickerContent(
+      content: content,
+      pickerContent: { _ in DefaultAuthPickerContent() },
+      destination: destination,
+      signedInContent: { DefaultSignedInContent(state: $0) }
+    )
+  }
+
+  /// Replaces the screen the authentication sheet shows to signed-in users. Your content receives
+  /// a ``SignedInContentState``: the library still signs out, deletes the account behind its
+  /// confirmation and reauthentication, sends the verification email and navigates. Wrap
+  /// ``DefaultSignedInContent`` to keep the stock markup.
+  ///
+  /// ```swift
+  /// AuthPickerView { authenticatedApp }
+  ///   .signedInContent { state in
+  ///     MyAccountView(
+  ///       email: state.user?.email,
+  ///       onDeleteAccount: state.deleteAccount,
+  ///       onSignOut: state.signOut
+  ///     )
+  ///   }
+  /// ```
+  public func signedInContent<NewSignedInContent: View>(
+    @ViewBuilder _ signedInContent: @escaping (SignedInContentState) -> NewSignedInContent
+  ) -> AuthPickerContent<
+    Content,
+    DefaultAuthPickerContent<DefaultAuthMethodPicker>,
+    DefaultAuthPickerDestination,
+    NewSignedInContent
   > {
     AuthPickerContent(
       content: content,
       pickerContent: { _ in DefaultAuthPickerContent() },
-      destination: destination
+      destination: { DefaultAuthPickerDestination(screen: $0) },
+      signedInContent: signedInContent
     )
   }
 }