novac 2.0.1 → 2.2.0

package/kits/kitx11/kitx11_worker.js

@@ -0,0 +1,121 @@
+'use strict';
+// ============================================================
+//  kitx11_worker.js  —  Socket worker thread
+//
+//  Owns the TCP or Unix-domain socket to the X11 server.
+//  Uses a SharedArrayBuffer to signal connect success/failure
+//  back to the main thread synchronously via Atomics.
+//
+//  Message protocol (main → worker):
+//    { cmd: 'write', data: Uint8Array }        — send bytes (no reply)
+//    { cmd: 'read',  n: number, port: Port }   — read n bytes, reply on port
+//    { cmd: 'close' }                          — close and exit
+// ============================================================
+
+const { parentPort, workerData } = require('worker_threads');
+const net = require('net');
+
+const { display, signalSab } = workerData;
+const signal = new Int32Array(signalSab); // signal[0]: 0=waiting, 1=ok, 2=error
+
+let readBuf    = Buffer.alloc(0);
+let readWaiter = null; // { n, port }
+
+function tryFlushRead() {
+  if (!readWaiter) return;
+  const { n, port } = readWaiter;
+  if (readBuf.length >= n) {
+    const chunk = readBuf.slice(0, n);
+    readBuf     = readBuf.slice(n);
+    readWaiter  = null;
+    const arr   = new Uint8Array(chunk.buffer.slice(chunk.byteOffset, chunk.byteOffset + chunk.byteLength));
+    port.postMessage({ ok: true, data: arr }, [arr.buffer]);
+  }
+}
+
+function attachSocketHandlers(sock) {
+  sock.on('data', chunk => {
+    readBuf = Buffer.concat([readBuf, chunk]);
+    tryFlushRead();
+  });
+  sock.on('error', err => {
+    if (readWaiter) {
+      readWaiter.port.postMessage({ ok: false, error: err.message });
+      readWaiter = null;
+    }
+  });
+  sock.on('close', () => {
+    if (readWaiter) {
+      readWaiter.port.postMessage({ ok: false, error: 'Socket closed unexpectedly' });
+      readWaiter = null;
+    }
+  });
+}
+
+function signalError(msg) {
+  parentPort.postMessage({ ok: false, error: msg });
+  Atomics.store(signal, 0, 2);
+  Atomics.notify(signal, 0);
+}
+
+function signalConnected() {
+  Atomics.store(signal, 0, 1);
+  Atomics.notify(signal, 0);
+}
+
+// ── Parse DISPLAY string ─────────────────────────────────────
+const m = display.match(/^(.*):(\d+)(?:\.(\d+))?$/);
+if (!m) { signalError(`Cannot parse DISPLAY string: "${display}"`); process.exit(1); }
+
+const host       = m[1] || '';
+const displayNum = parseInt(m[2], 10);
+const unixPath   = `/tmp/.X11-unix/X${displayNum}`;
+const tcpPort    = 6000 + displayNum;
+const tcpHost    = host || '127.0.0.1';
+
+// ── Connect: try Unix socket first, then TCP ─────────────────
+let socket;
+
+function tryTCP() {
+  socket = net.createConnection({ host: tcpHost, port: tcpPort });
+  attachSocketHandlers(socket);
+  socket.once('connect', () => { listenForCommands(); signalConnected(); });
+  socket.once('error', err => {
+    signalError(
+      `Could not connect via unix:${unixPath} or tcp:${tcpHost}:${tcpPort} — ${err.message}`
+    );
+  });
+}
+
+socket = net.createConnection({ path: unixPath });
+attachSocketHandlers(socket);
+socket.once('connect', () => { listenForCommands(); signalConnected(); });
+socket.once('error', () => { socket.removeAllListeners(); tryTCP(); });
+
+// ── Command handler ───────────────────────────────────────────
+function listenForCommands() {
+  parentPort.on('message', ({ cmd, data, n, port }) => {
+    switch (cmd) {
+      case 'write': {
+        socket.write(Buffer.from(data.buffer, data.byteOffset, data.byteLength));
+        break;
+      }
+      case 'read': {
+        if (readBuf.length >= n) {
+          const chunk = readBuf.slice(0, n);
+          readBuf     = readBuf.slice(n);
+          const arr   = new Uint8Array(chunk.buffer.slice(chunk.byteOffset, chunk.byteOffset + chunk.byteLength));
+          port.postMessage({ ok: true, data: arr }, [arr.buffer]);
+        } else {
+          readWaiter = { n, port };
+        }
+        break;
+      }
+      case 'close': {
+        socket.destroy();
+        process.exit(0);
+        break;
+      }
+    }
+  });
+}