forgot tools/workerlib

src/tools/testlib.ts

@@ -4,7 +4,7 @@
 
 import * as fs from 'fs';
 import type { WorkerResult, WorkerMessage, WorkerErrorResult, WorkerOutputResult } from "../common/workertypes";
-import { setupNodeEnvironment, handleMessage, store, TOOL_PRELOADFS } from "../worker/workerlib";
+import { setupNodeEnvironment, handleMessage, store, TOOL_PRELOADFS } from "./workerlib";
 import { PLATFORM_PARAMS } from "../worker/platforms";
 import { TOOLS } from "../worker/workertools";
 

src/tools/workerlib.ts

@@ -0,0 +1,140 @@
+
+// workerlib.ts - Node.js-friendly entry point for the worker build system
+// Re-exports core worker functionality without Web Worker onmessage/postMessage wiring
+
+import * as fs from 'fs';
+import * as path from 'path';
+import type { WorkerResult, WorkerMessage } from "../common/workertypes";
+import { getBasePlatform, getRootBasePlatform } from "../common/util";
+import { TOOL_PRELOADFS } from "./workertools";
+import { store, builder } from "./builder";
+import { emglobal, fsMeta, loadFilesystem } from "./wasmutils";
+import { setupRequireFunction } from "./workermain";
+
+export { store, builder, TOOL_PRELOADFS };
+export { PLATFORM_PARAMS } from "./platforms";
+export { TOOLS } from "./workertools";
+
+class Blob {
+  private data: string;
+  size: number;
+  length: number;
+  constructor(data: string) {
+    this.data = data;
+    this.size = data.length;
+    this.length = data.length;
+  }
+  slice(a: number, b: number): Blob {
+    return new Blob(this.data.slice(a, b));
+  }
+  arrayBuffer(): Uint8Array {
+    return this.asArrayBuffer();
+  }
+  asArrayBuffer(): Uint8Array {
+    var buf = new ArrayBuffer(this.data.length);
+    var arr = new Uint8Array(buf);
+    for (var i = 0; i < this.data.length; i++)
+      arr[i] = this.data.charCodeAt(i);
+    return arr;
+  }
+}
+
+/**
+ * Set up the Node.js environment to provide XMLHttpRequest, fetch, and other
+ * browser globals that the worker build system expects.
+ * Call this once before using handleMessage/builder.
+ */
+export function setupNodeEnvironment() {
+  // Basic globals expected by various parts of the worker system
+  emglobal.window = emglobal;
+  emglobal.exports = {};
+  emglobal.self = emglobal;
+  emglobal.location = { href: '.' };
+  emglobal.path = path;
+  emglobal.btoa = require('btoa');
+  emglobal.atob = require('atob');
+  try { emglobal.navigator = emglobal; } catch (e) { /* read-only in newer Node */ }
+  emglobal.ResizeObserver = class { observe() { } };
+
+  // XMLHttpRequest shim that reads from the local filesystem
+  emglobal.XMLHttpRequest = function () {
+    this.open = function (method: string, url: string, async?: boolean) {
+      if (this.responseType == 'json') {
+        var txt = fs.readFileSync('src/worker/' + url, 'utf-8');
+        this.response = JSON.parse(txt);
+      } else if (this.responseType == 'blob') {
+        var data = fs.readFileSync('src/worker/' + url, { encoding: 'binary' });
+        this.response = new Blob(data);
+      } else if (this.responseType == 'arraybuffer') {
+        var data = fs.readFileSync('src/worker/' + url, { encoding: 'binary' });
+        this.response = new Blob(data).asArrayBuffer();
+      }
+      this.status = this.response ? 200 : 404;
+    };
+    this.send = function () { };
+  };
+
+  // FileReaderSync shim
+  emglobal.FileReaderSync = function () {
+    this.readAsArrayBuffer = function (blob: any) {
+      return blob.asArrayBuffer();
+    };
+  };
+
+  // fetch shim that reads from the local filesystem
+  emglobal.fetch = function (filepath: string) {
+    return new Promise((resolve, reject) => {
+      try {
+        var bin = fs.readFileSync(filepath, { encoding: 'binary' });
+        var response = new Blob(bin);
+        resolve(response);
+      } catch (e) {
+        reject(e);
+      }
+    });
+  };
+
+  // importScripts shim - runs scripts in the global context like a web worker would.
+  // In the web worker, importScripts loads relative to the worker bundle.
+  // PWORKER is "../../src/worker/", so paths like "../../src/worker/asmjs/dasm.js"
+  // need to be resolved to the actual file on disk.
+  var vm = require('vm');
+  emglobal.importScripts = function (scriptPath: string) {
+    // Strip the PWORKER prefix and load from src/worker/
+    var resolved = scriptPath.replace(/^\.\.\/\.\.\//, '');
+    var fullPath = path.resolve(process.cwd(), resolved);
+    var code = fs.readFileSync(fullPath, 'utf-8');
+    vm.runInThisContext(code, fullPath);
+  };
+
+  // Suppress onmessage/postMessage (not used in Node mode)
+  emglobal.onmessage = null;
+  emglobal.postMessage = null;
+
+  // Set up the require function for WASM modules
+  setupRequireFunction();
+}
+
+/**
+ * Handle a worker message (preload, reset, or build).
+ * Same logic as workermain.ts handleMessage but exported for direct use.
+ */
+export async function handleMessage(data: WorkerMessage): Promise<WorkerResult> {
+  // preload file system
+  if (data.preload) {
+    var fsName = TOOL_PRELOADFS[data.preload];
+    if (!fsName && data.platform)
+      fsName = TOOL_PRELOADFS[data.preload + '-' + getBasePlatform(data.platform)];
+    if (!fsName && data.platform)
+      fsName = TOOL_PRELOADFS[data.preload + '-' + getRootBasePlatform(data.platform)];
+    if (fsName && !fsMeta[fsName])
+      loadFilesystem(fsName);
+    return;
+  }
+  // clear filesystem?
+  if (data.reset) {
+    store.reset();
+    return;
+  }
+  return builder.handleMessage(data);
+}