@google-cloud/cloud-run-mcp 1.3.0 → 1.5.0

package/{tools.js → tools/register-tools.js}

@@ -3,7 +3,7 @@ Copyright 2025 Google LLC
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
-you may obtain a copy of the License at
+You may obtain a copy of the License at
 
     https://www.apache.org/licenses/LICENSE-2.0
 
@@ -15,18 +15,18 @@ limitations under the License.
 */
 
 import { z } from 'zod';
-import { deploy, deployImage } from './lib/cloud-run-deploy.js';
-import {
-  listServices,
-  getService,
-  getServiceLogs,
-} from './lib/cloud-run-services.js';
-import {
-  listProjects,
-  createProjectAndAttachBilling,
-} from './lib/gcp-projects.js';
-
-import { ResourceTemplate } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { listProjects, createProjectAndAttachBilling } from '../lib/gcp-projects.js';
+import { listServices, getService, getServiceLogs } from '../lib/cloud-run-services.js';
+import { deploy, deployImage } from '../lib/cloud-run-deploy.js';
+
+function createProgressCallback(sendNotification) {
+  return (progress) => {
+    sendNotification({
+      method: 'notifications/message',
+      params: { level: progress.level || 'info', data: progress.data },
+    });
+  };
+}
 
 function gcpTool(gcpCredentialsAvailable, fn) {
   if (!gcpCredentialsAvailable) {
@@ -42,24 +42,15 @@ function gcpTool(gcpCredentialsAvailable, fn) {
   return fn;
 }
 
-export const registerTools = (
-  server,
-  {
-    defaultProjectId,
-    defaultRegion,
-    defaultServiceName,
-    skipIamCheck,
-    gcpCredentialsAvailable,
-  } = {}
-) => {
-  // Tool to list GCP projects
+// Tool to list GCP projects
+function registerListProjectsTool(server, options) {
   server.registerTool(
     'list_projects',
     {
       description: 'Lists available GCP projects',
       inputSchema: {},
     },
-    gcpTool(gcpCredentialsAvailable, async () => {
+    gcpTool(options.gcpCredentialsAvailable, async () => {
       try {
         const projects = await listProjects();
         return {
@@ -82,8 +73,10 @@ export const registerTools = (
       }
     })
   );
+}
 
-  // Tool to create a new GCP project
+// Tool to create a new GCP project
+function registerCreateProjectTool(server, options) {
   server.registerTool(
     'create_project',
     {
@@ -98,7 +91,7 @@ export const registerTools = (
           ),
       },
     },
-    gcpTool(gcpCredentialsAvailable, async ({ projectId }) => {
+    gcpTool(options.gcpCredentialsAvailable, async ({ projectId }) => {
       if (
         projectId !== undefined &&
         (typeof projectId !== 'string' || projectId.trim() === '')
@@ -134,8 +127,10 @@ export const registerTools = (
       }
     })
   );
+}
 
-  // Listing Cloud Run services
+// Listing Cloud Run services
+function registerListServicesTool(server, options) {
   server.registerTool(
     'list_services',
     {
@@ -144,14 +139,14 @@ export const registerTools = (
         project: z
           .string()
           .describe('Google Cloud project ID')
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .describe('Region where the services are located')
-          .default(defaultRegion),
+          .default(options.defaultRegion),
       },
     },
-    gcpTool(gcpCredentialsAvailable, async ({ project, region }) => {
+    gcpTool(options.gcpCredentialsAvailable, async ({ project, region }) => {
       if (typeof project !== 'string') {
         return {
           content: [
@@ -191,8 +186,10 @@ export const registerTools = (
       }
     })
   );
+}
 
-  // Dynamic resource for getting a specific service
+// Dynamic resource for getting a specific service
+function registerGetServiceTool(server, options) {
   server.registerTool(
     'get_service',
     {
@@ -201,18 +198,18 @@ export const registerTools = (
         project: z
           .string()
           .describe('Google Cloud project ID containing the service')
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .describe('Region where the service is located')
-          .default(defaultRegion),
+          .default(options.defaultRegion),
         service: z
           .string()
           .describe('Name of the Cloud Run service')
-          .default(defaultServiceName),
+          .default(options.defaultServiceName),
       },
     },
-    gcpTool(gcpCredentialsAvailable, async ({ project, region, service }) => {
+    gcpTool(options.gcpCredentialsAvailable, async ({ project, region, service }) => {
       if (typeof project !== 'string') {
         return {
           content: [
@@ -260,8 +257,10 @@ export const registerTools = (
       }
     })
   );
+}
 
-  // Logs for a service
+// Get logs for a service
+function registerGetServiceLogTool(server, options) {
   server.registerTool(
     'get_service_log',
     {
@@ -271,18 +270,18 @@ export const registerTools = (
         project: z
           .string()
           .describe('Google Cloud project ID containing the service')
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .describe('Region where the service is located')
-          .default(defaultRegion),
+          .default(options.defaultRegion),
         service: z
           .string()
           .describe('Name of the Cloud Run service')
-          .default(defaultServiceName),
+          .default(options.defaultServiceName),
       },
     },
-    gcpTool(gcpCredentialsAvailable, async ({ project, region, service }) => {
+    gcpTool(options.gcpCredentialsAvailable, async ({ project, region, service }) => {
       let allLogs = [];
       let requestOptions;
       try {
@@ -323,83 +322,10 @@ export const registerTools = (
       }
     })
   );
+}
 
-  server.registerTool(
-    'deploy_local_files',
-    {
-      description:
-        'Deploy local files to Cloud Run. Takes an array of absolute file paths from the local filesystem that will be deployed. Use this tool if the files exists on the user local filesystem.',
-      inputSchema: {
-        project: z
-          .string()
-          .describe(
-            'Google Cloud project ID. Do not select it yourself, make sure the user provides or confirms the project ID.'
-          )
-          .default(defaultProjectId),
-        region: z
-          .string()
-          .optional()
-          .default(defaultRegion)
-          .describe('Region to deploy the service to'),
-        service: z
-          .string()
-          .optional()
-          .default(defaultServiceName)
-          .describe('Name of the Cloud Run service to deploy to'),
-        files: z
-          .array(z.string())
-          .describe(
-            'Array of absolute file paths to deploy (e.g. ["/home/user/project/src/index.js", "/home/user/project/package.json"])'
-          ),
-      },
-    },
-    gcpTool(
-      gcpCredentialsAvailable,
-      async ({ project, region, service, files }) => {
-        if (typeof project !== 'string') {
-          throw new Error(
-            'Project must specified, please prompt the user for a valid existing Google Cloud project ID.'
-          );
-        }
-        if (typeof files !== 'object' || !Array.isArray(files)) {
-          throw new Error('Files must specified');
-        }
-        if (files.length === 0) {
-          throw new Error('No files specified for deployment');
-        }
-
-        // Deploy to Cloud Run
-        try {
-          // TODO: Should we return intermediate progress messages? we'd need to use sendNotification for that, see https://github.com/modelcontextprotocol/typescript-sdk/blob/main/src/examples/server/jsonResponseStreamableHttp.ts#L46C24-L46C41
-          const response = await deploy({
-            projectId: project,
-            serviceName: service,
-            region: region,
-            files: files,
-            skipIamCheck: skipIamCheck, // Pass the new flag
-          });
-          return {
-            content: [
-              {
-                type: 'text',
-                text: `Cloud Run service ${service} deployed in project ${project}\nCloud Console: https://console.cloud.google.com/run/detail/${region}/${service}?project=${project}\nService URL: ${response.uri}`,
-              },
-            ],
-          };
-        } catch (error) {
-          return {
-            content: [
-              {
-                type: 'text',
-                text: `Error deploying to Cloud Run: ${error.message || error}`,
-              },
-            ],
-          };
-        }
-      }
-    )
-  );
-
+// Tool to deploy to Cloud Run from local folder
+function registerDeployLocalFolderTool(server, options) {
   server.registerTool(
     'deploy_local_folder',
     {
@@ -411,16 +337,16 @@ export const registerTools = (
           .describe(
             'Google Cloud project ID. Do not select it yourself, make sure the user provides or confirms the project ID.'
           )
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .optional()
-          .default(defaultRegion)
+          .default(options.defaultRegion)
           .describe('Region to deploy the service to'),
         service: z
           .string()
           .optional()
-          .default(defaultServiceName)
+          .default(options.defaultServiceName)
           .describe('Name of the Cloud Run service to deploy to'),
         folderPath: z
           .string()
@@ -430,8 +356,11 @@ export const registerTools = (
       },
     },
     gcpTool(
-      gcpCredentialsAvailable,
-      async ({ project, region, service, folderPath }) => {
+      options.gcpCredentialsAvailable,
+      async (
+        { project, region, service, folderPath },
+        { sendNotification }
+      ) => {
         if (typeof project !== 'string') {
           throw new Error(
             'Project must be specified, please prompt the user for a valid existing Google Cloud project ID.'
@@ -443,14 +372,20 @@ export const registerTools = (
           );
         }
 
+        const progressCallback = createProgressCallback(sendNotification);
+
         // Deploy to Cloud Run
         try {
+          await progressCallback({
+            data: `Starting deployment of local folder for service ${service} in project ${project}...`,
+          });
           const response = await deploy({
             projectId: project,
             serviceName: service,
             region: region,
             files: [folderPath],
-            skipIamCheck: skipIamCheck, // Pass the new flag
+            skipIamCheck: options.skipIamCheck, // Pass the new flag
+            progressCallback,
           });
           return {
             content: [
@@ -473,7 +408,10 @@ export const registerTools = (
       }
     )
   );
+}
 
+// Tool to deploy to Cloud Run from file contents
+function registerDeployFileContentsTool(server, options) {
   server.registerTool(
     'deploy_file_contents',
     {
@@ -485,16 +423,16 @@ export const registerTools = (
           .describe(
             'Google Cloud project ID. Leave unset for the app to be deployed in a new project. If provided, make sure the user confirms the project ID they want to deploy to.'
           )
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .optional()
-          .default(defaultRegion)
+          .default(options.defaultRegion)
           .describe('Region to deploy the service to'),
         service: z
           .string()
           .optional()
-          .default(defaultServiceName)
+          .default(options.defaultServiceName)
           .describe('Name of the Cloud Run service to deploy to'),
         files: z
           .array(
@@ -514,8 +452,8 @@ export const registerTools = (
       },
     },
     gcpTool(
-      gcpCredentialsAvailable,
-      async ({ project, region, service, files }) => {
+      options.gcpCredentialsAvailable,
+      async ({ project, region, service, files }, { sendNotification }) => {
         if (typeof project !== 'string') {
           throw new Error(
             'Project must specified, please prompt the user for a valid existing Google Cloud project ID.'
@@ -535,14 +473,20 @@ export const registerTools = (
           }
         }
 
+        const progressCallback = createProgressCallback(sendNotification);
+
         // Deploy to Cloud Run
         try {
+          await progressCallback({
+            data: `Starting deployment of file contents for service ${service} in project ${project}...`,
+          });
           const response = await deploy({
             projectId: project,
             serviceName: service,
             region: region,
             files: files,
-            skipIamCheck: skipIamCheck, // Pass the new flag
+            skipIamCheck: options.skipIamCheck, // Pass the new flag
+            progressCallback,
           });
           return {
             content: [
@@ -565,7 +509,10 @@ export const registerTools = (
       }
     )
   );
+}
 
+// Tool to deploy to Cloud Run from container image
+function registerDeployContainerImageTool(server, options) {
   server.registerTool(
     'deploy_container_image',
     {
@@ -577,16 +524,16 @@ export const registerTools = (
           .describe(
             'Google Cloud project ID. Do not select it yourself, make sure the user provides or confirms the project ID.'
           )
-          .default(defaultProjectId),
+          .default(options.defaultProjectId),
         region: z
           .string()
           .optional()
-          .default(defaultRegion)
+          .default(options.defaultRegion)
           .describe('Region to deploy the service to'),
         service: z
           .string()
           .optional()
-          .default(defaultServiceName)
+          .default(options.defaultServiceName)
           .describe('Name of the Cloud Run service to deploy to'),
         imageUrl: z
           .string()
@@ -596,8 +543,8 @@ export const registerTools = (
       },
     },
     gcpTool(
-      gcpCredentialsAvailable,
-      async ({ project, region, service, imageUrl }) => {
+      options.gcpCredentialsAvailable,
+      async ({ project, region, service, imageUrl }, { sendNotification }) => {
         if (typeof project !== 'string') {
           throw new Error(
             'Project must specified, please prompt the user for a valid existing Google Cloud project ID.'
@@ -609,6 +556,8 @@ export const registerTools = (
           );
         }
 
+        const progressCallback = createProgressCallback(sendNotification);
+
         // Deploy to Cloud Run
         try {
           const response = await deployImage({
@@ -616,7 +565,8 @@ export const registerTools = (
             serviceName: service,
             region: region,
             imageUrl: imageUrl,
-            skipIamCheck: skipIamCheck,
+            skipIamCheck: options.skipIamCheck,
+            progressCallback,
           });
           return {
             content: [
@@ -639,331 +589,15 @@ export const registerTools = (
       }
     )
   );
-};
-
-export const registerToolsRemote = async (
-  server,
-  {
-    defaultProjectId,
-    defaultRegion,
-    defaultServiceName,
-    skipIamCheck = false,
-    gcpCredentialsAvailable,
-  } = {}
-) => {
-  // We no longer call checkGCP here; the effective defaults are passed in.
-  const currentProject = defaultProjectId; // Use the passed effective project ID
-  const currentRegion = defaultRegion; // Use the passed effective region
-
-  if (!currentProject) {
-    throw new Error(
-      'Cannot register remote tools: GCP project ID could not be determined. Please ensure GOOGLE_CLOUD_PROJECT environment variable is set or the server is running on GCP.'
-    );
-  }
-
-  // Listing Cloud Run services (Remote)
-  server.registerTool(
-    'list_services',
-    {
-      description: `Lists Cloud Run services in GCP project ${currentProject} and a given region.`,
-      inputSchema: {
-        region: z
-          .string()
-          .describe('Region where the services are located')
-          .default(currentRegion),
-      },
-    },
-    gcpTool(gcpCredentialsAvailable, async ({ region }) => {
-      try {
-        const services = await listServices(currentProject, region);
-        const serviceList = services
-          .map((s) => {
-            const serviceName = s.name.split('/').pop();
-            return `- ${serviceName} (URL: ${s.uri})`;
-          })
-          .join('\n');
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Services in project ${currentProject} (location ${region}):\n${serviceList}`,
-            },
-          ],
-        };
-      } catch (error) {
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Error listing services for project ${currentProject} (region ${currentRegion}): ${error.message}`,
-            },
-          ],
-        };
-      }
-    })
-  );
-
-  // Dynamic resource for getting a specific service (Remote)
-  server.registerTool(
-    'get_service',
-    {
-      description: `Gets details for a specific Cloud Run service in GCP project ${currentProject}.`,
-      inputSchema: {
-        region: z
-          .string()
-          .describe('Region where the service is located')
-          .default(currentRegion),
-        service: z
-          .string()
-          .describe('Name of the Cloud Run service')
-          .default(defaultServiceName),
-      },
-    },
-    gcpTool(gcpCredentialsAvailable, async ({ region, service }) => {
-      if (typeof service !== 'string') {
-        return {
-          content: [
-            { type: 'text', text: 'Error: Service name must be provided.' },
-          ],
-        };
-      }
-      try {
-        const serviceDetails = await getService(
-          currentProject,
-          region,
-          service
-        );
-        if (serviceDetails) {
-          return {
-            content: [
-              {
-                type: 'text',
-                text: `Name: ${service}\nRegion: ${region}\nProject: ${currentProject}\nURL: ${serviceDetails.uri}\nLast deployed by: ${serviceDetails.lastModifier}`,
-              },
-            ],
-          };
-        } else {
-          return {
-            content: [
-              {
-                type: 'text',
-                text: `Service ${service} not found in project ${currentProject} (region ${currentRegion}).`,
-              },
-            ],
-          };
-        }
-      } catch (error) {
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Error getting service ${service} in project ${currentProject} (region ${currentRegion}): ${error.message}`,
-            },
-          ],
-        };
-      }
-    })
-  );
-
-  // Logs for a service
-  server.registerTool(
-    'get_service_log',
-    {
-      description:
-        'Gets Logs and Error Messages for a specific Cloud Run service.',
-      inputSchema: {
-        project: z
-          .string()
-          .describe('Google Cloud project ID containing the service')
-          .default(currentProject), // Use currentProject
-        region: z
-          .string()
-          .describe('Region where the service is located')
-          .default(currentRegion), // Use currentRegion
-        service: z
-          .string()
-          .describe('Name of the Cloud Run service')
-          .default(defaultServiceName),
-      },
-    },
-    gcpTool(gcpCredentialsAvailable, async ({ project, region, service }) => {
-      let allLogs = [];
-      let requestOptions;
-      try {
-        do {
-          // Fetch a page of logs
-          const response = await getServiceLogs(
-            project,
-            region,
-            service,
-            requestOptions
-          );
-
-          if (response.logs) {
-            allLogs.push(response.logs);
-          }
-
-          // Set the requestOptions incl pagintion token for the next iteration
-
-          requestOptions = response.requestOptions;
-        } while (requestOptions); // Continue as long as there is a next page token
-        return {
-          content: [
-            {
-              type: 'text',
-              text: allLogs.join('\n'),
-            },
-          ],
-        };
-      } catch (error) {
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Error getting Logs for service ${service} in project ${project} (region ${region}): ${error.message}`,
-            },
-          ],
-        };
-      }
-    })
-  );
-
-  // Deploy file contents to Cloud Run (Remote)
-  server.registerTool(
-    'deploy_file_contents',
-    {
-      description: `Deploy files to Cloud Run by providing their contents directly to the GCP project ${currentProject}.`,
-      inputSchema: {
-        region: z
-          .string()
-          .optional()
-          .default(currentRegion)
-          .describe('Region to deploy the service to'),
-        service: z
-          .string()
-          .optional()
-          .default(defaultServiceName)
-          .describe('Name of the Cloud Run service to deploy to'), // Use defaultServiceName
-        files: z
-          .array(
-            z.object({
-              filename: z
-                .string()
-                .describe(
-                  'Name and path of the file (e.g. "src/index.js" or "data/config.json")'
-                ),
-              content: z.string().describe('Text content of the file'),
-            })
-          )
-          .describe('Array of file objects containing filename and content'),
-      },
-    },
-    gcpTool(gcpCredentialsAvailable, async ({ region, service, files }) => {
-      console.log(
-        `New deploy request (remote): ${JSON.stringify({ project: currentProject, region, service, files })}`
-      );
-
-      if (
-        typeof files !== 'object' ||
-        !Array.isArray(files) ||
-        files.length === 0
-      ) {
-        throw new Error('Files must be specified');
-      }
-
-      // Validate that each file has content
-      for (const file of files) {
-        if (!file.content) {
-          throw new Error(`File ${file.filename} must have content`);
-        }
-      }
-
-      // Deploy to Cloud Run
-      try {
-        const response = await deploy({
-          projectId: currentProject,
-          serviceName: service,
-          region: region,
-          files: files,
-          skipIamCheck: skipIamCheck, // Pass the new flag
-        });
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Cloud Run service ${service} deployed in project ${currentProject}\nCloud Console: https://console.cloud. google.com/run/detail/${region}/${service}?project=${currentProject}\nService URL: ${response.uri}`,
-            },
-          ],
-        };
-      } catch (error) {
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Error deploying to Cloud Run: ${error.message || error}`,
-            },
-          ],
-        };
-      }
-    })
-  );
-
-  server.registerTool(
-    'deploy_container_image',
-    {
-      description: `Deploys a container image to Cloud Run in the GCP project ${currentProject}. Use this tool if the user provides a container image URL.`,
-      inputSchema: {
-        region: z
-          .string()
-          .optional()
-          .default(currentRegion)
-          .describe('Region to deploy the service to'),
-        service: z
-          .string()
-          .optional()
-          .default(defaultServiceName)
-          .describe('Name of the Cloud Run service to deploy to'),
-        imageUrl: z
-          .string()
-          .describe(
-            'The URL of the container image to deploy (e.g. "gcr.io/cloudrun/hello")'
-          ),
-      },
-    },
-    gcpTool(gcpCredentialsAvailable, async ({ region, service, imageUrl }) => {
-      if (typeof imageUrl !== 'string' || imageUrl.trim() === '') {
-        throw new Error(
-          'Container image URL must be specified and be a non-empty string.'
-        );
-      }
+}
 
-      // Deploy to Cloud Run
-      try {
-        const response = await deployImage({
-          projectId: currentProject,
-          serviceName: service,
-          region: region,
-          imageUrl: imageUrl,
-          skipIamCheck: skipIamCheck,
-        });
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Cloud Run service ${service} deployed in project ${currentProject}\nCloud Console: https://console.cloud. google.com/run/detail/${region}/${service}?project=${currentProject}\nService URL: ${response.uri}`,
-            },
-          ],
-        };
-      } catch (error) {
-        return {
-          content: [
-            {
-              type: 'text',
-              text: `Error deploying to Cloud Run: ${error.message || error}`,
-            },
-          ],
-        };
-      }
-    })
-  );
+export {
+  registerListProjectsTool,
+  registerCreateProjectTool,
+  registerListServicesTool,
+  registerGetServiceTool,
+  registerGetServiceLogTool,
+  registerDeployLocalFolderTool,
+  registerDeployFileContentsTool,
+  registerDeployContainerImageTool,
 };