1
0
silverbullet/plugbox/environment/node_sandbox.ts

49 lines
1.0 KiB
TypeScript
Raw Normal View History

import { Worker } from "worker_threads";
2022-03-23 14:41:12 +00:00
import { safeRun } from "../util";
// @ts-ignore
2022-03-21 14:21:34 +00:00
import workerCode from "bundle-text:./node_worker.ts";
2022-03-23 14:41:12 +00:00
import { Sandbox } from "../sandbox";
import { WorkerLike } from "./worker";
2022-03-25 11:03:06 +00:00
import { Plug } from "../plug";
class NodeWorkerWrapper implements WorkerLike {
onMessage?: (message: any) => Promise<void>;
private worker: Worker;
2022-03-21 14:21:34 +00:00
ready: Promise<void>;
constructor(worker: Worker) {
this.worker = worker;
worker.on("message", (message: any) => {
safeRun(async () => {
await this.onMessage!(message);
});
});
2022-03-21 14:21:34 +00:00
this.ready = new Promise((resolve) => {
worker.once("online", resolve);
});
}
postMessage(message: any): void {
this.worker.postMessage(message);
}
terminate(): void {
this.worker.terminate();
}
}
2022-03-25 11:03:06 +00:00
export function createSandbox(plug: Plug<any>) {
2022-03-21 14:21:34 +00:00
let worker = new Worker(workerCode, {
eval: true,
});
return new Sandbox(
2022-03-25 11:03:06 +00:00
plug,
new NodeWorkerWrapper(
new Worker(workerCode, {
eval: true,
})
)
);
}