initial apphosting mcp tool

Author: bkendall

src/gcp/apphosting.ts

@@ -333,6 +333,19 @@ export async function getBackend(
   return res.body;
 }
 
+/**
+ * Gets traffic details.
+ */
+export async function getTraffic(
+  projectId: string,
+  location: string,
+  backendId: string,
+): Promise<Traffic> {
+  const name = `projects/${projectId}/locations/${location}/backends/${backendId}/traffic`;
+  const res = await client.get<Traffic>(name);
+  return res.body;
+}
+
 /**
  * List all backends present in a project and location.
  */

src/mcp/tools/apphosting/index.ts

@@ -0,0 +1,4 @@
+import { ServerTool } from "../../tool";
+import { list_backends } from "./list_backends";
+
+export const appHostingTools: ServerTool[] = [list_backends];

src/mcp/tools/apphosting/list_backends.ts

@@ -0,0 +1,46 @@
+import { z } from "zod";
+import { tool } from "../../tool.js";
+import { toContent } from "../../util.js";
+import { NO_PROJECT_ERROR } from "../../errors.js";
+import {
+  Backend,
+  getTraffic,
+  listBackends,
+  parseBackendName,
+  Traffic,
+} from "../../../gcp/apphosting.js";
+
+export const list_backends = tool(
+  {
+    name: "list_backends",
+    description:
+      "Retrieves a list of App Hosting backends in the current project. The `uri` is the public URL of the backend.",
+    inputSchema: z.object({
+      location: z
+        .string()
+        .optional()
+        .default("-")
+        .describe("Limit the listed backends to this region."),
+    }),
+    annotations: {
+      title: "List App Hosting backends.",
+      readOnlyHint: true,
+    },
+    _meta: {
+      requiresAuth: true,
+      requiresProject: true,
+    },
+  },
+  async ({ location } = {}, { projectId }) => {
+    if (!projectId) return NO_PROJECT_ERROR;
+    if (!location) location = "-";
+    const data: (Backend & { traffic: Traffic })[] = [];
+    const backends = await listBackends(projectId, location);
+    for (const backend of backends.backends) {
+      const { location, id } = parseBackendName(backend.name);
+      const traffic = await getTraffic(projectId, location, id);
+      data.push({ ...backend, traffic: traffic });
+    }
+    return toContent(data);
+  },
+);

src/mcp/tools/index.ts

@@ -7,6 +7,7 @@ import { coreTools } from "./core/index.js";
 import { storageTools } from "./storage/index.js";
 import { messagingTools } from "./messaging/index.js";
 import { remoteConfigTools } from "./remoteconfig/index.js";
+import { appHostingTools } from "./apphosting/index.js";
 
 /** availableTools returns the list of MCP tools available given the server flags */
 export function availableTools(activeFeatures?: ServerFeature[]): ServerTool[] {
@@ -28,6 +29,7 @@ const tools: Record<ServerFeature, ServerTool[]> = {
   storage: addFeaturePrefix("storage", storageTools),
   messaging: addFeaturePrefix("messaging", messagingTools),
   remoteconfig: addFeaturePrefix("remoteconfig", remoteConfigTools),
+  apphosting: addFeaturePrefix("apphosting", appHostingTools),
 };
 
 function addFeaturePrefix(feature: string, tools: ServerTool[]): ServerTool[] {

src/mcp/types.ts

@@ -5,5 +5,6 @@ export const SERVER_FEATURES = [
   "auth",
   "messaging",
   "remoteconfig",
+  "apphosting",
 ] as const;
 export type ServerFeature = (typeof SERVER_FEATURES)[number];

src/mcp/util.ts

@@ -4,6 +4,7 @@ import { dump } from "js-yaml";
 import { platform } from "os";
 import { ServerFeature } from "./types";
 import {
+  apphostingOrigin,
   authManagementOrigin,
   dataconnectOrigin,
   firestoreOrigin,
@@ -85,6 +86,7 @@ const SERVER_FEATURE_APIS: Record<ServerFeature, string> = {
   auth: authManagementOrigin(),
   messaging: messagingApiOrigin(),
   remoteconfig: remoteConfigApiOrigin(),
+  apphosting: apphostingOrigin(),
 };
 
 /**