From ccefa04690d9e21ada4bc7666e8c78c7c4b3b682 Mon Sep 17 00:00:00 2001
From: Percs <83934299+Percslol@users.noreply.github.com>
Date: Mon, 9 Dec 2024 03:24:55 -0600
Subject: [PATCH] fix: fetch fixes, origin header

---
 src/worker/fetch.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/worker/fetch.ts b/src/worker/fetch.ts
index 323ca82..ab8d1f3 100644
--- a/src/worker/fetch.ts
+++ b/src/worker/fetch.ts
@@ -127,7 +127,10 @@ export async function handleFetch(
 				// console.log(headers);
 			} else {
 				headers.set("Referer", clientURL.toString());
-				headers.set("Origin", clientURL.origin);
+				headers.set(
+					"Origin",
+					clientURL.origin ? `${clientURL.protocol}//${clientURL.host}` : "null"
+				);
 			}
 		}
 
@@ -320,6 +323,8 @@ async function rewriteBody(
 		case "document":
 			if (response.headers.get("content-type")?.startsWith("text/html")) {
 				// note from percs: i think this has the potential to be slow asf, but for right now its fine (we should probably look for a better solution)
+				// another note from percs: regex seems to be broken, gonna comment this out
+				/*
 				const buf = await response.arrayBuffer();
 				const decode = new TextDecoder("utf-8").decode(buf);
 				const charsetHeader = response.headers.get("content-type");
@@ -330,8 +335,8 @@ async function rewriteBody(
 				const htmlContent = charset
 					? new TextDecoder(charset).decode(buf)
 					: decode;
-
-				return rewriteHtml(htmlContent, cookieStore, meta, true);
+				*/
+				return rewriteHtml(await response.text(), cookieStore, meta, true);
 			} else {
 				return response.body;
 			}