@react-grab/cli 0.1.27 → 0.1.29

package/dist/cli.cjs

@@ -345,10 +345,29 @@ var scanDirectoryForProjects = (rootDirectory, ignorer, maxDepth, currentDepth =
 var MAX_SCAN_DEPTH = 2;
 var findReactProjects = (projectRoot) => {
   if (detectMonorepo(projectRoot)) {
-    return findWorkspaceProjects(projectRoot);
+    const workspaceProjects = findWorkspaceProjects(projectRoot);
+    if (workspaceProjects.length > 0) {
+      return workspaceProjects;
+    }
   }
   const ignorer = loadGitignore(projectRoot);
-  return scanDirectoryForProjects(projectRoot, ignorer, MAX_SCAN_DEPTH);
+  const scannedProjects = scanDirectoryForProjects(
+    projectRoot,
+    ignorer,
+    MAX_SCAN_DEPTH
+  );
+  if (scannedProjects.length > 0) {
+    return scannedProjects;
+  }
+  let currentDirectory = path.dirname(projectRoot);
+  while (currentDirectory !== path.dirname(currentDirectory)) {
+    const parentProject = buildReactProject(currentDirectory);
+    if (parentProject) {
+      return [parentProject];
+    }
+    currentDirectory = path.dirname(currentDirectory);
+  }
+  return [];
 };
 var hasReactGrabInFile = (filePath) => {
   if (!fs.existsSync(filePath)) return false;
@@ -414,7 +433,6 @@ var AGENT_PACKAGES = [
   "@react-grab/codex",
   "@react-grab/gemini",
   "@react-grab/amp",
-  "@react-grab/ami",
   "@react-grab/droid",
   "@react-grab/copilot",
   "@react-grab/mcp"
@@ -617,7 +635,8 @@ var installPackages = (packages, packageManager, projectRoot, isDev = true) => {
 `);
   child_process.execSync(fullCommand, {
     cwd: projectRoot,
-    stdio: "inherit"
+    stdio: "inherit",
+    env: { ...process.env, REACT_GRAB_INIT: "1" }
   });
 };
 var getPackagesToInstall = (agent, includeReactGrab = true) => {
@@ -890,7 +909,6 @@ var AGENTS = [
   "codex",
   "gemini",
   "amp",
-  "ami",
   "droid",
   "copilot"
 ];
@@ -901,7 +919,6 @@ var AGENT_NAMES = {
   codex: "Codex",
   gemini: "Gemini",
   amp: "Amp",
-  ami: "Ami",
   droid: "Droid",
   copilot: "Copilot"
 };
@@ -958,19 +975,15 @@ var NEXT_PAGES_ROUTER_SCRIPT_WITH_AGENT = (agent) => {
           />
         )}`;
 };
-var VITE_SCRIPT = `<script type="module">
-      if (import.meta.env.DEV) {
-        import("react-grab");
-      }
-    </script>`;
-var VITE_SCRIPT_WITH_AGENT = (agent) => {
-  if (agent === "none") return VITE_SCRIPT;
-  return `<script type="module">
-      if (import.meta.env.DEV) {
-        import("react-grab");
-        import("@react-grab/${agent}/client");
-      }
-    </script>`;
+var VITE_IMPORT = `if (import.meta.env.DEV) {
+  import("react-grab");
+}`;
+var VITE_IMPORT_WITH_AGENT = (agent) => {
+  if (agent === "none") return VITE_IMPORT;
+  return `if (import.meta.env.DEV) {
+  import("react-grab");
+  import("@react-grab/${agent}/client");
+}`;
 };
 var WEBPACK_IMPORT = `if (process.env.NODE_ENV === "development") {
   import("react-grab");
@@ -1166,51 +1179,7 @@ var addAgentToExistingNextApp = (originalContent, agent, filePath) => {
     message: "Could not find React Grab script to add agent after"
   };
 };
-var addAgentToExistingVite = (originalContent, agent, filePath) => {
-  if (agent === "none") {
-    return {
-      success: true,
-      filePath,
-      message: "React Grab is already configured",
-      noChanges: true
-    };
-  }
-  const agentPackage = `@react-grab/${agent}`;
-  if (originalContent.includes(agentPackage)) {
-    return {
-      success: true,
-      filePath,
-      message: `Agent ${agent} is already configured`,
-      noChanges: true
-    };
-  }
-  const agentImport = `import("${agentPackage}/client");`;
-  const reactGrabImportMatch = originalContent.match(
-    /import\s*\(\s*["']react-grab["']\s*\);?/
-  );
-  if (reactGrabImportMatch) {
-    const matchedText = reactGrabImportMatch[0];
-    const hasSemicolon = matchedText.endsWith(";");
-    const newContent = originalContent.replace(
-      matchedText,
-      `${hasSemicolon ? matchedText.slice(0, -1) : matchedText};
-        ${agentImport}`
-    );
-    return {
-      success: true,
-      filePath,
-      message: `Add ${agent} agent`,
-      originalContent,
-      newContent
-    };
-  }
-  return {
-    success: false,
-    filePath,
-    message: "Could not find React Grab import to add agent after"
-  };
-};
-var addAgentToExistingWebpack = (originalContent, agent, filePath) => {
+var addAgentToExistingImport = (originalContent, agent, filePath) => {
   if (agent === "none") {
     return {
       success: true,
@@ -1415,41 +1384,55 @@ ${SCRIPT_IMPORT}`
     newContent
   };
 };
+var checkExistingInstallation = (filePath, agent, reactGrabAlreadyConfigured) => {
+  const content = fs.readFileSync(filePath, "utf-8");
+  if (!hasReactGrabCode(content)) return null;
+  if (reactGrabAlreadyConfigured) {
+    return addAgentToExistingImport(content, agent, filePath);
+  }
+  return {
+    success: true,
+    filePath,
+    message: "React Grab is already installed in this file",
+    noChanges: true
+  };
+};
 var transformVite = (projectRoot, agent, reactGrabAlreadyConfigured, force = false) => {
-  const indexPath = findIndexHtml(projectRoot);
-  if (!indexPath) {
+  const entryPath = findEntryFile(projectRoot);
+  if (!force) {
+    const indexPath = findIndexHtml(projectRoot);
+    if (indexPath) {
+      const existingResult = checkExistingInstallation(
+        indexPath,
+        agent,
+        reactGrabAlreadyConfigured
+      );
+      if (existingResult) return existingResult;
+    }
+  }
+  if (!entryPath) {
     return {
       success: false,
       filePath: "",
-      message: "Could not find index.html"
-    };
-  }
-  const originalContent = fs.readFileSync(indexPath, "utf-8");
-  let newContent = originalContent;
-  const hasReactGrabInFile2 = hasReactGrabCode(originalContent);
-  if (!force && hasReactGrabInFile2 && reactGrabAlreadyConfigured) {
-    return addAgentToExistingVite(originalContent, agent, indexPath);
-  }
-  if (!force && hasReactGrabInFile2) {
-    return {
-      success: true,
-      filePath: indexPath,
-      message: "React Grab is already installed in this file",
-      noChanges: true
+      message: "Could not find entry file (src/index.tsx, src/main.tsx, etc.)"
     };
   }
-  const scriptBlock = VITE_SCRIPT_WITH_AGENT(agent);
-  const headMatch = newContent.match(/<head[^>]*>/i);
-  if (headMatch) {
-    newContent = newContent.replace(
-      headMatch[0],
-      `${headMatch[0]}
-    ${scriptBlock}`
+  if (!force) {
+    const existingResult = checkExistingInstallation(
+      entryPath,
+      agent,
+      reactGrabAlreadyConfigured
     );
+    if (existingResult) return existingResult;
   }
+  const originalContent = fs.readFileSync(entryPath, "utf-8");
+  const importBlock = VITE_IMPORT_WITH_AGENT(agent);
+  const newContent = `${importBlock}
+
+${originalContent}`;
   return {
     success: true,
-    filePath: indexPath,
+    filePath: entryPath,
     message: "Add React Grab" + (agent !== "none" ? ` with ${agent} agent` : ""),
     originalContent,
     newContent
@@ -1464,19 +1447,15 @@ var transformWebpack = (projectRoot, agent, reactGrabAlreadyConfigured, force =
       message: "Could not find entry file (src/index.tsx, src/main.tsx, etc.)"
     };
   }
-  const originalContent = fs.readFileSync(entryPath, "utf-8");
-  const hasReactGrabInFile2 = hasReactGrabCode(originalContent);
-  if (!force && hasReactGrabInFile2 && reactGrabAlreadyConfigured) {
-    return addAgentToExistingWebpack(originalContent, agent, entryPath);
-  }
-  if (!force && hasReactGrabInFile2) {
-    return {
-      success: true,
-      filePath: entryPath,
-      message: "React Grab is already installed in this file",
-      noChanges: true
-    };
+  if (!force) {
+    const existingResult = checkExistingInstallation(
+      entryPath,
+      agent,
+      reactGrabAlreadyConfigured
+    );
+    if (existingResult) return existingResult;
   }
+  const originalContent = fs.readFileSync(entryPath, "utf-8");
   const importBlock = WEBPACK_IMPORT_WITH_AGENT(agent);
   const newContent = `${importBlock}
 
@@ -1563,40 +1542,41 @@ ${newContent}`;
   };
 };
 var previewTransform = (projectRoot, framework, nextRouterType, agent, reactGrabAlreadyConfigured = false, force = false) => {
+  const resolvedAgent = agent === "mcp" ? "none" : agent;
   switch (framework) {
     case "next":
       if (nextRouterType === "app") {
         return transformNextAppRouter(
           projectRoot,
-          agent,
+          resolvedAgent,
           reactGrabAlreadyConfigured,
           force
         );
       }
       return transformNextPagesRouter(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "vite":
       return transformVite(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "tanstack":
       return transformTanStack(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "webpack":
       return transformWebpack(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
@@ -1841,8 +1821,17 @@ var findReactGrabFile = (projectRoot, framework, nextRouterType) => {
         return findLayoutFile(projectRoot);
       }
       return findDocumentFile(projectRoot);
-    case "vite":
-      return findIndexHtml(projectRoot);
+    case "vite": {
+      const entryFile = findEntryFile(projectRoot);
+      if (entryFile && hasReactGrabCode(fs.readFileSync(entryFile, "utf-8"))) {
+        return entryFile;
+      }
+      const indexHtml = findIndexHtml(projectRoot);
+      if (indexHtml && hasReactGrabCode(fs.readFileSync(indexHtml, "utf-8"))) {
+        return indexHtml;
+      }
+      return entryFile;
+    }
     case "tanstack":
       return findTanStackRootFile(projectRoot);
     case "webpack":
@@ -2253,7 +2242,7 @@ var previewCdnTransform = (projectRoot, framework, nextRouterType, targetCdnDoma
 };
 
 // src/commands/add.ts
-var VERSION = "0.1.27";
+var VERSION = "0.1.29";
 var formatInstalledAgentNames = (agents) => agents.map((agent) => AGENT_NAMES[agent] || agent).join(", ");
 var add = new commander.Command().name("add").alias("install").description("connect React Grab to your agent").argument("[agent]", `agent to connect (${AGENTS.join(", ")}, mcp)`).option("-y, --yes", "skip confirmation prompts", false).option(
   "-c, --cwd <cwd>",
@@ -2291,11 +2280,36 @@ var add = new commander.Command().name("add").alias("install").description("conn
     }
     let agentIntegration;
     let agentsToRemove = [];
-    if (agentArg) {
+    if (agentArg === "mcp") {
+      if (isNonInteractive) {
+        const results = installMcpServers();
+        const hasSuccess = results.some((result2) => result2.success);
+        if (!hasSuccess) {
+          logger.break();
+          logger.error("Failed to install MCP server.");
+          logger.break();
+          process.exit(1);
+        }
+      } else {
+        const didInstall = await promptMcpInstall();
+        if (!didInstall) {
+          logger.break();
+          process.exit(0);
+        }
+      }
+      logger.break();
+      logger.log(
+        `${highlighter.success("Success!")} MCP server has been configured.`
+      );
+      logger.log("Restart your agents to activate.");
+      logger.break();
+      agentIntegration = "mcp";
+      projectInfo.installedAgents = [...projectInfo.installedAgents, "mcp"];
+    } else if (agentArg) {
       if (!AGENTS.includes(agentArg)) {
         logger.break();
         logger.error(`Invalid agent: ${agentArg}`);
-        logger.error(`Available agents: ${AGENTS.join(", ")}`);
+        logger.error(`Available agents: ${AGENTS.join(", ")}, mcp`);
         logger.break();
         process.exit(1);
       }
@@ -2614,7 +2628,7 @@ var MAX_KEY_HOLD_DURATION_MS = 2e3;
 var MAX_CONTEXT_LINES = 50;
 
 // src/commands/configure.ts
-var VERSION2 = "0.1.27";
+var VERSION2 = "0.1.29";
 var isMac = process.platform === "darwin";
 var META_LABEL = isMac ? "Cmd" : "Win";
 var ALT_LABEL = isMac ? "Option" : "Alt";
@@ -3170,7 +3184,7 @@ var uninstallPackagesWithFeedback = (packages, packageManager, projectRoot) => {
 };
 
 // src/commands/init.ts
-var VERSION3 = "0.1.27";
+var VERSION3 = "0.1.29";
 var REPORT_URL = "https://react-grab.com/api/report-cli";
 var DOCS_URL = "https://github.com/aidenybai/react-grab";
 var reportToCli = (type, config, error) => {
@@ -3501,76 +3515,6 @@ var init = new commander.Command().name("init").description("initialize React Gr
           logger.break();
           logger.success("MCP server has been configured.");
           logger.log("Restart your agents to activate.");
-          agentIntegration2 = "mcp";
-          projectInfo.installedAgents = ["mcp"];
-          const result2 = previewTransform(
-            projectInfo.projectRoot,
-            projectInfo.framework,
-            projectInfo.nextRouterType,
-            agentIntegration2,
-            true
-          );
-          const packageJsonResult2 = previewPackageJsonTransform(
-            projectInfo.projectRoot,
-            agentIntegration2,
-            projectInfo.installedAgents,
-            projectInfo.packageManager
-          );
-          if (!result2.success) {
-            logger.break();
-            logger.error(result2.message);
-            logger.break();
-            process.exit(1);
-          }
-          const hasLayoutChanges2 = !result2.noChanges && result2.originalContent && result2.newContent;
-          const hasPackageJsonChanges2 = packageJsonResult2.success && !packageJsonResult2.noChanges && packageJsonResult2.originalContent && packageJsonResult2.newContent;
-          if (hasLayoutChanges2 || hasPackageJsonChanges2) {
-            logger.break();
-            if (hasLayoutChanges2) {
-              printDiff(
-                result2.filePath,
-                result2.originalContent,
-                result2.newContent
-              );
-            }
-            if (hasPackageJsonChanges2) {
-              if (hasLayoutChanges2) {
-                logger.break();
-              }
-              printDiff(
-                packageJsonResult2.filePath,
-                packageJsonResult2.originalContent,
-                packageJsonResult2.newContent
-              );
-            }
-            logger.break();
-            const { proceed } = await prompts({
-              type: "confirm",
-              name: "proceed",
-              message: "Apply these changes?",
-              initial: true
-            });
-            if (!proceed) {
-              logger.break();
-              logger.log("Agent addition cancelled.");
-            } else {
-              installPackagesWithFeedback(
-                getPackagesToInstall(agentIntegration2, false),
-                projectInfo.packageManager,
-                projectInfo.projectRoot
-              );
-              if (hasLayoutChanges2) {
-                applyTransformWithFeedback(result2);
-              }
-              if (hasPackageJsonChanges2) {
-                applyPackageJsonWithFeedback(packageJsonResult2);
-              }
-              logger.break();
-              logger.success(
-                `${getAgentName(agentIntegration2)} has been added.`
-              );
-            }
-          }
         } else {
           const { agent } = await prompts({
             type: "select",
@@ -4029,7 +3973,7 @@ var init = new commander.Command().name("init").description("initialize React Gr
     reportToCli("error", void 0, error);
   }
 });
-var VERSION4 = "0.1.27";
+var VERSION4 = "0.1.29";
 var remove = new commander.Command().name("remove").description("disconnect React Grab from your agent").argument("[agent]", `agent to disconnect (${AGENTS.join(", ")}, mcp)`).option("-y, --yes", "skip confirmation prompts", false).option(
   "-c, --cwd <cwd>",
   "working directory (defaults to current directory)",
@@ -4205,7 +4149,7 @@ var remove = new commander.Command().name("remove").description("disconnect Reac
 });
 
 // src/cli.ts
-var VERSION5 = "0.1.27";
+var VERSION5 = "0.1.29";
 var VERSION_API_URL = "https://www.react-grab.com/api/version";
 process.on("SIGINT", () => process.exit(0));
 process.on("SIGTERM", () => process.exit(0));

package/dist/cli.js

@@ -4,7 +4,7 @@ import pc from 'picocolors';
 import basePrompts from 'prompts';
 import { execSync } from 'child_process';
 import fs, { existsSync, readFileSync, writeFileSync, accessSync, constants, readdirSync } from 'fs';
-import path, { resolve, join, relative, basename } from 'path';
+import path, { resolve, join, dirname, relative, basename } from 'path';
 import { detect } from '@antfu/ni';
 import ignore from 'ignore';
 import os from 'os';
@@ -312,10 +312,29 @@ var scanDirectoryForProjects = (rootDirectory, ignorer, maxDepth, currentDepth =
 var MAX_SCAN_DEPTH = 2;
 var findReactProjects = (projectRoot) => {
   if (detectMonorepo(projectRoot)) {
-    return findWorkspaceProjects(projectRoot);
+    const workspaceProjects = findWorkspaceProjects(projectRoot);
+    if (workspaceProjects.length > 0) {
+      return workspaceProjects;
+    }
   }
   const ignorer = loadGitignore(projectRoot);
-  return scanDirectoryForProjects(projectRoot, ignorer, MAX_SCAN_DEPTH);
+  const scannedProjects = scanDirectoryForProjects(
+    projectRoot,
+    ignorer,
+    MAX_SCAN_DEPTH
+  );
+  if (scannedProjects.length > 0) {
+    return scannedProjects;
+  }
+  let currentDirectory = dirname(projectRoot);
+  while (currentDirectory !== dirname(currentDirectory)) {
+    const parentProject = buildReactProject(currentDirectory);
+    if (parentProject) {
+      return [parentProject];
+    }
+    currentDirectory = dirname(currentDirectory);
+  }
+  return [];
 };
 var hasReactGrabInFile = (filePath) => {
   if (!existsSync(filePath)) return false;
@@ -381,7 +400,6 @@ var AGENT_PACKAGES = [
   "@react-grab/codex",
   "@react-grab/gemini",
   "@react-grab/amp",
-  "@react-grab/ami",
   "@react-grab/droid",
   "@react-grab/copilot",
   "@react-grab/mcp"
@@ -584,7 +602,8 @@ var installPackages = (packages, packageManager, projectRoot, isDev = true) => {
 `);
   execSync(fullCommand, {
     cwd: projectRoot,
-    stdio: "inherit"
+    stdio: "inherit",
+    env: { ...process.env, REACT_GRAB_INIT: "1" }
   });
 };
 var getPackagesToInstall = (agent, includeReactGrab = true) => {
@@ -857,7 +876,6 @@ var AGENTS = [
   "codex",
   "gemini",
   "amp",
-  "ami",
   "droid",
   "copilot"
 ];
@@ -868,7 +886,6 @@ var AGENT_NAMES = {
   codex: "Codex",
   gemini: "Gemini",
   amp: "Amp",
-  ami: "Ami",
   droid: "Droid",
   copilot: "Copilot"
 };
@@ -925,19 +942,15 @@ var NEXT_PAGES_ROUTER_SCRIPT_WITH_AGENT = (agent) => {
           />
         )}`;
 };
-var VITE_SCRIPT = `<script type="module">
-      if (import.meta.env.DEV) {
-        import("react-grab");
-      }
-    </script>`;
-var VITE_SCRIPT_WITH_AGENT = (agent) => {
-  if (agent === "none") return VITE_SCRIPT;
-  return `<script type="module">
-      if (import.meta.env.DEV) {
-        import("react-grab");
-        import("@react-grab/${agent}/client");
-      }
-    </script>`;
+var VITE_IMPORT = `if (import.meta.env.DEV) {
+  import("react-grab");
+}`;
+var VITE_IMPORT_WITH_AGENT = (agent) => {
+  if (agent === "none") return VITE_IMPORT;
+  return `if (import.meta.env.DEV) {
+  import("react-grab");
+  import("@react-grab/${agent}/client");
+}`;
 };
 var WEBPACK_IMPORT = `if (process.env.NODE_ENV === "development") {
   import("react-grab");
@@ -1133,51 +1146,7 @@ var addAgentToExistingNextApp = (originalContent, agent, filePath) => {
     message: "Could not find React Grab script to add agent after"
   };
 };
-var addAgentToExistingVite = (originalContent, agent, filePath) => {
-  if (agent === "none") {
-    return {
-      success: true,
-      filePath,
-      message: "React Grab is already configured",
-      noChanges: true
-    };
-  }
-  const agentPackage = `@react-grab/${agent}`;
-  if (originalContent.includes(agentPackage)) {
-    return {
-      success: true,
-      filePath,
-      message: `Agent ${agent} is already configured`,
-      noChanges: true
-    };
-  }
-  const agentImport = `import("${agentPackage}/client");`;
-  const reactGrabImportMatch = originalContent.match(
-    /import\s*\(\s*["']react-grab["']\s*\);?/
-  );
-  if (reactGrabImportMatch) {
-    const matchedText = reactGrabImportMatch[0];
-    const hasSemicolon = matchedText.endsWith(";");
-    const newContent = originalContent.replace(
-      matchedText,
-      `${hasSemicolon ? matchedText.slice(0, -1) : matchedText};
-        ${agentImport}`
-    );
-    return {
-      success: true,
-      filePath,
-      message: `Add ${agent} agent`,
-      originalContent,
-      newContent
-    };
-  }
-  return {
-    success: false,
-    filePath,
-    message: "Could not find React Grab import to add agent after"
-  };
-};
-var addAgentToExistingWebpack = (originalContent, agent, filePath) => {
+var addAgentToExistingImport = (originalContent, agent, filePath) => {
   if (agent === "none") {
     return {
       success: true,
@@ -1382,41 +1351,55 @@ ${SCRIPT_IMPORT}`
     newContent
   };
 };
+var checkExistingInstallation = (filePath, agent, reactGrabAlreadyConfigured) => {
+  const content = readFileSync(filePath, "utf-8");
+  if (!hasReactGrabCode(content)) return null;
+  if (reactGrabAlreadyConfigured) {
+    return addAgentToExistingImport(content, agent, filePath);
+  }
+  return {
+    success: true,
+    filePath,
+    message: "React Grab is already installed in this file",
+    noChanges: true
+  };
+};
 var transformVite = (projectRoot, agent, reactGrabAlreadyConfigured, force = false) => {
-  const indexPath = findIndexHtml(projectRoot);
-  if (!indexPath) {
+  const entryPath = findEntryFile(projectRoot);
+  if (!force) {
+    const indexPath = findIndexHtml(projectRoot);
+    if (indexPath) {
+      const existingResult = checkExistingInstallation(
+        indexPath,
+        agent,
+        reactGrabAlreadyConfigured
+      );
+      if (existingResult) return existingResult;
+    }
+  }
+  if (!entryPath) {
     return {
       success: false,
       filePath: "",
-      message: "Could not find index.html"
-    };
-  }
-  const originalContent = readFileSync(indexPath, "utf-8");
-  let newContent = originalContent;
-  const hasReactGrabInFile2 = hasReactGrabCode(originalContent);
-  if (!force && hasReactGrabInFile2 && reactGrabAlreadyConfigured) {
-    return addAgentToExistingVite(originalContent, agent, indexPath);
-  }
-  if (!force && hasReactGrabInFile2) {
-    return {
-      success: true,
-      filePath: indexPath,
-      message: "React Grab is already installed in this file",
-      noChanges: true
+      message: "Could not find entry file (src/index.tsx, src/main.tsx, etc.)"
     };
   }
-  const scriptBlock = VITE_SCRIPT_WITH_AGENT(agent);
-  const headMatch = newContent.match(/<head[^>]*>/i);
-  if (headMatch) {
-    newContent = newContent.replace(
-      headMatch[0],
-      `${headMatch[0]}
-    ${scriptBlock}`
+  if (!force) {
+    const existingResult = checkExistingInstallation(
+      entryPath,
+      agent,
+      reactGrabAlreadyConfigured
     );
+    if (existingResult) return existingResult;
   }
+  const originalContent = readFileSync(entryPath, "utf-8");
+  const importBlock = VITE_IMPORT_WITH_AGENT(agent);
+  const newContent = `${importBlock}
+
+${originalContent}`;
   return {
     success: true,
-    filePath: indexPath,
+    filePath: entryPath,
     message: "Add React Grab" + (agent !== "none" ? ` with ${agent} agent` : ""),
     originalContent,
     newContent
@@ -1431,19 +1414,15 @@ var transformWebpack = (projectRoot, agent, reactGrabAlreadyConfigured, force =
       message: "Could not find entry file (src/index.tsx, src/main.tsx, etc.)"
     };
   }
-  const originalContent = readFileSync(entryPath, "utf-8");
-  const hasReactGrabInFile2 = hasReactGrabCode(originalContent);
-  if (!force && hasReactGrabInFile2 && reactGrabAlreadyConfigured) {
-    return addAgentToExistingWebpack(originalContent, agent, entryPath);
-  }
-  if (!force && hasReactGrabInFile2) {
-    return {
-      success: true,
-      filePath: entryPath,
-      message: "React Grab is already installed in this file",
-      noChanges: true
-    };
+  if (!force) {
+    const existingResult = checkExistingInstallation(
+      entryPath,
+      agent,
+      reactGrabAlreadyConfigured
+    );
+    if (existingResult) return existingResult;
   }
+  const originalContent = readFileSync(entryPath, "utf-8");
   const importBlock = WEBPACK_IMPORT_WITH_AGENT(agent);
   const newContent = `${importBlock}
 
@@ -1530,40 +1509,41 @@ ${newContent}`;
   };
 };
 var previewTransform = (projectRoot, framework, nextRouterType, agent, reactGrabAlreadyConfigured = false, force = false) => {
+  const resolvedAgent = agent === "mcp" ? "none" : agent;
   switch (framework) {
     case "next":
       if (nextRouterType === "app") {
         return transformNextAppRouter(
           projectRoot,
-          agent,
+          resolvedAgent,
           reactGrabAlreadyConfigured,
           force
         );
       }
       return transformNextPagesRouter(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "vite":
       return transformVite(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "tanstack":
       return transformTanStack(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
     case "webpack":
       return transformWebpack(
         projectRoot,
-        agent,
+        resolvedAgent,
         reactGrabAlreadyConfigured,
         force
       );
@@ -1808,8 +1788,17 @@ var findReactGrabFile = (projectRoot, framework, nextRouterType) => {
         return findLayoutFile(projectRoot);
       }
       return findDocumentFile(projectRoot);
-    case "vite":
-      return findIndexHtml(projectRoot);
+    case "vite": {
+      const entryFile = findEntryFile(projectRoot);
+      if (entryFile && hasReactGrabCode(readFileSync(entryFile, "utf-8"))) {
+        return entryFile;
+      }
+      const indexHtml = findIndexHtml(projectRoot);
+      if (indexHtml && hasReactGrabCode(readFileSync(indexHtml, "utf-8"))) {
+        return indexHtml;
+      }
+      return entryFile;
+    }
     case "tanstack":
       return findTanStackRootFile(projectRoot);
     case "webpack":
@@ -2220,7 +2209,7 @@ var previewCdnTransform = (projectRoot, framework, nextRouterType, targetCdnDoma
 };
 
 // src/commands/add.ts
-var VERSION = "0.1.27";
+var VERSION = "0.1.29";
 var formatInstalledAgentNames = (agents) => agents.map((agent) => AGENT_NAMES[agent] || agent).join(", ");
 var add = new Command().name("add").alias("install").description("connect React Grab to your agent").argument("[agent]", `agent to connect (${AGENTS.join(", ")}, mcp)`).option("-y, --yes", "skip confirmation prompts", false).option(
   "-c, --cwd <cwd>",
@@ -2258,11 +2247,36 @@ var add = new Command().name("add").alias("install").description("connect React
     }
     let agentIntegration;
     let agentsToRemove = [];
-    if (agentArg) {
+    if (agentArg === "mcp") {
+      if (isNonInteractive) {
+        const results = installMcpServers();
+        const hasSuccess = results.some((result2) => result2.success);
+        if (!hasSuccess) {
+          logger.break();
+          logger.error("Failed to install MCP server.");
+          logger.break();
+          process.exit(1);
+        }
+      } else {
+        const didInstall = await promptMcpInstall();
+        if (!didInstall) {
+          logger.break();
+          process.exit(0);
+        }
+      }
+      logger.break();
+      logger.log(
+        `${highlighter.success("Success!")} MCP server has been configured.`
+      );
+      logger.log("Restart your agents to activate.");
+      logger.break();
+      agentIntegration = "mcp";
+      projectInfo.installedAgents = [...projectInfo.installedAgents, "mcp"];
+    } else if (agentArg) {
       if (!AGENTS.includes(agentArg)) {
         logger.break();
         logger.error(`Invalid agent: ${agentArg}`);
-        logger.error(`Available agents: ${AGENTS.join(", ")}`);
+        logger.error(`Available agents: ${AGENTS.join(", ")}, mcp`);
         logger.break();
         process.exit(1);
       }
@@ -2581,7 +2595,7 @@ var MAX_KEY_HOLD_DURATION_MS = 2e3;
 var MAX_CONTEXT_LINES = 50;
 
 // src/commands/configure.ts
-var VERSION2 = "0.1.27";
+var VERSION2 = "0.1.29";
 var isMac = process.platform === "darwin";
 var META_LABEL = isMac ? "Cmd" : "Win";
 var ALT_LABEL = isMac ? "Option" : "Alt";
@@ -3137,7 +3151,7 @@ var uninstallPackagesWithFeedback = (packages, packageManager, projectRoot) => {
 };
 
 // src/commands/init.ts
-var VERSION3 = "0.1.27";
+var VERSION3 = "0.1.29";
 var REPORT_URL = "https://react-grab.com/api/report-cli";
 var DOCS_URL = "https://github.com/aidenybai/react-grab";
 var reportToCli = (type, config, error) => {
@@ -3468,76 +3482,6 @@ var init = new Command().name("init").description("initialize React Grab in your
           logger.break();
           logger.success("MCP server has been configured.");
           logger.log("Restart your agents to activate.");
-          agentIntegration2 = "mcp";
-          projectInfo.installedAgents = ["mcp"];
-          const result2 = previewTransform(
-            projectInfo.projectRoot,
-            projectInfo.framework,
-            projectInfo.nextRouterType,
-            agentIntegration2,
-            true
-          );
-          const packageJsonResult2 = previewPackageJsonTransform(
-            projectInfo.projectRoot,
-            agentIntegration2,
-            projectInfo.installedAgents,
-            projectInfo.packageManager
-          );
-          if (!result2.success) {
-            logger.break();
-            logger.error(result2.message);
-            logger.break();
-            process.exit(1);
-          }
-          const hasLayoutChanges2 = !result2.noChanges && result2.originalContent && result2.newContent;
-          const hasPackageJsonChanges2 = packageJsonResult2.success && !packageJsonResult2.noChanges && packageJsonResult2.originalContent && packageJsonResult2.newContent;
-          if (hasLayoutChanges2 || hasPackageJsonChanges2) {
-            logger.break();
-            if (hasLayoutChanges2) {
-              printDiff(
-                result2.filePath,
-                result2.originalContent,
-                result2.newContent
-              );
-            }
-            if (hasPackageJsonChanges2) {
-              if (hasLayoutChanges2) {
-                logger.break();
-              }
-              printDiff(
-                packageJsonResult2.filePath,
-                packageJsonResult2.originalContent,
-                packageJsonResult2.newContent
-              );
-            }
-            logger.break();
-            const { proceed } = await prompts({
-              type: "confirm",
-              name: "proceed",
-              message: "Apply these changes?",
-              initial: true
-            });
-            if (!proceed) {
-              logger.break();
-              logger.log("Agent addition cancelled.");
-            } else {
-              installPackagesWithFeedback(
-                getPackagesToInstall(agentIntegration2, false),
-                projectInfo.packageManager,
-                projectInfo.projectRoot
-              );
-              if (hasLayoutChanges2) {
-                applyTransformWithFeedback(result2);
-              }
-              if (hasPackageJsonChanges2) {
-                applyPackageJsonWithFeedback(packageJsonResult2);
-              }
-              logger.break();
-              logger.success(
-                `${getAgentName(agentIntegration2)} has been added.`
-              );
-            }
-          }
         } else {
           const { agent } = await prompts({
             type: "select",
@@ -3996,7 +3940,7 @@ var init = new Command().name("init").description("initialize React Grab in your
     reportToCli("error", void 0, error);
   }
 });
-var VERSION4 = "0.1.27";
+var VERSION4 = "0.1.29";
 var remove = new Command().name("remove").description("disconnect React Grab from your agent").argument("[agent]", `agent to disconnect (${AGENTS.join(", ")}, mcp)`).option("-y, --yes", "skip confirmation prompts", false).option(
   "-c, --cwd <cwd>",
   "working directory (defaults to current directory)",
@@ -4172,7 +4116,7 @@ var remove = new Command().name("remove").description("disconnect React Grab fro
 });
 
 // src/cli.ts
-var VERSION5 = "0.1.27";
+var VERSION5 = "0.1.29";
 var VERSION_API_URL = "https://www.react-grab.com/api/version";
 process.on("SIGINT", () => process.exit(0));
 process.on("SIGTERM", () => process.exit(0));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@react-grab/cli",
-  "version": "0.1.27",
+  "version": "0.1.29",
   "bin": {
     "react-grab": "./dist/cli.js"
   },