feat: storage tools

Author: Ngineer101

packages/mcp-server-supabase/src/platform/api-platform.ts

@@ -30,6 +30,7 @@ import {
   type GetLogsOptions,
   type ResetBranchOptions,
   type SupabasePlatform,
+  type StorageConfigUpdate,
 } from './index.js';
 
 const { version } = packageJson;
@@ -600,6 +601,69 @@ export function createSupabaseApiPlatform(
 
       assertSuccess(response, 'Failed to rebase branch');
     },
+
+    // Storage methods
+    async listAllBuckets(project_id: string) {
+      const response = await managementApiClient.GET(
+        '/v1/projects/{ref}/storage/buckets',
+        {
+          params: {
+            path: {
+              ref: project_id,
+            },
+          },
+        }
+      );
+
+      assertSuccess(response, 'Failed to list storage buckets');
+
+      return response.data;
+    },
+
+    async getStorageConfig(project_id: string) {
+      const response = await managementApiClient.GET(
+        '/v1/projects/{ref}/config/storage',
+        {
+          params: {
+            path: {
+              ref: project_id,
+            },
+          },
+        }
+      );
+
+      assertSuccess(response, 'Failed to get storage config');
+
+      return response.data;
+    },
+
+    async updateStorageConfig(projectId: string, config: StorageConfigUpdate) {
+      const response = await managementApiClient.PATCH(
+        '/v1/projects/{ref}/config/storage',
+        {
+          params: {
+            path: {
+              ref: projectId,
+            },
+          },
+          body: {
+            fileSizeLimit: config.fileSizeLimit,
+            features: {
+              imageTransformation: {
+                enabled: config.features.imageTransformation.enabled,
+              },
+              s3Protocol: {
+                enabled: config.features.s3Protocol.enabled,
+              },
+            },
+          },
+        }
+      );
+
+      assertSuccess(response, 'Failed to update storage config');
+
+      return response.data;
+    },
   };
 
   return platform;

packages/mcp-server-supabase/src/platform/types.ts

@@ -2,6 +2,23 @@ import { z } from 'zod';
 import { AWS_REGION_CODES } from '../regions.js';
 import type { InitData } from '@supabase/mcp-utils';
 
+export const v1StorageBucketResponseSchema = z.object({
+  id: z.string(),
+  name: z.string(),
+  owner: z.string(),
+  created_at: z.string(),
+  updated_at: z.string(),
+  public: z.boolean(),
+});
+
+export const storageConfigResponseSchema = z.object({
+  fileSizeLimit: z.number(),
+  features: z.object({
+    imageTransformation: z.object({ enabled: z.boolean() }),
+    s3Protocol: z.object({ enabled: z.boolean() }),
+  }),
+});
+
 export const organizationSchema = z.object({
   id: z.string(),
   name: z.string(),
@@ -135,6 +152,10 @@ export type GenerateTypescriptTypesResult = z.infer<
   typeof generateTypescriptTypesResultSchema
 >;
 
+export type StorageConfigResponse = z.infer<typeof storageConfigResponseSchema>;
+export type StorageConfigUpdate = z.infer<typeof storageConfigResponseSchema>;
+export type StorageBucketResponse = z.infer<typeof v1StorageBucketResponseSchema>;
+
 export type SupabasePlatform = {
   init?(info: InitData): Promise<void>;
 
@@ -186,4 +207,9 @@ export type SupabasePlatform = {
   mergeBranch(branchId: string): Promise<void>;
   resetBranch(branchId: string, options: ResetBranchOptions): Promise<void>;
   rebaseBranch(branchId: string): Promise<void>;
+
+  // Storage
+  getStorageConfig(projectId: string): Promise<StorageConfigResponse>;
+  updateStorageConfig(projectId: string, config: StorageConfigUpdate): Promise<void>;
+  listAllBuckets(projectId: string): Promise<StorageBucketResponse[]>;
 };

packages/mcp-server-supabase/src/server.ts

@@ -7,6 +7,7 @@ import { getDebuggingTools } from './tools/debugging-tools.js';
 import { getDevelopmentTools } from './tools/development-tools.js';
 import { getEdgeFunctionTools } from './tools/edge-function-tools.js';
 import { getProjectManagementTools } from './tools/project-management-tools.js';
+import { getStorageTools } from './tools/storage-tools.js';
 
 const { version } = packageJson;
 
@@ -57,7 +58,8 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
         getEdgeFunctionTools({ platform, projectId }),
         getDebuggingTools({ platform, projectId }),
         getDevelopmentTools({ platform, projectId }),
-        getBranchingTools({ platform, projectId })
+        getBranchingTools({ platform, projectId }),
+        getStorageTools({ platform, projectId })
       );
 
       return tools;

packages/mcp-server-supabase/src/tools/storage-tools.ts

@@ -0,0 +1,55 @@
+import { z } from 'zod';
+import type { SupabasePlatform } from '../platform/types.js';
+import { injectableTool } from './util.js';
+
+export type StorageToolsOptions = {
+  platform: SupabasePlatform;
+  projectId?: string;
+};
+
+export function getStorageTools({
+  platform,
+  projectId,
+}: StorageToolsOptions) {
+  const project_id = projectId;
+
+  return {
+    list_storage_buckets: injectableTool({
+      description: 'Lists all storage buckets in a Supabase project.',
+      parameters: z.object({
+        project_id: z.string(),
+      }),
+      inject: { project_id },
+      execute: async ({ project_id }) => {
+        return await platform.listAllBuckets(project_id);
+      },
+    }),
+    get_storage_config: injectableTool({
+      description: 'Get the storage config for a Supabase project.',
+      parameters: z.object({
+        project_id: z.string(),
+      }),
+      inject: { project_id },
+      execute: async ({ project_id }) => {
+        return await platform.getStorageConfig(project_id);
+      },
+    }),
+    update_storage_config: injectableTool({
+      description: 'Update the storage config for a Supabase project.',
+      parameters: z.object({
+        project_id: z.string(),
+        config: z.object({
+          fileSizeLimit: z.number(),
+          features: z.object({
+            imageTransformation: z.object({ enabled: z.boolean() }),
+            s3Protocol: z.object({ enabled: z.boolean() }),
+          }),
+        })
+      }),
+      inject: { project_id },
+      execute: async ({ project_id, config }) => {
+        return await platform.updateStorageConfig(project_id, config);
+      },
+    }),
+  };
+}