1 files changed, 121 insertions, 0 deletions
diff --git a/mobile/android/geckoview_example/src/main/java/org/mozilla/geckoview_example/TabSessionManager.java b/mobile/android/geckoview_example/src/main/java/org/mozilla/geckoview_example/TabSessionManager.java
new file mode 100644
index 0000000000..83bd4fa14e
--- /dev/null
+++ b/mobile/android/geckoview_example/src/main/java/org/mozilla/geckoview_example/TabSessionManager.java
@@ -0,0 +1,121 @@
+/* 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/. */
+
+package org.mozilla.geckoview_example;
+
+import androidx.annotation.Nullable;
+import java.util.ArrayList;
+import org.mozilla.geckoview.GeckoSession;
+import org.mozilla.geckoview.GeckoSessionSettings;
+import org.mozilla.geckoview.WebExtension;
+
+public class TabSessionManager {
+  private static ArrayList<TabSession> mTabSessions = new ArrayList<>();
+  private int mCurrentSessionIndex = 0;
+  private TabObserver mTabObserver;
+  private boolean mTrackingProtection;
+
+  public interface TabObserver {
+    void onCurrentSession(TabSession session);
+  }
+
+  public TabSessionManager() {}
+
+  public void unregisterWebExtension() {
+    for (final TabSession session : mTabSessions) {
+      session.action = null;
+    }
+  }
+
+  public void setWebExtensionDelegates(
+      WebExtension extension,
+      WebExtension.ActionDelegate actionDelegate,
+      WebExtension.SessionTabDelegate tabDelegate) {
+    for (final TabSession session : mTabSessions) {
+      final WebExtension.SessionController sessionController = session.getWebExtensionController();
+      sessionController.setActionDelegate(extension, actionDelegate);
+      sessionController.setTabDelegate(extension, tabDelegate);
+    }
+  }
+
+  public void setUseTrackingProtection(boolean trackingProtection) {
+    if (trackingProtection == mTrackingProtection) {
+      return;
+    }
+    mTrackingProtection = trackingProtection;
+
+    for (final TabSession session : mTabSessions) {
+      session.getSettings().setUseTrackingProtection(trackingProtection);
+    }
+  }
+
+  public void setTabObserver(TabObserver observer) {
+    mTabObserver = observer;
+  }
+
+  public void addSession(TabSession session) {
+    mTabSessions.add(session);
+  }
+
+  public TabSession getSession(int index) {
+    if (index >= mTabSessions.size() || index < 0) {
+      return null;
+    }
+    return mTabSessions.get(index);
+  }
+
+  public TabSession getCurrentSession() {
+    return getSession(mCurrentSessionIndex);
+  }
+
+  public TabSession getSession(GeckoSession session) {
+    int index = mTabSessions.indexOf(session);
+    if (index == -1) {
+      return null;
+    }
+    return getSession(index);
+  }
+
+  public void setCurrentSession(TabSession session) {
+    int index = mTabSessions.indexOf(session);
+    if (index == -1) {
+      mTabSessions.add(session);
+      index = mTabSessions.size() - 1;
+    }
+    mCurrentSessionIndex = index;
+
+    if (mTabObserver != null) {
+      mTabObserver.onCurrentSession(session);
+    }
+  }
+
+  private boolean isCurrentSession(TabSession session) {
+    return session == getCurrentSession();
+  }
+
+  public void closeSession(@Nullable TabSession session) {
+    if (session == null) {
+      return;
+    }
+    if (isCurrentSession(session) && mCurrentSessionIndex == mTabSessions.size() - 1) {
+      --mCurrentSessionIndex;
+    }
+    session.close();
+    mTabSessions.remove(session);
+  }
+
+  public TabSession newSession(GeckoSessionSettings settings) {
+    TabSession tabSession = new TabSession(settings);
+    mTabSessions.add(tabSession);
+    return tabSession;
+  }
+
+  public int sessionCount() {
+    return mTabSessions.size();
+  }
+
+  public ArrayList<TabSession> getSessions() {
+    return mTabSessions;
+  }
+}