import * as CompileConfig from "../config.json"; import Config from "./config"; import { Registration } from "./types"; import "content-scripts-register-polyfill"; import { sendRealRequestToCustomServer, setupBackgroundRequestProxy } from "../maze-utils/src/background-request-proxy"; import { setupTabUpdates } from "../maze-utils/src/tab-updates"; import { generateUserID } from "../maze-utils/src/setup"; import Utils from "./utils"; import { getExtensionIdsToImportFrom } from "./utils/crossExtension"; import { isFirefoxOrSafari, waitFor } from "../maze-utils/src"; import { injectUpdatedScripts } from "../maze-utils/src/cleanup"; import { logWarn } from "./utils/logger"; import { chromeP } from "../maze-utils/src/browserApi"; const utils = new Utils({ registerFirefoxContentScript, unregisterFirefoxContentScript }); const popupPort: Record = {}; // Used only on Firefox, which does not support non persistent background pages. const contentScriptRegistrations = {}; // Register content script if needed utils.wait(() => Config.isReady()).then(function() { if (Config.config.supportInvidious) utils.setupExtraSiteContentScripts(); }); setupBackgroundRequestProxy(); setupTabUpdates(Config); chrome.runtime.onMessage.addListener(function (request, sender, callback) { switch(request.message) { case "openConfig": chrome.tabs.create({url: chrome.runtime.getURL('options/options.html' + (request.hash ? '#' + request.hash : ''))}); return false; case "openHelp": chrome.tabs.create({url: chrome.runtime.getURL('help/index.html')}); return false; case "openPage": chrome.tabs.create({url: chrome.runtime.getURL(request.url)}); return false; case "submitVote": submitVote(request.type, request.UUID, request.category).then(callback); //this allows the callback to be called later return true; case "registerContentScript": registerFirefoxContentScript(request); return false; case "unregisterContentScript": unregisterFirefoxContentScript(request.id) return false; case "tabs": { chrome.tabs.query({ active: true, currentWindow: true }, tabs => { chrome.tabs.sendMessage( tabs[0].id, request.data, (response) => { callback(response); } ); }); return true; } case "time": case "infoUpdated": case "videoChanged": if (sender.tab) { try { popupPort[sender.tab.id]?.postMessage(request); } catch (e) { // This can happen if the popup is closed } } return false; default: return false; } }); chrome.runtime.onMessageExternal.addListener((request, sender, callback) => { if (getExtensionIdsToImportFrom().includes(sender.id)) { if (request.message === "requestConfig") { callback({ userID: Config.config.userID, allowExpirements: Config.config.allowExpirements, showDonationLink: Config.config.showDonationLink, showUpsells: Config.config.showUpsells, darkMode: Config.config.darkMode, }) } } }); chrome.runtime.onConnect.addListener((port) => { if (port.name === "popup") { chrome.tabs.query({ active: true, currentWindow: true }, tabs => { popupPort[tabs[0].id] = port; }); } }); //add help page on install chrome.runtime.onInstalled.addListener(function () { // This let's the config sync to run fully before checking. // This is required on Firefox setTimeout(async () => { const userID = Config.config.userID; // If there is no userID, then it is the first install. if (!userID && !Config.local.alreadyInstalled){ //open up the install page chrome.tabs.create({url: chrome.runtime.getURL("/help/index.html")}); //generate a userID const newUserID = generateUserID(); //save this UUID Config.config.userID = newUserID; Config.local.alreadyInstalled = true; // Don't show update notification Config.config.categoryPillUpdate = true; } if (Config.config.supportInvidious) { if (!(await utils.containsInvidiousPermission())) { chrome.tabs.create({url: chrome.runtime.getURL("/permissions/index.html")}); } } }, 1500); if (!isFirefoxOrSafari()) { injectUpdatedScripts().catch(logWarn); waitFor(() => Config.isReady()).then(() => { if (Config.config.supportInvidious) { injectUpdatedScripts([ utils.getExtraSiteRegistration() ]) } }).catch(logWarn); } }); /** * Only works on Firefox. * Firefox requires that it be applied after every extension restart. * * @param {JSON} options */ async function registerFirefoxContentScript(options: Registration) { if ("scripting" in chrome && "getRegisteredContentScripts" in chrome.scripting) { const existingRegistrations = await chromeP.scripting.getRegisteredContentScripts({ ids: [options.id] }).catch(() => []); if (existingRegistrations && existingRegistrations.length > 0 && options.matches.every((match) => existingRegistrations[0].matches.includes(match))) { // No need to register another script, already registered return; } } await unregisterFirefoxContentScript(options.id); if ("scripting" in chrome && "getRegisteredContentScripts" in chrome.scripting) { await chromeP.scripting.registerContentScripts([{ id: options.id, runAt: "document_start", matches: options.matches, allFrames: options.allFrames, js: options.js, css: options.css, persistAcrossSessions: true, }]); } else { chrome.contentScripts.register({ allFrames: options.allFrames, js: options.js?.map?.(file => ({file})), css: options.css?.map?.(file => ({file})), matches: options.matches }).then((registration) => void (contentScriptRegistrations[options.id] = registration)); } } /** * Only works on Firefox. * Firefox requires that this is handled by the background script */ async function unregisterFirefoxContentScript(id: string) { if ("scripting" in chrome && "getRegisteredContentScripts" in chrome.scripting) { try { await chromeP.scripting.unregisterContentScripts({ ids: [id] }); } catch (e) { // Not registered yet } } else { if (contentScriptRegistrations[id]) { contentScriptRegistrations[id].unregister(); delete contentScriptRegistrations[id]; } } } async function submitVote(type: number, UUID: string, category: string) { let userID = Config.config.userID; if (userID == undefined || userID === "undefined") { //generate one userID = generateUserID(); Config.config.userID = userID; } const typeSection = (type !== undefined) ? "&type=" + type : "&category=" + category; try { const response = await asyncRequestToServer("POST", "/api/voteOnSponsorTime?UUID=" + UUID + "&userID=" + userID + typeSection); if (response.ok) { return { successType: 1, responseText: await response.text() }; } else if (response.status == 405) { //duplicate vote return { successType: 0, statusCode: response.status, responseText: await response.text() }; } else { //error while connect return { successType: -1, statusCode: response.status, responseText: await response.text() }; } } catch (e) { console.error(e); return { successType: -1, statusCode: -1, responseText: "" }; } } async function asyncRequestToServer(type: string, address: string, data = {}) { const serverAddress = Config.config.testingServer ? CompileConfig.testingServerAddress : Config.config.serverAddress; return await (sendRealRequestToCustomServer(type, serverAddress + address, data)); }