[MCP] Organize MCP tools && update README (#8523)

Author: fredzqm

src/mcp/README.md

@@ -29,15 +29,29 @@ In `.cursor/mcp.json` in your workspace directory, add the Firebase MCP server:
 
 ### Command Line Options
 
-- `--dir <absolute_dir_path>`: The absolute path of a directory containing `firebase.json` to set a project context for the MCP server. If unspecified, the `{get|set}_firebase_directory` tools will become available and the default directory will be the working directory where the MCP server was started.
+- `--dir <absolute_dir_path>`: The absolute path of a directory containing `firebase.json` to set a project context for the MCP server. If unspecified, the `{get|set}_project_directory` tools will become available and the default directory will be the working directory where the MCP server was started.
 - `--only <feature1,feature2>`: A comma-separated list of feature groups to activate. Use this to limit the tools exposed to only features you are actively using.
 
 ## Tools
 
-| Tool Name                | Feature Group | Description                                                                                                         |
-| ------------------------ | ------------- | ------------------------------------------------------------------------------------------------------------------- |
-| `get_firebase_directory` | `core`        | When running without the `--dir` command, retrieves the current directory (defaults to current working directory).  |
-| `set_firebase_directory` | `core`        | When running without the `--dir` command, sets the current project directory (i.e. one with `firebase.json` in it). |
-| `get_project`            | `project`     | Get basic information about the active project in the current Firebase directory.                                   |
-| `list_apps`              | `project`     | List registered apps for the currently active project.                                                              |
-| `get_sdk_config`         | `project`     | Get an Firebase client SDK config for a specific platform.                                                          |
+| Tool Name                        | Feature Group | Description                                                                                                                                                                                                    |
+| -------------------------------- | ------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| `get_project_directory`          | `directory`   | When running without the `--dir` command, retrieves the current directory (defaults to current working directory).                                                                                             |
+| `set_project_directory`          | `directory`   | When running without the `--dir` command, sets the current project directory (i.e. one with `firebase.json` in it).                                                                                            |
+| `firebase_get_project`           | `core`        | Get basic information about the active project in the current Firebase directory.                                                                                                                              |
+| `firebase_list_apps`             | `core`        | List registered apps for the currently active project.                                                                                                                                                         |
+| `firebase_get_sdk_config`        | `core`        | Get an Firebase client SDK config for a specific platform.                                                                                                                                                     |
+| `firebase_consult_assistant`     | `core`        | Consult Gemini in Firebase agent for help                                                                                                                                                                      |
+| `firestore_list_collections`     | `firestore`   | Retrieves a list of collections from a Firestore database in the current project.                                                                                                                              |
+| `firestore_get_documents`        | `firestore`   | Retrieves one or more Firestore documents from a database in the current project by full document paths. Use this if you know the exact path of a document.                                                    |
+| `firestore_get_rules`            | `firestore`   | Retrieves the active Firestore security rules for the current project.                                                                                                                                         |
+| `auth_get_user`                  | `auth`        | Retrieves a user based on an email address, phone number, or UID.                                                                                                                                              |
+| `auth_disable_user`              | `auth`        | Disables or enables a user based on a UID.                                                                                                                                                                     |
+| `auth_set_claims`                | `auth`        | Sets custom claims on a specific user's account. Use to create trusted values associated with a user e.g. marking them as an admin. Claims are limited in size and should be succinct in name and value.       |
+| `auth_set_sms_region_policy`     | `auth`        | Sets an SMS Region Policy for Firebase Auth to restrict the regions which can receive text messages based on an ALLOW or DENY list of country codes. This policy will override any existing policies when set. |
+| `dataconnect_list_services`      | `dataconnect` | List the Firebase Data Connect services available in the current project.                                                                                                                                      |
+| `dataconnect_generate_schema`    | `dataconnect` | Generates a Firebase Data Connect Schema based on the users description of an app.                                                                                                                             |
+| `dataconnect_generate_operation` | `dataconnect` | Generates a single Firebase Data Connect query or mutation based on the currently deployed schema and the provided prompt.                                                                                     |
+| `dataconnect_get_schema`         | `dataconnect` | List the Firebase Data Connect Schema in the project, which includes Cloud SQL data sources and the GraphQL Schema describing what tables are available.                                                       |
+| `dataconnect_get_connector`      | `dataconnect` | Get the Firebase Data Connect Connectors in the project, which includes the pre-defined GraphQL queries accessible to client SDKs.                                                                             |
+| `storage_get_rules`              | `storage`     | Retrieves the Firebase Cloud Storage Rules for the default bucket.                                                                                                                                             |

src/mcp/index.ts

@@ -9,10 +9,9 @@ import {
 } from "@modelcontextprotocol/sdk/types.js";
 import { mcpError } from "./util.js";
 import { ServerFeature } from "./types.js";
-import { tools } from "./tools/index.js";
+import { availableTools } from "./tools/index.js";
 import { ServerTool } from "./tool.js";
 import { configstore } from "../configstore.js";
-import { coreTools } from "./tools/core/index.js";
 import { Command } from "../command.js";
 import { requireAuth } from "../requireAuth.js";
 import { Options } from "../options.js";
@@ -47,14 +46,7 @@ export class FirebaseMcpServer {
   }
 
   get availableTools(): ServerTool[] {
-    const toolDefs: ServerTool[] = this.fixedRoot ? [] : [...coreTools];
-    const activeFeatures = this.activeFeatures?.length
-      ? this.activeFeatures
-      : (Object.keys(tools) as ServerFeature[]);
-    for (const key of activeFeatures || []) {
-      toolDefs.push(...tools[key]);
-    }
-    return toolDefs;
+    return availableTools(!!this.fixedRoot, this.activeFeatures);
   }
 
   getTool(name: string): ServerTool | null {

src/mcp/tools/core/consult_firebase_assistant.ts renamed to src/mcp/tools/core/consult_assistant.ts

@@ -3,9 +3,9 @@ import { tool } from "../../tool.js";
 import { toContent } from "../../util.js";
 import { chatWithFirebase } from "../../../gif/fdcExperience.js";
 
-export const consult_firebase_assistant = tool(
+export const consult_assistant = tool(
   {
-    name: "consult_firebase_assistant",
+    name: "consult_assistant",
     description:
       "Send a question to an AI assistant specifically enhanced to answer Firebase questions.",
     inputSchema: z.object({
@@ -20,7 +20,7 @@ export const consult_firebase_assistant = tool(
     _meta: {
       requiresProject: true,
       requiresAuth: true,
-      // TODO: Create an endpoint to check for GiF activiation.
+      // TODO: Create an endpoint to check for GiF activation.
     },
   },
   async ({ prompt }, { projectId }) => {

src/mcp/tools/project/get_project.ts renamed to src/mcp/tools/core/get_project.ts

@@ -1,9 +1,8 @@
-import { consult_firebase_assistant } from "./consult_firebase_assistant";
-import { get_firebase_directory } from "./get_firebase_directory";
-import { set_firebase_directory } from "./set_firebase_directory";
+import type { ServerTool } from "../../tool.js";
 
-export const coreTools = [
-  get_firebase_directory,
-  set_firebase_directory,
-  consult_firebase_assistant,
-];
+import { consult_assistant } from "./consult_assistant.js";
+import { get_project } from "./get_project.js";
+import { get_sdk_config } from "./get_sdk_config.js";
+import { list_apps } from "./list_apps.js";
+
+export const coreTools: ServerTool[] = [get_project, list_apps, get_sdk_config, consult_assistant];

src/mcp/tools/project/get_sdk_config.ts renamed to src/mcp/tools/core/get_sdk_config.ts

@@ -3,11 +3,11 @@ import { tool } from "../../tool.js";
 import { mcpError, toContent } from "../../util.js";
 import { detectProjectRoot } from "../../../detectProjectRoot.js";
 
-export const get_firebase_directory = tool(
+export const get_project_directory = tool(
   {
-    name: "get_firebase_directory",
+    name: "get_project_directory",
     description:
-      "Gets the current Firebase project directory. If this has been set using the `set_firebase_directory` tool it will return that, otherwise it will look for a PROJECT_ROOT environment variable or the current working directory of the running Firebase MCP server.",
+      "Gets the current Firebase project directory. If this has been set using the `set_project_directory` tool it will return that, otherwise it will look for a PROJECT_ROOT environment variable or the current working directory of the running Firebase MCP server.",
     inputSchema: z.object({}),
     annotations: {
       title: "Get Firebase Project Directory",
@@ -18,7 +18,7 @@ export const get_firebase_directory = tool(
     if (!detectProjectRoot({ cwd: host.projectRoot }))
       return Promise.resolve(
         mcpError(
-          `There is no detected 'firebase.json' in directory '${host.projectRoot}'. Please use the 'set_firebase_directory' tool to activate a Firebase project directory.`,
+          `There is no detected 'firebase.json' in directory '${host.projectRoot}'. Please use the 'set_project_directory' tool to activate a Firebase project directory.`,
         ),
       );
     return Promise.resolve(

src/mcp/tools/core/index.ts

@@ -0,0 +1,4 @@
+import { get_project_directory } from "./get_project_directory";
+import { set_project_directory } from "./set_project_directory";
+
+export const directoryTools = [get_project_directory, set_project_directory];

src/mcp/tools/project/list_apps.ts renamed to src/mcp/tools/core/list_apps.ts

@@ -4,9 +4,9 @@ import { mcpError, toContent } from "../../util.js";
 import { existsSync } from "fs";
 import { join } from "path";
 
-export const set_firebase_directory = tool(
+export const set_project_directory = tool(
   {
-    name: "set_firebase_directory",
+    name: "set_project_directory",
     description:
       "Sets the project directory for the Firebase MCP server to utilize for project detection and authentication. This should be a directory with a `firebase.json` file in it. This information is persisted between sessions.",
     inputSchema: z.object({

src/mcp/tools/core/get_firebase_directory.ts renamed to src/mcp/tools/directory/get_project_directory.ts

@@ -3,11 +3,28 @@ import { ServerFeature } from "../types.js";
 import { authTools } from "./auth/index.js";
 import { dataconnectTools } from "./dataconnect/index.js";
 import { firestoreTools } from "./firestore/index.js";
-import { projectTools } from "./project/index.js";
+import { directoryTools } from "./directory/index.js";
+import { coreTools } from "./core/index.js";
 import { storageTools } from "./storage/index.js";
 
-export const tools: Record<ServerFeature, ServerTool[]> = {
-  project: projectTools,
+/** availableTools returns the list of MCP tools available given the server flags */
+export function availableTools(fixedRoot: boolean, activeFeatures?: ServerFeature[]): ServerTool[] {
+  // Core tools are always present.
+  const toolDefs: ServerTool[] = addPrefixToToolName("firebase_", coreTools);
+  if (!fixedRoot) {
+    // Present if the root is not fixed.
+    toolDefs.push(...directoryTools);
+  }
+  if (!activeFeatures || !activeFeatures.length) {
+    activeFeatures = Object.keys(tools) as ServerFeature[];
+  }
+  for (const key of activeFeatures) {
+    toolDefs.push(...tools[key]);
+  }
+  return toolDefs;
+}
+
+const tools: Record<ServerFeature, ServerTool[]> = {
   firestore: addPrefixToToolName("firestore_", firestoreTools),
   auth: addPrefixToToolName("auth_", authTools),
   dataconnect: addPrefixToToolName("dataconnect_", dataconnectTools),

src/mcp/tools/directory/index.ts

@@ -1,2 +1,2 @@
-export const SERVER_FEATURES = ["project", "firestore", "storage", "dataconnect", "auth"] as const;
+export const SERVER_FEATURES = ["firestore", "storage", "dataconnect", "auth"] as const;
 export type ServerFeature = (typeof SERVER_FEATURES)[number];