Fix #8146: Fixes Tab Selection and mode switching (#8147)

brave · Sep 25, 2023 · 6b03a95 · 6b03a95
1 parent 825a7a3
commit 6b03a95
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Sources/Brave/Frontend/Browser/Tabs/TabTray/TabTrayController.swift b/Sources/Brave/Frontend/Browser/Tabs/TabTray/TabTrayController.swift
@@ -648,11 +648,12 @@ class TabTrayController: AuthenticationController {
           tabManager.addTabAndSelect(isPrivate: true)
         }
 
-        let privateModeTabSelected = tabManager.allTabs[safe: tabManager.privateTabSelectedIndex]
+        let privateModeTabSelected = tabManager.tabsForCurrentMode[safe: tabManager.privateTabSelectedIndex] ?? tabManager.tabsForCurrentMode.last
 
         if Preferences.Privacy.persistentPrivateBrowsing.value {
           tabManager.selectTab(privateModeTabSelected)
         }
+
         tabTrayView.hidePrivateModeInfo()
         tabTrayView.collectionView.reloadData()
 
@@ -668,8 +669,8 @@ class TabTrayController: AuthenticationController {
 
       // When you go back from private mode, a previous current tab is selected
       // Reloding the collection view in order to mark the selecte the tab
-      let normalModeTabSelected = tabManager.allTabs[safe: tabManager.normalTabSelectedIndex]
-      
+      let normalModeTabSelected = tabManager.tabsForCurrentMode[safe: tabManager.normalTabSelectedIndex] ?? tabManager.tabsForCurrentMode.last
+
       tabManager.selectTab(normalModeTabSelected)
       tabTrayView.collectionView.reloadData()