@nitronjs/framework 0.3.3 → 0.3.4

package/lib/HMR/Server.js

@@ -1,170 +1,93 @@
-import { Server as SocketServer } from "socket.io";
-import { createRequire } from "module";
-import path from "path";
-import fs from "fs";
-import { fileURLToPath } from "url";
-
-const __dirname = path.dirname(fileURLToPath(import.meta.url));
-
-/**
- * HMR (Hot Module Replacement) server for development mode.
- * Manages WebSocket connections and broadcasts file change events to connected clients.
- * 
- * @example
- * // In HTTP Server
- * HMR.registerRoutes(fastify);
- * HMR.setup(httpServer);
- *
- * // Emit updates
- * HMR.emitChange({ changeType: "page", file: "Site/Home.tsx" });
- * HMR.emitChange({ changeType: "css", file: "global.css" });
- */
-class HMRServer {
-    #io = null;
-    #connections = 0;
-    #clientScript = null;
-
-    // Public Methods
-
-    /**
-     * Registers socket.io client script route in Fastify.
-     * Must be called before Router.setup() to ensure the route is registered.
-     * @param {import("fastify").FastifyInstance} fastify
-     */
-    registerRoutes(fastify) {
-        this.#clientScript = this.#findSocketIoClient();
-        const cachedScript = this.#clientScript ? fs.readFileSync(this.#clientScript, "utf-8") : null;
-
-        fastify.get("/__nitron_client/socket.io.js", (req, reply) => {
-            if (!cachedScript) {
-                return reply.code(503).send("// HMR disabled: socket.io client not found");
-            }
-
-            reply.type("application/javascript").send(cachedScript);
-        });
-    }
-
-    /**
-     * Initializes the WebSocket server.
-     * @param {import("http").Server} httpServer - Node.js HTTP server instance.
-     */
-    setup(httpServer) {
-        if (this.#io) return;
-
-        this.#io = new SocketServer(httpServer, {
-            path: "/__nitron_hmr",
-            transports: ["websocket"],
-            cors: { origin: "*" },
-            pingTimeout: 60000,
-            pingInterval: 25000,
-            serveClient: false,
-            allowEIO3: true
-        });
-
-        this.#io.on("connection", (socket) => {
-            this.#connections++;
-            socket.on("disconnect", () => this.#connections--);
-        });
-    }
-
-    /**
-     * Whether the HMR server is ready and accepting connections.
-     * @returns {boolean}
-     */
-    get isReady() {
-        return this.#io !== null;
-    }
-
-    /**
-     * Emits a unified change event for RSC-based hot updates.
-     * @param {object} data
-     * @param {string} data.changeType - "page" | "layout" | "css"
-     * @param {boolean} [data.cssChanged] - Whether CSS also changed (Tailwind rebuild)
-     * @param {string} [data.file] - Relative path of the changed file (for logging)
-     */
-    emitChange(data) {
-        if (!this.#io) return;
-
-        this.#io.emit("hmr:change", {
-            changeType: data.changeType,
-            cssChanged: data.cssChanged || false,
-            file: data.file || null,
-            timestamp: Date.now()
-        });
-    }
-
-    /**
-     * Emits a full page reload event.
-     * @param {string} reason - Reason for the reload (shown in dev tools).
-     */
-    emitReload(reason) {
-        if (!this.#io) return;
-
-        this.#io.emit("hmr:reload", {
-            reason,
-            timestamp: Date.now()
-        });
-    }
-
-    /**
-     * Emits a build error event to show error overlay in browser.
-     * @param {Error|string} error - The error that occurred.
-     * @param {string} [filePath] - Path to the file that caused the error.
-     */
-    emitError(error, filePath) {
-        if (!this.#io) return;
-
-        this.#io.emit("hmr:error", {
-            file: filePath,
-            message: String(error?.message || error),
-            timestamp: Date.now()
-        });
-    }
-
-    /**
-     * Closes the WebSocket server and cleans up resources.
-     */
-    close() {
-        if (this.#io) {
-            this.#io.close();
-            this.#io = null;
-        }
-
-        this.#connections = 0;
-    }
-
-    // Private Methods
-
-    /**
-     * Finds socket.io client script.
-     * Searches framework node_modules and walks up directory tree for monorepo support.
-     * @returns {string|null} Path to socket.io.min.js or null if not found
-     */
-    #findSocketIoClient() {
-        const clientFile = "client-dist/socket.io.min.js";
-
-        // 1. Try resolving from framework package location (most common case)
-        try {
-            const frameworkRequire = createRequire(import.meta.url);
-            const socketIoDir = path.dirname(frameworkRequire.resolve("socket.io/package.json"));
-            const clientPath = path.join(socketIoDir, clientFile);
-            if (fs.existsSync(clientPath)) return clientPath;
-        }
-        catch {}
-
-        // 2. Walk up from framework location for monorepo setups
-        let currentDir = __dirname;
-        for (let i = 0; i < 5; i++) {
-            const clientPath = path.join(currentDir, "node_modules", "socket.io", clientFile);
-            if (fs.existsSync(clientPath)) return clientPath;
-            
-            const parentDir = path.dirname(currentDir);
-            if (parentDir === currentDir) break;
-            currentDir = parentDir;
-        }
-
-        return null;
-    }
-}
-
-export default new HMRServer();
+import { WebSocketServer } from "ws";
+
+class HMRServer {
+    #wss = null;
+    #clients = new Set();
+
+    registerRoutes(fastify) {
+        // No routes needed — native WebSocket doesn't need a client script endpoint
+    }
+
+    setup(httpServer) {
+        if (this.#wss) return;
+
+        this.#wss = new WebSocketServer({
+            server: httpServer,
+            path: "/__nitron_hmr"
+        });
+
+        this.#wss.on("connection", (ws) => {
+            this.#clients.add(ws);
+
+            ws.on("close", () => {
+                this.#clients.delete(ws);
+            });
+        });
+    }
+
+    get isReady() {
+        return this.#wss !== null;
+    }
+
+    emitChange(data) {
+        this.#broadcast({
+            type: "change",
+            changeType: data.changeType,
+            cssChanged: data.cssChanged || false,
+            file: data.file || null,
+            timestamp: Date.now()
+        });
+    }
+
+    emitReload(reason) {
+        this.#broadcast({
+            type: "reload",
+            reason,
+            timestamp: Date.now()
+        });
+    }
+
+    emitFastRefresh(data) {
+        this.#broadcast({
+            type: "fast-refresh",
+            chunks: data.chunks || [],
+            cssChanged: data.cssChanged || false,
+            timestamp: data.timestamp || Date.now()
+        });
+    }
+
+    emitError(error, filePath) {
+        this.#broadcast({
+            type: "error",
+            file: filePath,
+            message: String(error?.message || error),
+            timestamp: Date.now()
+        });
+    }
+
+    close() {
+        if (this.#wss) {
+            for (const client of this.#clients) {
+                client.close();
+            }
+
+            this.#clients.clear();
+            this.#wss.close();
+            this.#wss = null;
+        }
+    }
+
+    #broadcast(data) {
+        if (!this.#wss) return;
+
+        const message = JSON.stringify(data);
+
+        for (const client of this.#clients) {
+            if (client.readyState === 1) {
+                client.send(message);
+            }
+        }
+    }
+}
+
+export default new HMRServer();

package/lib/Runtime/Entry.js

@@ -21,6 +21,9 @@ export async function start() {
         case "change":
           HMRServer.emitChange(msg);
           break;
+        case "fast-refresh":
+          HMRServer.emitFastRefresh(msg);
+          break;
         case "reload":
           HMRServer.emitReload(msg.reason);
           break;