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 + ); + }, + }); }