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

Syncs deletion of site assignments and reassignments

Browse source
This commit is contained in:
Kendall Werts 2019-12-31 13:59:00 -06:00
parent 0e45f06338
commit 0bdf8558f6
1 changed files with 39 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
src/js/background/assignManager.js
View file

@ -75,7 +75,7 @@ const assignManager = {
await this.area.set({
[siteStoreKey]: data
});
await backup();
await backup({undelete: siteStoreKey});
return;
},
@ -84,7 +84,7 @@ const assignManager = {
// When we remove an assignment we should clear all the exemptions
this.removeExempted(pageUrl);
await this.area.remove([siteStoreKey]);
await backup();
await backup({siteStoreKey});
return;
},
@ -587,6 +587,8 @@ async function backup(options) {
await updateCookieStoreIdMap();
await updateSyncSiteAssignments();
if (options && options.uuid) await updateDeletedIdentityList(options.uuid);
if (options && options.siteStoreKey) await addToDeletedSitesList(options.siteStoreKey);
if (options && options.undelete) await removeFromDeletedSitesList(options.undelete);
// for testing
const storage = await browser.storage.sync.get();
console.log("in sync: ", storage);
@ -616,10 +618,26 @@ async function updateSyncSiteAssignments() {
async function updateDeletedIdentityList(deletedIdentityUUID) {
let { deletedIdentityList } = await browser.storage.sync.get("deletedIdentityList");
if (!deletedIdentityList) deletedIdentityList = [];
if (deletedIdentityList.find(element => element === deletedIdentityUUID)) return;
deletedIdentityList.push(deletedIdentityUUID);
await browser.storage.sync.set({ deletedIdentityList });
}
async function addToDeletedSitesList(siteStoreKey) {
let { deletedSiteList } = await browser.storage.sync.get("deletedSiteList");
if (!deletedSiteList) deletedSiteList = [];
if (deletedSiteList.find(element => element === siteStoreKey)) return;
deletedSiteList.push(siteStoreKey);
await browser.storage.sync.set({ deletedSiteList });
}
async function removeFromDeletedSitesList(siteStoreKey) {
let { deletedSiteList } = await browser.storage.sync.get("deletedSiteList");
if (!deletedSiteList) return;
deletedSiteList = deletedSiteList.filter(element => element !== siteStoreKey);
await browser.storage.sync.set({ deletedSiteList });
}
browser.resetMAC1 = async function () {
// for debugging and testing: remove all containers except the default 4 and the first one created
browser.storage.onChanged.removeListener(syncOnChangedListener);
@ -801,23 +819,25 @@ async function reconcileSiteAssignments(inSync, firstSync = false) {
console.log("reconcileSiteAssignments");
const assignedSitesLocal = await assignManager.storageArea.getAssignedSites();
const assignedSitesFromSync = inSync.assignedSites;
if (inSync.hasOwnProperty("deletedSiteList")){
for(const siteStoreKey of inSync.deletedSiteList) {
if (assignedSitesLocal.hasOwnProperty(siteStoreKey)) {
assignManager.storageArea.remove(siteStoreKey.replace(/^siteContainerMap@@_/, "https://"));
}
}
}
for(const urlKey of Object.keys(assignedSitesFromSync)) {
if (assignedSitesLocal.hasOwnProperty(urlKey)) {
const syncCookieStoreId = "firefox-container-" + assignedSitesFromSync[urlKey].userContextId;
const syncUUID = await inSync.cookieStoreIDmap[syncCookieStoreId];
const assignedSite = assignedSitesLocal[urlKey];
const localCookieStoreId = "firefox-container-" + assignedSite.userContextId;
const localIdentityUUID = await identityState.storageArea.get(localCookieStoreId).macAddonUUID
const localIdentityUUID = await identityState.storageArea.get(localCookieStoreId).macAddonUUID;
if (syncUUID === localIdentityUUID) {
continue;
}
if (!firstSync) {
// overwrite with Sync data
await setAsignmentWithUUID(syncUUID, assignedSite, urlKey);
continue;
}
// TODO: on First Sync only, if uuids are not the same,
// ask user where to assign the site.
// overwrite with Sync data. Sync is the source of truth
await setAsignmentWithUUID(syncUUID, assignedSite, urlKey);
continue;
}
const assignedSite = assignedSitesFromSync[urlKey];
@ -829,11 +849,15 @@ async function reconcileSiteAssignments(inSync, firstSync = false) {
async function setAsignmentWithUUID (newUUID, assignedSite, urlKey) {
const cookieStoreId = await identityState.lookupCookieStoreId(newUUID);
assignedSite.userContextId = cookieStoreId.replace(/^firefox-container-/, "");
await assignManager.storageArea.set(
urlKey.replace(/^siteContainerMap@@_/, "https://"),
assignedSite
);
if (cookieStoreId) {
assignedSite.userContextId = cookieStoreId.replace(/^firefox-container-/, "");
await assignManager.storageArea.set(
urlKey.replace(/^siteContainerMap@@_/, "https://"),
assignedSite
);
return;
}
throw new Error ("No cookieStoreId found for: ", newUUID, assignedSite, urlKey);
}
async function runSync() {