42 lines
1.0 KiB
TypeScript
42 lines
1.0 KiB
TypeScript
import { safeRun } from "../util.ts";
|
|
import { Sandbox } from "../sandbox.ts";
|
|
import { WorkerLike } from "./worker.ts";
|
|
import type { Plug } from "../plug.ts";
|
|
|
|
class WebWorkerWrapper implements WorkerLike {
|
|
private worker: Worker;
|
|
onMessage?: (message: any) => Promise<void>;
|
|
ready: Promise<void>;
|
|
|
|
constructor(worker: Worker) {
|
|
this.worker = worker;
|
|
this.worker.addEventListener("message", (evt: any) => {
|
|
let data = evt.data;
|
|
if (!data) return;
|
|
safeRun(async () => {
|
|
await this.onMessage!(data);
|
|
});
|
|
});
|
|
this.ready = Promise.resolve();
|
|
}
|
|
postMessage(message: any): void {
|
|
this.worker.postMessage(message);
|
|
}
|
|
|
|
terminate() {
|
|
return this.worker.terminate();
|
|
}
|
|
}
|
|
|
|
export function createSandbox(plug: Plug<any>) {
|
|
const worker = new Worker(
|
|
import.meta.url
|
|
? new URL("sandbox_worker.ts", import.meta.url)
|
|
: new URL("worker.js", location.origin),
|
|
{
|
|
type: "module",
|
|
},
|
|
);
|
|
return new Sandbox(plug, new WebWorkerWrapper(worker));
|
|
}
|