fluxy-bot 0.4.17 → 0.4.19

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fluxy-bot",
-  "version": "0.4.17",
+  "version": "0.4.19",
   "description": "Self-hosted, self-evolving AI agent with its own dashboard.",
   "type": "module",
   "license": "MIT",

package/supervisor/index.ts

@@ -57,9 +57,13 @@ export async function startSupervisor() {
   const workerPort = getWorkerPort(config.port);
   const backendPort = getBackendPort(config.port);
 
-  // Start Vite dev server for dashboard HMR
+  // Create HTTP server first (Vite needs it for HMR WebSocket)
+  // The request handler is set up later via server.on('request')
+  const server = http.createServer();
+
+  // Start Vite dev server — pass supervisor server so Vite attaches HMR WebSocket directly
   console.log('[supervisor] Starting Vite dev server...');
-  const vitePorts = await startViteDevServers(config.port);
+  const vitePorts = await startViteDevServers(config.port, server);
   console.log(`[supervisor] Vite ready — dashboard :${vitePorts.dashboard}`);
 
   // Ensure file storage dirs exist
@@ -152,8 +156,8 @@ export async function startSupervisor() {
     });
   }
 
-  // HTTP server — proxies to Vite dev servers + worker API
-  const server = http.createServer(async (req, res) => {
+  // HTTP request handler — proxies to Vite dev servers + worker API
+  server.on('request', async (req, res) => {
     // Fluxy widget — served directly (not part of Vite build)
     if (req.url === '/fluxy/widget.js') {
       console.log('[supervisor] Serving /fluxy/widget.js directly');
@@ -421,8 +425,7 @@ export async function startSupervisor() {
               // Intercept bot:done — Vite HMR handles file changes automatically
               if (type === 'bot:done') {
                 if (eventData.usedFileTools) {
-                  console.log('[supervisor] File tools used — Vite HMR will apply changes automatically');
-                  console.log('[supervisor] Restarting backend...');
+                  console.log('[supervisor] File tools used — Vite HMR handles frontend, restarting backend');
                   resetBackendRestarts();
                   stopBackend();
                   spawnBackend(backendPort);
@@ -508,41 +511,27 @@ export async function startSupervisor() {
     });
   });
 
+  // Fluxy chat WebSocket — Vite HMR is handled automatically (hmr.server = this server)
   server.on('upgrade', async (req, socket: net.Socket, head) => {
+    if (!req.url?.startsWith('/fluxy/ws')) return; // Let Vite handle HMR upgrades
+
     console.log(`[supervisor] WebSocket upgrade: ${req.url}`);
 
-    if (req.url?.startsWith('/fluxy/ws')) {
-      // Auth check for WebSocket
-      const needsAuth = await isAuthRequired();
-      if (needsAuth) {
-        const urlObj = new URL(req.url, `http://${req.headers.host}`);
-        const token = urlObj.searchParams.get('token');
-        if (!token || !(await validateToken(token))) {
-          console.log('[supervisor] WS auth failed — rejecting');
-          socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
-          socket.destroy();
-          return;
-        }
+    // Auth check for WebSocket
+    const needsAuth = await isAuthRequired();
+    if (needsAuth) {
+      const urlObj = new URL(req.url, `http://${req.headers.host}`);
+      const token = urlObj.searchParams.get('token');
+      if (!token || !(await validateToken(token))) {
+        console.log('[supervisor] WS auth failed — rejecting');
+        socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
+        socket.destroy();
+        return;
       }
-
-      console.log('[supervisor] → Fluxy chat WebSocket');
-      fluxyWss.handleUpgrade(req, socket, head, (ws) => fluxyWss.emit('connection', ws, req));
-      return;
     }
 
-    // Route HMR WebSocket to dashboard Vite dev server
-    const targetPort = vitePorts.dashboard;
-    console.log(`[supervisor] → Vite HMR (dashboard) :${targetPort}`);
-
-    const proxy = net.connect(targetPort, () => {
-      const headers = Object.entries(req.headers).map(([k, v]) => `${k}: ${v}`).join('\r\n');
-      proxy.write(`GET ${req.url} HTTP/1.1\r\n${headers}\r\n\r\n`);
-      if (head.length > 0) proxy.write(head);
-      socket.pipe(proxy).pipe(socket);
-    });
-
-    proxy.on('error', () => { console.error(`[supervisor] WS proxy error → :${targetPort}`); socket.destroy(); });
-    socket.on('error', () => proxy.destroy());
+    console.log('[supervisor] → Fluxy chat WebSocket');
+    fluxyWss.handleUpgrade(req, socket, head, (ws) => fluxyWss.emit('connection', ws, req));
   });
 
   // Start

package/supervisor/vite-dev.ts

@@ -1,17 +1,18 @@
 import { createServer as createViteServer, type ViteDevServer } from 'vite';
+import type http from 'http';
 import path from 'path';
 import { PKG_DIR } from '../shared/paths.js';
 import { log } from '../shared/logger.js';
 
 let dashboardVite: ViteDevServer | null = null;
 
-export async function startViteDevServers(supervisorPort: number): Promise<{ dashboard: number }> {
+export async function startViteDevServers(supervisorPort: number, hmrServer: http.Server): Promise<{ dashboard: number }> {
   const ports = {
     dashboard: supervisorPort + 2,
   };
 
   console.log(`[vite-dev] Starting dashboard Vite dev server on :${ports.dashboard}`);
-  console.log(`[vite-dev] HMR clientPort → :${supervisorPort} (browser connects through supervisor proxy)`);
+  console.log(`[vite-dev] HMR attached to supervisor server (port ${supervisorPort})`);
   console.log(`[vite-dev] PKG_DIR = ${PKG_DIR}`);
 
   try {
@@ -22,7 +23,7 @@ export async function startViteDevServers(supervisorPort: number): Promise<{ das
         host: '127.0.0.1',
         strictPort: true,
         allowedHosts: true,
-        hmr: { clientPort: supervisorPort },
+        hmr: { server: hmrServer, clientPort: supervisorPort },
       },
       logLevel: 'info',
     });
@@ -51,6 +52,13 @@ export async function startViteDevServers(supervisorPort: number): Promise<{ das
   return ports;
 }
 
+/** Tell connected browsers to full-reload via Vite's HMR WebSocket */
+export function reloadDashboard(): void {
+  if (!dashboardVite) return;
+  console.log('[vite-dev] Sending full-reload to dashboard clients');
+  dashboardVite.hot.send({ type: 'full-reload', path: '*' });
+}
+
 export async function stopViteDevServers(): Promise<void> {
   console.log('[vite-dev] Stopping Vite dev servers...');
   if (dashboardVite) {