fix: misc fixes

MercuryWorkshop · Oct 26, 2024 · 5f03f8d · 5f03f8d
1 parent 4b32401
commit 5f03f8d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/client/dom/origin.ts b/src/client/dom/origin.ts
@@ -12,7 +12,7 @@ export default function (client: ScramjetClient, self: typeof window) {
 		},
 	});
 
-	client.Trap("document.URL", {
+	client.Trap("Document.prototype.URL", {
 		get() {
 			return client.url.href;
 		},
@@ -21,7 +21,7 @@ export default function (client: ScramjetClient, self: typeof window) {
 		},
 	});
 
-	client.Trap("document.documentURI", {
+	client.Trap("Document.prototype.documentURI", {
 		get() {
 			return client.url.href;
 		},
@@ -30,7 +30,7 @@ export default function (client: ScramjetClient, self: typeof window) {
 		},
 	});
 
-	client.Trap("document.domain", {
+	client.Trap("Document.prototype.domain", {
 		get() {
 			return client.url.hostname;
 		},

diff --git a/src/worker/fetch.ts b/src/worker/fetch.ts
@@ -65,10 +65,12 @@ export async function swfetch(
 			if (response.body) {
 				body = await rewriteBody(
 					response,
-					{
-						base: new URL(new URL(client.url).origin),
-						origin: new URL(new URL(client.url).origin),
-					},
+					client
+						? {
+								base: new URL(new URL(client.url).origin),
+								origin: new URL(new URL(client.url).origin),
+							}
+						: newmeta(new URL(unrewriteUrl(request.referrer))),
 					request.destination,
 					workertype,
 					this.cookieStore