vite-plugin-ai-annotator 1.14.7 → 1.14.9

package/dist/vite-plugin.js

@@ -1196,7 +1196,7 @@ function injectSourceLocations(code, id, root) {
   const s = new MagicString(code);
   const relativePath = relative(root, id);
   let hasChanges = false;
-  const tagRegex = /<([a-zA-Z][a-zA-Z0-9-]*)((?:\s+[^>]*?)?)\s*\/?>/g;
+  const tagRegex = /<([a-zA-Z][a-zA-Z0-9-]*)((?:\s+(?:[^>"'\/=]+(?:=(?:"[^"]*"|'[^']*'|[^\s>"']+))?|\s))*)\s*\/?>/g;
   const skipTags = /* @__PURE__ */ new Set([
     "script",
     "style",
@@ -1330,10 +1330,6 @@ var AiAnnotatorServer = class {
       env: process.env,
       stdio: this.options.verbose ? "inherit" : "pipe"
     });
-    if (!this.options.verbose && this.serverProcess.stdout) {
-      this.serverProcess.stdout.on("data", (_data) => {
-      });
-    }
     if (!this.options.verbose && this.serverProcess.stderr) {
       this.serverProcess.stderr.on("data", (data) => {
         console.error(`[ai-annotator-server] Error: ${data}`);
@@ -1348,7 +1344,12 @@ var AiAnnotatorServer = class {
       }
       this.serverProcess = null;
     });
-    await new Promise((resolve) => setTimeout(resolve, 1e3));
+    const deadline = Date.now() + 5e3;
+    while (Date.now() < deadline) {
+      await new Promise((resolve) => setTimeout(resolve, 100));
+      if (await this.isServerRunning()) return;
+    }
+    this.log("Server did not respond to health check within 5 seconds");
   }
   async stop() {
     if (this.serverProcess) {
@@ -1454,36 +1455,38 @@ function aiAnnotator(options = {}) {
     },
     // For SSR frameworks like Nuxt - intercept HTML responses
     configureServer(server) {
+      server.httpServer?.on("close", () => {
+        serverManager.stop();
+      });
       server.middlewares.use((_req, res, next) => {
         if (!serverManager) {
           return next();
         }
-        const originalWrite = res.write.bind(res);
         const originalEnd = res.end.bind(res);
         let chunks = [];
-        let isHtml = false;
-        res.write = function(chunk, ...args) {
-          const contentType = res.getHeader("content-type");
-          if (typeof contentType === "string" && contentType.includes("text/html")) {
-            isHtml = true;
-          }
-          if (isHtml && chunk) {
+        res.write = function(chunk, ..._args) {
+          if (chunk) {
             chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-            return true;
           }
-          return originalWrite(chunk, ...args);
+          return true;
         };
         res.end = function(chunk, ...args) {
           if (chunk) {
             chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
           }
+          const contentType = res.getHeader("content-type");
+          const isHtml = typeof contentType === "string" && contentType.includes("text/html");
           if (isHtml && chunks.length > 0) {
             const html = Buffer.concat(chunks).toString("utf-8");
             const injectedHtml = injectScriptIntoHtml(html, serverManager.getInjectScript());
             res.setHeader("content-length", Buffer.byteLength(injectedHtml));
             return originalEnd(injectedHtml, ...args);
           }
-          return originalEnd(chunk, ...args);
+          if (chunks.length > 0) {
+            const body = Buffer.concat(chunks);
+            return originalEnd(body, ...args);
+          }
+          return originalEnd(void 0, ...args);
         };
         next();
       });