Fix command handler to look up cookieStoreId per command type

Fix command handler to look up cookieStoreId per command type

Author: antondudakov

src/js/background/backgroundLogic.js

@@ -26,19 +26,19 @@ const backgroundLogic = {
       for (let i=0; i < backgroundLogic.NUMBER_OF_KEYBOARD_SHORTCUTS; i++) {
         const key = MAC_CONSTANTS.OPEN_CONTAINER_PREFIX + i;
         const reopenKey = MAC_CONSTANTS.REOPEN_IN_CONTAINER_PREFIX + i;
-        const cookieStoreId = identityState.keyboardShortcut[key];
-
-        if (cookieStoreId === "none") {
-          continue;
-        }
-
         if (command === key) {
-          browser.tabs.create({cookieStoreId});
+          const cookieStoreId = identityState.keyboardShortcut[key];
+          if (cookieStoreId && cookieStoreId !== "none") {
+            browser.tabs.create({cookieStoreId});
+          }
           return;
         }
 
         if (command === reopenKey) {
-          backgroundLogic.reopenInContainer(cookieStoreId);
+          const cookieStoreId = identityState.keyboardShortcut[reopenKey];
+          if (cookieStoreId && cookieStoreId !== "none") {
+            backgroundLogic.reopenInContainer(cookieStoreId);
+          }
           return;
         }
       }
@@ -83,14 +83,14 @@ const backgroundLogic = {
   },
 
   async reopenInContainer(cookieStoreId) {
-   const currentTab = await browser.tabs.query({ active: true, currentWindow: true })
+    const currentTab = await browser.tabs.query({ active: true, currentWindow: true });
 
     if (currentTab.length > 0) {
       const tab = currentTab[0];
 
       browser.tabs.create({
         url: tab.url,
-        cookieStoreId: cookieStoreId,
+        cookieStoreId,
         index: tab.index + 1,
         active: tab.active
       });