libredirect/src/pages/background/background.js

"use strict";

import youtubeHelper from "../../assets/javascripts/helpers/youtube/youtube.js";
import twitterHelper from "../../assets/javascripts/helpers/twitter.js";
import instagramHelper from "../../assets/javascripts/helpers/instagram.js";
import redditHelper from "../../assets/javascripts/helpers/reddit.js";
import searchHelper from "../../assets/javascripts/helpers/search.js";
import translateHelper from "../../assets/javascripts/helpers/translate.js";
import mapsHelper from "../../assets/javascripts/helpers/maps.js";
import wikipediaHelper from "../../assets/javascripts/helpers/wikipedia.js";
import mediumHelper from "../../assets/javascripts/helpers/medium.js";

window.browser = window.browser || window.chrome;

function wholeInit() {
  mapsHelper.init()
  searchHelper.init()
  translateHelper.init()
  instagramHelper.init()
  mediumHelper.init()
  redditHelper.init()
  twitterHelper.init()
  wikipediaHelper.init()
  youtubeHelper.init()
}

wholeInit();

browser.storage.onChanged.addListener(wholeInit);

browser.webRequest.onBeforeRequest.addListener(
  async (details) => {
    const url = new URL(details.url);
    let initiator;
    if (details.originUrl)
      initiator = new URL(details.originUrl);
    else if (details.initiator)
      initiator = new URL(details.initiator);

    var newUrl;

    if (youtubeHelper.isYoutube(url)) newUrl = youtubeHelper.redirect(url, initiator, details.type)

    else if (twitterHelper.isTwitter(url)) newUrl = twitterHelper.redirect(url, initiator);

    else if (instagramHelper.isInstagram(url)) newUrl = instagramHelper.redirect(url, initiator, details.type);

    else if (mapsHelper.isMaps(url)) newUrl = mapsHelper.redirect(url, initiator);

    else if (redditHelper.isReddit(url)) newUrl = redditHelper.redirect(url, initiator, details.type);

    else if (mediumHelper.isMedium(url)) newUrl = mediumHelper.redirect(url, initiator);

    else if (translateHelper.isTranslate(url)) newUrl = translateHelper.redirect(url, initiator);

    else if (searchHelper.isSearch(url)) newUrl = searchHelper.redirect(url, initiator)

    else if (wikipediaHelper.isWikipedia(url)) newUrl = wikipediaHelper.redirect(url, initiator);

    if (newUrl) {
      console.info("Redirecting", url.href, "=>", newUrl);
      return { redirectUrl: newUrl };
    }
    return null;
  },
  { urls: ["<all_urls>"], },
  ["blocking"]
);

browser.tabs.onUpdated.addListener((tabId, changeInfo, _) => {
  let url;
  try {
    url = new URL(changeInfo.url)
  } catch (_) {
    return;
  }
  var protocolHost = `${url.protocol}//${url.host}`;


  if (youtubeHelper.getRedirects().invidious.normal.includes(protocolHost)) {
    if (youtubeHelper.getPersistInvidiousPrefs())
      youtubeHelper.invidiousInitCookies(tabId);
  }

  var mightyList = [];
  mightyList.push(
    ...youtubeHelper.getRedirects().invidious.normal,
    ...youtubeHelper.getRedirects().piped.normal,
    ...twitterHelper.getRedirects().normal,
    ...instagramHelper.getRedirects().normal,
    ...redditHelper.getRedirects().libreddit.normal,
    ...redditHelper.getRedirects().teddit.normal,
    redditHelper.getRedirects().desktop,
    redditHelper.getRedirects().mobile,
    ...searchHelper.getRedirects().searx.normal,
    ...searchHelper.getRedirects().whoogle.normal,
    ...translateHelper.getRedirects().simplyTranslate.normal,
    ...translateHelper.getRedirects().lingva.normal,
    ...mediumHelper.getRedirects().normal,
    ...wikipediaHelper.getRedirects().normal
  );

  if (mightyList.includes(protocolHost)) browser.pageAction.show(tabId);
});


browser.pageAction.onClicked.addListener((tab) => {
  var tabUrl = new URL(tab.url);
  var protocolHost = `${tabUrl.protocol}//${tabUrl.host}`;
  var newUrl;

  if (
    youtubeHelper.getRedirects().invidious.normal.includes(protocolHost) ||
    youtubeHelper.getRedirects().piped.normal.includes(protocolHost)
  )
    newUrl = 'https://youtube.com';

  if (twitterHelper.getRedirects().normal.includes(protocolHost)) newUrl = 'https://twitter.com';

  if (instagramHelper.getRedirects().normal.includes(protocolHost)) newUrl = 'https://instagram.com';

  if (redditHelper.getRedirects().libreddit.normal.includes(protocolHost) || redditHelper.getRedirects().teddit.normal.includes(protocolHost)) {
    if (tabUrl.pathname.startsWith('/img')) {
      newUrl = "https://i.redd.it"
      tabUrl.href = tabUrl.href.replace("/img", "")
    }
    else
      newUrl = 'https://reddit.com';
  }

  if (
    searchHelper.getRedirects().searx.normal.includes(protocolHost) ||
    searchHelper.getRedirects().whoogle.normal.includes(protocolHost)
  ) newUrl = 'https://google.com';

  if (
    translateHelper.getRedirects().simplyTranslate.normal.includes(protocolHost) ||
    translateHelper.getRedirects().lingva.normal.includes(protocolHost)
  ) newUrl = 'https://translate.google.com';

  if (mediumHelper.getRedirects().normal.includes(protocolHost)) newUrl = 'https://medium.com';

  if (wikipediaHelper.getRedirects().normal.includes(protocolHost)) newUrl = 'https://wikipedia.com';

  if (newUrl) browser.tabs.update({ url: tabUrl.href.replace(protocolHost, newUrl) });
});