feat: support branching workflow via mcp

Author: sweatybridge

packages/mcp-server-supabase/src/management-api/types.ts

@@ -1346,6 +1346,9 @@ export interface components {
             /** @enum {string} */
             message: "ok";
         };
+        BranchActionBody: {
+            migration_version?: string;
+        };
         BranchUpdateResponse: {
             workflow_run_id: string;
             /** @enum {string} */
@@ -2364,6 +2367,10 @@ export interface components {
             version: string;
             name?: string;
         }[];
+        V1CreateMigrationBody: {
+            query: string;
+            name?: string;
+        };
         V1RunQueryBody: {
             query: string;
         };
@@ -2705,7 +2712,11 @@ export interface operations {
             };
             cookie?: never;
         };
-        requestBody?: never;
+        requestBody: {
+            content: {
+                "application/json": components["schemas"]["BranchActionBody"];
+            };
+        };
         responses: {
             201: {
                 headers: {
@@ -2733,7 +2744,11 @@ export interface operations {
             };
             cookie?: never;
         };
-        requestBody?: never;
+        requestBody: {
+            content: {
+                "application/json": components["schemas"]["BranchActionBody"];
+            };
+        };
         responses: {
             201: {
                 headers: {
@@ -2761,7 +2776,11 @@ export interface operations {
             };
             cookie?: never;
         };
-        requestBody?: never;
+        requestBody: {
+            content: {
+                "application/json": components["schemas"]["BranchActionBody"];
+            };
+        };
         responses: {
             201: {
                 headers: {
@@ -5310,7 +5329,11 @@ export interface operations {
             };
             cookie?: never;
         };
-        requestBody?: never;
+        requestBody: {
+            content: {
+                "application/json": components["schemas"]["V1CreateMigrationBody"];
+            };
+        };
         responses: {
             200: {
                 headers: {

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

@@ -267,7 +267,7 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
                 name,
                 query,
               },
-            } as any // TODO: remove once API spec updated to include body
+            }
           );
 
           assertSuccess(response, 'Failed to apply migration');
@@ -345,6 +345,153 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
 
           assertSuccess(response, 'Failed to fetch TypeScript types');
 
+          return response.data;
+        },
+      }),
+      // Experimental features
+      create_branch: tool({
+        description:
+          'Creates a development branch with migrations from production project.',
+        parameters: z.object({
+          project_id: z.string(),
+          name: z
+            .string()
+            .default('develop')
+            .describe('Name of the branch to create'),
+        }),
+        execute: async ({ project_id, name }) => {
+          const response = await managementApiClient.POST(
+            '/v1/projects/{ref}/branches',
+            {
+              params: {
+                path: {
+                  ref: project_id,
+                },
+              },
+              body: {
+                branch_name: name,
+              },
+            }
+          );
+
+          assertSuccess(response, 'Failed to create branch');
+
+          return response.data;
+        },
+      }),
+      list_branches: tool({
+        description: 'Lists all development branches.',
+        parameters: z.object({
+          project_id: z.string(),
+        }),
+        execute: async ({ project_id }) => {
+          const response = await managementApiClient.GET(
+            '/v1/projects/{ref}/branches',
+            {
+              params: {
+                path: {
+                  ref: project_id,
+                },
+              },
+            }
+          );
+
+          assertSuccess(response, 'Failed to list branches');
+
+          return response.data;
+        },
+      }),
+      delete_branch: tool({
+        description: 'Deletes a development branch.',
+        parameters: z.object({
+          branch_id: z.string(),
+        }),
+        execute: async ({ branch_id }) => {
+          const response = await managementApiClient.DELETE(
+            '/v1/branches/{branch_id}',
+            {
+              params: {
+                path: {
+                  branch_id,
+                },
+              },
+            }
+          );
+
+          assertSuccess(response, 'Failed to delete branch');
+
+          return response.data;
+        },
+      }),
+      merge_branch: tool({
+        description:
+          'Merges migrations and edge functions from a development branch to production.',
+        parameters: z.object({
+          branch_id: z.string(),
+        }),
+        execute: async ({ branch_id }) => {
+          const response = await managementApiClient.POST(
+            '/v1/branches/{branch_id}/merge',
+            {
+              params: {
+                path: {
+                  branch_id,
+                },
+              },
+              body: {},
+            }
+          );
+
+          assertSuccess(response, 'Failed to merge branch');
+
+          return response.data;
+        },
+      }),
+      reset_branch: tool({
+        description:
+          'Resets migrations on a development branch to a prior version.',
+        parameters: z.object({
+          branch_id: z.string(),
+        }),
+        execute: async ({ branch_id }) => {
+          const response = await managementApiClient.POST(
+            '/v1/branches/{branch_id}/reset',
+            {
+              params: {
+                path: {
+                  branch_id,
+                },
+              },
+              body: {},
+            }
+          );
+
+          assertSuccess(response, 'Failed to reset branch');
+
+          return response.data;
+        },
+      }),
+      rebase_branch: tool({
+        description:
+          'Rebases development branch back on production in case it drifted.',
+        parameters: z.object({
+          branchId: z.string(),
+        }),
+        execute: async ({ branchId }) => {
+          const response = await managementApiClient.POST(
+            '/v1/branches/{branch_id}/push',
+            {
+              params: {
+                path: {
+                  branch_id: branchId,
+                },
+              },
+              body: {},
+            }
+          );
+
+          assertSuccess(response, 'Failed to rebase branch');
+
           return response.data;
         },
       }),