@gitroll/tacitt-cli 0.0.0-stage → 0.1.0

package/dist/session-process.mjs

@@ -0,0 +1,90 @@
+import { Worker } from "node:worker_threads";
+import { getHeapStatistics } from "node:v8";
+//#region apps/cli/src/session-process.ts
+function stop() {
+	process.kill(process.pid, "SIGKILL");
+	throw new Error("PROCESS_TERMINATION_FAILED");
+}
+process.once("disconnect", stop);
+if (!process.connected) stop();
+const entry = process.argv[2];
+const heapMb = Number(process.argv[3]);
+if (!entry || !Number.isSafeInteger(heapMb) || heapMb < 1) stop();
+const heapLimit = getHeapStatistics().heap_size_limit;
+if (heapLimit < heapMb * 1024 ** 2 || heapLimit > (heapMb + 96) * 1024 ** 2) stop();
+const worker = new Worker(new URL(entry), {
+	stdout: true,
+	stderr: true,
+	resourceLimits: {
+		maxOldGenerationSizeMb: heapMb,
+		maxYoungGenerationSizeMb: 64
+	}
+});
+worker.stdout.resume();
+worker.stderr.resume();
+let busy = false;
+let terminal = false;
+let sending = false;
+const pending = [];
+function sendNext() {
+	if (sending) return;
+	const message = pending.shift();
+	if (!message) {
+		if (terminal) stop();
+		return;
+	}
+	if (!process.connected || !process.send) stop();
+	sending = true;
+	process.send(message, (error) => {
+		sending = false;
+		if (error) stop();
+		sendNext();
+	});
+}
+function enqueue(message) {
+	if (pending.length >= 4) stop();
+	pending.push(message);
+	sendNext();
+}
+function fail(code = "ARTIFACT_PROCESSING_FAILED") {
+	if (terminal) return;
+	terminal = true;
+	enqueue({
+		kind: "worker-error",
+		code
+	});
+}
+process.on("message", (message) => {
+	if (busy || terminal) {
+		fail();
+		return;
+	}
+	busy = true;
+	try {
+		worker.postMessage(message);
+	} catch {
+		fail();
+	}
+});
+worker.on("message", (message) => {
+	if (terminal) return;
+	if (!busy || typeof message !== "object" || message === null || !("kind" in message)) {
+		fail();
+		return;
+	}
+	if (message.kind === "source" && "bytes" in message && typeof message.bytes === "number") {
+		const last = pending.at(-1);
+		if (last && "kind" in last && last.kind === "source" && "bytes" in last && typeof last.bytes === "number") {
+			last.bytes += message.bytes;
+			return;
+		}
+	}
+	if (message.kind === "result" || message.kind === "error") busy = false;
+	enqueue(message);
+});
+worker.on("error", (error) => {
+	fail(Reflect.get(error, "code") === "ERR_WORKER_OUT_OF_MEMORY" ? "ARTIFACT_WORKER_MEMORY" : "ARTIFACT_PROCESSING_FAILED");
+});
+worker.on("exit", () => fail());
+//#endregion
+export {};