@cfasim-ui/pyodide 0.1.3 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfasim-ui/pyodide",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "type": "module",
   "description": "Pyodide (Python-in-browser) integration for cfasim-ui",
   "license": "Apache-2.0",
@@ -9,6 +9,9 @@
     "url": "https://github.com/CDCgov/cfa-simulator.git",
     "directory": "cfasim-ui/pyodide"
   },
+  "publishConfig": {
+    "access": "public"
+  },
   "files": [
     "src"
   ],
@@ -18,7 +21,7 @@
   },
   "dependencies": {
     "pyodide": "^0.29.3",
-    "@cfasim-ui/shared": "0.1.3"
+    "@cfasim-ui/shared": "0.1.5"
   },
   "peerDependencies": {
     "vue": "^3.5.0"

package/src/pyodide.worker.ts

@@ -44,14 +44,26 @@ const pyodideReadyPromise = (async () => {
   return pyodide;
 })();
 
+let wheelsInstalled = false;
+
+async function installAllWheels() {
+  if (wheelsInstalled) return;
+  wheelsInstalled = true;
+  const urls = Object.values(wheelMap).map(
+    (f) => `${self.location.origin}${baseUrl}${f}`,
+  );
+  if (urls.length > 0) {
+    await micropip.install(urls, { deps: false });
+  }
+}
+
 async function ensureModule(
   pyodide: Awaited<typeof pyodideReadyPromise>,
   moduleName: string,
 ) {
   if (loadedModules.has(moduleName)) return;
-  const filename = wheelMap[moduleName];
-  if (!filename) throw new Error(`Unknown module: ${moduleName}`);
-  await micropip.install(`${self.location.origin}${baseUrl}${filename}`);
+  if (!wheelMap[moduleName]) throw new Error(`Unknown module: ${moduleName}`);
+  await installAllWheels();
   pyodide.pyimport(moduleName);
   loadedModules.add(moduleName);
 }

package/src/vitePlugin.ts

@@ -6,6 +6,8 @@ import type { Plugin } from "vite";
 interface CfasimPyodideOptions {
   /** Path to the Python model directory (default: "model") */
   model?: string;
+  /** Additional local package directories to build as wheels */
+  deps?: string[];
 }
 
 /**
@@ -18,6 +20,12 @@ export function cfasimPyodide(options?: CfasimPyodideOptions): Plugin {
   function build(root: string) {
     const publicDir = resolve(root, "public");
     mkdirSync(publicDir, { recursive: true });
+    for (const dep of options?.deps ?? []) {
+      execSync(`uv build ${dep} --wheel --out-dir public`, {
+        cwd: root,
+        stdio: "pipe",
+      });
+    }
     execSync(`uv build ${modelDir} --wheel --out-dir public`, {
       cwd: root,
       stdio: "pipe",