From 6e9ce2238c85b809089154547f53966a382c5fa2 Mon Sep 17 00:00:00 2001
From: velzie <velzie@velzie.rip>
Date: Sun, 1 Sep 2024 20:39:33 -0400
Subject: [PATCH] document.write why not

---
 src/client/dom/element.ts | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/client/dom/element.ts b/src/client/dom/element.ts
index ce29ed2..d534cec 100644
--- a/src/client/dom/element.ts
+++ b/src/client/dom/element.ts
@@ -234,4 +234,27 @@ export default function (client: ScramjetClient, self: typeof window) {
 			return scram.documentProxy;
 		},
 	});
+
+	client.Proxy("document.write", {
+		apply(ctx) {
+			ctx.args[0] = rewriteHtml(
+				ctx.args[0],
+				client.cookieStore,
+				client.meta,
+				true
+			);
+		},
+	});
+
+	client.Proxy("document.writeln", {
+		apply(ctx) {
+			// this injects scramjet multiple times but who gaf
+			ctx.args[0] = rewriteHtml(
+				ctx.args[0],
+				client.cookieStore,
+				client.meta,
+				true
+			);
+		},
+	});
 }