@arkade-os/sdk 0.4.1 → 0.4.2

package/dist/cjs/worker/messageBus.js

@@ -9,7 +9,7 @@ const identity_1 = require("../identity");
 const wallet_1 = require("../wallet/wallet");
 const base_1 = require("@scure/base");
 class MessageBus {
-    constructor(walletRepository, contractRepository, { messageHandlers, tickIntervalMs = 10000, debug = false, buildServices, }) {
+    constructor(walletRepository, contractRepository, { messageHandlers, tickIntervalMs = 10000, messageTimeoutMs = 30000, debug = false, buildServices, }) {
         this.walletRepository = walletRepository;
         this.contractRepository = contractRepository;
         this.running = false;
@@ -17,8 +17,10 @@ class MessageBus {
         this.tickInProgress = false;
         this.debug = false;
         this.initialized = false;
+        this.boundOnMessage = this.onMessage.bind(this);
         this.handlers = new Map(messageHandlers.map((u) => [u.messageTag, u]));
         this.tickIntervalMs = tickIntervalMs;
+        this.messageTimeoutMs = messageTimeoutMs;
         this.debug = debug;
         this.buildServicesFn = buildServices ?? this.buildServices.bind(this);
     }
@@ -29,7 +31,7 @@ class MessageBus {
         if (this.debug)
             console.log("MessageBus starting");
         // Hook message routing
-        self.addEventListener("message", this.onMessage.bind(this));
+        self.addEventListener("message", this.boundOnMessage);
         // activate service worker immediately
         self.addEventListener("install", () => {
             self.skipWaiting();
@@ -52,7 +54,7 @@ class MessageBus {
             self.clearTimeout(this.tickTimeout);
             this.tickTimeout = null;
         }
-        self.removeEventListener("message", this.onMessage.bind(this));
+        self.removeEventListener("message", this.boundOnMessage);
         await Promise.all(Array.from(this.handlers.values()).map((updater) => updater.stop()));
     }
     scheduleNextTick() {
@@ -78,7 +80,7 @@ class MessageBus {
             const now = Date.now();
             for (const updater of this.handlers.values()) {
                 try {
-                    const response = await updater.tick(now);
+                    const response = await this.withTimeout(updater.tick(now), `${updater.messageTag}:tick`);
                     if (this.debug)
                         console.log(`[${updater.messageTag}] outgoing tick response:`, response);
                     if (response && response.length > 0) {
@@ -169,12 +171,25 @@ class MessageBus {
             throw new Error("Missing privateKey or publicKey in configuration object");
         }
     }
-    async onMessage(event) {
+    onMessage(event) {
+        // Keep the service worker alive while async work is pending.
+        // Without this, the browser may terminate the SW mid-operation,
+        // causing all pending responses to be lost silently.
+        const promise = this.processMessage(event);
+        if (typeof event.waitUntil === "function") {
+            event.waitUntil(promise);
+        }
+        return promise;
+    }
+    async processMessage(event) {
         const { id, tag, broadcast } = event.data;
         if (tag === "INITIALIZE_MESSAGE_BUS") {
             if (this.debug) {
                 console.log("Init Command received");
             }
+            // Intentionally not wrapped with withTimeout: initialization
+            // performs network calls (buildServices) and handler startup
+            // that may legitimately exceed the message timeout.
             await this.waitForInit(event.data.config);
             event.source?.postMessage({ id, tag });
             if (this.debug) {
@@ -208,7 +223,7 @@ class MessageBus {
         }
         if (broadcast) {
             const updaters = Array.from(this.handlers.values());
-            const results = await Promise.allSettled(updaters.map((updater) => updater.handleMessage(event.data)));
+            const results = await Promise.allSettled(updaters.map((updater) => this.withTimeout(updater.handleMessage(event.data), updater.messageTag)));
             results.forEach((result, index) => {
                 const updater = updaters[index];
                 if (result.status === "fulfilled") {
@@ -239,7 +254,7 @@ class MessageBus {
             return;
         }
         try {
-            const response = await updater.handleMessage(event.data);
+            const response = await this.withTimeout(updater.handleMessage(event.data), tag);
             if (this.debug)
                 console.log(`[${tag}] outgoing response:`, response);
             if (response) {
@@ -253,6 +268,27 @@ class MessageBus {
             event.source?.postMessage({ id, tag, error });
         }
     }
+    /**
+     * Race `promise` against a timeout. Note: this does NOT cancel the
+     * underlying work — the original promise keeps running. This is safe
+     * here because only the caller (not the handler) posts the response.
+     */
+    withTimeout(promise, label) {
+        if (this.messageTimeoutMs <= 0)
+            return promise;
+        return new Promise((resolve, reject) => {
+            const timer = self.setTimeout(() => {
+                reject(new Error(`Message handler timed out after ${this.messageTimeoutMs}ms (${label})`));
+            }, this.messageTimeoutMs);
+            promise.then((val) => {
+                self.clearTimeout(timer);
+                resolve(val);
+            }, (err) => {
+                self.clearTimeout(timer);
+                reject(err);
+            });
+        });
+    }
     /**
      * Returns the registered SW for the path.
      * It uses the functions in `service-worker-manager.ts` module.

package/dist/esm/worker/messageBus.js

@@ -6,7 +6,7 @@ import { ReadonlySingleKey, SingleKey } from '../identity/index.js';
 import { ReadonlyWallet, Wallet } from '../wallet/wallet.js';
 import { hex } from "@scure/base";
 export class MessageBus {
-    constructor(walletRepository, contractRepository, { messageHandlers, tickIntervalMs = 10000, debug = false, buildServices, }) {
+    constructor(walletRepository, contractRepository, { messageHandlers, tickIntervalMs = 10000, messageTimeoutMs = 30000, debug = false, buildServices, }) {
         this.walletRepository = walletRepository;
         this.contractRepository = contractRepository;
         this.running = false;
@@ -14,8 +14,10 @@ export class MessageBus {
         this.tickInProgress = false;
         this.debug = false;
         this.initialized = false;
+        this.boundOnMessage = this.onMessage.bind(this);
         this.handlers = new Map(messageHandlers.map((u) => [u.messageTag, u]));
         this.tickIntervalMs = tickIntervalMs;
+        this.messageTimeoutMs = messageTimeoutMs;
         this.debug = debug;
         this.buildServicesFn = buildServices ?? this.buildServices.bind(this);
     }
@@ -26,7 +28,7 @@ export class MessageBus {
         if (this.debug)
             console.log("MessageBus starting");
         // Hook message routing
-        self.addEventListener("message", this.onMessage.bind(this));
+        self.addEventListener("message", this.boundOnMessage);
         // activate service worker immediately
         self.addEventListener("install", () => {
             self.skipWaiting();
@@ -49,7 +51,7 @@ export class MessageBus {
             self.clearTimeout(this.tickTimeout);
             this.tickTimeout = null;
         }
-        self.removeEventListener("message", this.onMessage.bind(this));
+        self.removeEventListener("message", this.boundOnMessage);
         await Promise.all(Array.from(this.handlers.values()).map((updater) => updater.stop()));
     }
     scheduleNextTick() {
@@ -75,7 +77,7 @@ export class MessageBus {
             const now = Date.now();
             for (const updater of this.handlers.values()) {
                 try {
-                    const response = await updater.tick(now);
+                    const response = await this.withTimeout(updater.tick(now), `${updater.messageTag}:tick`);
                     if (this.debug)
                         console.log(`[${updater.messageTag}] outgoing tick response:`, response);
                     if (response && response.length > 0) {
@@ -166,12 +168,25 @@ export class MessageBus {
             throw new Error("Missing privateKey or publicKey in configuration object");
         }
     }
-    async onMessage(event) {
+    onMessage(event) {
+        // Keep the service worker alive while async work is pending.
+        // Without this, the browser may terminate the SW mid-operation,
+        // causing all pending responses to be lost silently.
+        const promise = this.processMessage(event);
+        if (typeof event.waitUntil === "function") {
+            event.waitUntil(promise);
+        }
+        return promise;
+    }
+    async processMessage(event) {
         const { id, tag, broadcast } = event.data;
         if (tag === "INITIALIZE_MESSAGE_BUS") {
             if (this.debug) {
                 console.log("Init Command received");
             }
+            // Intentionally not wrapped with withTimeout: initialization
+            // performs network calls (buildServices) and handler startup
+            // that may legitimately exceed the message timeout.
             await this.waitForInit(event.data.config);
             event.source?.postMessage({ id, tag });
             if (this.debug) {
@@ -205,7 +220,7 @@ export class MessageBus {
         }
         if (broadcast) {
             const updaters = Array.from(this.handlers.values());
-            const results = await Promise.allSettled(updaters.map((updater) => updater.handleMessage(event.data)));
+            const results = await Promise.allSettled(updaters.map((updater) => this.withTimeout(updater.handleMessage(event.data), updater.messageTag)));
             results.forEach((result, index) => {
                 const updater = updaters[index];
                 if (result.status === "fulfilled") {
@@ -236,7 +251,7 @@ export class MessageBus {
             return;
         }
         try {
-            const response = await updater.handleMessage(event.data);
+            const response = await this.withTimeout(updater.handleMessage(event.data), tag);
             if (this.debug)
                 console.log(`[${tag}] outgoing response:`, response);
             if (response) {
@@ -250,6 +265,27 @@ export class MessageBus {
             event.source?.postMessage({ id, tag, error });
         }
     }
+    /**
+     * Race `promise` against a timeout. Note: this does NOT cancel the
+     * underlying work — the original promise keeps running. This is safe
+     * here because only the caller (not the handler) posts the response.
+     */
+    withTimeout(promise, label) {
+        if (this.messageTimeoutMs <= 0)
+            return promise;
+        return new Promise((resolve, reject) => {
+            const timer = self.setTimeout(() => {
+                reject(new Error(`Message handler timed out after ${this.messageTimeoutMs}ms (${label})`));
+            }, this.messageTimeoutMs);
+            promise.then((val) => {
+                self.clearTimeout(timer);
+                resolve(val);
+            }, (err) => {
+                self.clearTimeout(timer);
+                reject(err);
+            });
+        });
+    }
     /**
      * Returns the registered SW for the path.
      * It uses the functions in `service-worker-manager.ts` module.

package/dist/types/worker/messageBus.d.ts

@@ -48,6 +48,7 @@ export interface MessageHandler<REQ extends RequestEnvelope = RequestEnvelope, R
 type Options = {
     messageHandlers: MessageHandler[];
     tickIntervalMs?: number;
+    messageTimeoutMs?: number;
     debug?: boolean;
     buildServices?: (config: Initialize["config"]) => Promise<{
         arkProvider: ArkProvider;
@@ -76,13 +77,15 @@ export declare class MessageBus {
     private readonly contractRepository;
     private handlers;
     private tickIntervalMs;
+    private messageTimeoutMs;
     private running;
     private tickTimeout;
     private tickInProgress;
     private debug;
     private initialized;
     private readonly buildServicesFn;
-    constructor(walletRepository: WalletRepository, contractRepository: ContractRepository, { messageHandlers, tickIntervalMs, debug, buildServices, }: Options);
+    private readonly boundOnMessage;
+    constructor(walletRepository: WalletRepository, contractRepository: ContractRepository, { messageHandlers, tickIntervalMs, messageTimeoutMs, debug, buildServices, }: Options);
     start(): Promise<void>;
     stop(): Promise<void>;
     private scheduleNextTick;
@@ -90,6 +93,13 @@ export declare class MessageBus {
     private waitForInit;
     private buildServices;
     private onMessage;
+    private processMessage;
+    /**
+     * Race `promise` against a timeout. Note: this does NOT cancel the
+     * underlying work — the original promise keeps running. This is safe
+     * here because only the caller (not the handler) posts the response.
+     */
+    private withTimeout;
     /**
      * Returns the registered SW for the path.
      * It uses the functions in `service-worker-manager.ts` module.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arkade-os/sdk",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "description": "Bitcoin wallet SDK with Taproot and Ark integration",
   "type": "module",
   "main": "./dist/cjs/index.js",