console.log("PROBLEMATIC REQUESTS");

// const isThirdParty = (arg) => arg.urlClassification.thirdParty.length > 0;
async function isThirdParty(request) {
  const request_url = new URL(request.url);
  const origin_url = new URL(await getOrigin(request));
  return (
    request_url.origin != origin_url.origin ||
    request.urlClassification.thirdParty.length > 0
  );
}
const hasCookie = (arg) => arg.requestHeaders.some((h) => h.name === "Cookie");
const hasReferer = (arg) =>
  arg.requestHeaders.some((h) => h.name === "Referer");

const getReferer = (arg) =>
  arg.requestHeaders.filter((h) => h.name === "Referer")[0].value;

const getOrigin = async (arg) => {
  let url;
  if (arg.tabId) {
    const tab = await browser.tabs.get(arg.tabId);
    url = tab.url;
  } else {
    url = arg.frameAncestors[0].url;
  }

  return url;
};

const exposesOrigin = async (arg) => {
  return getReferer(arg).includes(new URL(await getOrigin(arg)).host);
};

browser.webRequest.onBeforeSendHeaders.addListener(
  async (request) => {
    if (
      (await isThirdParty(request)) &&
      hasReferer(request) &&
      (await exposesOrigin(request))
    ) {
      const has_cookie = hasCookie(request);
      fn = has_cookie ? console.warn : console.log;
      fn(
        "Leaked referrer! Has cookie:",
        hasCookie(request),
        request.url,
        "referer was",
        getReferer(request)
      );
    }
  },
  { urls: ["<all_urls>"] },
  ["requestHeaders"]
);