feat: dynamically fetch content api graphql schema

Author: gregnr

package-lock.json

@@ -1,7 +1,5 @@
 {
-  "workspaces": [
-    "packages/*"
-  ],
+  "workspaces": ["packages/*"],
   "scripts": {
     "build": "npm run build --workspace @supabase/mcp-utils --workspace @supabase/mcp-server-supabase",
     "test": "npm run test --workspace @supabase/mcp-utils --workspace @supabase/mcp-server-supabase",

package.json

@@ -8,16 +8,13 @@
   "types": "dist/index.d.ts",
   "sideEffects": false,
   "scripts": {
-    "prebuild": "npm run generate:content-api-schema",
     "build": "tsup --clean",
     "prepublishOnly": "npm run build",
-    "pretest": "npm run generate:content-api-schema",
     "test": "vitest",
     "test:unit": "vitest --project unit",
     "test:e2e": "vitest --project e2e",
     "test:integration": "vitest --project integration",
     "test:coverage": "vitest --coverage",
-    "generate:content-api-schema": "tsx scripts/fetch-content-api-schema.ts",
     "generate:management-api-types": "openapi-typescript https://api.supabase.com/api/v1-json -o ./src/management-api/types.ts"
   },
   "files": ["dist/**/*"],
@@ -41,6 +38,7 @@
     "@modelcontextprotocol/sdk": "^1.11.0",
     "@supabase/mcp-utils": "0.2.1",
     "common-tags": "^1.8.2",
+    "graphql": "^16.11.0",
     "openapi-fetch": "^0.13.5",
     "zod": "^3.24.1"
   },
@@ -54,7 +52,6 @@
     "ai": "^4.3.4",
     "date-fns": "^4.1.0",
     "dotenv": "^16.5.0",
-    "graphql": "^16.11.0",
     "msw": "^2.7.3",
     "nanoid": "^5.1.5",
     "openapi-typescript": "^7.5.0",

packages/mcp-server-supabase/package.json

@@ -0,0 +1,88 @@
+import { stripIndent } from 'common-tags';
+import { describe, expect, it } from 'vitest';
+import { GraphQLClient } from './graphql.js';
+
+describe('graphql client', () => {
+  it('should load schema', async () => {
+    const schema = stripIndent`
+      schema {
+        query: RootQueryType
+      }
+      type RootQueryType {
+        message: String!
+      }
+    `;
+
+    const graphqlClient = new GraphQLClient({
+      url: 'dummy-url',
+      loadSchema: async () => schema,
+    });
+
+    const { source } = await graphqlClient.schemaLoaded;
+
+    expect(source).toBe(schema);
+  });
+
+  it('should throw error if validation requested but loadSchema not provided', async () => {
+    const graphqlClient = new GraphQLClient({
+      url: 'dummy-url',
+    });
+
+    await expect(
+      graphqlClient.query(
+        { query: '{ getHelloWorld }' },
+        { validateSchema: true }
+      )
+    ).rejects.toThrow('No schema loader provided');
+  });
+
+  it('should throw for invalid query regardless of schema', async () => {
+    const graphqlClient = new GraphQLClient({
+      url: 'dummy-url',
+    });
+
+    await expect(
+      graphqlClient.query({ query: 'invalid graphql query' })
+    ).rejects.toThrow(
+      'Invalid GraphQL query: Syntax Error: Unexpected Name "invalid"'
+    );
+  });
+
+  it("should throw error if query doesn't match schema", async () => {
+    const schema = stripIndent`
+      schema {
+        query: RootQueryType
+      }
+      type RootQueryType {
+        message: String!
+      }
+    `;
+
+    const graphqlClient = new GraphQLClient({
+      url: 'dummy-url',
+      loadSchema: async () => schema,
+    });
+
+    await expect(
+      graphqlClient.query(
+        { query: '{ invalidField }' },
+        { validateSchema: true }
+      )
+    ).rejects.toThrow(
+      'Invalid GraphQL query: Cannot query field "invalidField" on type "RootQueryType"'
+    );
+  });
+
+  it('bubbles up loadSchema errors', async () => {
+    const graphqlClient = new GraphQLClient({
+      url: 'dummy-url',
+      loadSchema: async () => {
+        throw new Error('Failed to load schema');
+      },
+    });
+
+    await expect(graphqlClient.schemaLoaded).rejects.toThrow(
+      'Failed to load schema'
+    );
+  });
+});