vite-plugin-ai-annotator 1.14.7 → 1.14.9

package/dist/mcp-tools.d.ts

@@ -0,0 +1,25 @@
+/**
+ * Shared MCP tool definitions for AI Annotator
+ *
+ * Registers all 9 MCP tools on a McpServer instance.
+ * Used by both ws-server.ts (HTTP MCP) and mcp-stdio.ts (stdio MCP).
+ */
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { BrowserSession, PageContext, ElementData, SelectionResult, ScreenshotResult, InjectResult, ConsoleEntry } from './rpc/define';
+import type { RpcError } from './rpc/types.generated';
+export interface AnnotatorConnection {
+    sessionId: string;
+    getPageContext(timeout?: number): Promise<PageContext | RpcError>;
+    getSelectedElements(timeout?: number): Promise<ElementData[] | RpcError>;
+    triggerSelection(mode: 'inspect' | 'selector', selector?: string, selectorType?: 'css' | 'xpath', timeout?: number): Promise<SelectionResult | RpcError>;
+    captureScreenshot(type: 'viewport' | 'element', selector?: string, quality?: number, timeout?: number): Promise<ScreenshotResult | RpcError>;
+    clearSelection(): void;
+    injectCSS(css: string, timeout?: number): Promise<InjectResult | RpcError>;
+    injectJS(code: string, timeout?: number): Promise<InjectResult | RpcError>;
+    getConsole(clear?: boolean, timeout?: number): Promise<ConsoleEntry[] | RpcError>;
+}
+export type GetConnection = (sessionId?: string) => AnnotatorConnection | {
+    error: string;
+};
+export type ListSessions = () => Promise<BrowserSession[]>;
+export declare function registerMcpTools(mcp: McpServer, listSessions: ListSessions, getConnection: GetConnection): void;

package/dist/nuxt-module.js

@@ -1,5 +1,5 @@
 // src/nuxt-module.ts
-import { defineNuxtModule, addVitePlugin, addWebpackPlugin } from "@nuxt/kit";
+import { defineNuxtModule, addVitePlugin } from "@nuxt/kit";
 
 // src/vite-plugin.ts
 import { spawn } from "node:child_process";
@@ -1199,7 +1199,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",
@@ -1333,10 +1333,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}`);
@@ -1351,7 +1347,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) {
@@ -1457,36 +1458,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();
       });
@@ -1518,7 +1521,6 @@ var nuxt_module_default = defineNuxtModule({
       return;
     }
     addVitePlugin(aiAnnotator(options));
-    addWebpackPlugin(aiAnnotator(options));
     console.log("[ai-annotator] Nuxt module initialized");
   }
 });