Fix InternalExecution interface

.changeset/long-dancers-judge.md

@@ -0,0 +1,5 @@
+---
+"@defer/client": patch
+---
+
+Fix InternalExecution interface

src/backend/local.ts

@@ -47,7 +47,7 @@ import { KV } from "./local/kv.js";
 interface InternalExecution<F extends DeferableFunction> {
   id: string;
   args: string;
-  func: F;
+  func: DeferredFunction<F>;
   functionId: string;
   functionName: string;
   state: ExecutionState;
@@ -232,9 +232,7 @@ async function loop(shouldRun: () => boolean): Promise<void> {
         const execution = await executionsStore.transaction(
           executionId,
           async (execution) => {
-            const func = execution.func as DeferredFunction<
-              typeof execution.func
-            >;
+            const func = execution.func;
             shouldDiscard =
               execution.state === "created" &&
               execution.discardAfter !== undefined &&
@@ -348,7 +346,7 @@ export async function enqueue<F extends DeferableFunction>(
   }
 
   const now = new Date();
-  const execution: InternalExecution<any> = {
+  const execution: InternalExecution<F> = {
     id: randomUUID(),
     state: "created",
     functionId: functionId,
@@ -464,7 +462,7 @@ export async function reRunExecution(
     throw new ExecutionNotFound(`cannot find execution "${id}"`);
 
   const now = new Date();
-  const newExecution: InternalExecution<any> = {
+  const newExecution: InternalExecution<typeof execution.func.__fn> = {
     id: randomUUID(),
     state: "created",
     functionId: execution.functionId,
@@ -490,12 +488,10 @@ export async function listExecutions(
   const data = new Map<string, Execution>();
 
   for (const executionId of executionIds) {
-    const execution = (await executionsStore.get(
-      executionId,
-    )) as InternalExecution<any>;
+    const execution = await executionsStore.get(executionId);
 
-    if (isExecutionMatchFilter(filters, execution))
-      data.set(executionId, buildExecution(execution));
+    if (isExecutionMatchFilter(filters, execution!))
+      data.set(executionId, buildExecution(execution!));
   }
 
   return paginate(pageRequest, data);
@@ -510,15 +506,13 @@ export async function listExecutionAttempts(
   const data = new Map<string, Execution>();
 
   for (const executionId of executionIds) {
-    const execution = (await executionsStore.get(
-      executionId,
-    )) as InternalExecution<any>;
+    const execution = await executionsStore.get(executionId);
 
     if (
-      (execution.id === id || execution.retryOf === id) &&
-      isExecutionMatchFilter(filters, execution)
+      (execution!.id === id || execution!.retryOf === id) &&
+      isExecutionMatchFilter(filters, execution!)
     )
-      data.set(executionId, buildExecution(execution));
+      data.set(executionId, buildExecution(execution!));
   }
 
   return paginate(pageRequest, data);

src/index.ts

@@ -467,10 +467,10 @@ export async function listExecutions(
  * @throws {DeferError} when error is unknown
  * @returns {Promise<Awaited<F>>}
  */
-export function awaitResult<F extends DeferableFunction>(
+export function awaitResult<F extends DeferableFunction, R = ReturnType<F>>(
   fn: DeferredFunction<F>,
-): (...args: Parameters<F>) => Promise<Awaited<F>> {
-  return async function (...args: Parameters<F>): Promise<Awaited<F>> {
+): (...args: Parameters<F>) => Promise<Awaited<R>> {
+  return async function (...args: Parameters<F>): Promise<Awaited<R>> {
     const enqueueResponse = await enqueue(fn, ...args);
     await sleep(1000);
 
@@ -493,7 +493,7 @@ export function awaitResult<F extends DeferableFunction>(
           throw error;
         }
         case "succeed":
-          return await getExecutionResult<Awaited<F>>(enqueueResponse.id);
+          return await getExecutionResult<Awaited<R>>(enqueueResponse.id);
         case "aborted":
         case "cancelled":
         case "discarded":