fixed mozilla#1564: cleanup updates requested for bookmark context menu

kendallcorner · Oct 30, 2019 · 05bea37 · 05bea37
1 parent f159e42
commit 05bea37
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 43 deletions.
diff --git a/src/js/background/assignManager.js b/src/js/background/assignManager.js
@@ -244,15 +244,18 @@ const assignManager = {
   },
 
   async getPermissions() {
-    const {permissions} = await browser.permissions.getAll();
-    permissions.includes("bookmarks") ? this.makeBookmarksMenu() : browser.contextMenus.remove(this.OPEN_IN_CONTAINER);
-  },
-
-  makeBookmarksMenu() {
-    this.initBookmarksMenu();
-    browser.contextualIdentities.onCreated.addListener(this.contextualIdentityCreated);
-    browser.contextualIdentities.onUpdated.addListener(this.contextualIdentityUpdated);
-    browser.contextualIdentities.onRemoved.addListener(this.contextualIdentityRemoved);
+    const hasPermission = await browser.permissions.contains({permissions: ["bookmarks"]});
+    if (hasPermission) {
+      this.initBookmarksMenu();
+      browser.contextualIdentities.onCreated.addListener(this.contextualIdentityCreated);
+      browser.contextualIdentities.onUpdated.addListener(this.contextualIdentityUpdated);
+      browser.contextualIdentities.onRemoved.addListener(this.contextualIdentityRemoved);
+    } else { 
+      this.removeBookmarksMenu();
+      browser.contextualIdentities.onCreated.removeListener(this.contextualIdentityCreated);
+      browser.contextualIdentities.onUpdated.removeListener(this.contextualIdentityUpdated);
+      browser.contextualIdentities.onRemoved.removeListener(this.contextualIdentityRemoved);
+    }
   },
 
   contextualIdentityCreated(changeInfo) {
@@ -324,7 +327,7 @@ const assignManager = {
         if(openInReaderMode) {
           try {
             const parsed = new URL(bookmark.url);
-            bookmark.url = parsed.searchParams.get("url") + parsed.hash; // can't believe const lets me do this ...
+            bookmark.url = parsed.searchParams.get("url") + parsed.hash;
           } catch (err) {
             return err.message;
           }
@@ -521,6 +524,14 @@ const assignManager = {
         icons: { "16": `img/usercontext.svg#${identity.icon}` }
       });
     }
+  },
+
+  async removeBookmarksMenu() {
+    browser.contextMenus.remove(this.OPEN_IN_CONTAINER);
+    const identities = await browser.contextualIdentities.query({});
+    for (const identity of identities) {
+      browser.contextMenus.remove(identity.cookieStoreId);
+    }
   }
 };
 

diff --git a/src/js/options.js b/src/js/options.js
@@ -1,39 +1,20 @@
 
-function requestPermissions() {
+async function requestPermissions() {
   const checkbox = document.querySelector("#bookmarksPermissions");
   if (checkbox.checked) {
-    browser.permissions.request({permissions: ["bookmarks"]}).
-    then((response) => {
-      if (response) {
-        browser.runtime.sendMessage({ method: "resetBookmarksContext" });
-      } else {
-        checkbox.checked = false;
-      }
-    }).
-    catch((err) => {
-      return err.message;
-    });
+    const response = await browser.permissions.request({permissions: ["bookmarks"]});
+    if (!response) { checkbox.checked = false; }
   } else {
-    browser.permissions.remove({permissions: ["bookmarks"]}).
-    then(() => {
-      browser.runtime.sendMessage({ method: "resetBookmarksContext" });
-    }).
-    catch((err) => {
-      return err.message;
-    });
+    await browser.permissions.remove({permissions: ["bookmarks"]});
   }
+  browser.runtime.sendMessage({ method: "resetBookmarksContext" });
 }
 
-function restoreOptions() {
-  browser.permissions.getAll()
-    .then((permissions) => {
-      if (permissions.permissions.includes("bookmarks")) {
-        document.querySelector("#bookmarksPermissions").checked = true;
-      }
-    }).
-    catch((err) => {
-      return err.message;
-    });
+async function restoreOptions() {
+  const hasPermission = await browser.permissions.contains({permissions: ["bookmarks"]});
+  if (hasPermission) {
+    document.querySelector("#bookmarksPermissions").checked = true;
+  }
 }
 
 

diff --git a/src/options.html b/src/options.html
@@ -2,13 +2,15 @@
 
 <html>
   <head>
-    <meta charset="utf-8"/>
+    <meta charset="utf-8">
   </head>
 
   <body>
     <form>
-      <input type="checkbox" id="bookmarksPermissions"/>
-      <label>Enable Bookmark Menus</label>
+      <label>
+        <input type="checkbox" id="bookmarksPermissions">
+        Enable Bookmark Menus
+      </label>
       <p>This setting allows you to open a bookmark or folder of bookmarks in a container.</p>
     </form>
     <script src="js/options.js"></script>

diff --git a/test/browser.mock.js b/test/browser.mock.js
@@ -93,7 +93,7 @@ module.exports = () => {
       getURL: sinon.stub().returns("moz-extension://multi-account-containers/confirm-page.html")
     },
     permissions: {
-      getAll: sinon.stub().returns({"permissions": ["bookmarks"]})
+      contains: sinon.stub().returns(true)
     }
   };