Sandbox (#1610) (#1611)

projects/sandbox/src/worker/utils.ts

@@ -0,0 +1,47 @@
+import { type MessagePort, Worker } from 'worker_threads';
+import * as path from 'path';
+
+export enum WorkerNameEnum {
+  runJs = 'runJs',
+  runPy = 'runPy'
+}
+
+type WorkerResponseType = { type: 'success' | 'error'; data: any };
+
+export const getWorker = (name: WorkerNameEnum) => {
+  const baseUrl =
+    process.env.NODE_ENV === 'production' ? 'projects/sandbox/dist/worker' : 'dist/worker';
+  const workerPath = path.join(process.cwd(), baseUrl, `${name}.js`);
+  return new Worker(workerPath);
+};
+
+export const runWorker = <T = any>(name: WorkerNameEnum, params?: Record<string, any>) => {
+  return new Promise<T>((resolve, reject) => {
+    const worker = getWorker(name);
+
+    worker.postMessage(params);
+
+    worker.on('message', (msg: WorkerResponseType) => {
+      if (msg.type === 'error') return reject(msg.data);
+
+      resolve(msg.data);
+      worker.terminate();
+    });
+
+    worker.on('error', (err) => {
+      reject(err);
+      worker.terminate();
+    });
+    worker.on('messageerror', (err) => {
+      reject(err);
+      worker.terminate();
+    });
+  });
+};
+
+export const workerResponse = ({
+  parentPort,
+  ...data
+}: WorkerResponseType & { parentPort?: MessagePort }) => {
+  parentPort?.postMessage(data);
+};