@nimiq/core/nodejs/worker.mjs

Nimiq's Rust-to-WASM web client

// The worker has its own scope and no direct access to functions/objects of the
// global scope. We import the generated JS file to make `wasm_bindgen`
// available which we need to initialize our WASM code.
import { parentPort } from 'node:worker_threads';
import Comlink from 'comlink';
import nodeEndpoint from 'comlink/dist/esm/node-adapter.mjs';
import websocket from 'websocket';
import { Client } from './worker-wasm/index.js';

// WebSocket was added to Node in v22. Polyfill it for older versions.
if (!global.WebSocket) {
    console.debug("Polyfilling WebSocket");
    // Provide a global WebSocket implementation, which is expected by the WASM code built for browsers.
    global.WebSocket = websocket.w3cwebsocket;
}
// Workaround for Node.js as it currently lacks support for Web Workers by pretending there is
// a WorkerGlobalScope object available which is checked within the libp2p's websocket-websys transport.
global.WorkerGlobalScope = global;

// Defined both here and in main thread exports.js
Comlink.transferHandlers.set('function', {
    canHandle: (_obj) => false, // Cannot send functions to main thread
    deserialize(port) {
        return Comlink.transferHandlers.get('proxy').deserialize(port);
    },
});

Comlink.transferHandlers.set('plain', {
    canHandle: (_obj) => false, // Cannot send class instances to main thread
    deserialize(plain) {
        return plain;
    },
});

let initialized = false;

async function init(config) {
    if (initialized) throw new Error('Already initialized');
    initialized = true;

    console.log('Initializing client WASM worker');

    const client = await Client.create(config);
    Comlink.expose(client, nodeEndpoint(parentPort));
};

parentPort.addListener('message', async (data) => {
    if (data === 'NIMIQ_CHECKREADY') {
        parentPort.postMessage('NIMIQ_READY');
        return;
    }

    const { type, config } = data;
    if (type !== 'NIMIQ_INIT') return;

    if (!config || typeof config !== 'object') config = {};

    try {
        await init(config);
        parentPort.postMessage({ ok: true });
    } catch (error) {
        parentPort.postMessage({ ok: false, error: error.message, stack: error.stack });
    }
});

console.debug('Client WASM worker ready');