fix: cloudflare waitUntil not working as inteded with getRuntime (#7419)

Co-authored-by: Bjorn Lu <bjornlu.dev@gmail.com>

2 files changed, 10 insertions, 3 deletions

diff --git a/packages/integrations/cloudflare/src/server.advanced.ts b/packages/integrations/cloudflare/src/server.advanced.ts
index 510e4e7e5..9758b8b19 100644
--- a/packages/integrations/cloudflare/src/server.advanced.ts
+++ b/packages/integrations/cloudflare/src/server.advanced.ts
@@ -1,4 +1,4 @@
-import type { Request as CFRequest } from '@cloudflare/workers-types';
+import type { Request as CFRequest, ExecutionContext } from '@cloudflare/workers-types';
 import type { SSRManifest } from 'astro';
 import { App } from 'astro/app';
 import { getProcessEnvProxy, isNode } from './util.js';
@@ -15,7 +15,7 @@ type Env = {
 export function createExports(manifest: SSRManifest) {
 	const app = new App(manifest);
 
-	const fetch = async (request: Request & CFRequest, env: Env, context: any) => {
+	const fetch = async (request: Request & CFRequest, env: Env, context: ExecutionContext) => {
 		process.env = env as any;
 
 		const { pathname } = new URL(request.url);
@@ -38,6 +38,9 @@ export function createExports(manifest: SSRManifest) {
 				caches,
 				cf: request.cf,
 				...context,
+				waitUntil: (promise: Promise<any>) => {
+					context.waitUntil(promise);
+				},
 			});
 			let response = await app.render(request, routeData);
 
diff --git a/packages/integrations/cloudflare/src/server.directory.ts b/packages/integrations/cloudflare/src/server.directory.ts
index a0f3299fd..649d0d99b 100644
--- a/packages/integrations/cloudflare/src/server.directory.ts
+++ b/packages/integrations/cloudflare/src/server.directory.ts
@@ -1,4 +1,4 @@
-import type { Request as CFRequest } from '@cloudflare/workers-types';
+import type { Request as CFRequest, EventContext } from '@cloudflare/workers-types';
 import type { SSRManifest } from 'astro';
 import { App } from 'astro/app';
 import { getProcessEnvProxy, isNode } from './util.js';
@@ -17,6 +17,7 @@ export function createExports(manifest: SSRManifest) {
 	}: {
 		request: Request & CFRequest;
 		next: (request: Request) => void;
+		waitUntil: EventContext<unknown, any, unknown>['waitUntil'];
 	} & Record<string, unknown>) => {
 		process.env = runtimeEnv.env as any;
 
@@ -35,6 +36,9 @@ export function createExports(manifest: SSRManifest) {
 			);
 			Reflect.set(request, Symbol.for('runtime'), {
 				...runtimeEnv,
+				waitUntil: (promise: Promise<any>) => {
+					runtimeEnv.waitUntil(promise);
+				},
 				name: 'cloudflare',
 				next,
 				caches,