From 592e8d183fb270197621e55603db47b851983e61 Mon Sep 17 00:00:00 2001
From: velzie <velzie@velzie.rip>
Date: Thu, 10 Oct 2024 17:42:47 -0400
Subject: [PATCH] add style proxy

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

diff --git a/src/client/dom/element.ts b/src/client/dom/element.ts
index 526f40a..d949432 100644
--- a/src/client/dom/element.ts
+++ b/src/client/dom/element.ts
@@ -12,6 +12,10 @@ import {
 import type { URLMeta } from "../../shared/rewriters/url";
 
 export default function (client: ScramjetClient, self: typeof window) {
+	const nativeGetAttribute = self.Element.prototype.getAttribute;
+	const nativeSetAttribute = self.Element.prototype.setAttribute;
+	const nativeHasAttribute = self.Element.prototype.hasAttribute;
+
 	const attrObject = {
 		nonce: [self.HTMLElement],
 		integrity: [self.HTMLScriptElement, self.HTMLLinkElement],
@@ -60,6 +64,7 @@ export default function (client: ScramjetClient, self: typeof window) {
 				},
 
 				set(value) {
+					nativeSetAttribute.call(this, "data-scramjet-" + attr, value);
 					if (["nonce", "integrity", "csp"].includes(attr)) {
 						return;
 					} else if (
@@ -160,6 +165,39 @@ export default function (client: ScramjetClient, self: typeof window) {
 		},
 	});
 
+	client.Trap("HTMLElement.prototype.style", {
+		get(ctx) {
+			// unfortunate and dumb hack. we have to trap every property of this
+			// since the prototype chain is fucked
+
+			const style = ctx.get() as CSSStyleDeclaration;
+			return new Proxy(style, {
+				get(t, p) {
+					let v = Reflect.get(t, p);
+					if (typeof v === "function") {
+						return new Proxy(v, {
+							apply(target, thisArg, argArray) {
+								return Reflect.apply(target, style, argArray);
+							},
+						});
+					}
+
+					// todo properly unrewrite whatever
+					return v;
+				},
+				set(t, p, v) {
+					if (v == "" || typeof v !== "string") {
+						return Reflect.set(t, p, v);
+					}
+					return Reflect.set(t, p, rewriteCss(v, client.meta));
+				},
+			});
+		},
+		set(ctx, v: string) {
+			ctx.set(rewriteCss(v, client.meta));
+		},
+	});
+
 	client.Trap("Element.prototype.innerHTML", {
 		set(ctx, value: string) {
 			let newval;