Merge pull request #14 from supabase-community/feat/reset

feat: support resetting branch to a specific migration version

Author: gregnr

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

@@ -825,6 +825,91 @@ describe('tools', () => {
     );
   });
 
+  test('revert migrations', async () => {
+    const { callTool } = await setup();
+    const project = mockProjects.values().next().value!;
+
+    await callTool({
+      name: 'enable_branching',
+      arguments: {
+        project_id: project.id,
+      },
+    });
+
+    const branch = await callTool({
+      name: 'create_branch',
+      arguments: {
+        project_id: project.id,
+        name: 'test-branch',
+      },
+    });
+
+    const migrationName = 'sample_migration';
+    const migrationQuery =
+      'create table sample (id integer generated always as identity primary key)';
+    await callTool({
+      name: 'apply_migration',
+      arguments: {
+        project_id: branch.project_ref,
+        name: migrationName,
+        query: migrationQuery,
+      },
+    });
+
+    // Check that migration has been applied to the branch
+    const firstListResult = await callTool({
+      name: 'list_migrations',
+      arguments: {
+        project_id: branch.project_ref,
+      },
+    });
+
+    expect(firstListResult).toContainEqual({
+      name: migrationName,
+      version: expect.stringMatching(/^\d{14}$/),
+    });
+
+    const firstTablesResult = await callTool({
+      name: 'list_tables',
+      arguments: {
+        project_id: branch.project_ref,
+      },
+    });
+
+    expect(firstTablesResult).toContainEqual(
+      expect.objectContaining({ name: 'sample' })
+    );
+
+    await callTool({
+      name: 'reset_branch',
+      arguments: {
+        branch_id: branch.id,
+        migration_version: '0',
+      },
+    });
+
+    // Check that all migrations have been reverted
+    const secondListResult = await callTool({
+      name: 'list_migrations',
+      arguments: {
+        project_id: branch.project_ref,
+      },
+    });
+
+    expect(secondListResult).toStrictEqual([]);
+
+    const secondTablesResult = await callTool({
+      name: 'list_tables',
+      arguments: {
+        project_id: branch.project_ref,
+      },
+    });
+
+    expect(secondTablesResult).not.toContainEqual(
+      expect.objectContaining({ name: 'sample' })
+    );
+  });
+
   test('rebase branch', async () => {
     const { callTool } = await setup();
     const project = mockProjects.values().next().value!;

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

@@ -592,8 +592,14 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
           'Resets migrations of a development branch. Any untracked data or schema changes will be lost.',
         parameters: z.object({
           branch_id: z.string(),
+          migration_version: z
+            .string()
+            .optional()
+            .describe(
+              'Reset your development branch to a specific migration version.'
+            ),
         }),
-        execute: async ({ branch_id }) => {
+        execute: async ({ branch_id, migration_version }) => {
           const response = await managementApiClient.POST(
             '/v1/branches/{branch_id}/reset',
             {
@@ -602,7 +608,9 @@ export function createSupabaseMcpServer(options: SupabaseMcpServerOptions) {
                   branch_id,
                 },
               },
-              body: {},
+              body: {
+                migration_version,
+              },
             }
           );
 

packages/mcp-server-supabase/test/mocks.ts

@@ -436,9 +436,9 @@ export const mockManagementApi = [
   /**
    * Resets a branch and re-runs migrations
    */
-  http.post<{ branchId: string }>(
+  http.post<{ branchId: string }, { migration_version?: string }>(
     `${API_URL}/v1/branches/:branchId/reset`,
-    async ({ params }) => {
+    async ({ params, request }) => {
       const branch = mockBranches.get(params.branchId);
       if (!branch) {
         return HttpResponse.json(
@@ -455,6 +455,15 @@ export const mockManagementApi = [
         );
       }
 
+      // Clear migrations below the specified version
+      const body = await request.json();
+      if (body.migration_version) {
+        const target = body.migration_version;
+        project.migrations = project.migrations.filter(
+          (m) => m.version <= target
+        );
+      }
+
       // Reset the DB a re-run migrations
       await project.resetDb();
       try {