I2P_in_Private_Browsing_Mode_for_Firefox/scrub.js

var titlepref = chrome.i18n.getMessage("titlePreface");
var titleprefpriv = chrome.i18n.getMessage("titlePrefacePrivate");
var webpref = chrome.i18n.getMessage("webPreface");
var webprefpriv = chrome.i18n.getMessage("webPrefacePrivate");
var routerpref = chrome.i18n.getMessage("routerPreface");
var routerprefpriv = chrome.i18n.getMessage("routerPrefacePrivate");
var mailpref = chrome.i18n.getMessage("mailPreface");
var mailprefpriv = chrome.i18n.getMessage("mailPrefacePrivate");
var torrentpref = chrome.i18n.getMessage("torrentPreface");
var torrentprefpriv = chrome.i18n.getMessage("torrentPrefacePrivate");
var tunnelpref = chrome.i18n.getMessage("i2ptunnelPreface");
var tunnelprefpriv = chrome.i18n.getMessage("i2ptunnelPrefacePrivate");
var localpref = chrome.i18n.getMessage("i2ptunnelPreface");

var contextScrub = async function(requestDetails) {
  console.log("(scrub)Scrubbing info from contextualized request");
  try {
    var headerScrub = function(context) {
      if (!context) {
        console.error("Context not found", context);
      } else if (context.name == titlepref) {
        var ua = "MYOB/6.66 (AN/ON)";
        if (i2pHost(requestDetails.url)) {
          for (var header of requestDetails.requestHeaders) {
            if (header.name.toLowerCase() === "user-agent") {
              header.value = ua;
              console.log("(scrub)User-Agent header modified", header.value);
            }
          }
        }
        return {
          requestHeaders: requestDetails.requestHeaders
        };
      } else if (context.name == routerpref) {
        var ua = "MYOB/6.66 (AN/ON)";
        if (i2pHost(requestDetails.url)) {
          for (var header of requestDetails.requestHeaders) {
            if (header.name.toLowerCase() === "user-agent") {
              header.value = ua;
              console.log("(scrub)User-Agent header modified", header.value);
            }
          }
        } else if (routerHost(requestDetails.url)) {
        }
        return {
          requestHeaders: requestDetails.requestHeaders
        };
      }
    };
    var contextGet = async function(tabInfo) {
      try {
        console.log("(scrub)Tab info from Function", tabInfo);
        let context = await browser.contextualIdentities.get(
          tabInfo.cookieStoreId
        );
        return context;
      } catch (error) {
        return undefined;
      }
    };
    var tabGet = async function(tabId) {
      try {
        console.log("(scrub)Tab ID from Request", tabId);
        let tabInfo = await browser.tabs.get(tabId);
        return tabInfo;
      } catch (error) {
        let tabInfo = await browser.tabs.getCurrent();
        return tabInfo;
      }
    };
    if (requestDetails.tabId > 0) {
      var tab = {};
      var context = {};
      var req = {};
      if (i2pHost(requestDetails.url)) {
        console.log("(scrub)I2P URL detected, ");
        tab = tabGet(requestDetails.tabId);
        context = tab.then(contextGet, onError);
        req = await context.then(headerScrub, onError);
        console.log("(scrub)Scrubbing I2P Request", req);
        return req;
      } else if (routerHost(requestDetails.url)) {
        tab = tabGet(requestDetails.tabId);
        context = tab.then(contextGet, onError);
        req = await context.then(headerScrub, onError);
        console.log("(scrub)Scrubbing non-I2P Request", req);
        return req;
      }
      return req;
    }
  } catch (error) {
    console.log("(scrub)Not scrubbing non-I2P request.", error);
  }
};

var contextSetup = async function(requestDetails) {
  console.log("(isolate)Forcing I2P requests into context");
  try {
    var i2pTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: titlepref
        });
        console.log(context[0].cookieStoreId);
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log(
                  "(isolate) Closing old, un-isolated tab",
                  tabId.id,
                  "in favor of",
                  tab.id
                );
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var routerTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: routerpref
        });
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log("(isolate) Closing old, un-isolated tab");
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var i2ptunnelTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: tunnelpref
        });
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log("(isolate) Closing old, un-isolated tab");
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var snarkTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: torrentpref
        });
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log("(isolate) Closing old, un-isolated tab");
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var mailTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: mailpref
        });
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log("(isolate) Closing old, un-isolated tab");
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var localTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: localpref
        });
        if (tabId.cookieStoreId != context[0].cookieStoreId) {
          function Create(currentTab) {
            function onCreated(tab) {
              if (tabId.id != tab.id) {
                console.log("(isolate) Closing old, un-isolated tab");
                browser.tabs.remove(tabId.id);
              }
            }
            var created = browser.tabs.create({
              active: true,
              cookieStoreId: context[0].cookieStoreId,
              url: requestDetails.url
            });
            created.then(onCreated, onError);
          }
          var getting = browser.tabs.getCurrent();
          getting.then(Create, onError);
          return tabId;
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var anyTabFind = async function(tabId) {
      try {
        var context = await browser.contextualIdentities.query({
          name: webpref
        });
        console.log("(ISOLATE)", tabId.cookieStoreId);
        if (
          tabId.cookieStoreId == "firefox-default" ||
          tabId.cookieStoreId == "firefox-private"
        ) {
          if (tabId.cookieStoreId != context[0].cookieStoreId) {
            function Create(currentTab) {
              function onCreated(tab) {
                if (tabId.id != tab.id) {
                  console.log("(isolate) Closing old, un-isolated tab");
                  browser.tabs.remove(tabId.id);
                }
              }
              var created = browser.tabs.create({
                active: true,
                cookieStoreId: context[0].cookieStoreId,
                url: requestDetails.url
              });
              created.then(onCreated, onError);
            }
            var getting = browser.tabs.getCurrent();
            getting.then(Create, onError);
            return tabId;
          }
        }
      } catch (error) {
        console.log("(isolate)Context Error", error);
      }
    };
    var tabGet = async function(tabId) {
      try {
        console.log("(isolate)Tab ID from Request", tabId);
        let tabInfo = await browser.tabs.get(tabId);
        return tabInfo;
      } catch (error) {
        console.log("(isolate)Tab error", error);
      }
    };
    if (requestDetails == undefined) {
      return requestDetails;
    }
    if (requestDetails.tabId > 0) {
      if (proxyHost(requestDetails.url)) {
        setcookie = browser.cookies.set({
          firstPartyDomain: i2pHostName(requestDetails.url),
          url: requestDetails.url,
          secure: true
        });
        setcookie.then(onContextGotLog, onError);
        return requestDetails;
      }
      console.log("(isolate)Request Details", requestDetails);
      if (extensionHost(requestDetails.url)) {
        var tab = tabGet(requestDetails.tabId);
        var mtab = tab.then(anyTabFind, onError);
        return requestDetails;
      }
      if (i2pHost(requestDetails.url)) {
        var setcookie = browser.cookies.set({
          firstPartyDomain: i2pHostName(requestDetails.url),
          url: requestDetails.url,
          secure: true
        });
        setcookie.then(onContextGotLog, onError);
        var tab = tabGet(requestDetails.tabId);
        var mtab = tab.then(i2pTabFind, onError);
        return requestDetails;
      }
      let localhost = localHost(requestDetails.url);
      let routerhost = routerHost(requestDetails.url);
      if (localhost && !routerhost) {
        var tab = tabGet(requestDetails.tabId);
        var mtab = tab.then(localTabFind, onError);
        return requestDetails;
      }
      if (routerhost) {
        if (routerhost === "i2ptunnelmgr") {
          var tab = tabGet(requestDetails.tabId);
          var mtab = tab.then(i2ptunnelTabFind, onError);
          return requestDetails;
        } else if (routerhost === "i2psnark") {
          var tab = tabGet(requestDetails.tabId);
          var mtab = tab.then(snarkTabFind, onError);
          return requestDetails;
        } else if (routerhost === "webmail") {
          var tab = tabGet(requestDetails.tabId);
          var mtab = tab.then(mailTabFind, onError);
          return requestDetails;
        } else if (routerhost === "routerconsole") {
          var tab = tabGet(requestDetails.tabId);
          var mtab = tab.then(routerTabFind, onError);
          return requestDetails;
        }
      } else {
        var tab = tabGet(requestDetails.tabId);
        var mtab = tab.then(anyTabFind, onError);
        return requestDetails;
      }
    }
    if (typeof requestDetails == "number") {
      tab = tabGet(requestDetails);
      var mtab = tab.then(anyTabFind);
    } else if (typeof requestDetails != undefined) {
      if (typeof requestDetails.tabId > 0) {
        tab = tabGet(requestDetails.tabId);
        var mtab = tab.then(anyTabFind);
      }
    }
    return requestDetails;
  } catch (error) {
    console.log("(isolate)Not an I2P request, blackholing", error);
  }
};

browser.webRequest.onBeforeRequest.addListener(
  contextSetup,
  { urls: ["<all_urls>"] },
  ["blocking"]
);

browser.webRequest.onBeforeSendHeaders.addListener(
  contextScrub,
  { urls: ["<all_urls>"] },
  ["blocking", "requestHeaders"]
);