libzapitu 1.0.0-alpha.25 → 1.0.0-alpha.26

package/lib/worker/child.js

@@ -77,8 +77,8 @@ function proxyCallback(socketId, key, log, ...args) {
 // ---------------------------------------------------------------------------
 /**
  * Recursively clone a value for postMessage, replacing non-cloneable types
- * (Error, Function) with plain serializable representations and converting
- * protobufjs message instances to JSON.
+ * (Error, Function) with plain serializable representations, preserving
+ * built-in value objects, and converting protobufjs message instances to JSON.
  *
  * Protobuf messages (e.g. proto.Message, proto.WebMessageInfo) carry a
  * toJSON() method that serializes byte fields as base64 strings and 64-bit
@@ -90,11 +90,42 @@ function proxyCallback(socketId, key, log, ...args) {
  */
 function serializeForPostMessage(value, seen = new WeakSet()) {
     if (value instanceof Error) {
+        // Boom carries critical metadata (statusCode, output, data, isServer).
+        // Preserve it so the parent can reconstruct a real Boom instance.
+        if (value.isBoom) {
+            return {
+                __error__: true,
+                __boom__: true,
+                message: value.message,
+                name: value.name,
+                stack: value.stack,
+                data: value.data,
+                output: value.output,
+                isServer: value.isServer
+            };
+        }
         return { __error__: true, message: value.message, name: value.name, stack: value.stack };
     }
     if (typeof value === 'function') {
         return '__fn__';
     }
+    // Preserve built-in value objects that structured clone handles, but tag
+    // them so the parent can reconstruct the exact same runtime type.
+    if (value instanceof Date) {
+        return { __date__: true, value: value.toISOString() };
+    }
+    if (value instanceof RegExp) {
+        return { __regexp__: true, source: value.source, flags: value.flags };
+    }
+    if (value instanceof URL) {
+        return { __url__: true, href: value.href };
+    }
+    if (value instanceof Map) {
+        return { __map__: true, entries: serializeForPostMessage(Array.from(value.entries()), seen) };
+    }
+    if (value instanceof Set) {
+        return { __set__: true, entries: serializeForPostMessage(Array.from(value.values()), seen) };
+    }
     if (value !== null && typeof value === 'object') {
         if (seen.has(value))
             return '[Circular]';
@@ -214,6 +245,8 @@ function createSocket(socketId, rawConfig) {
     log.info('socket created successfully');
     // Send user and authState to the parent when they become available.
     // These are needed for wsocket.user and wsocket.authState on the proxy side.
+    // authState.keys contains forwarding functions, so we must serialize the
+    // info before postMessage or the whole message is dropped.
     const sendSocketInfo = () => {
         try {
             const info = {};
@@ -222,7 +255,11 @@ function createSocket(socketId, rawConfig) {
             if (sock.authState)
                 info.authState = sock.authState;
             if (Object.keys(info).length > 0) {
-                worker_threads_1.parentPort.postMessage({ type: 'socket-info', socketId, info });
+                worker_threads_1.parentPort.postMessage({
+                    type: 'socket-info',
+                    socketId,
+                    info: serializeForPostMessage(info)
+                });
             }
         }
         catch (_) {

package/lib/worker/proxy.js

@@ -36,6 +36,7 @@ const stream_1 = require("stream");
 const util_1 = require("util");
 const os_2 = require("os");
 const long_1 = __importDefault(require("long"));
+const boom_1 = require("@hapi/boom");
 const WAProto_1 = require("../../WAProto");
 const pipelineAsync = (0, util_1.promisify)(stream_1.pipeline);
 // ---------------------------------------------------------------------------
@@ -254,6 +255,7 @@ const PROTOBUF_TYPE_REGISTRY = buildProtobufTypeRegistry(WAProto_1.proto);
  * to direct mode (base64 strings for bytes, strings for 64-bit integers).
  */
 function revivePostMessage(value) {
+    var _a;
     // After structured clone, Buffer subclasses become plain Uint8Arrays.
     // Restore them to Buffer so downstream code gets the same type as in
     // direct mode for non-protobuf binary fields.
@@ -262,11 +264,42 @@ function revivePostMessage(value) {
     }
     if (value !== null && typeof value === 'object') {
         if (value.__error__) {
+            // Boom errors carry status codes, payloads, and extra data. Rebuild
+            // a real Boom instance so callers can inspect err.output.statusCode.
+            if (value.__boom__) {
+                const boom = new boom_1.Boom(value.message, {
+                    statusCode: (_a = value.output) === null || _a === void 0 ? void 0 : _a.statusCode,
+                    data: value.data
+                });
+                // Restore the exact output payload/headers from the worker so the
+                // status code and error name match direct mode.
+                if (value.output) {
+                    Object.assign(boom.output, value.output);
+                }
+                boom.stack = value.stack || boom.stack;
+                return boom;
+            }
             const err = new Error(value.message);
             err.name = value.name;
             err.stack = value.stack;
             return err;
         }
+        // Revive built-in value objects tagged by the child.
+        if (value.__date__) {
+            return new Date(value.value);
+        }
+        if (value.__regexp__) {
+            return new RegExp(value.source, value.flags);
+        }
+        if (value.__url__) {
+            return new URL(value.href);
+        }
+        if (value.__map__) {
+            return new Map(revivePostMessage(value.entries));
+        }
+        if (value.__set__) {
+            return new Set(revivePostMessage(value.entries));
+        }
         // Revive protobuf messages tagged by the child. fromObject() will
         // recursively recreate nested protobuf instances and convert base64
         // strings back into Buffer instances.
@@ -279,7 +312,7 @@ function revivePostMessage(value) {
                 try {
                     return type.fromObject(copy);
                 }
-                catch (_a) {
+                catch (_b) {
                     // Fall back to the plain JSON object if fromObject fails;
                     // downstream code will see base64 strings instead of Buffers.
                 }
@@ -292,7 +325,7 @@ function revivePostMessage(value) {
             return new long_1.default(value.low, value.high, value.unsigned);
         }
         if (Array.isArray(value)) {
-            // Arrays: recurse to revive any Error sentinels / Longs / Uint8Arrays / protobuf messages inside
+            // Arrays: recurse to revive any Error sentinels / Longs / Uint8Arrays / protobuf messages / built-ins inside
             let changed = false;
             const result = value.map(v => {
                 const revived = revivePostMessage(v);
@@ -303,7 +336,7 @@ function revivePostMessage(value) {
             return changed ? result : value;
         }
         // Plain objects: only recurse if they might contain __error__ sentinels,
-        // Long-like objects, Uint8Arrays, or protobuf markers.
+        // Long-like objects, Uint8Arrays, protobuf markers, or built-in tags.
         if (value.constructor === Object || value.constructor === undefined) {
             let changed = false;
             const result = {};
@@ -515,7 +548,12 @@ async function handleCallbackCall(slot, socketId, msg) {
         try {
             log === null || log === void 0 ? void 0 : log.debug({ keystoreMethod: method, argCount: args.length }, 'invoking keystore method');
             const result = await keystore[method](...args);
-            slot.worker.postMessage({ type: 'callback-result', socketId, id, result });
+            slot.worker.postMessage({
+                type: 'callback-result',
+                socketId,
+                id,
+                result: deepStripFunctions(result)
+            });
         }
         catch (err) {
             log === null || log === void 0 ? void 0 : log.error({ err, keystoreMethod: method }, 'keystore method threw error');
@@ -543,7 +581,12 @@ async function handleCallbackCall(slot, socketId, msg) {
     try {
         log === null || log === void 0 ? void 0 : log.debug({ callbackKey: key, argCount: args.length }, 'invoking proxied callback');
         const result = await fn(...args);
-        slot.worker.postMessage({ type: 'callback-result', socketId, id, result });
+        slot.worker.postMessage({
+            type: 'callback-result',
+            socketId,
+            id,
+            result: deepStripFunctions(result)
+        });
     }
     catch (err) {
         log === null || log === void 0 ? void 0 : log.error({ err, callbackKey: key }, 'proxied callback threw error');
@@ -616,8 +659,30 @@ async function drainPool() {
  * Recursively clone a value, replacing all functions with the sentinel
  * string '__proxy_fn__'. This ensures the config can be sent via
  * postMessage (structured clone).
+ *
+ * Built-in value objects (Date, RegExp, URL, Map, Set) are preserved in
+ * their serializable tagged form so they survive the boundary instead of
+ * being reduced to empty objects by Object.keys().
  */
 function deepStripFunctions(value, seen = new WeakSet()) {
+    if (typeof value === 'function') {
+        return '__proxy_fn__';
+    }
+    if (value instanceof Date) {
+        return { __date__: true, value: value.toISOString() };
+    }
+    if (value instanceof RegExp) {
+        return { __regexp__: true, source: value.source, flags: value.flags };
+    }
+    if (value instanceof URL) {
+        return { __url__: true, href: value.href };
+    }
+    if (value instanceof Map) {
+        return { __map__: true, entries: deepStripFunctions(Array.from(value.entries()), seen) };
+    }
+    if (value instanceof Set) {
+        return { __set__: true, entries: deepStripFunctions(Array.from(value.values()), seen) };
+    }
     if (typeof value === 'function') {
         return '__proxy_fn__';
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "libzapitu",
-  "version": "1.0.0-alpha.25",
+  "version": "1.0.0-alpha.26",
   "description": "A WebSockets library for interacting with WhatsApp Web",
   "keywords": [
     "whatsapp",