multi-account-containers/index.js

/* global require */
const {ContextualIdentityService} = require('resource://gre/modules/ContextualIdentityService.jsm');
const { Cc, Ci, Cu, Cr } = require('chrome');

Cu.import("resource://gre/modules/Services.jsm");

const tabs = require('sdk/tabs');
const webExtension = require('sdk/webextension');

/* Let's start enabling Containers */
var prefs = [
  [ "privacy.userContext.enabled", true ],
  [ "privacy.userContext.ui.enabled", true ],
  [ "privacy.usercontext.about_newtab_segregation.enabled", true ],
  [ "privacy.usercontext.longPressBehavior", 1 ]
];

const prefService = require("sdk/preferences/service");
prefs.forEach((pref) => {
  prefService.set(pref[0], pref[1]);
});

const CONTAINER_STORE = 'firefox-container-';

const identitiesState = {
};

function getCookieStoreIdForContainer(containerId) {
  return CONTAINER_STORE + containerId;
}

function convert(identity) {
  const cookieStoreId = getCookieStoreIdForContainer(identity.userContextId);
  let hiddenTabUrls = [];

  if (cookieStoreId in identitiesState) {
    hiddenTabUrls = identitiesState[cookieStoreId].hiddenTabUrls;
  }
  const result = {
    name: ContextualIdentityService.getUserContextLabel(identity.userContextId),
    icon: identity.icon,
    color: identity.color,
    cookieStoreId: cookieStoreId,
    hiddenTabUrls: hiddenTabUrls
  };

  return result;
}

function isContainerCookieStoreId(storeId) {
  return storeId !== null && storeId.startsWith(CONTAINER_STORE);
}

function getContainerForCookieStoreId(storeId) {
  if (!isContainerCookieStoreId(storeId)) {
    return null;
  }

  const containerId = storeId.substring(CONTAINER_STORE.length);

  if (ContextualIdentityService.getIdentityFromId(containerId)) {
    return parseInt(containerId, 10);
  }

  return null;
}

function getContainer(cookieStoreId) {
  const containerId = getContainerForCookieStoreId(cookieStoreId);

  if (!containerId) {
    return Promise.resolve(null);
  }

  const identity = ContextualIdentityService.getIdentityFromId(containerId);

  return Promise.resolve(convert(identity));
}

function queryContainers(details) {
  const identities = [];

  ContextualIdentityService.getIdentities().forEach(identity=> {
    if (details && details.name &&
        ContextualIdentityService.getUserContextLabel(identity.userContextId) !== details.name) {
      return;
    }

    const convertedIdentity = convert(identity);

    identities.push(convertedIdentity);
    if (!(convertedIdentity.cookieStoreId in identitiesState)) {
      identitiesState[convertedIdentity.cookieStoreId] = {hiddenTabUrls: []};
    }
  });

  return Promise.resolve(identities);
}

function createContainer(details) {
  const identity = ContextualIdentityService.create(details.name,
                                                  details.icon,
                                                  details.color);

  return Promise.resolve(convert(identity));
}

function updateContainer(cookieStoreId, details) {
  const containerId = getContainerForCookieStoreId(cookieStoreId);

  if (!containerId) {
    return Promise.resolve(null);
  }

  const identity = ContextualIdentityService.getIdentityFromId(containerId);

  if (!identity) {
    return Promise.resolve(null);
  }

  if (details.name !== null) {
    identity.name = details.name;
  }

  if (details.color !== null) {
    identity.color = details.color;
  }

  if (details.icon !== null) {
    identity.icon = details.icon;
  }

  if (!ContextualIdentityService.update(identity.userContextId,
                                        identity.name, identity.icon,
                                        identity.color)) {
    return Promise.resolve(null);
  }

  return Promise.resolve(convert(identity));
}

function removeContainer(cookieStoreId) {
  const containerId = getContainerForCookieStoreId(cookieStoreId);

  if (!containerId) {
    return Promise.resolve(null);
  }

  const identity = ContextualIdentityService.getIdentityFromId(containerId);

  if (!identity) {
    return Promise.resolve(null);
  }

  // We have to create the identity object before removing it.
  const convertedIdentity = convert(identity);

  if (!ContextualIdentityService.remove(identity.userContextId)) {
    return Promise.resolve(null);
  }

  return Promise.resolve(convertedIdentity);
}

const contextualIdentities = {
  get: getContainer,
  query: queryContainers,
  create: createContainer,
  update: updateContainer,
  remove: removeContainer
};

function openTab(args) {
  let browserWin = Services.wm.getMostRecentWindow('navigator:browser');

  // This should not really happen.
  if (!browserWin || !browserWin.gBrowser) {
    return Promise.resolve(false);
  }

  let userContextId = 0;
  if ('cookieStoreId' in args) {
    userContextId = getContainerForCookieStoreId(args.cookieStoreId);
  }

  let tab = browserWin.gBrowser.addTab(args.url || null,
                                       { userContextId: userContextId })
  browserWin.gBrowser.selectedTab = tab;
  return Promise.resolve(true);
}

function handleWebExtensionMessage(message, sender, sendReply) {
  switch (message.method) {
      case 'queryIdentities':
        sendReply(contextualIdentities.query(message.arguments));
        break;
      case 'hideTab':
        identitiesState[message.cookieStoreId].hiddenTabUrls = message.tabUrlsToSave;
        break;
      case 'showTab':
        sendReply(identitiesState[message.cookieStoreId].hiddenTabUrls);
        identitiesState[message.cookieStoreId].hiddenTabUrls = [];
        break;
      case 'getTab':
        sendReply(contextualIdentities.get(message.arguments));
        break;
      case 'createTab':
        sendReply(contextualIdentities.create(message.arguments));
        break;
      case 'updateTab':
        sendReply(contextualIdentities.update(message.arguments));
        break;
      case 'removeTab':
        sendReply(contextualIdentities.remove(message.arguments));
        break;
      case 'getIdentitiesState':
        sendReply(identitiesState);
        break;
      case 'openContainersPreferences':
        tabs.open('about:preferences#containers');
        sendReply({content: 'opened'});
        break;
      case 'openTab':
        sendReply(openTab(message));
        break;
  }
}

webExtension.startup().then(api=> {
  const {browser} = api;

  browser.runtime.onMessage.addListener(handleWebExtensionMessage);
});