webext-messenger 0.31.0 → 0.32.0

package/distribution/logging.js

@@ -1,4 +1,3 @@
-/* Warning: Do not use import browser-polyfill directly or indirectly */
 // .bind preserves the call location in the console
 const debug = console.debug.bind(console, "Messenger:");
 const warn = console.warn.bind(console, "Messenger:");

package/distribution/receiver.js

@@ -1,4 +1,3 @@
-import browser from "webextension-polyfill";
 import { serializeError } from "serialize-error";
 import { getContextName } from "webext-detect";
 import { messenger } from "./sender.js";
@@ -13,8 +12,14 @@ export function isMessengerMessage(message) {
         message["__webextMessenger"] === true &&
         Array.isArray(message["args"]));
 }
-// MUST NOT be `async` or Promise-returning-only
-function onMessageListener(message, sender) {
+/**
+ * Decides what to do with a message and sends a response (value or error) back to the sender.
+ *
+ * @warn This function cannot return a Promise.
+ * @warn Limit the amount of logic here because errors won't make it to `sendResponse`
+ */
+//
+function onMessageListener(message, sender, sendResponse) {
     if (!isMessengerMessage(message)) {
         // TODO: Add test for this eventuality: ignore unrelated messages
         return;
@@ -29,21 +34,10 @@ function onMessageListener(message, sender) {
         });
         return;
     }
-    return handleMessage(message, sender, action);
-}
-// This function can only be called when the message *will* be handled locally.
-// Returning "undefined" or throwing an error will still handle it.
-async function handleMessage(message, sender, 
-// Once messages reach this function they cannot be "ignored", they're already being handled
-action) {
     const { type, target, args, options = {} } = message;
     const { trace = [], seq } = options;
-    trace.push(sender);
-    const meta = { trace };
-    let handleMessage;
     if (action === "forward") {
         log.debug(type, seq, "🔀 forwarded", { sender, target });
-        handleMessage = async () => messenger(type, meta, target, ...args);
     }
     else {
         log.debug(type, seq, "↘️ received in", getContextName(), {
@@ -51,24 +45,40 @@ action) {
             args,
             wasForwarded: trace.length > 1,
         });
-        const localHandler = handlers.get(type);
-        if (!localHandler) {
-            if (!didUserRegisterMethods()) {
-                // TODO: Test the handling of __getTabData in contexts that have no registered methods
-                // https://github.com/pixiebrix/webext-messenger/pull/82
-                throw new MessengerError(`No handlers registered in ${getContextName()}`);
-            }
-            throw new MessengerError(`No handler registered for ${type} in ${getContextName()}`);
+    }
+    // Prepare the response asynchronously because the listener must return `true` synchronously
+    (async () => {
+        try {
+            trace.push(sender);
+            const value = await prepareResponse(message, action, { trace });
+            log.debug(type, seq, "↗️ responding", { value });
+            sendResponse({ __webextMessenger, value });
         }
-        handleMessage = async () => localHandler.apply(meta, args);
+        catch (error) {
+            log.debug(type, seq, "↗️ responding", { error });
+            sendResponse({ __webextMessenger, error: serializeError(error) });
+        }
+    })();
+    // This indicates that the message is being handled and a response will be sent asynchronously
+    // TODO: Just return a promise if this is ever implemented https://issues.chromium.org/issues/40753031
+    return true;
+}
+/** Generates the value or error to return to the sender; does not include further messaging logic */
+async function prepareResponse(message, action, meta) {
+    const { type, target, args } = message;
+    if (action === "forward") {
+        return messenger(type, meta, target, ...args);
+    }
+    const localHandler = handlers.get(type);
+    if (localHandler) {
+        return localHandler.apply(meta, args);
+    }
+    if (didUserRegisterMethods()) {
+        throw new MessengerError(`No handler registered for ${type} in ${getContextName()}`);
     }
-    const response = await handleMessage().then((value) => ({ value }), (error) => ({
-        // Errors must be serialized because the stack traces are currently lost on Chrome
-        // and https://github.com/mozilla/webextension-polyfill/issues/210
-        error: serializeError(error),
-    }));
-    log.debug(type, seq, "↗️ responding", response);
-    return { ...response, __webextMessenger };
+    // TODO: Test the handling of __getTabData in contexts that have no registered methods
+    // https://github.com/pixiebrix/webext-messenger/pull/82
+    throw new MessengerError(`No handlers registered in ${getContextName()}`);
 }
 export function registerMethods(methods) {
     for (const [type, method] of Object.entries(methods)) {
@@ -78,7 +88,7 @@ export function registerMethods(methods) {
         log.debug("Registered", type);
         handlers.set(type, method);
     }
-    browser.runtime.onMessage.addListener(onMessageListener);
+    chrome.runtime.onMessage.addListener(onMessageListener);
 }
 /** Ensure/document that the current function was called via Messenger */
 export function assertMessengerCall(_this) { }

package/distribution/sender.js

@@ -1,4 +1,3 @@
-import browser from "webextension-polyfill";
 import pRetry from "p-retry";
 import { isBackground } from "webext-detect";
 import { deserializeError } from "serialize-error";
@@ -7,7 +6,6 @@ import { log } from "./logging.js";
 import { handlers } from "./handlers.js";
 import { events } from "./events.js";
 const _errorNonExistingTarget = "Could not establish connection. Receiving end does not exist.";
-// https://github.com/mozilla/webextension-polyfill/issues/384
 const _errorTargetClosedEarly = "A listener indicated an asynchronous response by returning true, but the message channel closed before a response was received";
 export const errorTargetClosedEarly = "The target was closed before receiving a response";
 export const errorTabDoesntExist = "The tab doesn't exist";
@@ -100,9 +98,9 @@ async function manageMessage(type, target, seq, retry, sendMessage) {
                 String(error.message).startsWith("No handlers registered in ")))) {
                 throw error;
             }
-            if (browser.tabs && typeof target.tabId === "number") {
+            if (chrome.tabs && typeof target.tabId === "number") {
                 try {
-                    const tabInfo = await browser.tabs.get(target.tabId);
+                    const tabInfo = await chrome.tabs.get(target.tabId);
                     if (tabInfo.discarded) {
                         throw new Error(errorTabWasDiscarded);
                     }
@@ -155,15 +153,15 @@ function messenger(type, options, target, ...args) {
         }
         const sendMessage = async (attemptCount) => {
             log.debug(type, seq, "↗️ sending message to runtime", attemptLog(attemptCount));
-            return browser.runtime.sendMessage(makeMessage(type, args, target, options));
+            return chrome.runtime.sendMessage(makeMessage(type, args, target, options));
         };
         return manageConnection(type, options, target, sendMessage);
     }
     // Contexts without direct Tab access must go through background
-    if (!browser.tabs) {
+    if (!chrome.tabs) {
         return manageConnection(type, options, target, async (attemptCount) => {
             log.debug(type, seq, "↗️ sending message to runtime", attemptLog(attemptCount));
-            return browser.runtime.sendMessage(makeMessage(type, args, target, options));
+            return chrome.runtime.sendMessage(makeMessage(type, args, target, options));
         });
     }
     // `frameId` must be specified. If missing, the message is sent to every frame
@@ -171,7 +169,7 @@ function messenger(type, options, target, ...args) {
     // Message tab directly
     return manageConnection(type, options, target, async (attemptCount) => {
         log.debug(type, seq, "↗️ sending message to tab", tabId, "frame", frameId, attemptLog(attemptCount));
-        return browser.tabs.sendMessage(tabId, makeMessage(type, args, target, options), frameId === "allFrames"
+        return chrome.tabs.sendMessage(tabId, makeMessage(type, args, target, options), frameId === "allFrames"
             ? {}
             : {
                 frameId,

package/distribution/targetLogic.test.js

@@ -11,6 +11,11 @@ const tab = {
     pinned: false,
     highlighted: true,
     incognito: false,
+    discarded: false,
+    frozen: false,
+    selected: true,
+    autoDiscardable: false,
+    groupId: -1,
 };
 const senders = {
     background: { page: "background" },

package/distribution/types.d.ts

@@ -1,4 +1,3 @@
-import { type Runtime } from "webextension-polyfill";
 import { type Asyncify, type ValueOf } from "type-fest";
 import { type ErrorObject } from "serialize-error";
 /**
@@ -48,9 +47,7 @@ export type Message<LocalArguments extends Arguments = Arguments> = {
     /** If the message is being sent to an intermediary receiver, also set the options */
     options?: Options;
 };
-export type Sender = Runtime.MessageSender & {
-    origin?: string;
-};
+export type Sender = chrome.runtime.MessageSender;
 export type MessengerMessage = Message & {
     /** Guarantees that a message is meant to be handled by this library */
     __webextMessenger: true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "webext-messenger",
-  "version": "0.31.0",
+  "version": "0.32.0",
   "description": "Browser Extension component messaging framework",
   "keywords": [],
   "repository": "pixiebrix/webext-messenger",
@@ -29,9 +29,8 @@
     "p-event": "^6.0.1",
     "p-retry": "^6.2.1",
     "serialize-error": "^12.0.0",
-    "type-fest": "^4.33.0",
-    "webext-detect": "^5.3.2",
-    "webext-events": "^3.1.1"
+    "type-fest": "^4.36.0",
+    "webext-detect": "^5.3.2"
   },
   "@parcel/resolver-default": {
     "packageExports": true
@@ -39,9 +38,8 @@
   "devDependencies": {
     "@parcel/config-webextension": "^2.11.0",
     "@sindresorhus/tsconfig": "^7.0.0",
-    "@types/chrome": "^0.0.301",
+    "@types/chrome": "^0.0.307",
     "@types/tape": "^5.8.1",
-    "@types/webextension-polyfill": "^0.12.1",
     "buffer": "^6.0.3",
     "eslint": "^8.57.0",
     "eslint-config-pixiebrix": "^0.41.1",
@@ -52,10 +50,8 @@
     "process": "^0.11.10",
     "stream-browserify": "^3.0.0",
     "tape": "^5.9.0",
-    "typescript": "^5.7.3",
-    "vitest": "^3.0.5",
-    "webext-content-scripts": "^2.7.2",
-    "webextension-polyfill": "^0.12.0"
+    "typescript": "^5.8.2",
+    "vitest": "^3.0.7"
   },
   "targets": {
     "main": false,