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

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

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

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

@ -75,6 +75,13 @@ const messageHandler = {
return response; 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 // Handles messages from sdk code
const port = browser.runtime.connect(); const port = browser.runtime.connect();
port.onMessage.addListener(m => { port.onMessage.addListener(m => {