Onion_In_Container_Browsing_Mode_Firefox/background.js

function onGot(contexts) {
  var ids = [];
  for (let context of contexts) {
    console.log(`Name: ${context.name}`);
    ids.push(context.name);
  }
  console.log("Checking new contexts");
  if (ids.indexOf("onionbrowser") == -1) {
    browser.contextualIdentities
      .create({
        name: "onionbrowser",
        color: "purple",
        icon: "fingerprint"
      })
      .then(onCreated, onError);
  }
  if (ids.indexOf("routerconsole") == -1) {
    browser.contextualIdentities
      .create({
        name: "routerconsole",
        color: "turquoise",
        icon: "briefcase"
      })
      .then(onCreated, onError);
  }
}

function onCreated(context) {
  console.log(`New identity's ID: ${context.cookieStoreId}.`);
}

function onError(e) {
  console.error(e);
}

browser.contextualIdentities.query({}).then(onGot, onError);

chrome.windows.onCreated.addListener(themeWindow);
chrome.windows.onFocusChanged.addListener(themeWindow);
chrome.windows.onRemoved.addListener(themeWindow);
chrome.tabs.onUpdated.addListener(themeWindowByTab);
chrome.tabs.onActivated.addListener(themeWindowByTab);

var titlepref = chrome.i18n.getMessage("titlePreface");
var titleprefpriv = chrome.i18n.getMessage("titlePrefaceContainer");

function themeWindowByTab(tabId) {
  function tabWindow(tab) {
    getwindow = browser.windows.get(tab.windowId);
    getwindow.then(themeWindow);
  }
  if (typeof tabId === "number") {
    tab = browser.tabs.get(tabId);
    tab.then(tabWindow);
  } else {
    tabWindow(tabId);
  }
}

function themeWindow(window) {
  // Check if the window is in private browsing
  function logTabs(tabInfo) {
    function onGot(context) {
      if (context.name == "onionbrowser") {
        console.log("Active in Onion window");
        if (window.incognito) {
          chrome.theme.update(window.id, {
            colors: {
              frame: "#2D4470",
              toolbar: "#2D4470"
            }
          });
        } else {
          chrome.theme.update(window.id, {
            colors: {
              frame: "#9DABD5",
              toolbar: "#9DABD5"
            }
          });
        }
      } else if (context.name == "routerconsole") {
        console.log("Active in Onion window");
        if (window.incognito) {
          chrome.theme.update(window.id, {
            colors: {
              frame: "#00CED1",
              toolbar: "#00CED1"
            }
          });
        } else {
          chrome.theme.update(window.id, {
            colors: {
              frame: "#40E0D0",
              toolbar: "#40E0D0"
            }
          });
        }
      } else {
        console.log("Not active in Onion window");
      }
    }
    if (tabInfo[0].cookieStoreId != "firefox-default") {
      browser.contextualIdentities
        .get(tabInfo[0].cookieStoreId)
        .then(onGot, onError);
    } else {
      chrome.theme.reset(window.id);
    }
  }

  var querying = browser.tabs.query({
    currentWindow: true,
    active: true
  });
  querying.then(logTabs, onError);
}

function setTitle(window) {
  function logTabs(tabInfo) {
    console.log(tabInfo);

    function onGot(context) {
      if (context.name == "onionbrowser") {
        console.log("Active in Onion window");

        console.log("Active in Onion window");
        if (window.incognito) {
          chrome.windows.update(window.id, {
            titlePreface: titleprefpriv
          });
        } else {
          chrome.windows.update(window.id, {
            titlePreface: titlepref
          });
        }
      }
    }

    if (tabInfo[0].cookieStoreId != "firefox-default")
      browser.contextualIdentities
        .get(tabInfo[0].cookieStoreId)
        .then(onGot, onError);
  }

  var querying = browser.tabs.query({
    currentWindow: true,
    active: true
  });
  querying.then(logTabs, onError);
}

chrome.windows.onCreated.addListener(() => {
  //var gettingStoredSettings = chrome.storage.local.get();
  //gettingStoredSettings.then(setupProxy, onError);
  chrome.storage.local.get(function(got) {
    setupProxy();
  });
});

chrome.tabs.onCreated.addListener(() => {
  var getting = browser.windows.getCurrent({
    populate: true
  });
  getting.then(setTitle, onError);
});