From 2c1e2b7adfc428096a530f7a5310f01da75bedd5 Mon Sep 17 00:00:00 2001
From: velzie <velzie@velzie.rip>
Date: Thu, 18 Jul 2024 21:34:08 -0400
Subject: [PATCH] handlers i'll delete later

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

diff --git a/src/client/element.ts b/src/client/element.ts
index cc7162f..f319606 100644
--- a/src/client/element.ts
+++ b/src/client/element.ts
@@ -6,6 +6,7 @@ import {
 	rewriteJs,
 	rewriteSrcset,
 } from "./shared";
+import { documentProxy } from "./window";
 
 const attrObject = {
 	nonce: [HTMLElement],
@@ -125,3 +126,19 @@ Object.defineProperty(Element.prototype, "innerHTML", {
 		return innerHTML.set.call(this, value);
 	},
 });
+
+MutationObserver.prototype.observe = new Proxy(MutationObserver.prototype.observe, {
+	apply(target, thisArg, argArray) {
+		if (argArray[0] === documentProxy) argArray[0] = document;
+
+		return Reflect.apply(target, thisArg, argArray);
+	}
+});
+
+document.createTreeWalker = new Proxy(document.createTreeWalker, {
+	apply(target, thisArg, argArray) {
+		if (argArray[0] === documentProxy) argArray[0] = document;
+
+		return Reflect.apply(target, thisArg, argArray);
+	}
+});