[fenix] 🎲 For https://github.com/mozilla-mobile/fenix/issues/10332 - Adds undo snackbar when closing tabs in the tab tray (https://github.com/mozilla-mobile/fenix/pull/10456)

app/src/main/java/org/mozilla/fenix/tabtray/TabTrayFragment.kt

@@ -12,6 +12,7 @@ import android.view.View
 import androidx.core.view.isVisible
 import mozilla.components.concept.engine.prompt.ShareData
 import androidx.fragment.app.Fragment
+import androidx.lifecycle.lifecycleScope
 import mozilla.components.feature.tabs.tabstray.TabsFeature
 import kotlinx.android.synthetic.main.fragment_tab_tray.tabsTray
 import kotlinx.android.synthetic.main.fragment_tab_tray.view.*
@@ -32,6 +33,7 @@ import org.mozilla.fenix.ext.components
 import org.mozilla.fenix.ext.nav
 import org.mozilla.fenix.ext.sessionsOfType
 import org.mozilla.fenix.ext.showToolbar
+import org.mozilla.fenix.utils.allowUndo
 
 @SuppressWarnings("TooManyFunctions", "LargeClass")
 class TabTrayFragment : Fragment(R.layout.fragment_tab_tray), TabsTray.Observer, UserInteractionHandler {
@@ -130,16 +132,47 @@ class TabTrayFragment : Fragment(R.layout.fragment_tab_tray), TabsTray.Observer,
                 true
             }
             R.id.tab_tray_close_menu_item -> {
-                val tabs = getListOfSessions()
+                closeAllTabs()
-                tabs.forEach {
-                    sessionManager.remove(it)
-                }
                 true
             }
             else -> super.onOptionsItemSelected(item)
         }
     }
 
+    private fun closeAllTabs() {
+        val tabs = getListOfSessions()
+
+        val selectedIndex = sessionManager
+            .selectedSession?.let { sessionManager.sessions.indexOf(it) } ?: 0
+
+        val snapshot = tabs
+            .map(sessionManager::createSessionSnapshot)
+            .map { it.copy(engineSession = null, engineSessionState = it.engineSession?.saveState()) }
+            .let { SessionManager.Snapshot(it, selectedIndex) }
+
+        tabs.forEach {
+            sessionManager.remove(it)
+        }
+
+        val isPrivate = (activity as HomeActivity).browsingModeManager.mode.isPrivate
+        val snackbarMessage = if (isPrivate) {
+            getString(R.string.snackbar_private_tabs_closed)
+        } else {
+            getString(R.string.snackbar_tabs_closed)
+        }
+
+        viewLifecycleOwner.lifecycleScope.allowUndo(
+            requireView(),
+            snackbarMessage,
+            getString(R.string.snackbar_deleted_undo),
+            {
+                sessionManager.restore(snapshot)
+            },
+            operation = { },
+            anchorView = view?.tab_tray_controls
+        )
+    }
+
     private fun saveToCollection() {
         val tabs = getListOfSessions()
         val tabIds = tabs.map { it.id }.toList().toTypedArray()
@@ -169,15 +202,15 @@ class TabTrayFragment : Fragment(R.layout.fragment_tab_tray), TabsTray.Observer,
     override fun onStart() {
         super.onStart()
 
-        tabsFeature?.start()
         tabsTray.register(this)
+        tabsFeature?.start()
     }
 
     override fun onStop() {
         super.onStop()
 
-        tabsFeature?.stop()
         tabsTray.unregister(this)
+        tabsFeature?.stop()
     }
 
     override fun onBackPressed(): Boolean {
@@ -196,7 +229,30 @@ class TabTrayFragment : Fragment(R.layout.fragment_tab_tray), TabsTray.Observer,
     }
 
     override fun onTabClosed(tab: Tab) {
-        // noop
+        val snapshot = sessionManager
+            .findSessionById(tab.id)?.let {
+            sessionManager.createSessionSnapshot(it)
+        } ?: return
+
+        val state = snapshot.engineSession?.saveState()
+        val isSelected = tab.id == requireComponents.core.store.state.selectedTabId ?: false
+
+        val snackbarMessage = if (snapshot.session.private) {
+            getString(R.string.snackbar_private_tab_closed)
+        } else {
+            getString(R.string.snackbar_tab_closed)
+        }
+
+        viewLifecycleOwner.lifecycleScope.allowUndo(
+            requireView(),
+            snackbarMessage,
+            getString(R.string.snackbar_deleted_undo),
+            {
+                sessionManager.add(snapshot.session, isSelected, engineSessionState = state)
+            },
+            operation = { },
+            anchorView = view?.tab_tray_controls
+        )
     }
 
     override fun onTabSelected(tab: Tab) {

app/src/main/res/layout/fragment_tab_tray.xml

@@ -2,9 +2,13 @@
 <!-- This Source Code Form is subject to the terms of the Mozilla Public
    - License, v. 2.0. If a copy of the MPL was not distributed with this
    - file, You can obtain one at http://mozilla.org/MPL/2.0/. -->
-<androidx.constraintlayout.widget.ConstraintLayout xmlns:android="http://schemas.android.com/apk/res/android"
+<androidx.coordinatorlayout.widget.CoordinatorLayout
+    xmlns:android="http://schemas.android.com/apk/res/android"
     xmlns:app="http://schemas.android.com/apk/res-auto"
     xmlns:mozac="http://schemas.android.com/apk/res-auto"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent">
+    <androidx.constraintlayout.widget.ConstraintLayout
         android:layout_width="match_parent"
         android:layout_height="match_parent"
         android:background="?tabTrayItemBackground">
@@ -108,3 +112,4 @@
             app:layout_constraintEnd_toEndOf="@+id/tab_tray_controls"
             app:layout_constraintStart_toEndOf="@+id/tab_tray_go_home" />
     </androidx.constraintlayout.widget.ConstraintLayout>
+</androidx.coordinatorlayout.widget.CoordinatorLayout>