HMR

omochi · May 1, 2024 · 5abee2b · 5abee2b
1 parent e3ff655
commit 5abee2b
Show file tree

Hide file tree

Showing 7 changed files with 228 additions and 10 deletions.
diff --git a/index.html b/index.html
@@ -6,7 +6,8 @@
     <link rel="stylesheet" type="text/css" href="/style.css">
     <script type="module">
         import { load } from "/src/loader/load.ts";
-        load("/.build/wasm32-unknown-wasi/debug/StringCounter.wasm");
+        import wasmURL from "/.build/wasm32-unknown-wasi/debug/StringCounter.wasm?url";
+        load(wasmURL);
     </script>
 </head>
 <body>

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -18,5 +18,8 @@
     "path-browserify": "^1.0.1",
     "typescript": "^5.4.5",
     "vite": "^5.2.10"
+  },
+  "devDependencies": {
+    "chokidar": "^3.6.0"
   }
 }
diff --git a/src/build/SwiftPlugin.ts b/src/build/SwiftPlugin.ts
@@ -0,0 +1,36 @@
+import { Plugin } from 'vite';
+import { watch } from 'chokidar';
+import { exec } from 'child_process';
+import { promisify } from 'util';
+
+const execAsync = promisify(exec);
+
+const SwiftPlugin = (): Plugin => {
+  return {
+    name: "vite-plugin-swift",
+    configureServer(server) {
+      const watcher = watch("Sources/**/*.swift", {
+        persistent: true
+      });
+
+      watcher.on("change", async (path) => {
+        console.log(`${path} changed, rebuilding swift...`);
+        try {
+          const { stdout, stderr } = await execAsync("bin/build");
+          console.log(stdout);
+          if(stderr) {
+            console.error(stderr);
+          }
+        } catch (error) {
+          console.error("build failed", error);
+        }
+      });
+
+      server.httpServer?.on("close", () => {
+        watcher.close();
+      });
+    }
+  }
+};
+
+export default SwiftPlugin;
diff --git a/src/loader/load.ts b/src/loader/load.ts
@@ -15,9 +15,9 @@
 import { WasmRunner } from "./WasmRunner";
 import { SwiftRuntime } from "../JavaScriptKit/Runtime/src/index";
 
-const startWasiTask = async (exePath: string) => {
+const startWasiTask = async (wasmURL: string) => {
   // Fetch our Wasm File
-  const response = await fetch(exePath);
+  const response = await fetch(wasmURL);
   const responseArrayBuffer = await response.arrayBuffer();
 
   const wasmRunner = WasmRunner(false, SwiftRuntime);
@@ -34,9 +34,9 @@ function handleError(e: any) {
   }
 }
 
-export function load(exePath: string) {
+export function load(wasmURL: string) {
   try {
-    startWasiTask(exePath).catch(handleError);
+    startWasiTask(wasmURL).catch(handleError);
   } catch (e) {
     handleError(e);
   }

diff --git a/vite.config.js b/vite.config.js
diff --git a/vite.config.ts b/vite.config.ts
@@ -0,0 +1,8 @@
+import { defineConfig } from 'vite';
+import SwiftPlugin from './src/build/SwiftPlugin';
+
+export default defineConfig({
+    plugins: [
+        SwiftPlugin()
+    ]
+});