can1357
e9e106da1b
- Refactored WASM module loading to use import.meta.resolve() instead of URL constructor for better module resolution.
69 lines
2.4 KiB
TypeScript
69 lines
2.4 KiB
TypeScript
import { $ } from "bun";
|
|
import * as path from "node:path";
|
|
import * as fs from "node:fs/promises";
|
|
|
|
const repoRoot = path.join(import.meta.dir, "../../..");
|
|
const rustDir = path.join(repoRoot, "crates/pi-natives");
|
|
const wasmDir = path.join(import.meta.dir, "../wasm");
|
|
|
|
await fs.mkdir(wasmDir, { recursive: true });
|
|
|
|
console.log("Building Rust WASM with wasm-bindgen...");
|
|
|
|
// Build with wasm-pack (wasm-opt disabled in Cargo.toml)
|
|
const buildResult = await $`wasm-pack build --target deno --release --out-dir pkg`
|
|
.cwd(rustDir)
|
|
.nothrow();
|
|
|
|
if (buildResult.exitCode !== 0) {
|
|
console.error("wasm-pack build failed");
|
|
console.error(buildResult.stderr.toString());
|
|
process.exit(1);
|
|
}
|
|
|
|
const pkgDir = path.join(rustDir, "pkg");
|
|
|
|
// Crate name is pi-natives -> output files are pi_natives_*
|
|
const wasmSrc = path.join(pkgDir, "pi_natives_bg.wasm");
|
|
const wasmDst = path.join(wasmDir, "pi_natives_bg.wasm");
|
|
|
|
// Optimize with wasm-opt with bulk memory enabled
|
|
const optResult = await $`wasm-opt \
|
|
--enable-bulk-memory \
|
|
--enable-nontrapping-float-to-int \
|
|
-O3 \
|
|
--strip-debug \
|
|
${wasmSrc} \
|
|
-o ${wasmDst}`.nothrow();
|
|
|
|
if (optResult.exitCode !== 0) {
|
|
console.log("wasm-opt failed or not available, copying unoptimized build");
|
|
await fs.copyFile(wasmSrc, wasmDst);
|
|
}
|
|
|
|
// Copy and patch JS bindings for Bun compiled binary compatibility
|
|
// wasm-bindgen uses `new URL('file.wasm', import.meta.url)` which doesn't work
|
|
// in Bun's single-file executable. We use asset import + import.meta.resolve() instead.
|
|
const jsDst = path.join(wasmDir, "pi_natives.js");
|
|
await fs.copyFile(path.join(pkgDir, "pi_natives.js"), jsDst);
|
|
let jsContent = await Bun.file(jsDst).text();
|
|
jsContent = jsContent.replace(
|
|
"/* @ts-self-types=",
|
|
'import wasmPath from "./pi_natives_bg.wasm";\n\n/* @ts-self-types='
|
|
);
|
|
jsContent = jsContent.replace(
|
|
"const wasmUrl = new URL('pi_natives_bg.wasm', import.meta.url);",
|
|
"const wasmUrl = import.meta.resolve(wasmPath);"
|
|
);
|
|
await Bun.write(jsDst, jsContent);
|
|
|
|
// Copy d.ts bindings
|
|
await fs.copyFile(path.join(pkgDir, "pi_natives.d.ts"), path.join(wasmDir, "pi_natives.d.ts"));
|
|
await fs.copyFile(path.join(pkgDir, "pi_natives_bg.wasm.d.ts"), path.join(wasmDir, "pi_natives_bg.wasm.d.ts"));
|
|
|
|
const stat = await fs.stat(wasmDst);
|
|
console.log(`Done: ${wasmDst} (${(stat.size / 1024).toFixed(0)}KB)`);
|
|
|
|
// Clean up pkg directory
|
|
await fs.rm(pkgDir, { recursive: true, force: true });
|