maga init

Author: fredzqm

src/init/index.ts

@@ -72,16 +72,7 @@ const featureMap = new Map(featuresList.map((feature) => [feature.name, feature]
 export async function init(setup: Setup, config: any, options: any): Promise<any> {
   const nextFeature = setup.features?.shift();
   if (nextFeature) {
-    const f = featureMap.get(nextFeature);
-    if (!f) {
-      const availableFeatures = Object.keys(features)
-        .filter((f) => f !== "project")
-        .join(", ");
-      throw new FirebaseError(
-        `${clc.bold(nextFeature)} is not a valid feature. Must be one of ${availableFeatures}`,
-      );
-    }
-
+    const f = lookupFeature(nextFeature);
     logger.info(clc.bold(`\n${clc.white("===")} ${capitalize(nextFeature)} Setup`));
 
     if (f.doSetup) {
@@ -100,3 +91,35 @@ export async function init(setup: Setup, config: any, options: any): Promise<any
     return init(setup, config, options);
   }
 }
+
+export async function actuate(setup: Setup, config: any, options: any): Promise<any> {
+  const nextFeature = setup.features?.shift();
+  if (nextFeature) {
+    const f = lookupFeature(nextFeature);
+    logger.info(clc.bold(`\n${clc.white("===")} ${capitalize(nextFeature)} Setup Actuation`));
+
+    if (f.doSetup) {
+      throw new FirebaseError(
+        `The feature ${nextFeature} does not support actuate yet. Please run ${clc.bold("firebase init " + nextFeature)} instead.`,
+      );
+    } else {
+      if (f.actuate) {
+        await f.actuate(setup, config, options);
+      }
+    }
+    return actuate(setup, config, options);
+  }
+}
+
+function lookupFeature(feature: string,): Feature {
+  const f = featureMap.get(feature);
+  if (!f) {
+    const availableFeatures = Object.keys(features)
+      .filter((f) => f !== "project")
+      .join(", ");
+    throw new FirebaseError(
+      `${clc.bold(feature)} is not a valid feature. Must be one of ${availableFeatures}`,
+    );
+  }
+  return f;
+}

src/mcp/tools/core/init.ts

@@ -0,0 +1,45 @@
+import { z } from "zod";
+import { tool } from "../../tool.js";
+import { toContent } from "../../util.js";
+import { actuate, Setup } from "../../../init/index.js";
+
+export const init = tool(
+  {
+    name: "init",
+    description: "Initialize the Firebase Products.",
+    inputSchema: z.object({
+      features: z.object({
+        // TODO: Add all the features here.
+        dataconnect: z.object({
+          serviceId: z.string().optional().describe("The Firebase Data Connect service ID to setup."),
+          locationId: z.string().default("us-central1").describe("The GCP region ID to set up the Firebase Data Connect service. For example, us-central1."),
+        }),
+      }),
+    }),
+    annotations: {
+      title: "List the Firebase Data Connect Services that's available in the backend",
+      readOnlyHint: false,
+    },
+    _meta: {
+      requiresProject: false, // Can start from stratch.
+      requiresAuth: false, // Will throw error if the specific feature needs it.
+    },
+  },
+  async ({features}, { projectId, config, rc }) => {
+    const setup: Setup = {
+      config: config.src,
+      rcfile: rc.data,
+      projectId,
+      features: [],
+      featureInfo: {},
+    };
+    if (features.dataconnect) {
+      setup.features.push("dataconnect");
+      setup.featureInfo.dataconnect = features.dataconnect;
+    }
+    await actuate(setup, config, {});
+    return toContent(
+      `The Firebase Data Connect Services has been initialized. You can now use the Firebase Data Connect Services.`,
+    );
+  },
+);