koguma 2.2.4 → 2.3.0

package/src/vite.ts

@@ -0,0 +1,78 @@
+/**
+ * koguma/vite — Vite plugin for koguma projects.
+ *
+ * Usage in vite.config.ts:
+ *
+ *   import { kogumaHMR } from 'koguma/vite';
+ *
+ *   export default defineConfig({
+ *     plugins: [kogumaHMR()],
+ *   });
+ */
+
+import { createHash } from 'node:crypto';
+import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
+import { resolve } from 'node:path';
+import type { Plugin } from 'vite';
+
+/**
+ * Watches the wrangler local D1 SQLite file and triggers a full browser reload
+ * when content actually changes. Rather than watching the binary SQLite directly
+ * (which churns 3-4× per operation due to WAL writes), it writes a SHA-256 hash
+ * of the DB to `.koguma/dbhash` and watches that text file as the reload trigger.
+ *
+ * Also suppresses Vite HMR for the `content/` directory, since those edits sync
+ * through koguma's own file watcher → D1 → API, not through the Vite module graph.
+ *
+ * Pair with the `server.proxy` pointing `/api` to `http://localhost:8787` so the
+ * browser refetches fresh content on each reload.
+ */
+export function kogumaHMR(): Plugin {
+  const HASH_FILE = '.koguma/dbhash';
+
+  return {
+    name: 'koguma-hmr',
+
+    configureServer(server) {
+      const root = server.config.root;
+      const hashPath = resolve(root, HASH_FILE);
+      const kogumaDir = resolve(root, '.koguma');
+
+      // Watch the wrangler local D1 directory for SQLite changes
+      server.watcher.add(resolve(root, '.wrangler/state/v3/d1/**/*.sqlite'));
+
+      server.watcher.on('change', (file: string) => {
+        if (
+          !file.includes('.wrangler/state/v3/d1') ||
+          !file.endsWith('.sqlite')
+        )
+          return;
+        try {
+          const hash = createHash('sha256')
+            .update(readFileSync(file))
+            .digest('hex');
+          if (!existsSync(kogumaDir)) mkdirSync(kogumaDir, { recursive: true });
+          writeFileSync(hashPath, hash);
+        } catch {
+          // SQLite may be locked mid-write — the next write will succeed
+        }
+      });
+
+      // Watch the hash file as the actual reload trigger.
+      // Vite's default behaviour for a watched file not in the module graph
+      // is a full-reload — no handleHotUpdate override needed.
+      server.watcher.add(hashPath);
+    },
+
+    config() {
+      return {
+        server: {
+          watch: {
+            // content/ changes go through koguma's sync pipeline, not Vite's module graph
+            ignored: ['**/content/**']
+          }
+        }
+      };
+    }
+  };
+}