Add Reopen in Container context menu for tab

src/js/background/index.html

@@ -11,6 +11,7 @@
       "js/background/badge.js",
       "js/background/identityState.js",
       "js/background/messageHandler.js",
+      "js/background/reopen.js",
     ]
 -->
     <script type="text/javascript" src="backgroundLogic.js"></script>
@@ -18,5 +19,6 @@
     <script type="text/javascript" src="badge.js"></script>
     <script type="text/javascript" src="identityState.js"></script>
     <script type="text/javascript" src="messageHandler.js"></script>
+    <script type="text/javascript" src="reopen.js"></script>
   </body>
 </html>

src/js/background/reopen.js

@@ -0,0 +1,69 @@
+const reopenIn = {
+  // Map from menuItemId to cookieStoreId.
+  cookieStoreIds: new Map(),
+
+  init() {
+    browser.menus.onShown.addListener(async (info, tab) => {
+      if (info.contexts.length !== 1) {
+        return;
+      }
+      if (info.contexts[0] !== "tab") {
+        return;
+      }
+
+      await this.rebuildMenu(tab);
+    });
+
+    browser.menus.onClicked.addListener((info, tab) => {
+      browser.tabs.create({
+        url: tab.url,
+        index: tab.index + 1,
+        cookieStoreId: this.cookieStoreIds.get(info.menuItemId)
+      });
+    });
+  },
+
+  async rebuildMenu(tab) {
+    browser.menus.removeAll();
+
+    const containers = await browser.contextualIdentities.query({});
+
+    const folderId = browser.menus.create({
+      title: "Reopen in Container",
+      contexts: ["tab"],
+    });
+
+    if (tab.cookieStoreId !== "firefox-default") {
+      const menuItemId = "openin-firefox-default";
+      this.cookieStoreIds.set(menuItemId, "firefox-default");
+      browser.menus.create({
+        id: menuItemId,
+        title: "No Container",
+        parentId: folderId,
+      });
+      browser.menus.create({
+        type: "separator",
+        parentId: folderId,
+      });
+    }
+
+    for (const [i, container] of containers.entries()) {
+      if (container.cookieStoreId === tab.cookieStoreId)
+        continue;
+
+      const menuItemId = "openin-" + i;
+      this.cookieStoreIds.set(menuItemId, container.cookieStoreId);
+      browser.menus.create({
+        id: menuItemId,
+        title: container.name,
+        icons: {
+          "16": container.iconUrl,
+        },
+        parentId: folderId,
+      });
+    }
+
+    browser.menus.refresh();
+  },
+};
+reopenIn.init();

src/manifest.json

@@ -27,6 +27,7 @@
     "history",
     "idle",
     "management",
+    "menus",
     "storage",
     "tabs",
     "webRequestBlocking",

test/browser.mock.js

@@ -80,6 +80,17 @@ module.exports = () => {
         addListener: sinon.stub()
       }
     },
+    menus: {
+      create: sinon.stub(),
+      onClicked: {
+        addListener: sinon.stub()
+      },
+      onShown: {
+        addListener: sinon.stub()
+      },
+      refresh: sinon.stub(),
+      removeAll: sinon.stub()
+    },
     extension: {
       getURL: sinon.stub().returns("moz-extension://multi-account-containers/confirm-page.html")
     }