From 4f7f633345eea8ff1ace99f48aed1cd989557565 Mon Sep 17 00:00:00 2001
From: Percs <83934299+Percslol@users.noreply.github.com>
Date: Sat, 14 Dec 2024 01:29:26 -0600
Subject: [PATCH] feat: add annotations to all calls of rewritejs

---
 rewriter/wasm/build.sh        | 3 ++-
 src/client/shared/eval.ts     | 6 ++++--
 src/client/shared/function.ts | 6 +++++-
 src/shared/rewriters/url.ts   | 3 ++-
 4 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/rewriter/wasm/build.sh b/rewriter/wasm/build.sh
index 26b0a9a..56e36ba 100644
--- a/rewriter/wasm/build.sh
+++ b/rewriter/wasm/build.sh
@@ -32,8 +32,9 @@ cd ../../
 # shellcheck disable=SC2086
 time wasm-opt $WASMOPTFLAGS --converge -tnh -O4 --vacuum --dce --enable-threads --enable-bulk-memory --enable-simd rewriter/wasm/out/wasm_bg.wasm -o rewriter/wasm/out/optimized.wasm
 
-mkdir dist/ || true
+mkdir -p dist/
 
+cp rewriter/wasm/out/optimized.wasm dist/scramjet.wasm.wasm
 {
 
 cat <<EOF
diff --git a/src/client/shared/eval.ts b/src/client/shared/eval.ts
index 93d1ad4..d11d1e6 100644
--- a/src/client/shared/eval.ts
+++ b/src/client/shared/eval.ts
@@ -8,7 +8,7 @@ export default function (client: ScramjetClient, self: Self) {
 		value: function (js: any) {
 			if (typeof js !== "string") return js;
 
-			const rewritten = rewriteJs(js, null, client.meta);
+			const rewritten = rewriteJs(js, "(direct eval proxy)", client.meta);
 
 			return rewritten;
 		},
@@ -23,5 +23,7 @@ export function indirectEval(this: ScramjetClient, js: any) {
 
 	const indirection = this.global.eval;
 
-	return indirection(rewriteJs(js, null, this.meta) as string);
+	return indirection(
+		rewriteJs(js, "(indirect eval proxy)", this.meta) as string
+	);
 }
diff --git a/src/client/shared/function.ts b/src/client/shared/function.ts
index 0dddb42..6f35d40 100644
--- a/src/client/shared/function.ts
+++ b/src/client/shared/function.ts
@@ -4,7 +4,11 @@ import { rewriteJs } from "../../shared";
 function rewriteFunction(ctx: ProxyCtx, client: ScramjetClient) {
 	const stringifiedFunction = ctx.call().toString();
 
-	const content = rewriteJs(`return ${stringifiedFunction}`, null, client.meta);
+	const content = rewriteJs(
+		`return ${stringifiedFunction}`,
+		"(function proxy)",
+		client.meta
+	);
 	ctx.return(ctx.fn(content)());
 }
 
diff --git a/src/shared/rewriters/url.ts b/src/shared/rewriters/url.ts
index 67296d1..50bb276 100644
--- a/src/shared/rewriters/url.ts
+++ b/src/shared/rewriters/url.ts
@@ -33,7 +33,8 @@ export function rewriteUrl(url: string | URL, meta: URLMeta) {
 
 	if (url.startsWith("javascript:")) {
 		return (
-			"javascript:" + rewriteJs(url.slice("javascript:".length), null, meta)
+			"javascript:" +
+			rewriteJs(url.slice("javascript:".length), "(javascript: url)", meta)
 		);
 	} else if (url.startsWith("blob:")) {
 		return location.origin + $scramjet.config.prefix + url;