From ff8a343a5428e855a2c3f87cc9399954cb1173d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Samy=20Pess=C3=A9?= <samypesse@gmail.com>
Date: Tue, 12 Dec 2023 10:56:07 +0100
Subject: [PATCH] Fix new cache on Cloudflare (#29)

* Try with next.revalidate = 0

* Remove log

* Format
---
 src/lib/api.ts    | 5 ++++-
 src/middleware.ts | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/lib/api.ts b/src/lib/api.ts
index 4ad18dd12..c5d0924e4 100644
--- a/src/lib/api.ts
+++ b/src/lib/api.ts
@@ -68,7 +68,10 @@ export const getPublishedContentByUrl = cache(
             secure: false,
             format: 'json',
             signal: signal,
-            cache: 'no-store',
+            // Cloudflare doesn't support the `cache` directive before next-on-pages patches the fetch function
+            // https://github.com/cloudflare/workerd/issues/698
+            // cache: 'no-store',
+            next: { revalidate: 0 },
         });
 
         return cacheResponse(response);
diff --git a/src/middleware.ts b/src/middleware.ts
index af0b0b704..948694480 100644
--- a/src/middleware.ts
+++ b/src/middleware.ts
@@ -53,6 +53,7 @@ export async function middleware(request: NextRequest) {
     console.log(`rendering ${resolved.space} ${resolved.pathname}`);
 
     const headers = new Headers(request.headers);
+    headers.set('origin', inputURL.origin);
     headers.set('x-gitbook-token', resolved.apiToken);
     headers.set('x-gitbook-basepath', joinPath(proxyBasePath, resolved.basePath));
     if (apiEndpoint) {