support multi account containers

andreicristianpetcu · Aug 18, 2019 · ae1d150 · ae1d150
1 parent 68f6de8
commit ae1d150
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/scripts/StorageService.js b/scripts/StorageService.js
@@ -33,12 +33,13 @@ class StorageService {
         return domainData.hasCSP;
     }
 
-    static async setLangCookie(domain) {
+    static async setLangCookie(domain, storeId) {
         const langCookie = await getGoogtransCookie();
-        browser.cookies.set({
+        await browser.cookies.set({   
             url: domain,
             name: "googtrans",
-            value: langCookie.value
+            value: langCookie.value,
+            storeId
         });
     }
 

diff --git a/scripts/background.js b/scripts/background.js
@@ -178,14 +178,14 @@ function getDomain(url) {
 }
 
 function translateTab(tabId) {
-  setTimeout(function () {
-    browser.tabs.get(tabId).then(function (tabInfo) {
-      StorageService.setLangCookie(tabInfo.url);
-      browser.tabs.executeScript(tabId, {
+  browser.tabs.get(tabId).then(async function (tabInfo) {
+    await StorageService.setLangCookie(tabInfo.url, tabInfo.cookieStoreId);
+    setTimeout(function () {
+      return browser.tabs.executeScript(tabId, {
         file: 'scripts/inject_google_translate_content.js'
       });
-    });
-  }, 250);
+    }, 250);
+  });
 }
 
 async function updateMenuForDomain() {