Non-interactive mode for migrate.ts

src/firebase_studio/migrate.spec.ts

@@ -25,12 +25,12 @@
 
   describe("extractMetadata", () => {
     beforeEach(() => {
       sandbox.stub(fs, "readFile").callsFake(async (p: any) => {
         const pStr = p.toString();
         if (pStr.endsWith("metadata.json")) {
           return JSON.stringify({ projectId: "original-project" });
         }
         if (pStr.endsWith("blueprint.md")) {
           return "# **App Name**: Test App";
         }
         return "";
@@ -63,7 +63,7 @@
     let spawnStub: sinon.SinonStub;
 
     beforeEach(() => {
       sandbox.stub(fs, "stat").resolves({ isDirectory: () => true } as any);
       const cp = require("child_process");
       sandbox.stub(cp, "spawnSync").returns({ status: 0 });
       sandbox.stub(process, "platform").value("darwin");
@@ -724,18 +724,31 @@
       expect(writeFileStub.calledWith(mcpConfigPath, sinon.match(/"dart":/))).to.be.true;
     });
 
-    it("should prompt user for Dart MCP server and NOT configure if they decline", async () => {
-      accessStub.withArgs(sinon.match("pubspec.yaml")).resolves();
-      commandStub.withArgs("dart").returns(true);
-      confirmStub
-        .withArgs(sinon.match({ message: sinon.match(/Dart MCP server/) }))
-        .resolves(false);
+    it("should respect non-interactive mode and not prompt", async () => {
+      commandStub.withArgs("agy").returns(true);
+      commandStub.withArgs("npx").returns(true);
 
-      await migrate(testRoot);
+      await migrate(testRoot, { startAntigravity: true, nonInteractive: true });
 
-      const mcpConfigDir = path.join(require("os").homedir(), ".gemini", "antigravity");
-      const mcpConfigPath = path.join(mcpConfigDir, "mcp_config.json");
-      expect(writeFileStub.calledWith(mcpConfigPath, sinon.match(/"dart":/))).to.be.false;
+      // Verify that prompts were called with nonInteractive: true
+      expect(
+        confirmStub.calledWith(
+          sinon.match({ message: sinon.match(/Firebase MCP server/), nonInteractive: true }),
+        ),
+      ).to.be.true;
+      expect(
+        selectStub.calledWith(
+          sinon.match({
+            message: sinon.match(/install Firebase project skills/),
+            nonInteractive: true,
+          }),
+        ),
+      ).to.be.true;
+      expect(
+        confirmStub.calledWith(
+          sinon.match({ message: sinon.match(/open it in Antigravity now/), nonInteractive: true }),
+        ),
+      ).to.be.true;
     });
   });
 

src/firebase_studio/migrate.ts

@@ -18,6 +18,7 @@ import { installAgentSkills } from "../agentSkills";
 export interface MigrateOptions {
   project?: string;
   startAntigravity?: boolean;
+  nonInteractive?: boolean;
 }
 
 interface McpServerConfig {
@@ -29,7 +30,11 @@ interface McpConfig {
   mcpServers: Record<string, McpServerConfig>;
 }
 
-async function setupAntigravityMcpServer(rootPath: string, appType?: AppType): Promise<void> {
+async function setupAntigravityMcpServer(
+  rootPath: string,
+  appType?: AppType,
+  nonInteractive?: boolean,
+): Promise<void> {
   const mcpConfigDir = path.join(os.homedir(), ".gemini", "antigravity");
   const mcpConfigPath = path.join(mcpConfigDir, "mcp_config.json");
 
@@ -59,6 +64,7 @@ async function setupAntigravityMcpServer(rootPath: string, appType?: AppType): P
         const confirmFirebase = await prompt.confirm({
           message: "Would you like to enable the Firebase MCP server for Antigravity?",
           default: true,
+          nonInteractive,
         });
 
         if (confirmFirebase) {
@@ -82,6 +88,7 @@ async function setupAntigravityMcpServer(rootPath: string, appType?: AppType): P
           const confirmDart = await prompt.confirm({
             message: "Would you like to enable the Dart MCP server for Antigravity?",
             default: true,
+            nonInteractive,
           });
 
           if (confirmDart) {
@@ -284,6 +291,7 @@ async function injectAntigravityContext(
   rootPath: string,
   projectId: string | undefined,
   appName: string,
+  nonInteractive?: boolean,
 ): Promise<void> {
   const agentDir = path.join(rootPath, ".agents");
   const rulesDir = path.join(agentDir, "rules");
@@ -302,7 +310,7 @@ async function injectAntigravityContext(
       { name: "Globally for all projects", value: "global" },
     ],
     default: "local",
-    nonInteractive: process.env.NODE_ENV === "test",
+    nonInteractive: nonInteractive || process.env.NODE_ENV === "test",
   });
 
   await installAgentSkills({
@@ -386,6 +394,7 @@ async function getAgyCommand(startAgy?: boolean): Promise<string | undefined> {
 async function createFirebaseConfigs(
   rootPath: string,
   projectId: string | undefined,
+  nonInteractive?: boolean,
 ): Promise<void> {
   if (!projectId) {
     return;
@@ -426,7 +435,7 @@ async function createFirebaseConfigs(
         const confirmBackend = await prompt.confirm({
           message: `Would you like to use the App Hosting backend "${selectedBackendId}"?`,
           default: true,
-          nonInteractive: process.env.NODE_ENV === "test",
+          nonInteractive: nonInteractive || process.env.NODE_ENV === "test",
         });
 
         if (confirmBackend) {
@@ -697,6 +706,7 @@ async function askToOpenAntigravity(
   rootPath: string,
   appName: string,
   startAntigravity?: boolean,
+  nonInteractive?: boolean,
 ): Promise<void> {
   const agyCommand = await getAgyCommand(startAntigravity);
 
@@ -721,6 +731,7 @@ async function askToOpenAntigravity(
   const answer = await prompt.confirm({
     message: "Would you like to open it in Antigravity now?",
     default: true,
+    nonInteractive,
   });
 
   if (answer) {
@@ -771,12 +782,12 @@ export async function migrate(
   }
 
   await updateReadme(rootPath, appType);
-  await createFirebaseConfigs(rootPath, projectId);
+  await createFirebaseConfigs(rootPath, projectId, options.nonInteractive);
   await uploadSecrets(rootPath, projectId);
   await upgradeGenkitVersion(rootPath);
-  await injectAntigravityContext(rootPath, projectId, appName);
+  await injectAntigravityContext(rootPath, projectId, appName, options.nonInteractive);
   await writeAntigravityConfigs(rootPath, appType);
-  await setupAntigravityMcpServer(rootPath, appType);
+  await setupAntigravityMcpServer(rootPath, appType, options.nonInteractive);
   await cleanupUnusedFiles(rootPath);
 
   // Suggest renaming if we are in the 'download' folder
@@ -788,5 +799,5 @@ export async function migrate(
   }
 
   await track.trackGA4("firebase_studio_migrate", { app_type: appType, result: "success" });
-  await askToOpenAntigravity(rootPath, appName, options.startAntigravity);
+  await askToOpenAntigravity(rootPath, appName, options.startAntigravity, options.nonInteractive);
 }