fix rewriter

This commit is contained in:
velzie 2024-07-27 16:40:50 -04:00
parent 0e37a7f267
commit 396170e4ea
No known key found for this signature in database
GPG key ID: 048413F95F0DDE1F
7 changed files with 115 additions and 43 deletions

View file

@ -8,6 +8,7 @@ import {
rewrite_js_from_arraybuffer,
} from "../../../rewriter/out/rewriter.js";
import "../../../static/wasm.js";
import { importfn, wrapfn } from "../../client/shared/wrap";
initSync(
new WebAssembly.Module(
@ -25,9 +26,25 @@ export function rewriteJs(js: string | ArrayBuffer, origin?: URL) {
const before = performance.now();
if (typeof js === "string") {
js = new TextDecoder().decode(rewrite_js(js, origin.toString()));
js = new TextDecoder().decode(
rewrite_js(
js,
origin.toString(),
self.$scramjet.config.prefix,
self.$scramjet.config.codec.encode as any,
wrapfn,
importfn
)
);
} else {
js = rewrite_js_from_arraybuffer(new Uint8Array(js), origin.toString());
js = rewrite_js_from_arraybuffer(
new Uint8Array(js),
origin.toString(),
self.$scramjet.config.prefix,
self.$scramjet.config.codec.encode as any,
wrapfn,
importfn
);
}
const after = performance.now();