vite-plugin-ai-annotator 1.0.0

package/dist/utils/sourcemap.d.ts

@@ -0,0 +1,72 @@
+/**
+ * Source Map utilities for mapping runtime elements back to source code
+ */
+export interface SourceMapPosition {
+    line: number;
+    column: number;
+    source?: string;
+    name?: string;
+}
+export interface SourceMapInfo {
+    file: string;
+    line: number;
+    column: number;
+    endLine?: number;
+    endColumn?: number;
+    originalSource?: string;
+    sourcesContent?: string[];
+}
+/**
+ * Basic source map consumer for parsing VLQ-encoded mappings
+ * Note: This is a simplified implementation. For production use,
+ * consider using the 'source-map' library for full compatibility.
+ */
+export declare class BasicSourceMapConsumer {
+    private sources;
+    private sourcesContent?;
+    constructor(sourceMap: any);
+    /**
+     * Find original position for generated position
+     * This is a simplified implementation that looks for approximate mappings
+     */
+    originalPositionFor(line: number, column: number): SourceMapPosition | null;
+    /**
+     * Get source content for a given source file
+     */
+    sourceContentFor(source: string): string | null;
+    /**
+     * Get all available sources
+     */
+    getSources(): string[];
+}
+/**
+ * Attempt to find and parse source maps from various sources
+ */
+export declare class SourceMapResolver {
+    private sourceMapCache;
+    /**
+     * Try to find source map for a given file URL
+     */
+    findSourceMapForFile(fileUrl: string): Promise<BasicSourceMapConsumer | null>;
+    /**
+     * Resolve original position from generated position using source maps
+     */
+    resolvePosition(fileUrl: string, line: number, column: number): Promise<SourceMapPosition | null>;
+    /**
+     * Get enhanced location info by combining runtime data with source maps
+     */
+    getEnhancedLocationInfo(fileUrl: string, line: number, column: number): Promise<SourceMapInfo | null>;
+    /**
+     * Clear the source map cache
+     */
+    clearCache(): void;
+}
+export declare const sourceMapResolver: SourceMapResolver;
+/**
+ * Helper function to extract source location from browser stack traces
+ */
+export declare function extractLocationFromStackTrace(error: Error): {
+    file: string;
+    line: number;
+    column: number;
+} | null;

package/dist/utils/xpath.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Enhanced XPath and CSS selector utilities for browser environment
+ */
+export interface ElementSelector {
+    xpath: string;
+    cssSelector: string;
+    uniqueId?: string;
+    position?: {
+        index: number;
+        total: number;
+    };
+}
+export declare class XPathUtils {
+    static generateXPath(element: Element): string;
+    private static getXPathStep;
+    private static getXPathIndex;
+    private static isValidId;
+    /**
+     * Generate optimized CSS selector using optimal-select library
+     */
+    static generateOptimalSelector(element: Element): string;
+    /**
+     * Generate enhanced CSS selector optimized for browser environment
+     */
+    static generateEnhancedCSSSelector(element: Element): string;
+    /**
+     * Validate XPath by testing if it uniquely identifies the element
+     */
+    static validateXPath(xpath: string, targetElement: Element): boolean;
+    /**
+     * Validate CSS selector by testing if it uniquely identifies the element
+     */
+    static validateCSSSelector(selector: string, targetElement: Element): boolean;
+    /**
+     * Find element using XPath
+     */
+    static findElementByXPath(xpath: string): Element | null;
+    /**
+     * Generate comprehensive selector information for an element
+     */
+    static generateElementSelector(element: Element): ElementSelector;
+    /**
+     * Check if an ID is unique in the document
+     */
+    private static isUniqueId;
+    /**
+     * Generate multiple selector strategies for robust element identification
+     */
+    static generateRobustSelectors(element: Element): {
+        primary: ElementSelector;
+        fallbacks: string[];
+        confidence: 'high' | 'medium' | 'low';
+    };
+}

package/dist/vite-plugin.d.ts

@@ -0,0 +1,25 @@
+import type { Plugin } from 'vite';
+export interface AiAnnotatorOptions {
+    /**
+     * Port to run the server on
+     * @default 7318
+     */
+    port?: number;
+    /**
+     * Address for the server to listen on
+     * @default 'localhost'
+     */
+    listenAddress?: string;
+    /**
+     * Public URL for reverse proxy scenarios
+     * @default Automatically determined from listenAddress and port
+     */
+    publicAddress?: string;
+    /**
+     * Verbose logging
+     * @default false
+     */
+    verbose?: boolean;
+}
+export declare function aiAnnotator(options?: AiAnnotatorOptions): Plugin;
+export default aiAnnotator;

package/dist/vite-plugin.js

@@ -0,0 +1,219 @@
+// src/vite-plugin.ts
+import { spawn } from "node:child_process";
+import { fileURLToPath } from "node:url";
+import { dirname, join } from "node:path";
+import { existsSync } from "node:fs";
+var AiAnnotatorServer = class {
+  constructor(options = {}) {
+    this.serverProcess = null;
+    const port = options.port ?? 7318;
+    const listenAddress = options.listenAddress ?? "localhost";
+    this.options = {
+      port,
+      listenAddress,
+      publicAddress: options.publicAddress ?? `http://${listenAddress}:${port}`,
+      verbose: options.verbose ?? false
+    };
+    const currentFileDir = dirname(fileURLToPath(import.meta.url));
+    this.isDevelopment = currentFileDir.endsWith("/src") || currentFileDir.endsWith("\\src");
+    if (this.isDevelopment) {
+      this.packageDir = dirname(currentFileDir);
+    } else {
+      this.packageDir = dirname(currentFileDir);
+    }
+    this.log(`Package directory: ${this.packageDir}`);
+    this.log(`Running in ${this.isDevelopment ? "development" : "production"} mode`);
+  }
+  async start() {
+    if (this.serverProcess) {
+      return;
+    }
+    const isRunning = await this.isServerRunning();
+    if (isRunning) {
+      this.log(`Server already running on port ${this.options.port}, skipping spawn`);
+      return;
+    }
+    let serverFile;
+    let cmd;
+    let args;
+    if (this.isDevelopment) {
+      serverFile = join(this.packageDir, "src", "index.ts");
+      cmd = "bun";
+      args = [serverFile];
+    } else {
+      serverFile = join(this.packageDir, "dist", "index.cjs");
+      if (!existsSync(serverFile)) {
+        const fallbackFile = join(this.packageDir, "src", "index.ts");
+        if (existsSync(fallbackFile)) {
+          this.log("dist/index.cjs not found, falling back to src/index.ts");
+          serverFile = fallbackFile;
+          cmd = "bun";
+          args = [serverFile];
+        } else {
+          throw new Error(`Annotator server file not found at ${serverFile} or ${fallbackFile}`);
+        }
+      } else {
+        cmd = "node";
+        args = [serverFile];
+      }
+    }
+    args.push("--port", String(this.options.port));
+    args.push("--listen", this.options.listenAddress);
+    args.push("--public-address", this.options.publicAddress);
+    if (this.options.verbose) {
+      args.push("--verbose");
+    }
+    this.log(`Starting annotator server: ${cmd} ${args.join(" ")}`);
+    this.log(`Working directory: ${this.packageDir}`);
+    this.serverProcess = spawn(cmd, args, {
+      cwd: this.packageDir,
+      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}`);
+      });
+    }
+    this.serverProcess.on("error", (error) => {
+      console.error("[ai-annotator-server] Failed to start:", error);
+    });
+    this.serverProcess.on("exit", (code) => {
+      if (code !== 0 && code !== null) {
+        console.error(`[ai-annotator-server] Process exited with code ${code}`);
+      }
+      this.serverProcess = null;
+    });
+    await new Promise((resolve) => setTimeout(resolve, 1e3));
+  }
+  async stop() {
+    if (this.serverProcess) {
+      this.log("Stopping annotator server...");
+      this.serverProcess.kill("SIGTERM");
+      await new Promise((resolve) => {
+        if (!this.serverProcess) {
+          resolve();
+          return;
+        }
+        const timeout = setTimeout(() => {
+          if (this.serverProcess) {
+            this.log("Force killing annotator server...");
+            this.serverProcess.kill("SIGKILL");
+          }
+          resolve();
+        }, 5e3);
+        this.serverProcess.on("exit", () => {
+          clearTimeout(timeout);
+          resolve();
+        });
+      });
+      this.serverProcess = null;
+    }
+  }
+  async isServerRunning() {
+    try {
+      const response = await fetch(`http://${this.options.listenAddress}:${this.options.port}/health`, {
+        signal: AbortSignal.timeout(1e3)
+      });
+      return response.ok;
+    } catch {
+      return false;
+    }
+  }
+  log(message) {
+    if (this.options.verbose) {
+      console.log(`[ai-annotator] ${message}`);
+    }
+  }
+  getInjectScript() {
+    return `<script src="${this.options.publicAddress}/annotator-toolbar.js" type="module" async></script>`;
+  }
+  shouldInject() {
+    return true;
+  }
+};
+function injectScriptIntoHtml(html, scriptTag) {
+  if (html.includes("</body>")) {
+    return html.replace("</body>", `${scriptTag}
+</body>`);
+  } else if (html.includes("</html>")) {
+    return html.replace("</html>", `${scriptTag}
+</html>`);
+  }
+  return html + scriptTag;
+}
+function aiAnnotator(options = {}) {
+  let serverManager;
+  return {
+    name: "vite-plugin-ai-annotator",
+    // Only apply plugin during development (serve command)
+    apply: "serve",
+    configResolved() {
+      serverManager = new AiAnnotatorServer(options);
+    },
+    async buildStart() {
+      await serverManager.start();
+    },
+    // For regular Vite apps (SPA)
+    transformIndexHtml(html) {
+      if (!serverManager || !serverManager.shouldInject()) {
+        return html;
+      }
+      return injectScriptIntoHtml(html, serverManager.getInjectScript());
+    },
+    // For SSR frameworks like Nuxt - intercept HTML responses
+    configureServer(server) {
+      server.middlewares.use((_req, res, next) => {
+        if (!serverManager || !serverManager.shouldInject()) {
+          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) {
+            chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+            return true;
+          }
+          return originalWrite(chunk, ...args);
+        };
+        res.end = function(chunk, ...args) {
+          if (chunk) {
+            chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
+          }
+          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);
+        };
+        next();
+      });
+    },
+    async closeBundle() {
+      await serverManager.stop();
+    },
+    async buildEnd() {
+      if (this.meta.watchMode === false) {
+        await serverManager.stop();
+      }
+    }
+  };
+}
+var vite_plugin_default = aiAnnotator;
+export {
+  aiAnnotator,
+  vite_plugin_default as default
+};
+//# sourceMappingURL=vite-plugin.js.map

package/dist/vite-plugin.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../src/vite-plugin.ts"],
+  "sourcesContent": ["import type { Plugin, ViteDevServer } from 'vite';\nimport { spawn, ChildProcess } from 'node:child_process';\nimport { fileURLToPath } from 'node:url';\nimport { dirname, join } from 'node:path';\nimport { existsSync } from 'node:fs';\n\nexport interface AiAnnotatorOptions {\n  /**\n   * Port to run the server on\n   * @default 7318\n   */\n  port?: number;\n  /**\n   * Address for the server to listen on\n   * @default 'localhost'\n   */\n  listenAddress?: string;\n  /**\n   * Public URL for reverse proxy scenarios\n   * @default Automatically determined from listenAddress and port\n   */\n  publicAddress?: string;\n  /**\n   * Verbose logging\n   * @default false\n   */\n  verbose?: boolean;\n}\n\nclass AiAnnotatorServer {\n  private serverProcess: ChildProcess | null = null;\n  private options: Required<AiAnnotatorOptions>;\n  private packageDir: string;\n  private isDevelopment: boolean;\n\n  constructor(options: AiAnnotatorOptions = {}) {\n    const port = options.port ?? 7318;\n    const listenAddress = options.listenAddress ?? 'localhost';\n    \n    this.options = {\n      port,\n      listenAddress,\n      publicAddress: options.publicAddress ?? `http://${listenAddress}:${port}`,\n      verbose: options.verbose ?? false,\n    };\n  \n\n    // Detect if we're running from source or from installed package\n    const currentFileDir = dirname(fileURLToPath(import.meta.url));\n\n    // Check if we're in src directory (development) or dist directory (production)\n    this.isDevelopment = currentFileDir.endsWith('/src') || currentFileDir.endsWith('\\\\src');\n\n    // Get the package root directory\n    if (this.isDevelopment) {\n      // In development: current file is in src/, package root is one level up\n      this.packageDir = dirname(currentFileDir);\n    } else {\n      // In production: current file is in dist/, package root is one level up\n      this.packageDir = dirname(currentFileDir);\n    }\n\n    this.log(`Package directory: ${this.packageDir}`);\n    this.log(`Running in ${this.isDevelopment ? 'development' : 'production'} mode`);\n  }\n\n  async start(): Promise<void> {\n    if (this.serverProcess) {\n      return;\n    }\n\n    // Check if server is already running on the port\n    const isRunning = await this.isServerRunning();\n    if (isRunning) {\n      this.log(`Server already running on port ${this.options.port}, skipping spawn`);\n      return;\n    }\n\n    // Determine the server file to run\n    let serverFile: string;\n    let cmd: string;\n    let args: string[];\n\n    if (this.isDevelopment) {\n      // Development: run TypeScript file directly with bun\n      serverFile = join(this.packageDir, 'src', 'index.ts');\n      cmd = 'bun';\n      args = [serverFile];\n    } else {\n      // Production: run compiled JavaScript file\n      serverFile = join(this.packageDir, 'dist', 'index.cjs');\n\n      // Check if dist/index.cjs exists, if not try to use bun with src/index.ts as fallback\n      if (!existsSync(serverFile)) {\n        const fallbackFile = join(this.packageDir, 'src', 'index.ts');\n        if (existsSync(fallbackFile)) {\n          this.log('dist/index.cjs not found, falling back to src/index.ts');\n          serverFile = fallbackFile;\n          cmd = 'bun';\n          args = [serverFile];\n        } else {\n          throw new Error(`Annotator server file not found at ${serverFile} or ${fallbackFile}`);\n        }\n      } else {\n        // Use node for compiled CJS in production\n        cmd = 'node';\n        args = [serverFile];\n      }\n    }\n\n    // Add CLI arguments\n    args.push('--port', String(this.options.port));\n    args.push('--listen', this.options.listenAddress);\n    args.push('--public-address', this.options.publicAddress);\n    if (this.options.verbose) {\n      args.push('--verbose');\n    }\n\n    this.log(`Starting annotator server: ${cmd} ${args.join(' ')}`);\n    this.log(`Working directory: ${this.packageDir}`);\n\n    // Start the server process\n    this.serverProcess = spawn(cmd, args, {\n      cwd: this.packageDir,\n      env: process.env,\n      stdio: this.options.verbose ? 'inherit' : 'pipe',\n    });\n\n    if (!this.options.verbose && this.serverProcess.stdout) {\n      this.serverProcess.stdout.on('data', (_data) => {\n        // Pipe output for debugging if needed\n      });\n    }\n\n    if (!this.options.verbose && this.serverProcess.stderr) {\n      this.serverProcess.stderr.on('data', (data) => {\n        console.error(`[ai-annotator-server] Error: ${data}`);\n      });\n    }\n\n    this.serverProcess.on('error', (error) => {\n      console.error('[ai-annotator-server] Failed to start:', error);\n    });\n\n    this.serverProcess.on('exit', (code) => {\n      if (code !== 0 && code !== null) {\n        console.error(`[ai-annotator-server] Process exited with code ${code}`);\n      }\n      this.serverProcess = null;\n    });\n\n    // Give server a moment to start\n    await new Promise(resolve => setTimeout(resolve, 1000));\n  }\n\n  async stop(): Promise<void> {\n    if (this.serverProcess) {\n      this.log('Stopping annotator server...');\n\n      // Send SIGTERM for graceful shutdown\n      this.serverProcess.kill('SIGTERM');\n\n      // Wait for process to exit\n      await new Promise<void>((resolve) => {\n        if (!this.serverProcess) {\n          resolve();\n          return;\n        }\n\n        const timeout = setTimeout(() => {\n          // Force kill if not exited after 5 seconds\n          if (this.serverProcess) {\n            this.log('Force killing annotator server...');\n            this.serverProcess.kill('SIGKILL');\n          }\n          resolve();\n        }, 5000);\n\n        this.serverProcess.on('exit', () => {\n          clearTimeout(timeout);\n          resolve();\n        });\n      });\n\n      this.serverProcess = null;\n    }\n  }\n\n\n  private async isServerRunning(): Promise<boolean> {\n    try {\n      const response = await fetch(`http://${this.options.listenAddress}:${this.options.port}/health`, {\n        signal: AbortSignal.timeout(1000),\n      });\n      return response.ok;\n    } catch {\n      return false;\n    }\n  }\n\n  private log(message: string): void {\n    if (this.options.verbose) {\n      console.log(`[ai-annotator] ${message}`);\n    }\n  }\n\n  getInjectScript(): string {\n    return `<script src=\"${this.options.publicAddress}/annotator-toolbar.js\" type=\"module\" async></script>`;\n  }\n\n  shouldInject(): boolean {\n    return true;\n  }\n\n}\n\nfunction injectScriptIntoHtml(html: string, scriptTag: string): string {\n  if (html.includes('</body>')) {\n    return html.replace('</body>', `${scriptTag}\\n</body>`);\n  } else if (html.includes('</html>')) {\n    return html.replace('</html>', `${scriptTag}\\n</html>`);\n  }\n  return html + scriptTag;\n}\n\nexport function aiAnnotator(options: AiAnnotatorOptions = {}): Plugin {\n  let serverManager: AiAnnotatorServer;\n\n  return {\n    name: 'vite-plugin-ai-annotator',\n    // Only apply plugin during development (serve command)\n    apply: 'serve',\n\n    configResolved() {\n      serverManager = new AiAnnotatorServer(options);\n    },\n\n    async buildStart() {\n      await serverManager.start();\n    },\n\n    // For regular Vite apps (SPA)\n    transformIndexHtml(html: string) {\n      if (!serverManager || !serverManager.shouldInject()) {\n        return html;\n      }\n      return injectScriptIntoHtml(html, serverManager.getInjectScript());\n    },\n\n    // For SSR frameworks like Nuxt - intercept HTML responses\n    configureServer(server: ViteDevServer) {\n      server.middlewares.use((_req, res, next) => {\n        if (!serverManager || !serverManager.shouldInject()) {\n          return next();\n        }\n\n        // Only intercept HTML responses\n        const originalWrite = res.write.bind(res);\n        const originalEnd = res.end.bind(res);\n        let chunks: Buffer[] = [];\n        let isHtml = false;\n\n        res.write = function(chunk: any, ...args: any[]) {\n          // Check content-type on first write\n          const contentType = res.getHeader('content-type');\n          if (typeof contentType === 'string' && contentType.includes('text/html')) {\n            isHtml = true;\n          }\n\n          if (isHtml && chunk) {\n            chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));\n            return true;\n          }\n          return originalWrite(chunk, ...args);\n        } as any;\n\n        res.end = function(chunk?: any, ...args: any[]) {\n          if (chunk) {\n            chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));\n          }\n\n          if (isHtml && chunks.length > 0) {\n            const html = Buffer.concat(chunks).toString('utf-8');\n            const injectedHtml = injectScriptIntoHtml(html, serverManager.getInjectScript());\n\n            // Update content-length header\n            res.setHeader('content-length', Buffer.byteLength(injectedHtml));\n            return originalEnd(injectedHtml, ...args);\n          }\n\n          return originalEnd(chunk, ...args);\n        } as any;\n\n        next();\n      });\n    },\n\n    async closeBundle() {\n      await serverManager.stop();\n    },\n\n    async buildEnd() {\n      // Stop server when build ends (in build mode)\n      if (this.meta.watchMode === false) {\n        await serverManager.stop();\n      }\n    },\n  };\n}\n\n// Default export for convenience\nexport default aiAnnotator;"],
+  "mappings": ";AACA,SAAS,aAA2B;AACpC,SAAS,qBAAqB;AAC9B,SAAS,SAAS,YAAY;AAC9B,SAAS,kBAAkB;AAyB3B,IAAM,oBAAN,MAAwB;AAAA,EAMtB,YAAY,UAA8B,CAAC,GAAG;AAL9C,SAAQ,gBAAqC;AAM3C,UAAM,OAAO,QAAQ,QAAQ;AAC7B,UAAM,gBAAgB,QAAQ,iBAAiB;AAE/C,SAAK,UAAU;AAAA,MACb;AAAA,MACA;AAAA,MACA,eAAe,QAAQ,iBAAiB,UAAU,aAAa,IAAI,IAAI;AAAA,MACvE,SAAS,QAAQ,WAAW;AAAA,IAC9B;AAIA,UAAM,iBAAiB,QAAQ,cAAc,YAAY,GAAG,CAAC;AAG7D,SAAK,gBAAgB,eAAe,SAAS,MAAM,KAAK,eAAe,SAAS,OAAO;AAGvF,QAAI,KAAK,eAAe;AAEtB,WAAK,aAAa,QAAQ,cAAc;AAAA,IAC1C,OAAO;AAEL,WAAK,aAAa,QAAQ,cAAc;AAAA,IAC1C;AAEA,SAAK,IAAI,sBAAsB,KAAK,UAAU,EAAE;AAChD,SAAK,IAAI,cAAc,KAAK,gBAAgB,gBAAgB,YAAY,OAAO;AAAA,EACjF;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,KAAK,eAAe;AACtB;AAAA,IACF;AAGA,UAAM,YAAY,MAAM,KAAK,gBAAgB;AAC7C,QAAI,WAAW;AACb,WAAK,IAAI,kCAAkC,KAAK,QAAQ,IAAI,kBAAkB;AAC9E;AAAA,IACF;AAGA,QAAI;AACJ,QAAI;AACJ,QAAI;AAEJ,QAAI,KAAK,eAAe;AAEtB,mBAAa,KAAK,KAAK,YAAY,OAAO,UAAU;AACpD,YAAM;AACN,aAAO,CAAC,UAAU;AAAA,IACpB,OAAO;AAEL,mBAAa,KAAK,KAAK,YAAY,QAAQ,WAAW;AAGtD,UAAI,CAAC,WAAW,UAAU,GAAG;AAC3B,cAAM,eAAe,KAAK,KAAK,YAAY,OAAO,UAAU;AAC5D,YAAI,WAAW,YAAY,GAAG;AAC5B,eAAK,IAAI,wDAAwD;AACjE,uBAAa;AACb,gBAAM;AACN,iBAAO,CAAC,UAAU;AAAA,QACpB,OAAO;AACL,gBAAM,IAAI,MAAM,sCAAsC,UAAU,OAAO,YAAY,EAAE;AAAA,QACvF;AAAA,MACF,OAAO;AAEL,cAAM;AACN,eAAO,CAAC,UAAU;AAAA,MACpB;AAAA,IACF;AAGA,SAAK,KAAK,UAAU,OAAO,KAAK,QAAQ,IAAI,CAAC;AAC7C,SAAK,KAAK,YAAY,KAAK,QAAQ,aAAa;AAChD,SAAK,KAAK,oBAAoB,KAAK,QAAQ,aAAa;AACxD,QAAI,KAAK,QAAQ,SAAS;AACxB,WAAK,KAAK,WAAW;AAAA,IACvB;AAEA,SAAK,IAAI,8BAA8B,GAAG,IAAI,KAAK,KAAK,GAAG,CAAC,EAAE;AAC9D,SAAK,IAAI,sBAAsB,KAAK,UAAU,EAAE;AAGhD,SAAK,gBAAgB,MAAM,KAAK,MAAM;AAAA,MACpC,KAAK,KAAK;AAAA,MACV,KAAK,QAAQ;AAAA,MACb,OAAO,KAAK,QAAQ,UAAU,YAAY;AAAA,IAC5C,CAAC;AAED,QAAI,CAAC,KAAK,QAAQ,WAAW,KAAK,cAAc,QAAQ;AACtD,WAAK,cAAc,OAAO,GAAG,QAAQ,CAAC,UAAU;AAAA,MAEhD,CAAC;AAAA,IACH;AAEA,QAAI,CAAC,KAAK,QAAQ,WAAW,KAAK,cAAc,QAAQ;AACtD,WAAK,cAAc,OAAO,GAAG,QAAQ,CAAC,SAAS;AAC7C,gBAAQ,MAAM,gCAAgC,IAAI,EAAE;AAAA,MACtD,CAAC;AAAA,IACH;AAEA,SAAK,cAAc,GAAG,SAAS,CAAC,UAAU;AACxC,cAAQ,MAAM,0CAA0C,KAAK;AAAA,IAC/D,CAAC;AAED,SAAK,cAAc,GAAG,QAAQ,CAAC,SAAS;AACtC,UAAI,SAAS,KAAK,SAAS,MAAM;AAC/B,gBAAQ,MAAM,kDAAkD,IAAI,EAAE;AAAA,MACxE;AACA,WAAK,gBAAgB;AAAA,IACvB,CAAC;AAGD,UAAM,IAAI,QAAQ,aAAW,WAAW,SAAS,GAAI,CAAC;AAAA,EACxD;AAAA,EAEA,MAAM,OAAsB;AAC1B,QAAI,KAAK,eAAe;AACtB,WAAK,IAAI,8BAA8B;AAGvC,WAAK,cAAc,KAAK,SAAS;AAGjC,YAAM,IAAI,QAAc,CAAC,YAAY;AACnC,YAAI,CAAC,KAAK,eAAe;AACvB,kBAAQ;AACR;AAAA,QACF;AAEA,cAAM,UAAU,WAAW,MAAM;AAE/B,cAAI,KAAK,eAAe;AACtB,iBAAK,IAAI,mCAAmC;AAC5C,iBAAK,cAAc,KAAK,SAAS;AAAA,UACnC;AACA,kBAAQ;AAAA,QACV,GAAG,GAAI;AAEP,aAAK,cAAc,GAAG,QAAQ,MAAM;AAClC,uBAAa,OAAO;AACpB,kBAAQ;AAAA,QACV,CAAC;AAAA,MACH,CAAC;AAED,WAAK,gBAAgB;AAAA,IACvB;AAAA,EACF;AAAA,EAGA,MAAc,kBAAoC;AAChD,QAAI;AACF,YAAM,WAAW,MAAM,MAAM,UAAU,KAAK,QAAQ,aAAa,IAAI,KAAK,QAAQ,IAAI,WAAW;AAAA,QAC/F,QAAQ,YAAY,QAAQ,GAAI;AAAA,MAClC,CAAC;AACD,aAAO,SAAS;AAAA,IAClB,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEQ,IAAI,SAAuB;AACjC,QAAI,KAAK,QAAQ,SAAS;AACxB,cAAQ,IAAI,kBAAkB,OAAO,EAAE;AAAA,IACzC;AAAA,EACF;AAAA,EAEA,kBAA0B;AACxB,WAAO,gBAAgB,KAAK,QAAQ,aAAa;AAAA,EACnD;AAAA,EAEA,eAAwB;AACtB,WAAO;AAAA,EACT;AAEF;AAEA,SAAS,qBAAqB,MAAc,WAA2B;AACrE,MAAI,KAAK,SAAS,SAAS,GAAG;AAC5B,WAAO,KAAK,QAAQ,WAAW,GAAG,SAAS;AAAA,QAAW;AAAA,EACxD,WAAW,KAAK,SAAS,SAAS,GAAG;AACnC,WAAO,KAAK,QAAQ,WAAW,GAAG,SAAS;AAAA,QAAW;AAAA,EACxD;AACA,SAAO,OAAO;AAChB;AAEO,SAAS,YAAY,UAA8B,CAAC,GAAW;AACpE,MAAI;AAEJ,SAAO;AAAA,IACL,MAAM;AAAA;AAAA,IAEN,OAAO;AAAA,IAEP,iBAAiB;AACf,sBAAgB,IAAI,kBAAkB,OAAO;AAAA,IAC/C;AAAA,IAEA,MAAM,aAAa;AACjB,YAAM,cAAc,MAAM;AAAA,IAC5B;AAAA;AAAA,IAGA,mBAAmB,MAAc;AAC/B,UAAI,CAAC,iBAAiB,CAAC,cAAc,aAAa,GAAG;AACnD,eAAO;AAAA,MACT;AACA,aAAO,qBAAqB,MAAM,cAAc,gBAAgB,CAAC;AAAA,IACnE;AAAA;AAAA,IAGA,gBAAgB,QAAuB;AACrC,aAAO,YAAY,IAAI,CAAC,MAAM,KAAK,SAAS;AAC1C,YAAI,CAAC,iBAAiB,CAAC,cAAc,aAAa,GAAG;AACnD,iBAAO,KAAK;AAAA,QACd;AAGA,cAAM,gBAAgB,IAAI,MAAM,KAAK,GAAG;AACxC,cAAM,cAAc,IAAI,IAAI,KAAK,GAAG;AACpC,YAAI,SAAmB,CAAC;AACxB,YAAI,SAAS;AAEb,YAAI,QAAQ,SAAS,UAAe,MAAa;AAE/C,gBAAM,cAAc,IAAI,UAAU,cAAc;AAChD,cAAI,OAAO,gBAAgB,YAAY,YAAY,SAAS,WAAW,GAAG;AACxE,qBAAS;AAAA,UACX;AAEA,cAAI,UAAU,OAAO;AACnB,mBAAO,KAAK,OAAO,SAAS,KAAK,IAAI,QAAQ,OAAO,KAAK,KAAK,CAAC;AAC/D,mBAAO;AAAA,UACT;AACA,iBAAO,cAAc,OAAO,GAAG,IAAI;AAAA,QACrC;AAEA,YAAI,MAAM,SAAS,UAAgB,MAAa;AAC9C,cAAI,OAAO;AACT,mBAAO,KAAK,OAAO,SAAS,KAAK,IAAI,QAAQ,OAAO,KAAK,KAAK,CAAC;AAAA,UACjE;AAEA,cAAI,UAAU,OAAO,SAAS,GAAG;AAC/B,kBAAM,OAAO,OAAO,OAAO,MAAM,EAAE,SAAS,OAAO;AACnD,kBAAM,eAAe,qBAAqB,MAAM,cAAc,gBAAgB,CAAC;AAG/E,gBAAI,UAAU,kBAAkB,OAAO,WAAW,YAAY,CAAC;AAC/D,mBAAO,YAAY,cAAc,GAAG,IAAI;AAAA,UAC1C;AAEA,iBAAO,YAAY,OAAO,GAAG,IAAI;AAAA,QACnC;AAEA,aAAK;AAAA,MACP,CAAC;AAAA,IACH;AAAA,IAEA,MAAM,cAAc;AAClB,YAAM,cAAc,KAAK;AAAA,IAC3B;AAAA,IAEA,MAAM,WAAW;AAEf,UAAI,KAAK,KAAK,cAAc,OAAO;AACjC,cAAM,cAAc,KAAK;AAAA,MAC3B;AAAA,IACF;AAAA,EACF;AACF;AAGA,IAAO,sBAAQ;",
+  "names": []
+}

package/dist/ws-server.d.ts

@@ -0,0 +1,26 @@
+import type { Express } from 'express';
+import type { Server } from 'node:http';
+import { Server as SocketIOServer, Socket } from 'socket.io';
+import { type RpcServer } from './rpc/server.generated';
+import type { BrowserSession } from './rpc/define';
+export interface ServerInstance {
+    app: Express;
+    server: Server;
+    io: SocketIOServer;
+    port: number;
+    listenAddress: string;
+    publicAddress: string;
+    verbose: boolean;
+}
+export interface BrowserConnection {
+    socket: Socket;
+    rpc: RpcServer;
+    session: BrowserSession;
+}
+export declare function getAllSessions(): BrowserSession[];
+export declare function getRpc(sessionId?: string): {
+    rpc: RpcServer;
+    sessionId: string;
+} | null;
+export declare function startServer(port: number, listenAddress: string, publicAddress: string, verbose?: boolean): Promise<ServerInstance>;
+export declare function stopServer(serverInstance: ServerInstance): Promise<void>;

package/package.json

@@ -0,0 +1,86 @@
+{
+  "name": "vite-plugin-ai-annotator",
+  "version": "1.0.0",
+  "description": "AI-powered element annotator for Vite - Pick elements and get instant AI code modifications",
+  "type": "module",
+  "main": "dist/vite-plugin.js",
+  "types": "dist/vite-plugin.d.ts",
+  "bin": {
+    "ai-annotator": "./dist/index.cjs"
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/vite-plugin.d.ts",
+      "import": "./dist/vite-plugin.js"
+    },
+    "./vite-plugin": {
+      "types": "./dist/vite-plugin.d.ts",
+      "import": "./dist/vite-plugin.js"
+    }
+  },
+  "scripts": {
+    "dev": "bun src/index.ts",
+    "build:server": "bun scripts/build-server.js",
+    "build:annotator": "bun scripts/build-annotator.js",
+    "build:vite-plugin": "bun scripts/build-vite-plugin.js",
+    "build:types": "bun scripts/build-types.js",
+    "build": "bun run build:server && bun run build:annotator && bun run build:vite-plugin && bun run build:types",
+    "typecheck": "tsc --noEmit",
+    "prepublishOnly": "bun run build"
+  },
+  "files": [
+    "dist/**/*",
+    "README.md"
+  ],
+  "keywords": [
+    "vite",
+    "vite-plugin",
+    "ai",
+    "annotator",
+    "element-picker",
+    "claude",
+    "inspector",
+    "web-development",
+    "frontend"
+  ],
+  "author": "Nguyen Van Duoc <nguyenvanduocit@gmail.com>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/nguyenvanduocit/vite-plugin-ai-annotator.git"
+  },
+  "homepage": "https://github.com/nguyenvanduocit/vite-plugin-ai-annotator#readme",
+  "bugs": {
+    "url": "https://github.com/nguyenvanduocit/vite-plugin-ai-annotator/issues"
+  },
+  "dependencies": {
+    "@floating-ui/dom": "^1.7.4",
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "@types/js-yaml": "^4.0.9",
+    "cors": "^2.8.5",
+    "express": "^5.1.0",
+    "html-to-image": "^1.11.13",
+    "js-yaml": "^4.1.0",
+    "lit": "^3.3.1",
+    "optimal-select": "^4.0.1",
+    "socket.io": "^4.8.1",
+    "socket.io-client": "^4.8.1",
+    "zod": "^4.0.17"
+  },
+  "devDependencies": {
+    "@types/bun": "^1.2.20",
+    "@types/cors": "^2.8.17",
+    "@types/express": "^5.0.2",
+    "@types/node": "^22.13.4",
+    "esbuild": "^0.25.9",
+    "tsx": "^4.7.0",
+    "typescript": "^5.7.3",
+    "vite": "^6.0.7"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "peerDependencies": {
+    "vite": ">=2.0.0"
+  }
+}