@cfasim-ui/wasm 0.1.1 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfasim-ui/wasm",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "type": "module",
   "description": "WebAssembly integration for cfasim-ui",
   "license": "Apache-2.0",
@@ -13,10 +13,11 @@
     "src"
   ],
   "exports": {
-    ".": "./src/index.ts"
+    ".": "./src/index.ts",
+    "./vite": "./src/vitePlugin.ts"
   },
   "dependencies": {
-    "@cfasim-ui/shared": "0.1.1"
+    "@cfasim-ui/shared": "0.1.3"
   },
   "peerDependencies": {
     "vue": "^3.5.0"

package/src/vitePlugin.ts

@@ -0,0 +1,34 @@
+import { execSync } from "node:child_process";
+import { basename, resolve } from "node:path";
+import type { Plugin } from "vite";
+
+interface CfasimWasmOptions {
+  /** Path to the Rust model directory (default: "model") */
+  model?: string;
+  /** Output name used for the wasm module (default: project directory name) */
+  name?: string;
+}
+
+/**
+ * Vite plugin that builds a Rust model to WebAssembly via wasm-pack
+ * and outputs it to public/wasm/{name}/.
+ */
+export function cfasimWasm(options?: CfasimWasmOptions): Plugin {
+  const modelDir = options?.model ?? "model";
+
+  function build(root: string) {
+    const name = options?.name ?? basename(root);
+    const outDir = resolve(root, "public", "wasm", name);
+    execSync(`wasm-pack build ${modelDir} --target web --out-dir ${outDir}`, {
+      cwd: root,
+      stdio: "pipe",
+    });
+  }
+
+  return {
+    name: "cfasim-wasm",
+    configResolved(config) {
+      build(config.root);
+    },
+  };
+}

package/src/wasm.worker.ts

@@ -14,12 +14,14 @@ interface WorkerMessage {
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 const loadedModules = new Map<string, Record<string, any>>();
 
+const baseUrl = import.meta.env.BASE_URL ?? "/";
+
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 async function ensureModule(model: string): Promise<Record<string, any>> {
   const cached = loadedModules.get(model);
   if (cached) return cached;
 
-  const url = `${self.location.origin}/wasm/${model}/${model}.js`;
+  const url = `${self.location.origin}${baseUrl}wasm/${model}/${model}.js`;
   const mod = await import(/* @vite-ignore */ url);
   await mod.default();
   loadedModules.set(model, mod);