Add export from zip support' (#10051)

* Add export from zip support'

* Emojis are fun

* More minor fixes

Author: joehan

src/commands/studio-export.ts

@@ -5,18 +5,45 @@ import { migrate, MigrateOptions } from "../firebase_studio/migrate";
 import * as path from "path";
 import * as experiments from "../experiments";
 import { FirebaseError } from "../error";
+import { unzip } from "../unzip";
+import * as fs from "fs";
 
 export const command = new Command("studio:export <path>")
   .description(
-    "Bootstrap Firebase Studio apps for migration to Antigravity. Run on the unzipped folder from the Firebase Studio download.",
+    "Bootstrap Firebase Studio apps for migration to Antigravity. Run on the unzipped folder from the Firebase Studio download, or directly on the downloaded zip file.",
   )
   .option("--no-start-agy", "skip starting the Antigravity IDE after migration")
   .action(async (exportPath: string, options: Options) => {
     experiments.assertEnabled("studioexport", "export Studio apps");
     if (!exportPath) {
       throw new FirebaseError("Must specify a path for migration.", { exit: 1 });
     }
-    const rootPath = path.resolve(exportPath);
-    logger.info(`Exporting Studio apps from ${rootPath} to Antigravity...`);
+
+    let rootPath = path.resolve(exportPath);
+
+    if (fs.existsSync(rootPath) && fs.statSync(rootPath).isFile() && rootPath.endsWith(".zip")) {
+      logger.info(`⏳ Unzipping ${rootPath}...`);
+      const parsedPath = path.parse(rootPath);
+      let extractDirName = parsedPath.name;
+      if (!extractDirName || extractDirName === ".") {
+        extractDirName = "studio-export";
+      }
+      const extractPath = path.join(parsedPath.dir, extractDirName);
+      await unzip(rootPath, extractPath);
+
+      // Studio exports usually contain a single top-level directory.
+      // E.g., `Export-12345/`. Let's check if we should dive into it
+      const extractedItems = fs.readdirSync(extractPath);
+      if (
+        extractedItems.length === 1 &&
+        fs.statSync(path.join(extractPath, extractedItems[0])).isDirectory()
+      ) {
+        rootPath = path.join(extractPath, extractedItems[0]);
+      } else {
+        rootPath = extractPath;
+      }
+    }
+
+    logger.info(`⏳ Exporting Studio apps from ${rootPath} to Antigravity...`);
     await migrate(rootPath, options as MigrateOptions);
   });

src/firebase_studio/migrate.ts

@@ -136,7 +136,9 @@ export async function extractMetadata(
     logger.info(`✅ Detected Firebase Project: ${projectId}`);
   } else {
     // TODO need a mitigation here
-    logger.info(`✅ Failed to determine the Firebase Project ID`);
+    logger.info(
+      `❌ Failed to determine the Firebase Project ID. You can set a project later with 'firebase use <project-id>' or by setting the '--project' flag.`,
+    );
   }
 
   // Extract App Name and Blueprint Content