add tools based on features

Author: Ngineer101

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

@@ -48,8 +48,31 @@ export type SupabaseMcpServerOptions = {
    * Executes database queries in read-only mode if true.
    */
   readOnly?: boolean;
+
+  /**
+   * Features to enable.
+   * Options: 'account', 'branching', 'database', 'debug', 'development', 'docs', 'functions', 'storage'
+   */
+  features?: string[];
 };
 
+export type FeatureGroup = 'account' | 'branching' | 'database' | 'debug' | 'development' | 'docs' | 'functions' | 'storage';
+
+// Single source of truth for valid feature values
+export const VALID_FEATURES: readonly FeatureGroup[] = [
+  'account',
+  'branching',
+  'database',
+  'debug',
+  'development',
+  'docs',
+  'functions',
+  'storage'
+] as const;
+
+const DEFAULT_ACCOUNT_FEATURES: FeatureGroup[] = ['account', 'database', 'debug', 'docs', 'functions'];
+const DEFAULT_PROJECT_FEATURES: FeatureGroup[] = ['database', 'debug', 'docs', 'functions'];
+
 /**
  * Creates an MCP server for interacting with Supabase.
  */
@@ -58,11 +81,27 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
     platform,
     projectId,
     readOnly,
+    features,
     contentApiUrl = 'https://supabase.com/docs/api/graphql',
   } = options;
 
   const contentApiClientPromise = createContentApiClient(contentApiUrl);
 
+  const enabledFeatures = new Set<FeatureGroup>();
+
+  if (features && features.length > 0) {
+    // Use explicitly provided features
+    features.forEach(feature => {
+      if (VALID_FEATURES.includes(feature as FeatureGroup)) {
+        enabledFeatures.add(feature as FeatureGroup);
+      }
+    });
+  } else {
+    // Use defaults based on mode
+    const defaultFeatures = projectId ? DEFAULT_PROJECT_FEATURES : DEFAULT_ACCOUNT_FEATURES;
+    defaultFeatures.forEach(feature => enabledFeatures.add(feature));
+  }
+
   const server = createMcpServer({
     name: 'supabase',
     version,
@@ -75,21 +114,38 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
       const contentApiClient = await contentApiClientPromise;
       const tools: Record<string, Tool> = {};
 
-      // Add account-level tools only if projectId is not provided
-      if (!projectId) {
-        Object.assign(tools, getProjectManagementTools({ platform }));
+      // Add feature-based tools
+      if (enabledFeatures.has('account') && !projectId) {
+        Object.assign(tools, getAccountTools({ platform }));
+      }
+
+      if (enabledFeatures.has('branching')) {
+        Object.assign(tools, getBranchingTools({ platform, projectId }));
       }
 
-      // Add project-level tools
-      Object.assign(
-        tools,
-        getDatabaseOperationTools({ platform, projectId, readOnly }),
-        getEdgeFunctionTools({ platform, projectId }),
-        getDebuggingTools({ platform, projectId }),
-        getDevelopmentTools({ platform, projectId }),
-        getBranchingTools({ platform, projectId }),
-        getDocsTools({ contentApiClient })
-      );
+      if (enabledFeatures.has('database')) {
+        Object.assign(tools, getDatabaseOperationTools({ platform, projectId, readOnly }));
+      }
+
+      if (enabledFeatures.has('debug')) {
+        Object.assign(tools, getDebuggingTools({ platform, projectId }));
+      }
+
+      if (enabledFeatures.has('development')) {
+        Object.assign(tools, getDevelopmentTools({ platform, projectId }));
+      }
+
+      if (enabledFeatures.has('docs')) {
+        Object.assign(tools, getDocsTools({ contentApiClient }));
+      }
+
+      if (enabledFeatures.has('functions')) {
+        Object.assign(tools, getEdgeFunctionTools({ platform, projectId }));
+      }
+
+      if (enabledFeatures.has('storage')) {
+        Object.assign(tools, getStorageTools({ platform, projectId }));
+      }
 
       return tools;
     },

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

@@ -16,6 +16,7 @@ async function main() {
       ['read-only']: readOnly,
       ['api-url']: apiUrl,
       ['version']: showVersion,
+      ['features']: features,
     },
   } = parseArgs({
     options: {
@@ -35,6 +36,9 @@ async function main() {
       ['version']: {
         type: 'boolean',
       },
+      ['features']: {
+        type: 'string',
+      },
     },
   });
 
@@ -46,6 +50,8 @@ async function main() {
   // Use access token from CLI argument or environment variable
   const accessToken = cliAccessToken ?? process.env.SUPABASE_ACCESS_TOKEN;
 
+  const featureList = features?.split(',') ?? [];
+
   if (!accessToken) {
     console.error(
       'Please provide a personal access token (PAT) with the --access-token flag or set the SUPABASE_ACCESS_TOKEN environment variable'
@@ -62,6 +68,7 @@ async function main() {
     platform,
     projectId,
     readOnly,
+    features: featureList,
   });
 
   const transport = new StdioServerTransport();