feat: add worker typescript support

Author: moonrailgun

src/client/components/CodeEditor/main.tsx

@@ -126,7 +126,7 @@ export const CodeEditor: React.FC<CodeEditorProps> = React.memo((props) => {
         readOnly,
         automaticLayout: true,
         scrollBeyondLastLine: false,
-        wordWrap: 'on',
+        wordWrap: 'off',
         wrappingIndent: 'indent',
         formatOnPaste: true,
         formatOnType: true,

src/client/components/worker/WorkerEditForm.tsx

@@ -67,7 +67,7 @@ const formSchema = z
     }
   );
 
-const defaultCode = `async function fetch(payload, ctx) {
+const defaultCode = `async function fetch(payload: Record<string, any>, ctx: FetchContext) {
   return 'Hello, World!';
 }
 `;

src/server/utils/vm/index.spec.ts

@@ -187,4 +187,17 @@ describe('runCodeInIVM', () => {
     expect(result.logger[0][5]).toEqual({ key: 'value' });
     expect(result.result).toBe('complete');
   });
+
+  test.only('should support typescript code', async () => {
+    const code = `
+      (async () => {
+        const a: number = 1;
+        const b: number = 2;
+
+        return a + b;
+      })()
+    `;
+    const result = await runCodeInIVM(code);
+    expect(result.result).toBe(3);
+  });
 });

src/server/utils/vm/index.ts

@@ -3,6 +3,7 @@ import { buildSandbox, environmentScript } from './sandbox.js';
 import { env } from '../env.js';
 import { runCodeInVM2 } from './sandbox-vm2.js';
 import { logger } from '../logger.js';
+import { transformTypescriptCode } from './utils.js';
 
 if (env.sandbox.useVM2) {
   logger.warn(
@@ -52,11 +53,12 @@ export async function runCodeInVM(_code: string): Promise<{
 export async function runCodeInIVM(_code: string) {
   const start = Date.now();
   const isolate = new ivm.Isolate({ memoryLimit: env.sandbox.memoryLimit });
+  const transformedCode = await transformTypescriptCode(_code);
 
   // avoid end comment with line break
   const code = `${environmentScript}
 
-${_code}`;
+${transformedCode}`;
 
   const [context, script] = await Promise.all([
     isolate.createContext(),

src/server/utils/vm/utils.ts

@@ -0,0 +1,22 @@
+import { transform as swcTransform } from '@swc/core';
+
+/**
+ * Transform Typescript code to ES6 code
+ */
+export async function transformTypescriptCode(sourceCode: string) {
+  const res = await swcTransform(sourceCode, {
+    jsc: {
+      parser: {
+        syntax: 'typescript',
+        tsx: false,
+      },
+      target: 'esnext',
+    },
+    module: {
+      type: 'es6',
+    },
+    minify: false,
+  });
+
+  return res.code;
+}