yo-bug 0.1.2 → 0.1.4

package/dist/src/server/html-injector.d.ts

@@ -2,6 +2,7 @@ import type { IncomingMessage, ServerResponse } from 'http';
 /**
  * Middleware that intercepts HTML responses from the proxy
  * and injects the vibe-feedback SDK script before </body>.
+ * Only intercepts 2xx HTML responses — redirects and errors pass through.
  */
 export declare function createHtmlInjector(proxyRes: IncomingMessage, res: ServerResponse): {
     onData(chunk: Buffer): void;

package/dist/src/server/html-injector.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"html-injector.d.ts","sourceRoot":"","sources":["../../../src/server/html-injector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AAI5D;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,EAAE,cAAc;kBAc/D,MAAM;;SAkBvB"}
+{"version":3,"file":"html-injector.d.ts","sourceRoot":"","sources":["../../../src/server/html-injector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AAI5D;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,eAAe,EAAE,GAAG,EAAE,cAAc;kBAmB/D,MAAM;;SAiBvB"}

package/dist/src/server/html-injector.js

@@ -2,14 +2,19 @@ const INJECT_SCRIPT = `<script src="/vibe-feedback.js"></script>`;
 /**
  * Middleware that intercepts HTML responses from the proxy
  * and injects the vibe-feedback SDK script before </body>.
+ * Only intercepts 2xx HTML responses — redirects and errors pass through.
  */
 export function createHtmlInjector(proxyRes, res) {
+    const statusCode = proxyRes.statusCode || 200;
     const contentType = proxyRes.headers['content-type'] || '';
-    if (!contentType.includes('text/html')) {
-        return null; // Not HTML, don't intercept
+    // Only inject into 2xx HTML responses
+    // Don't touch redirects (3xx), errors (4xx/5xx), or non-HTML
+    if (statusCode < 200 || statusCode >= 300 || !contentType.includes('text/html')) {
+        return null;
     }
     // Copy all response headers except content-length (we'll modify the body)
-    res.writeHead(proxyRes.statusCode || 200, Object.fromEntries(Object.entries(proxyRes.headers).filter(([key]) => key !== 'content-length')));
+    const headers = Object.fromEntries(Object.entries(proxyRes.headers).filter(([key]) => key !== 'content-length'));
+    res.writeHead(statusCode, headers);
     const chunks = [];
     return {
         onData(chunk) {
@@ -17,7 +22,6 @@ export function createHtmlInjector(proxyRes, res) {
         },
         onEnd() {
             let html = Buffer.concat(chunks).toString('utf-8');
-            // Inject before </body> or at end if no </body>
             if (html.includes('</body>')) {
                 html = html.replace('</body>', `${INJECT_SCRIPT}\n</body>`);
             }

package/dist/src/server/html-injector.js.map

@@ -1 +1 @@
-{"version":3,"file":"html-injector.js","sourceRoot":"","sources":["../../../src/server/html-injector.ts"],"names":[],"mappings":"AAEA,MAAM,aAAa,GAAG,2CAA2C,CAAC;AAElE;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAyB,EAAE,GAAmB;IAC/E,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAC3D,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC,CAAC,4BAA4B;IAC3C,CAAC;IAED,0EAA0E;IAC1E,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,IAAI,GAAG,EAAE,MAAM,CAAC,WAAW,CAC1D,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,gBAAgB,CAAC,CAC7E,CAAC,CAAC;IAEH,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,OAAO;QACL,MAAM,CAAC,KAAa;YAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,KAAK;YACH,IAAI,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAEnD,gDAAgD;YAChD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;YAC9D,CAAC;iBAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACpC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACN,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAC/B,CAAC;YAED,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"html-injector.js","sourceRoot":"","sources":["../../../src/server/html-injector.ts"],"names":[],"mappings":"AAEA,MAAM,aAAa,GAAG,2CAA2C,CAAC;AAElE;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,QAAyB,EAAE,GAAmB;IAC/E,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,IAAI,GAAG,CAAC;IAC9C,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAE3D,sCAAsC;IACtC,6DAA6D;IAC7D,IAAI,UAAU,GAAG,GAAG,IAAI,UAAU,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QAChF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0EAA0E;IAC1E,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAChC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,gBAAgB,CAAC,CAC7E,CAAC;IACF,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAEnC,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,OAAO;QACL,MAAM,CAAC,KAAa;YAClB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,KAAK;YACH,IAAI,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAEnD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;YAC9D,CAAC;iBAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACpC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACN,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAC/B,CAAC;YAED,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/server/proxy-server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"proxy-server.d.ts","sourceRoot":"","sources":["../../../src/server/proxy-server.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAUpD,wBAAsB,gBAAgB,CACpC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,aAAa,GACnB,OAAO,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC,CA0ElD;AAED,wBAAgB,eAAe,IAAI,IAAI,CAStC;AAED,wBAAgB,cAAc,IAAI,OAAO,CAExC"}
+{"version":3,"file":"proxy-server.d.ts","sourceRoot":"","sources":["../../../src/server/proxy-server.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAUpD,wBAAsB,gBAAgB,CACpC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,aAAa,GACnB,OAAO,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC,CAwGlD;AAED,wBAAgB,eAAe,IAAI,IAAI,CAStC;AAED,wBAAgB,cAAc,IAAI,OAAO,CAExC"}

package/dist/src/server/proxy-server.js

@@ -3,8 +3,8 @@ import cors from 'cors';
 import httpProxy from 'http-proxy';
 import { createFeedbackRouter } from './feedback-api.js';
 import { createSdkRouter } from './sdk-serve.js';
-import { createHtmlInjector } from './html-injector.js';
 const PROXY_PORT = 3695;
+const INJECT_SCRIPT = `<script src="/vibe-feedback.js"></script>`;
 let server = null;
 let proxy = null;
 export async function startProxyServer(targetPort, store) {
@@ -14,45 +14,77 @@ export async function startProxyServer(targetPort, store) {
     const targetUrl = `http://localhost:${targetPort}`;
     const proxyUrl = `http://localhost:${PROXY_PORT}`;
     const app = express();
-    // CORS for SDK to submit feedback
     app.use(cors());
-    // Parse JSON for feedback API
     app.use(express.json({ limit: '10mb' }));
-    // Serve SDK JS file
+    // Our own routes — served directly, not proxied
     app.use(createSdkRouter());
-    // Feedback API routes
     app.use(createFeedbackRouter(store));
-    // Create proxy
+    // Create proxy — let it handle responses by default (no selfHandleResponse)
     proxy = httpProxy.createProxyServer({
         target: targetUrl,
-        ws: true, // WebSocket support for HMR
+        ws: true,
         changeOrigin: true,
     });
     proxy.on('error', (_err, _req, res) => {
         if (res && 'writeHead' in res) {
-            res.writeHead?.(502, { 'Content-Type': 'text/plain' });
-            res.end?.('Dev server not responding');
+            try {
+                res.writeHead?.(502, { 'Content-Type': 'text/plain' });
+                res.end?.('Dev server not responding');
+            }
+            catch { }
         }
     });
-    // Intercept proxy responses to inject SDK into HTML
+    // Inject SDK into HTML responses using the modifyResponse approach
+    // We check Accept header on REQUEST to decide if we need to intercept
+    app.all('*', (req, res) => {
+        const accept = req.headers['accept'] || '';
+        const isHtmlRequest = accept.includes('text/html');
+        if (isHtmlRequest) {
+            // Potentially an HTML page — intercept to inject SDK
+            proxy.web(req, res, { selfHandleResponse: true });
+        }
+        else {
+            // Static asset (JS/CSS/images/fonts/etc) — let proxy handle directly
+            proxy.web(req, res);
+        }
+    });
+    // Handle HTML injection in proxyRes (only fires for selfHandleResponse requests)
     proxy.on('proxyRes', (proxyRes, _req, res) => {
-        const injector = createHtmlInjector(proxyRes, res);
-        if (injector) {
-            // HTML response — buffer and inject
-            proxyRes.on('data', (chunk) => injector.onData(chunk));
-            proxyRes.on('end', () => injector.onEnd());
+        const statusCode = proxyRes.statusCode || 200;
+        const contentType = proxyRes.headers['content-type'] || '';
+        // Only inject into 2xx HTML responses
+        if (statusCode >= 200 && statusCode < 300 && contentType.includes('text/html')) {
+            // Buffer response, inject SDK, send
+            const chunks = [];
+            proxyRes.on('data', (chunk) => chunks.push(chunk));
+            proxyRes.on('end', () => {
+                let html = Buffer.concat(chunks).toString('utf-8');
+                if (html.includes('</body>')) {
+                    html = html.replace('</body>', `${INJECT_SCRIPT}\n</body>`);
+                }
+                else if (html.includes('</html>')) {
+                    html = html.replace('</html>', `${INJECT_SCRIPT}\n</html>`);
+                }
+                else {
+                    html += `\n${INJECT_SCRIPT}`;
+                }
+                // Copy headers except content-length (body size changed)
+                const headers = { ...proxyRes.headers };
+                delete headers['content-length'];
+                // Remove content-encoding since we decoded the buffer as utf-8
+                delete headers['content-encoding'];
+                delete headers['transfer-encoding'];
+                headers['content-length'] = String(Buffer.byteLength(html));
+                res.writeHead(statusCode, headers);
+                res.end(html);
+            });
         }
         else {
-            // Non-HTML — pipe through directly
+            // Redirect or non-HTML — pass through as-is
+            res.writeHead(statusCode, proxyRes.headers);
             proxyRes.pipe(res);
         }
     });
-    // All other requests → proxy to dev server
-    app.all('*', (req, res) => {
-        proxy.web(req, res, {
-            selfHandleResponse: true, // We handle response in proxyRes event
-        });
-    });
     // Start server
     await new Promise((resolve, reject) => {
         server = app.listen(PROXY_PORT, () => resolve());

package/dist/src/server/proxy-server.js.map

@@ -1 +1 @@
-{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../../src/server/proxy-server.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,SAAS,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,MAAM,UAAU,GAAG,IAAI,CAAC;AAExB,IAAI,MAAM,GAAkB,IAAI,CAAC;AACjC,IAAI,KAAK,GAAqB,IAAI,CAAC;AAEnC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,UAAkB,EAClB,KAAoB;IAEpB,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,oBAAoB,UAAU,EAAE,CAAC;IACnD,MAAM,QAAQ,GAAG,oBAAoB,UAAU,EAAE,CAAC;IAElD,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IAEtB,kCAAkC;IAClC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAEhB,8BAA8B;IAC9B,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAEzC,oBAAoB;IACpB,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC;IAE3B,sBAAsB;IACtB,GAAG,CAAC,GAAG,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;IAErC,eAAe;IACf,KAAK,GAAG,SAAS,CAAC,iBAAiB,CAAC;QAClC,MAAM,EAAE,SAAS;QACjB,EAAE,EAAE,IAAI,EAAE,4BAA4B;QACtC,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACpC,IAAI,GAAG,IAAI,WAAW,IAAI,GAAG,EAAE,CAAC;YAC7B,GAAW,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YAC/D,GAAW,CAAC,GAAG,EAAE,CAAC,2BAA2B,CAAC,CAAC;QAClD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,oDAAoD;IACpD,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC3C,MAAM,QAAQ,GAAG,kBAAkB,CAAC,QAAQ,EAAE,GAAU,CAAC,CAAC;QAC1D,IAAI,QAAQ,EAAE,CAAC;YACb,oCAAoC;YACpC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACvD,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,mCAAmC;YACnC,QAAQ,CAAC,IAAI,CAAC,GAAU,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,2CAA2C;IAC3C,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACxB,KAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE;YACnB,kBAAkB,EAAE,IAAI,EAAE,uCAAuC;SAClE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,eAAe;IACf,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACjD,MAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;YACjD,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,UAAU,wDAAwD,CAAC,CAAC,CAAC;YAChG,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,mCAAmC;IACnC,MAAO,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAQ,EAAE,MAAW,EAAE,IAAS,EAAE,EAAE;QACzD,KAAM,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,KAAK,GAAG,IAAI,CAAC;IACf,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,MAAM,KAAK,IAAI,CAAC;AACzB,CAAC"}
+{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../../src/server/proxy-server.ts"],"names":[],"mappings":"AAAA,OAAO,OAAO,MAAM,SAAS,CAAC;AAC9B,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,SAAS,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,MAAM,UAAU,GAAG,IAAI,CAAC;AACxB,MAAM,aAAa,GAAG,2CAA2C,CAAC;AAElE,IAAI,MAAM,GAAkB,IAAI,CAAC;AACjC,IAAI,KAAK,GAAqB,IAAI,CAAC;AAEnC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,UAAkB,EAClB,KAAoB;IAEpB,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IAED,MAAM,SAAS,GAAG,oBAAoB,UAAU,EAAE,CAAC;IACnD,MAAM,QAAQ,GAAG,oBAAoB,UAAU,EAAE,CAAC;IAElD,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IAChB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAEzC,gDAAgD;IAChD,GAAG,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC;IAC3B,GAAG,CAAC,GAAG,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;IAErC,4EAA4E;IAC5E,KAAK,GAAG,SAAS,CAAC,iBAAiB,CAAC;QAClC,MAAM,EAAE,SAAS;QACjB,EAAE,EAAE,IAAI;QACR,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;IAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QACpC,IAAI,GAAG,IAAI,WAAW,IAAI,GAAG,EAAE,CAAC;YAC9B,IAAI,CAAC;gBACF,GAAW,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC/D,GAAW,CAAC,GAAG,EAAE,CAAC,2BAA2B,CAAC,CAAC;YAClD,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACZ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,mEAAmE;IACnE,sEAAsE;IACtE,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;QACxB,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAEnD,IAAI,aAAa,EAAE,CAAC;YAClB,qDAAqD;YACrD,KAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,qEAAqE;YACrE,KAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,iFAAiF;IACjF,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;QAC3C,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,IAAI,GAAG,CAAC;QAC9C,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAE3D,sCAAsC;QACtC,IAAI,UAAU,IAAI,GAAG,IAAI,UAAU,GAAG,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAC/E,oCAAoC;YACpC,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACnD,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACtB,IAAI,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAEnD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC7B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;gBAC9D,CAAC;qBAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;oBACpC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,aAAa,WAAW,CAAC,CAAC;gBAC9D,CAAC;qBAAM,CAAC;oBACN,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;gBAC/B,CAAC;gBAED,yDAAyD;gBACzD,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACxC,OAAO,OAAO,CAAC,gBAAgB,CAAC,CAAC;gBACjC,+DAA+D;gBAC/D,OAAO,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACnC,OAAO,OAAO,CAAC,mBAAmB,CAAC,CAAC;gBACpC,OAAO,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;gBAE5D,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACnC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,4CAA4C;YAC5C,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC5C,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,eAAe;IACf,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QACjD,MAAO,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;YACjD,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,UAAU,wDAAwD,CAAC,CAAC,CAAC;YAChG,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,mCAAmC;IACnC,MAAO,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAQ,EAAE,MAAW,EAAE,IAAS,EAAE,EAAE;QACzD,KAAM,CAAC,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,KAAK,EAAE,CAAC;QACd,KAAK,GAAG,IAAI,CAAC;IACf,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,MAAM,KAAK,IAAI,CAAC;AACzB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yo-bug",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "MCP Server for visual test feedback in vibe coding — QA capability as a protocol",
   "type": "module",
   "bin": {

package/src/server/proxy-server.ts

@@ -1,13 +1,13 @@
 import express from 'express';
 import cors from 'cors';
 import httpProxy from 'http-proxy';
-import type { Server } from 'http';
+import type { Server, IncomingMessage, ServerResponse } from 'http';
 import { FeedbackStore } from '../storage/store.js';
 import { createFeedbackRouter } from './feedback-api.js';
 import { createSdkRouter } from './sdk-serve.js';
-import { createHtmlInjector } from './html-injector.js';
 
 const PROXY_PORT = 3695;
+const INJECT_SCRIPT = `<script src="/vibe-feedback.js"></script>`;
 
 let server: Server | null = null;
 let proxy: httpProxy | null = null;
@@ -24,51 +24,81 @@ export async function startProxyServer(
   const proxyUrl = `http://localhost:${PROXY_PORT}`;
 
   const app = express();
-
-  // CORS for SDK to submit feedback
   app.use(cors());
-
-  // Parse JSON for feedback API
   app.use(express.json({ limit: '10mb' }));
 
-  // Serve SDK JS file
+  // Our own routes — served directly, not proxied
   app.use(createSdkRouter());
-
-  // Feedback API routes
   app.use(createFeedbackRouter(store));
 
-  // Create proxy
+  // Create proxy — let it handle responses by default (no selfHandleResponse)
   proxy = httpProxy.createProxyServer({
     target: targetUrl,
-    ws: true, // WebSocket support for HMR
+    ws: true,
     changeOrigin: true,
   });
 
   proxy.on('error', (_err, _req, res) => {
     if (res && 'writeHead' in res) {
-      (res as any).writeHead?.(502, { 'Content-Type': 'text/plain' });
-      (res as any).end?.('Dev server not responding');
+      try {
+        (res as any).writeHead?.(502, { 'Content-Type': 'text/plain' });
+        (res as any).end?.('Dev server not responding');
+      } catch {}
     }
   });
 
-  // Intercept proxy responses to inject SDK into HTML
-  proxy.on('proxyRes', (proxyRes, _req, res) => {
-    const injector = createHtmlInjector(proxyRes, res as any);
-    if (injector) {
-      // HTML response — buffer and inject
-      proxyRes.on('data', (chunk) => injector.onData(chunk));
-      proxyRes.on('end', () => injector.onEnd());
+  // Inject SDK into HTML responses using the modifyResponse approach
+  // We check Accept header on REQUEST to decide if we need to intercept
+  app.all('*', (req, res) => {
+    const accept = req.headers['accept'] || '';
+    const isHtmlRequest = accept.includes('text/html');
+
+    if (isHtmlRequest) {
+      // Potentially an HTML page — intercept to inject SDK
+      proxy!.web(req, res, { selfHandleResponse: true });
     } else {
-      // Non-HTML — pipe through directly
-      proxyRes.pipe(res as any);
+      // Static asset (JS/CSS/images/fonts/etc) — let proxy handle directly
+      proxy!.web(req, res);
     }
   });
 
-  // All other requests → proxy to dev server
-  app.all('*', (req, res) => {
-    proxy!.web(req, res, {
-      selfHandleResponse: true, // We handle response in proxyRes event
-    });
+  // Handle HTML injection in proxyRes (only fires for selfHandleResponse requests)
+  proxy.on('proxyRes', (proxyRes, _req, res) => {
+    const statusCode = proxyRes.statusCode || 200;
+    const contentType = proxyRes.headers['content-type'] || '';
+
+    // Only inject into 2xx HTML responses
+    if (statusCode >= 200 && statusCode < 300 && contentType.includes('text/html')) {
+      // Buffer response, inject SDK, send
+      const chunks: Buffer[] = [];
+      proxyRes.on('data', (chunk) => chunks.push(chunk));
+      proxyRes.on('end', () => {
+        let html = Buffer.concat(chunks).toString('utf-8');
+
+        if (html.includes('</body>')) {
+          html = html.replace('</body>', `${INJECT_SCRIPT}\n</body>`);
+        } else if (html.includes('</html>')) {
+          html = html.replace('</html>', `${INJECT_SCRIPT}\n</html>`);
+        } else {
+          html += `\n${INJECT_SCRIPT}`;
+        }
+
+        // Copy headers except content-length (body size changed)
+        const headers = { ...proxyRes.headers };
+        delete headers['content-length'];
+        // Remove content-encoding since we decoded the buffer as utf-8
+        delete headers['content-encoding'];
+        delete headers['transfer-encoding'];
+        headers['content-length'] = String(Buffer.byteLength(html));
+
+        res.writeHead(statusCode, headers);
+        res.end(html);
+      });
+    } else {
+      // Redirect or non-HTML — pass through as-is
+      res.writeHead(statusCode, proxyRes.headers);
+      proxyRes.pipe(res);
+    }
   });
 
   // Start server

package/src/server/html-injector.ts

@@ -1,41 +0,0 @@
-import type { IncomingMessage, ServerResponse } from 'http';
-
-const INJECT_SCRIPT = `<script src="/vibe-feedback.js"></script>`;
-
-/**
- * Middleware that intercepts HTML responses from the proxy
- * and injects the vibe-feedback SDK script before </body>.
- */
-export function createHtmlInjector(proxyRes: IncomingMessage, res: ServerResponse) {
-  const contentType = proxyRes.headers['content-type'] || '';
-  if (!contentType.includes('text/html')) {
-    return null; // Not HTML, don't intercept
-  }
-
-  // Copy all response headers except content-length (we'll modify the body)
-  res.writeHead(proxyRes.statusCode || 200, Object.fromEntries(
-    Object.entries(proxyRes.headers).filter(([key]) => key !== 'content-length')
-  ));
-
-  const chunks: Buffer[] = [];
-
-  return {
-    onData(chunk: Buffer) {
-      chunks.push(chunk);
-    },
-    onEnd() {
-      let html = Buffer.concat(chunks).toString('utf-8');
-
-      // Inject before </body> or at end if no </body>
-      if (html.includes('</body>')) {
-        html = html.replace('</body>', `${INJECT_SCRIPT}\n</body>`);
-      } else if (html.includes('</html>')) {
-        html = html.replace('</html>', `${INJECT_SCRIPT}\n</html>`);
-      } else {
-        html += `\n${INJECT_SCRIPT}`;
-      }
-
-      res.end(html);
-    },
-  };
-}