vimd 0.3.0 → 0.3.1

package/dist/core/websocket-server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"websocket-server.d.ts","sourceRoot":"","sources":["../../src/core/websocket-server.ts"],"names":[],"mappings":"AAQA;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,OAAO,CAAC;CACtB;AA0CD;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,UAAU,CAA4B;IAC9C,OAAO,CAAC,QAAQ,CAAyB;IACzC,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,KAAK,CAAS;gBAEV,OAAO,EAAE,sBAAsB;IAQ3C;;OAEG;IACH,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,iBAAiB,CAAC;IAwGzC;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA8B3B;;OAEG;IACH,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI;IAU7C;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAc3B"}
+{"version":3,"file":"websocket-server.d.ts","sourceRoot":"","sources":["../../src/core/websocket-server.ts"],"names":[],"mappings":"AAUA;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,OAAO,CAAC;CACtB;AA0CD;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,UAAU,CAA4B;IAC9C,OAAO,CAAC,QAAQ,CAAyB;IACzC,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,KAAK,CAAS;gBAEV,OAAO,EAAE,sBAAsB;IAQ3C;;OAEG;IACH,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,iBAAiB,CAAC;IA4GzC;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA8B3B;;OAEG;IACH,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI;IAU7C;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAc3B"}

package/dist/core/websocket-server.js

@@ -1,5 +1,7 @@
 // src/core/websocket-server.ts
 import http from 'http';
+import fs from 'fs';
+import path from 'path';
 import polka from 'polka';
 import sirv from 'sirv';
 import { WebSocketServer as WSServer, WebSocket } from 'ws';
@@ -84,31 +86,38 @@ export class WebSocketServer {
         // Create polka app with HTML injection middleware
         const app = polka();
         // Middleware to inject reload script into HTML
+        // For HTML files, read directly to avoid sirv streaming issues
         app.use((req, res, next) => {
             // Only process GET requests for HTML files
             if (req.method !== 'GET' || !req.url?.endsWith('.html')) {
                 return serve(req, res, next);
             }
-            // For HTML files, we need to inject the script
-            // Override res.end to modify the response
-            const originalEnd = res.end.bind(res);
-            let body = '';
-            res.write = (chunk) => {
-                body += chunk.toString();
-                return true;
-            };
-            res.end = ((chunk) => {
-                if (chunk) {
-                    body += chunk.toString();
-                }
-                // Inject reload script before </body> or </html>
-                const injectedBody = this.injectReloadScript(body);
-                // Update Content-Length header
-                const contentLength = Buffer.byteLength(injectedBody, 'utf8');
-                res.setHeader('Content-Length', contentLength);
-                return originalEnd(injectedBody);
-            });
-            serve(req, res, next);
+            // Parse the URL and resolve the file path
+            const urlPath = req.url || '/index.html';
+            const filePath = path.join(this.options.root, urlPath);
+            // Check if file exists
+            if (!fs.existsSync(filePath)) {
+                // Let sirv handle 404
+                return serve(req, res, next);
+            }
+            try {
+                // Read HTML file directly
+                const html = fs.readFileSync(filePath, 'utf8');
+                // Inject reload script
+                const injectedHtml = this.injectReloadScript(html);
+                // Send response with proper headers
+                res.writeHead(200, {
+                    'Content-Type': 'text/html; charset=utf-8',
+                    'Content-Length': Buffer.byteLength(injectedHtml, 'utf8'),
+                    'Cache-Control': 'no-cache, no-store, must-revalidate',
+                });
+                res.end(injectedHtml);
+            }
+            catch (error) {
+                // On error, fall back to sirv
+                Logger.warn(`Failed to read HTML file: ${filePath}`);
+                serve(req, res, next);
+            }
         });
         // Create HTTP server
         this.httpServer = http.createServer(app.handler);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vimd",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Real-time Markdown preview tool with pandoc (view markdown)",
   "type": "module",
   "keywords": [