firebase-tools 14.15.0 → 14.15.2

package/lib/mcp/tools/dataconnect/compile.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.compile = void 0;
+const zod_1 = require("zod");
+const tool_1 = require("../../tool");
+const load_1 = require("../../../dataconnect/load");
+const compile_1 = require("../../util/dataconnect/compile");
+exports.compile = (0, tool_1.tool)({
+    name: "build",
+    description: "Use this to compile Firebase Data Connect schema, operations, and/or connectors and check for build errors.",
+    inputSchema: zod_1.z.object({
+        error_filter: zod_1.z
+            .enum(["all", "schema", "operations"])
+            .describe("filter errors to a specific type only. defaults to `all` if omitted.")
+            .optional(),
+        service_id: zod_1.z
+            .string()
+            .optional()
+            .describe("The Firebase Data Connect service ID to look for. If omitted, builds all services defined in `firebase.json`."),
+    }),
+    annotations: {
+        title: "Compile Data Connect",
+        readOnlyHint: true,
+    },
+    _meta: {
+        requiresProject: false,
+        requiresAuth: false,
+    },
+}, async ({ service_id, error_filter }, { projectId, config }) => {
+    const serviceInfo = await (0, load_1.pickService)(projectId, config, service_id || undefined);
+    const errors = await (0, compile_1.compileErrors)(serviceInfo.sourceDirectory, error_filter);
+    if (errors)
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `The following errors were encountered while compiling Data Connect from directory \`${serviceInfo.sourceDirectory}\`:\n\n${errors}`,
+                },
+            ],
+            isError: true,
+        };
+    return { content: [{ type: "text", text: "Compiled successfully." }] };
+});

package/lib/mcp/tools/dataconnect/execute.js

@@ -0,0 +1,71 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.execute = void 0;
+const zod_1 = require("zod");
+const tool_1 = require("../../tool");
+const dataplane = require("../../../dataconnect/dataplaneClient");
+const load_1 = require("../../../dataconnect/load");
+const converter_1 = require("../../util/dataconnect/converter");
+const emulator_1 = require("../../util/dataconnect/emulator");
+exports.execute = (0, tool_1.tool)({
+    name: "execute",
+    description: "Executes a GraphQL operation against a Data Connect service or its emulator.",
+    inputSchema: zod_1.z.object({
+        query: zod_1.z.string().describe(`A Firebase Data Connect GraphQL query or mutation to execute.
+You can use the \`dataconnect_generate_operation\` tool to generate a query.
+Example Data Connect schema and example queries can be found in files ending in \`.graphql\` or \`.gql\`.
+`),
+        service_id: zod_1.z.string().optional()
+            .describe(`Data Connect Service ID to dis-ambulate if there are multiple.
+It's only necessary if there are multiple dataconnect sources in \`firebase.json\`.
+You can find candidate service_id in \`dataconnect.yaml\`
+`),
+        variables_json: zod_1.z
+            .string()
+            .optional()
+            .describe("GraphQL variables to pass into the query. MUST be a valid stringified JSON object."),
+        auth_token_json: zod_1.z
+            .string()
+            .optional()
+            .describe("Firebase Auth Token JWT to use in this query. MUST be a valid stringified JSON object." +
+            'Importantly, when executing queries with `@auth(level: USER)` or `auth.uid`, a valid Firebase Auth Token JWT with "sub" field is required. ' +
+            '"auth.uid" expression in the query evaluates to the value of "sub" field in Firebase Auth token.'),
+        use_emulator: zod_1.z
+            .boolean()
+            .default(false)
+            .describe("If true, target the DataConnect emulator. Run `firebase emulators:start` to start it"),
+    }),
+    annotations: {
+        title: "Execute Firebase Data Connect Query",
+    },
+    _meta: {
+        requiresProject: true,
+        requiresAuth: true,
+    },
+}, async ({ query, service_id, variables_json: unparsedVariables, use_emulator, auth_token_json: unparsedAuthToken, }, { projectId, config, host }) => {
+    const serviceInfo = await (0, load_1.pickService)(projectId, config, service_id || undefined);
+    let apiClient;
+    if (use_emulator) {
+        apiClient = await (0, emulator_1.getDataConnectEmulatorClient)(host);
+    }
+    else {
+        apiClient = dataplane.dataconnectDataplaneClient();
+    }
+    let executeGraphQL = dataplane.executeGraphQL;
+    if (query.startsWith("query")) {
+        executeGraphQL = dataplane.executeGraphQLRead;
+    }
+    const response = await executeGraphQL(apiClient, serviceInfo.serviceName, {
+        name: "",
+        query,
+        variables: (0, converter_1.parseVariables)(unparsedVariables),
+        extensions: {
+            impersonate: unparsedAuthToken
+                ? {
+                    authClaims: (0, converter_1.parseVariables)(unparsedAuthToken),
+                }
+                : undefined,
+        },
+    });
+    return (0, converter_1.graphqlResponseToToolResponse)(response.body);
+});

package/lib/mcp/tools/dataconnect/index.js

@@ -3,21 +3,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.dataconnectTools = void 0;
 const generate_operation_1 = require("./generate_operation");
 const generate_schema_1 = require("./generate_schema");
-const list_services_1 = require("./list_services");
-const get_schema_1 = require("./get_schema");
-const get_connector_1 = require("./get_connector");
-const execute_graphql_1 = require("./execute_graphql");
-const execute_graphql_read_1 = require("./execute_graphql_read");
-const execute_query_1 = require("./execute_query");
-const execute_mutation_1 = require("./execute_mutation");
+const info_1 = require("./info");
+const compile_1 = require("./compile");
+const execute_1 = require("./execute");
 exports.dataconnectTools = [
-    list_services_1.list_services,
+    compile_1.compile,
     generate_schema_1.generate_schema,
     generate_operation_1.generate_operation,
-    get_schema_1.get_schema,
-    get_connector_1.get_connectors,
-    execute_graphql_1.execute_graphql,
-    execute_graphql_read_1.execute_graphql_read,
-    execute_mutation_1.execute_mutation,
-    execute_query_1.execute_query,
+    info_1.info,
+    execute_1.execute,
 ];

package/lib/mcp/tools/dataconnect/info.js

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.info = void 0;
+const zod_1 = require("zod");
+const path = require("path");
+const tool_1 = require("../../tool");
+const util_1 = require("../../util");
+const client = require("../../../dataconnect/client");
+const load_1 = require("../../../dataconnect/load");
+const js_yaml_1 = require("js-yaml");
+const logger_1 = require("../../../logger");
+exports.info = (0, tool_1.tool)({
+    name: "info",
+    description: "Get information about the Firebase Data Connect local and deployed resources.",
+    inputSchema: zod_1.z.object({}),
+    annotations: {
+        title: "Get information about Firebase Data Connect",
+        readOnlyHint: true,
+    },
+    _meta: {
+        requiresProject: false,
+        requiresAuth: false,
+    },
+}, async (_, { projectId, config }) => {
+    const localServiceInfos = await (0, load_1.loadAll)(projectId, config);
+    const serviceInfos = new Map();
+    for (const l of localServiceInfos) {
+        serviceInfos.set(`locations/${l.dataConnectYaml.location}/services/${l.dataConnectYaml.serviceId}`, { local: l });
+    }
+    if (projectId) {
+        try {
+            const [services, schemas, connectors] = await Promise.all([
+                client.listAllServices(projectId),
+                client.listSchemas(`projects/${projectId}/locations/-/services/-`),
+                client.listConnectors(`projects/${projectId}/locations/-/services/-`),
+            ]);
+            console.log(services, schemas, connectors);
+            for (const s of services) {
+                const k = s.name.split("/").slice(2, 6).join("/");
+                const st = serviceInfos.get(k) || {};
+                st.deployed = st.deployed || {};
+                st.deployed.service = s;
+                serviceInfos.set(k, st);
+            }
+            for (const s of schemas) {
+                const k = s.name.split("/").slice(2, 6).join("/");
+                const st = serviceInfos.get(k) || {};
+                st.deployed = st.deployed || {};
+                st.deployed.schemas = st.deployed.schemas || [];
+                st.deployed.schemas.push(s);
+                serviceInfos.set(k, st);
+            }
+            for (const s of connectors) {
+                const k = s.name.split("/").slice(2, 6).join("/");
+                const st = serviceInfos.get(k) || {};
+                st.deployed = st.deployed || {};
+                st.deployed.connectors = st.deployed.connectors || [];
+                st.deployed.connectors.push(s);
+                serviceInfos.set(k, st);
+            }
+        }
+        catch (e) {
+            logger_1.logger.debug("cannot fetch dataconnect resources in the backend", e);
+        }
+    }
+    const localServices = Array.from(serviceInfos.values()).filter((s) => s.local);
+    const remoteOnlyServices = Array.from(serviceInfos.values()).filter((s) => !s.local);
+    const output = [];
+    function includeDeployedServiceInfo(deployed) {
+        var _a, _b;
+        if ((_a = deployed.schemas) === null || _a === void 0 ? void 0 : _a.length) {
+            output.push(`### Schemas`);
+            for (const s of deployed.schemas) {
+                clearCCFEFields(s);
+                output.push((0, js_yaml_1.dump)(s));
+            }
+        }
+        if ((_b = deployed.connectors) === null || _b === void 0 ? void 0 : _b.length) {
+            output.push(`### Connectors`);
+            for (const c of deployed.connectors) {
+                clearCCFEFields(c);
+                output.push((0, js_yaml_1.dump)(c));
+            }
+        }
+    }
+    if (localServices.length) {
+        output.push(`# Local Data Connect Sources`);
+        for (const s of localServices) {
+            const local = s.local;
+            output.push((0, js_yaml_1.dump)(local.dataConnectYaml));
+            const schemaDir = path.join(local.sourceDirectory, local.dataConnectYaml.schema.source);
+            output.push(`You can find all of schema sources under ${schemaDir}/`);
+            if (s.deployed) {
+                output.push(`It's already deployed in the backend:\n`);
+                includeDeployedServiceInfo(s.deployed);
+            }
+        }
+    }
+    if (remoteOnlyServices.length) {
+        output.push(`# Data Connect Services in project ${projectId}`);
+        for (const s of remoteOnlyServices) {
+            if (s.deployed) {
+                includeDeployedServiceInfo(s.deployed);
+            }
+        }
+    }
+    output.push(`\n# What's next?`);
+    if (!localServices.length) {
+        output.push(`- There is no local Data Connect service in the local workspace. Consider use the \`firebase_init\` MCP tool to setup one.`);
+    }
+    output.push(`- You can use the \`dataconnect_compile\` tool to compile all local Data Connect schemas and query sources.`);
+    output.push(`- You run \`firebase deploy\` in command line to deploy the Data Connect schemas, connector and perform SQL migrations.`);
+    return (0, util_1.toContent)(output.join("\n"));
+});
+function clearCCFEFields(r) {
+    const fieldsToClear = ["updateTime", "uid", "etag"];
+    for (const k of fieldsToClear) {
+        delete r[k];
+    }
+}

package/lib/mcp/tools/firestore/list_collections.js

@@ -5,7 +5,6 @@ const zod_1 = require("zod");
 const tool_1 = require("../../tool");
 const util_1 = require("../../util");
 const firestore_1 = require("../../../gcp/firestore");
-const errors_1 = require("../../errors");
 const types_1 = require("../../../emulator/types");
 exports.list_collections = (0, tool_1.tool)({
     name: "list_collections",
@@ -30,7 +29,5 @@ exports.list_collections = (0, tool_1.tool)({
     if (use_emulator) {
         emulatorUrl = await host.getEmulatorUrl(types_1.Emulators.FIRESTORE);
     }
-    if (!projectId)
-        return errors_1.NO_PROJECT_ERROR;
     return (0, util_1.toContent)(await (0, firestore_1.listCollectionIds)(projectId, database, emulatorUrl));
 });

package/lib/mcp/tools/index.js

@@ -16,6 +16,9 @@ function availableTools(activeFeatures) {
     if (!(activeFeatures === null || activeFeatures === void 0 ? void 0 : activeFeatures.length)) {
         activeFeatures = Object.keys(tools);
     }
+    if (!activeFeatures.includes("core")) {
+        activeFeatures.unshift("core");
+    }
     for (const key of activeFeatures) {
         toolDefs.push(...tools[key]);
     }

package/lib/mcp/util/dataconnect/compile.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.compileErrors = void 0;
+const graphqlError_1 = require("../../../dataconnect/graphqlError");
+const dataconnectEmulator_1 = require("../../../emulator/dataconnectEmulator");
+async function compileErrors(configDir, errorFilter) {
+    const errors = (await dataconnectEmulator_1.DataConnectEmulator.build({ configDir })).errors;
+    return ((errors === null || errors === void 0 ? void 0 : errors.filter((e) => {
+        var _a;
+        const isOperationError = ["query", "mutation"].includes((_a = e.path) === null || _a === void 0 ? void 0 : _a[0]);
+        if (errorFilter === "operations")
+            return isOperationError;
+        if (errorFilter === "schema")
+            return !isOperationError;
+        return true;
+    }).map(graphqlError_1.prettify).join("\n")) || "");
+}
+exports.compileErrors = compileErrors;