@cfasim-ui/pyodide 0.1.2 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfasim-ui/pyodide",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "type": "module",
   "description": "Pyodide (Python-in-browser) integration for cfasim-ui",
   "license": "Apache-2.0",
@@ -13,11 +13,12 @@
     "src"
   ],
   "exports": {
-    ".": "./src/index.ts"
+    ".": "./src/index.ts",
+    "./vite": "./src/vitePlugin.ts"
   },
   "dependencies": {
     "pyodide": "^0.29.3",
-    "@cfasim-ui/shared": "0.1.2"
+    "@cfasim-ui/shared": "0.1.3"
   },
   "peerDependencies": {
     "vue": "^3.5.0"

package/src/vitePlugin.ts

@@ -0,0 +1,35 @@
+import { execSync } from "node:child_process";
+import { readdirSync, writeFileSync, mkdirSync } from "node:fs";
+import { resolve } from "node:path";
+import type { Plugin } from "vite";
+
+interface CfasimPyodideOptions {
+  /** Path to the Python model directory (default: "model") */
+  model?: string;
+}
+
+/**
+ * Vite plugin that builds a Python wheel from a local model directory
+ * and generates wheels.json in the public directory.
+ */
+export function cfasimPyodide(options?: CfasimPyodideOptions): Plugin {
+  const modelDir = options?.model ?? "model";
+
+  function build(root: string) {
+    const publicDir = resolve(root, "public");
+    mkdirSync(publicDir, { recursive: true });
+    execSync(`uv build ${modelDir} --wheel --out-dir public`, {
+      cwd: root,
+      stdio: "pipe",
+    });
+    const wheels = readdirSync(publicDir).filter((f) => f.endsWith(".whl"));
+    writeFileSync(resolve(publicDir, "wheels.json"), JSON.stringify(wheels));
+  }
+
+  return {
+    name: "cfasim-pyodide",
+    configResolved(config) {
+      build(config.root);
+    },
+  };
+}