Fix missing actions on StackTab

Author: SubvertDev

Modules/Sources/AppFeature/Navigation/ProfileFlow/ProfileFlow.swift

@@ -31,8 +31,9 @@ public struct ProfileTab: View {
     public var body: some View {
         WithPerceptionTracking {
             switch store.case {
-            case let .loggedIn(store),
-                let .loggedOut(store):
+            case let .loggedIn(store):
+                StackTabView(store: store)
+            case let .loggedOut(store):
                 StackTabView(store: store)
             }
         }

Modules/Sources/AuthFeature/AuthScreen.swift

@@ -158,35 +158,8 @@ public struct AuthScreen: View {
                 }
             }
             ._toolbarTitleDisplayMode(.inline)
-            .toolbar {
-                // Profile is used as root in this case so we don't need close button
-                if store.openReason != .profile {
-                    ToolbarItem(placement: .topBarTrailing) {
-                        if #available(iOS 26, *) {
-                            Button(role: .close) {
-                                send(.closeButtonTapped)
-                            }
-                        } else {
-                            Button {
-                                send(.closeButtonTapped)
-                            } label: {
-                                Text("Close", bundle: .module)
-                                    .foregroundStyle(tintColor)
-                            }
-                        }
-                    }
-                }
-                
-                // We're showing app settings only if it's opened from profile tab
-                if store.openReason == .profile {
-                    ToolbarItem(placement: .topBarTrailing) {
-                        Button {
-                            send(.settingsButtonTapped)
-                        } label: {
-                            Image(systemSymbol: .gearshape)
-                        }
-                    }
-                }
+            .toolbar(id: "auth_toolbar") {
+                ToolbarButtons()
             }
             .alert($store.scope(state: \.alert, action: \.alert))
             .bind($store.focus, to: $focus)
@@ -208,6 +181,40 @@ public struct AuthScreen: View {
         }
     }
 
+    // MARK: - Toolbar Buttons
+    
+    @ToolbarContentBuilder
+    private func ToolbarButtons() -> some CustomizableToolbarContent {
+        // Profile is used as root in this case so we don't need close button
+        if store.openReason != .profile {
+            ToolbarItem(id: "auth_close_button", placement: .topBarTrailing) {
+                if #available(iOS 26, *) {
+                    Button(role: .close) {
+                        send(.closeButtonTapped)
+                    }
+                } else {
+                    Button {
+                        send(.closeButtonTapped)
+                    } label: {
+                        Text("Close", bundle: .module)
+                            .foregroundStyle(tintColor)
+                    }
+                }
+            }
+        }
+        
+        // We're showing app settings only if it's opened from profile tab
+        if store.openReason == .profile {
+            ToolbarItem(id: "auth_settings_button", placement: .topBarTrailing) {
+                Button {
+                    send(.settingsButtonTapped)
+                } label: {
+                    Image(systemSymbol: .gearshape)
+                }
+            }
+        }
+    }
+    
     // MARK: - Field
 
     @ViewBuilder

Modules/Sources/ProfileFeature/ProfileScreen.swift

@@ -70,7 +70,7 @@ public struct ProfileScreen: View {
             .alert($store.scope(state: \.destination?.alert, action: \.destination.alert))
             .navigationTitle(Text("Profile", bundle: .module))
             ._toolbarTitleDisplayMode(.large)
-            .toolbar {
+            .toolbar(id: "profile_toolbar") {
                 ToolbarButtons()
             }
             .onAppear {
@@ -82,9 +82,9 @@ public struct ProfileScreen: View {
     // MARK: - Toolbar Items
 
     @ToolbarContentBuilder
-    private func ToolbarButtons() -> some ToolbarContent {
+    private func ToolbarButtons() -> some CustomizableToolbarContent {
         if store.shouldShowToolbarButtons {
-            ToolbarItem {
+            ToolbarItem(id: "profile_logout_button") {
                 Button {
                     send(.logoutButtonTapped)
                 } label: {
@@ -96,7 +96,7 @@ public struct ProfileScreen: View {
                 ToolbarSpacer(.fixed)
             }
 
-            ToolbarItem {
+            ToolbarItem(id: "profile_settings_button") {
                 Button {
                     send(.settingsButtonTapped)
                 } label: {