@blotch/angular-lynx 22.2.7 → 22.2.8

package/fesm2022/blotch-angular-lynx.mjs

@@ -262,6 +262,7 @@ const callOnBackground = (name, ...args) => {
 	} catch (error) {
 		return Promise.reject(error);
 	}
+	listenInTestingEnvironment();
 	return new Promise((resolve, reject) => {
 		const id = nextCallId++;
 		pending.set(id, {
@@ -316,60 +317,74 @@ const unsubscribeSessionStorageFromMainThread = (relayId) => {
 	if (!sessionStorageListeners.delete(relayId)) return;
 	callOnBackground("unsubscribeSessionStorage", relayId).catch(() => {});
 };
-const installRelay = () => {
-	if (typeof lynx === "undefined") return;
-	try {
-		if (isMainThread()) {
-			const context = lynx.getJSContext();
-			context.addEventListener(RESULT_EVENT, (event) => {
-				const { id, value, error } = JSON.parse(event.data);
-				const call = pending.get(id);
-				if (!call) return;
-				pending.delete(id);
-				if (error !== void 0) call.reject(new Error(error));
-				else call.resolve(value);
-			});
-			context.addEventListener(SESSION_STORAGE_EVENT, (event) => {
-				const { relayId, value } = JSON.parse(event.data);
-				sessionStorageListeners.get(relayId)?.(value);
-			});
-			context.addEventListener(CALLBACK_EVENT, (event) => {
-				const { id, args } = JSON.parse(event.data);
-				relayedFunctions.get(id)?.(...args);
-			});
-			return;
+const relayContexts = /* @__PURE__ */ new WeakSet();
+const listenOnMainThread = (context) => {
+	if (relayContexts.has(context)) return;
+	relayContexts.add(context);
+	context.addEventListener(RESULT_EVENT, (event) => {
+		const { id, value, error } = JSON.parse(event.data);
+		const call = pending.get(id);
+		if (!call) return;
+		pending.delete(id);
+		if (error !== void 0) call.reject(new Error(error));
+		else call.resolve(value);
+	});
+	context.addEventListener(SESSION_STORAGE_EVENT, (event) => {
+		const { relayId, value } = JSON.parse(event.data);
+		sessionStorageListeners.get(relayId)?.(value);
+	});
+	context.addEventListener(CALLBACK_EVENT, (event) => {
+		const { id, args } = JSON.parse(event.data);
+		relayedFunctions.get(id)?.(...args);
+	});
+};
+const listenOnBackgroundThread = (context) => {
+	if (relayContexts.has(context)) return;
+	relayContexts.add(context);
+	context.addEventListener(CALL_EVENT, async (event) => {
+		const { id, name, args } = JSON.parse(event.data);
+		let reply;
+		try {
+			const handler = handlers[name];
+			if (!handler) throw new Error(`Unknown background call: ${name}`);
+			reply = {
+				id,
+				value: await handler(...args)
+			};
+		} catch (error) {
+			reply = {
+				id,
+				error: error instanceof Error ? error.message : String(error)
+			};
 		}
-		const context = lynx.getCoreContext();
-		context.addEventListener(CALL_EVENT, async (event) => {
-			const { id, name, args } = JSON.parse(event.data);
-			let reply;
-			try {
-				const handler = handlers[name];
-				if (!handler) throw new Error(`Unknown background call: ${name}`);
-				reply = {
-					id,
-					value: await handler(...args)
-				};
-			} catch (error) {
-				reply = {
-					id,
-					error: error instanceof Error ? error.message : String(error)
-				};
-			}
-			let data;
-			try {
-				data = JSON.stringify(reply);
-			} catch (error) {
-				data = JSON.stringify({
-					id,
-					error: `The result of ${name} can't be sent to the main thread: ${error.message}`
-				});
-			}
-			context.dispatchEvent({
-				type: RESULT_EVENT,
-				data
+		let data;
+		try {
+			data = JSON.stringify(reply);
+		} catch (error) {
+			data = JSON.stringify({
+				id,
+				error: `The result of ${name} can't be sent to the main thread: ${error.message}`
 			});
+		}
+		context.dispatchEvent({
+			type: RESULT_EVENT,
+			data
 		});
+	});
+};
+const listenInTestingEnvironment = () => {
+	const env = globalThis.lynxTestingEnv;
+	if (!env) return;
+	try {
+		listenOnMainThread(lynx.getJSContext());
+		listenOnBackgroundThread(env.backgroundThread.globalThis.lynx.getCoreContext());
+	} catch {}
+};
+const installRelay = () => {
+	if (typeof lynx === "undefined") return;
+	try {
+		if (isMainThread()) listenOnMainThread(lynx.getJSContext());
+		else listenOnBackgroundThread(lynx.getCoreContext());
 	} catch {}
 };
 installRelay();