Allow mapping of entrypoint arguments in bindings

virtualstate · Dec 23, 2023 · ad5559f · ad5559f
1 parent 68ae650
commit ad5559f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/worker/service-worker/configure/types.ts b/src/worker/service-worker/configure/types.ts
@@ -41,6 +41,7 @@ export interface Service extends ScriptWorker, WorkerOptions {
 
 export interface ServiceEntrypoint extends Service {
     entrypoint?: string;
+    entrypointArguments?: string[];
 }
 
 export interface NamedService extends Service {
@@ -121,7 +122,8 @@ const example: Config = {
                     name: "./requested/resource.json",
                     service: {
                         name: "named",
-                        entrypoint: "getRequestedResource"
+                        entrypoint: "getRequestedResource",
+                        entrypointArguments: ["request", "$event", "SPECIAL_CONTEXT_KEY"]
                     }
                 },
                 {

diff --git a/src/worker/service-worker/execute-fetch.ts b/src/worker/service-worker/execute-fetch.ts
@@ -10,6 +10,7 @@ import {ServiceWorkerWorkerData} from "./worker";
 export interface ServiceWorkerFetchOptions {
     tag?: string;
     entrypoint?: string;
+    entrypointArguments?: string[];
     dispatch?: string | DurableEventData;
 }
 
@@ -66,6 +67,7 @@ export async function executeServiceWorkerFetch(registration: DurableServiceWork
         virtual: true,
         dispatch: options?.dispatch,
         entrypoint: options?.entrypoint,
+        entrypointArguments: options?.entrypointArguments
     }, serviceWorkerInit);
 }
 

diff --git a/src/worker/service-worker/worker-fetch.ts b/src/worker/service-worker/worker-fetch.ts
@@ -81,7 +81,8 @@ export function createServiceWorkerWorkerFetch(data: ServiceWorkerWorkerData, se
         const entrypoint = getBindingServiceEntrypoint(binding);
         return fetch(input, {
             ...init,
-            entrypoint: entrypoint.entrypoint
+            entrypoint: entrypoint.entrypoint,
+            entrypointArguments: entrypoint.entrypointArguments
         });
     }