rowan/multi-account-containers
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle removing containers to refresh menus and remove assignments. Fixes: #761, Fixes: #752

Browse source
This commit is contained in:
Jonathan Kingston 2017-08-28 21:35:02 -07:00
parent 3cc40344af
commit b6a98fb83e
2 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
webextension/js/background/backgroundLogic.js
View file

@ -25,13 +25,12 @@ const backgroundLogic = {
return false;
},
async deleteContainer(userContextId) {
async deleteContainer(userContextId, removed = false) {
await this._closeTabs(userContextId);
await browser.contextualIdentities.remove(this.cookieStoreId(userContextId));
if (!removed) {
await browser.contextualIdentities.remove(this.cookieStoreId(userContextId));
}
assignManager.deleteContainer(userContextId);
await browser.runtime.sendMessage({
method: "forgetIdentityAndRefresh"
});
return {done: true, userContextId};
},
@ -158,13 +157,13 @@ const backgroundLogic = {
const cookieStoreId = this.cookieStoreId(userContextId);
let tabs;
/* if we have no windowId we are going to close all this container (used for deleting) */
if (windowId) {
if (windowId !== false) {
tabs = await browser.tabs.query({
cookieStoreId,
windowId
});
} else {
await browser.tabs.query({
tabs = await browser.tabs.query({
cookieStoreId
});
}

7
webextension/js/background/messageHandler.js
View file

@ -75,6 +75,13 @@ const messageHandler = {
return response;
});
if (browser.contextualIdentities.onRemoved) {
browser.contextualIdentities.onRemoved.addListener(({contextualIdentity}) => {
const userContextId = backgroundLogic.getUserContextIdFromCookieStoreId(contextualIdentity.cookieStoreId);
backgroundLogic.deleteContainer(userContextId, true);
});
}
// Handles messages from sdk code
const port = browser.runtime.connect();
port.onMessage.addListener(m => {