Merge pull request #21 from supabase-community/feat/project-cost-prompt

feat: cost confirmation flows

Author: gregnr

README.md

@@ -154,6 +154,11 @@ The following Supabase tools are available to the LLM:
 
 - `generate_typescript_types`: Generates TypeScript types based on the database schema. LLMs can save this to a file and use it in their code.
 
+#### Cost Confirmation
+
+- `get_cost`: Gets the cost of a new project or branch for an organization.
+- `confirm_cost`: Confirms the user's understanding of new project or branch costs. This is required to create a new project or branch.
+
 ## Other MCP servers
 
 ### `@supabase/mcp-server-postgrest`

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

@@ -0,0 +1,71 @@
+import {
+  assertSuccess,
+  type ManagementApiClient,
+} from './management-api/index.js';
+
+export const PROJECT_COST_MONTHLY = 10;
+export const BRANCH_COST_HOURLY = 0.01344;
+
+export type ProjectCost = {
+  type: 'project';
+  recurrence: 'monthly';
+  amount: number;
+};
+
+export type BranchCost = {
+  type: 'branch';
+  recurrence: 'hourly';
+  amount: number;
+};
+
+export type Cost = ProjectCost | BranchCost;
+
+/**
+ * Gets the cost of the next project in an organization.
+ */
+export async function getNextProjectCost(
+  managementApiClient: ManagementApiClient,
+  orgId: string
+): Promise<Cost> {
+  const orgResponse = await managementApiClient.GET(
+    '/v1/organizations/{slug}',
+    {
+      params: {
+        path: {
+          slug: orgId,
+        },
+      },
+    }
+  );
+
+  assertSuccess(orgResponse, 'Failed to fetch organization');
+
+  const projectsResponse = await managementApiClient.GET('/v1/projects');
+
+  assertSuccess(projectsResponse, 'Failed to fetch projects');
+
+  const org = orgResponse.data;
+  const activeProjects = projectsResponse.data.filter(
+    (project) =>
+      project.organization_id === orgId &&
+      !['INACTIVE', 'GOING_DOWN', 'REMOVED'].includes(project.status)
+  );
+
+  let amount = 0;
+
+  if (org.plan !== 'free') {
+    // If the organization is on a paid plan, the first project is included
+    if (activeProjects.length > 0) {
+      amount = PROJECT_COST_MONTHLY;
+    }
+  }
+
+  return { type: 'project', recurrence: 'monthly', amount };
+}
+
+/**
+ * Gets the cost for a database branch.
+ */
+export function getBranchCost(): Cost {
+  return { type: 'branch', recurrence: 'hourly', amount: BRANCH_COST_HOURLY };
+}