feat: add worker runtime context

Author: moonrailgun

src/client/components/CodeEditor/lib/sandbox.ts

@@ -50,7 +50,27 @@ interface RequestReturn {
   status: number;
 }
 
+interface FetchContext {
+  type: 'http' | 'cron' | 'manual' | 'test';
+  request?: {
+    method: string;
+    url: string;
+    headers: Record<string, string>;
+  };
+}
+
+interface Console {
+  log(...data: any[]): void;
+  info(...data: any[]): void;
+  warn(...data: any[]): void;
+  error(...data: any[]): void;
+}
+
+declare const console: Console;
+
 declare function request(config: AxiosConfig): Promise<RequestReturn>;
 
 const request = async (config: AxiosConfig): Promise<RequestReturn> => {};
+
+declare function fetch(params: Record<string, any>, ctx: FetchContext): Promise<string>;
 `;

src/client/components/CodeEditor/main.tsx

@@ -42,6 +42,18 @@ export const CodeEditor: React.FC<CodeEditorProps> = React.memo((props) => {
     editorRef.current = editor;
     monacoRef.current = monaco;
 
+    const tsDefaults = monaco.languages.typescript.typescriptDefaults;
+    tsDefaults.setCompilerOptions({
+      ...tsDefaults.getCompilerOptions(),
+      lib: ['es2021'],
+    });
+
+    const jsDefaults = monaco.languages.typescript.javascriptDefaults;
+    jsDefaults.setCompilerOptions({
+      ...jsDefaults.getCompilerOptions(),
+      lib: ['es2021'],
+    });
+
     monaco.languages.typescript.javascriptDefaults.addExtraLib(
       sandboxGlobal,
       'global.ts'

src/client/components/worker/WorkerExecutionDetail.tsx

@@ -120,7 +120,7 @@ export const WorkerExecutionDetail: React.FC<WorkerExecutionDetailProps> =
                                   {formatDate(log[1])}
                                 </span>
                               </div>
-                              <span>
+                              <span className="overflow-x-auto">
                                 {drop(log, 2).map((item, i) => (
                                   <DataRender
                                     key={i}
@@ -137,7 +137,7 @@ export const WorkerExecutionDetail: React.FC<WorkerExecutionDetailProps> =
                               <span className="opacity-60">
                                 {formatDate(log[1])}
                               </span>
-                              <span>
+                              <span className="overflow-x-auto">
                                 {drop(log, 2).map((item, i) => (
                                   <DataRender
                                     key={i}

src/server/model/worker/cronRunner.ts

@@ -35,7 +35,9 @@ export class WorkerCronRunner {
       );
 
       // Execute the worker
-      const result = await execWorker(worker.code, worker.id);
+      const result = await execWorker(worker.code, worker.id, undefined, {
+        type: 'cron',
+      });
 
       logger.info(
         `[Worker Cron] Worker ${worker.name}(${worker.id}) executed successfully`

src/server/model/worker/index.ts

@@ -9,18 +9,20 @@ import { isPlainObject } from 'lodash-es';
 export async function execWorker(
   code: string,
   workerId?: string,
-  requestPayload?: Record<string, any>
+  requestPayload?: Record<string, any>,
+  context?: Record<string, any>
 ) {
   const requestPayloadString = isPlainObject(requestPayload)
     ? JSON.stringify(requestPayload)
     : '{}';
+  const contextString = isPlainObject(context) ? JSON.stringify(context) : '{}';
 
   try {
     const { isolate, logger, result, usage } = await runCodeInIVM(`
       (async () => {
         ${code}
 
-        return typeof fetch === 'function' ? fetch(${requestPayloadString}) : 'fetch is not defined';
+        return typeof fetch === 'function' ? fetch(${requestPayloadString}, ${contextString}) : 'fetch is not defined';
       })()
     `);
 

src/server/router/worker.ts

@@ -51,7 +51,19 @@ workerRouter.all(
       }
 
       // Execute the worker
-      const execution = await execWorker(worker.code, workerId, requestPayload);
+      const execution = await execWorker(
+        worker.code,
+        workerId,
+        requestPayload,
+        {
+          type: 'http',
+          request: {
+            method: req.method,
+            url: req.url,
+            headers: { ...req.headers },
+          },
+        }
+      );
 
       const response = execution.responsePayload;
 

src/server/trpc/routers/worker.ts

@@ -252,7 +252,9 @@ export const workerRouter = router({
         throw new Error('Worker not found');
       }
 
-      const execution = await execWorker(worker.code, workerId);
+      const execution = await execWorker(worker.code, workerId, undefined, {
+        type: 'manual',
+      });
 
       await createAuditLog({
         workspaceId,
@@ -407,7 +409,9 @@ export const workerRouter = router({
         throw new Error('Function worker is not enabled');
       }
 
-      const execution = await execWorker(code);
+      const execution = await execWorker(code, undefined, undefined, {
+        type: 'test',
+      });
 
       return execution;
     }),