enable access to cloudflare runtime

withastro · Oct 21, 2022 · da2a317 · da2a317
1 parent 7e3a5e5
commit da2a317
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/.changeset/poor-frogs-clean.md b/.changeset/poor-frogs-clean.md
@@ -0,0 +1,5 @@
+---
+'@astrojs/cloudflare': patch
+---
+
+enable access to cloudflare runtime [KV, R2, Durable Objects]
diff --git a/packages/integrations/cloudflare/src/server.advanced.ts b/packages/integrations/cloudflare/src/server.advanced.ts
@@ -5,13 +5,15 @@ import { App } from 'astro/app';
 
 type Env = {
 	ASSETS: { fetch: (req: Request) => Promise<Response> };
+  name: string;
 };
 
 export function createExports(manifest: SSRManifest) {
 	const app = new App(manifest, false);
 
 	const fetch = async (request: Request, env: Env) => {
 		const { origin, pathname } = new URL(request.url);
+    env.name = "cloudflare";
 
 		// static assets
 		if (manifest.assets.has(pathname)) {
@@ -26,6 +28,11 @@ export function createExports(manifest: SSRManifest) {
 				Symbol.for('astro.clientAddress'),
 				request.headers.get('cf-connecting-ip')
 			);
+      Reflect.set(
+        request,
+        Symbol.for('runtime'),
+        env
+      );
 			let response = await app.render(request, routeData);
 
 			if (app.setCookieHeaders) {

diff --git a/packages/integrations/cloudflare/src/server.directory.ts b/packages/integrations/cloudflare/src/server.directory.ts
@@ -9,11 +9,13 @@ export function createExports(manifest: SSRManifest) {
 	const onRequest = async ({
 		request,
 		next,
+    ...runtimeEnv
 	}: {
 		request: Request;
 		next: (request: Request) => void;
-	}) => {
+	} & Record<string, unknown>) => {
 		const { origin, pathname } = new URL(request.url);
+    runtimeEnv.name = "cloudflare";
 
 		// static assets
 		if (manifest.assets.has(pathname)) {
@@ -28,6 +30,11 @@ export function createExports(manifest: SSRManifest) {
 				Symbol.for('astro.clientAddress'),
 				request.headers.get('cf-connecting-ip')
 			);
+      Reflect.set(
+        request,
+        Symbol.for('runtime'),
+        runtimeEnv
+      );
 			let response = await app.render(request, routeData);
 
 			if (app.setCookieHeaders) {