Added a check for Containers Mode in switchToNoList for #20

Cookie-AutoDelete · Jun 20, 2017 · 9e55c45 · 9e55c45
1 parent d7a02df
commit 9e55c45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/extension/popup/popup.js b/extension/popup/popup.js
@@ -152,7 +152,11 @@ document.getElementById("activeModeSwitch").addEventListener("click", () => {
 document.getElementById("switchToNoList").addEventListener("click", () => {
 	console.log("Removed from list");
 	if (hostUrl !== undefined) {
-		page.whiteList.removeURLFromLists(hostUrl, cookieStoreId);
+		if(page.contextualIdentitiesEnabled) {
+			page.whiteList.removeURLFromLists(hostUrl, cookieStoreId);
+		} else {
+			page.whiteList.removeURLFromLists(hostUrl, defaultWhiteList);
+		}
 		page.checkIfProtected(activeTab);
 	}
 });

diff --git a/src/services/WhiteListService.js b/src/services/WhiteListService.js
@@ -98,6 +98,8 @@ class WhiteListService {
 	removeURLFromLists(url, cookieStoreId = defaultWhiteList) {
 		let otherList = this.returnOtherList(cookieStoreId);
 		// console.log(this.cookieWhiteList);
+		// console.log(url);
+		// console.log(cookieStoreId + " " + otherList);
 		this.cookieWhiteList.get(cookieStoreId).delete(url);
 		this.cookieWhiteList.get(otherList).delete(url);
 	}