@payloadcms/plugin-mcp 4.0.0-internal.5b1e7cd → 4.0.0-internal.5f0cd13

package/src/mcp/sanitizeMCPConfig.ts

@@ -0,0 +1,239 @@
+import type {
+  CollectionConfig,
+  Config,
+  GlobalConfig,
+  SanitizedCollectionConfig,
+  SanitizedConfig,
+  SanitizedGlobalConfig,
+} from 'payload'
+
+import type {
+  CollectionTool,
+  GlobalTool,
+  MCPBuiltInToolOverride,
+  MCPItem,
+  MCPPluginCollectionConfig,
+  MCPPluginConfig,
+  MCPPluginGlobalConfig,
+  SanitizedMCPPluginConfig,
+} from '../types.js'
+
+import {
+  COLLECTION_AUTH_BUILTIN_ENTRIES,
+  COLLECTION_AUTH_BUILTINS,
+  COLLECTION_BUILTIN_ENTRIES,
+  COLLECTION_BUILTINS,
+  GLOBAL_BUILTIN_ENTRIES,
+  GLOBAL_BUILTINS,
+} from './builtinTools.js'
+
+/**
+ * Converts the user-configured `MCPPluginConfig` into a `SanitizedMCPPluginConfig`:
+ *  - Flattens `tools` / `prompts` / `resources` / per-collection / per-global
+ *    tool maps into a single `items` array.
+ *  - Applies built-in tools for collections and globals, respecting opt-out user overrides.
+ *  - Applies the `userCollection` default
+ *
+ * Called once during plugin init. After that, `plugins['@payloadcms/plugin-mcp']
+ * ?.options` holds the sanitized result
+ */
+export const sanitizeMCPConfig = ({
+  config,
+  pluginConfig,
+}: {
+  config: Config | SanitizedConfig
+  pluginConfig: MCPPluginConfig
+}): SanitizedMCPPluginConfig => {
+  const items: MCPItem[] = []
+
+  for (const collection of config.collections ?? []) {
+    items.push(...sanitizeCollectionConfig({ collection, pluginConfig }))
+  }
+
+  for (const global of config.globals ?? []) {
+    items.push(...sanitizeGlobalConfig({ global, pluginConfig }))
+  }
+
+  for (const [key, tool] of Object.entries(pluginConfig.tools ?? {})) {
+    items.push({
+      type: 'tool',
+      key,
+      label: key,
+      tool,
+    })
+  }
+
+  for (const [key, prompt] of Object.entries(pluginConfig.prompts ?? {})) {
+    items.push({
+      type: 'prompt',
+      key,
+      label: prompt.title ?? key,
+      prompt,
+    })
+  }
+
+  for (const [key, resource] of Object.entries(pluginConfig.resources ?? {})) {
+    items.push({
+      type: 'resource',
+      key,
+      label: resource.title ?? key,
+      resource,
+    })
+  }
+
+  // Mirror Payload's own admin.user detection (sanitize.ts) since plugins run first.
+  const firstCollectionWithAuth = config.collections!.find(({ auth }) => Boolean(auth))
+
+  return {
+    disabled: pluginConfig.disabled,
+    items,
+    mcp: pluginConfig.mcp,
+    overrideApiKeyCollection: pluginConfig.overrideApiKeyCollection,
+    overrideAuth: pluginConfig.overrideAuth,
+    userCollection:
+      pluginConfig.userCollection ?? config.admin?.user ?? firstCollectionWithAuth?.slug ?? 'users',
+  }
+}
+
+const sanitizeCollectionConfig = ({
+  collection,
+  pluginConfig,
+}: {
+  collection: CollectionConfig | SanitizedCollectionConfig
+  pluginConfig: MCPPluginConfig
+}): MCPItem[] => {
+  if (collection.slug === 'payload-mcp-api-keys') {
+    return []
+  }
+  const slug = collection.slug
+  const collectionPluginConfig = pluginConfig.collections?.[slug]
+  const items: MCPItem[] = []
+
+  for (const [toolKey, tool] of COLLECTION_BUILTIN_ENTRIES) {
+    const matchedConfigEntry = collectionPluginConfig?.tools?.[toolKey]
+    if (matchedConfigEntry === false) {
+      continue
+    }
+    items.push({
+      type: 'collectionTool',
+      collectionSlug: slug,
+      key: toolKey,
+      label: capitalize(toolKey),
+      tool: overrideBuiltinTool(tool, matchedConfigEntry, collectionPluginConfig),
+    })
+  }
+
+  if (collection.auth) {
+    for (const [authToolKey, { label, tool }] of COLLECTION_AUTH_BUILTIN_ENTRIES) {
+      const matchedConfigEntry = collectionPluginConfig?.tools?.[authToolKey]
+      if (!matchedConfigEntry) {
+        continue
+      }
+      // `true` means "enable, no override"; only the object form carries fields.
+      const override = typeof matchedConfigEntry === 'object' ? matchedConfigEntry : undefined
+      items.push({
+        type: 'collectionTool',
+        collectionSlug: slug,
+        key: authToolKey,
+        label,
+        tool: overrideBuiltinTool(tool, override, collectionPluginConfig),
+      })
+    }
+  }
+
+  // Cast: builtin keys are filtered out below, so the remaining values are
+  // always custom tools (`CollectionTool`) or undefined
+  const customEntries = Object.entries(collectionPluginConfig?.tools ?? {}) as Array<
+    [string, CollectionTool | undefined]
+  >
+  for (const [key, customTool] of customEntries) {
+    if (key in COLLECTION_BUILTINS || key in COLLECTION_AUTH_BUILTINS) {
+      continue
+    }
+    if (!customTool) {
+      continue
+    }
+    items.push({
+      type: 'collectionTool',
+      collectionSlug: slug,
+      key,
+      label: key,
+      tool: customTool,
+    })
+  }
+
+  return items
+}
+
+const sanitizeGlobalConfig = ({
+  global,
+  pluginConfig,
+}: {
+  global: GlobalConfig | SanitizedGlobalConfig
+  pluginConfig: MCPPluginConfig
+}): MCPItem[] => {
+  const slug = global.slug
+  const globalPluginConfig = pluginConfig.globals?.[slug]
+  const items: MCPItem[] = []
+
+  for (const [toolKey, baseTool] of GLOBAL_BUILTIN_ENTRIES) {
+    const matchedConfigEntry = globalPluginConfig?.tools?.[toolKey]
+    if (matchedConfigEntry === false) {
+      continue
+    }
+    items.push({
+      type: 'globalTool',
+      globalSlug: slug,
+      key: toolKey,
+      label: capitalize(toolKey),
+      tool: overrideBuiltinTool(baseTool, matchedConfigEntry, globalPluginConfig),
+    })
+  }
+
+  const customEntries = Object.entries(globalPluginConfig?.tools ?? {}) as Array<
+    [string, GlobalTool | undefined]
+  >
+  for (const [key, customTool] of customEntries) {
+    if (key in GLOBAL_BUILTINS) {
+      continue
+    }
+    if (!customTool) {
+      continue
+    }
+    items.push({
+      type: 'globalTool',
+      globalSlug: slug,
+      key,
+      label: key,
+      tool: customTool,
+    })
+  }
+
+  return items
+}
+
+const capitalize = (s: string): string => s.charAt(0).toUpperCase() + s.slice(1)
+
+/**
+ * Spread the static built-in tool and apply consumer-side overrides.
+ * Precedence: per-tool override > scope-level (collection/global) override >
+ * the static tool's defaults. `toolEntry` is whatever the user put under
+ * `tools: { find: ... }` — could be `true`, `false`, or an override object —
+ * so it's narrowed internally.
+ */
+const overrideBuiltinTool = <TTool extends CollectionTool | GlobalTool>(
+  tool: TTool,
+  toolOverride?: MCPBuiltInToolOverride,
+  entityPluginConfig?: TTool extends CollectionTool
+    ? MCPPluginCollectionConfig<any>
+    : MCPPluginGlobalConfig,
+): TTool => {
+  return {
+    ...tool,
+    description: toolOverride?.description ?? entityPluginConfig?.description ?? tool.description,
+    overrideResponse:
+      toolOverride?.overrideResponse ??
+      entityPluginConfig?.overrideResponse ??
+      tool.overrideResponse,
+  }
+}

package/src/stdio.ts

@@ -0,0 +1,98 @@
+import type { MaybePromise, Plugin, SanitizedConfig } from 'payload'
+
+/* eslint-disable no-console */
+import { StdioServerTransport } from '@modelcontextprotocol/server'
+import { fileURLToPath, pathToFileURL } from 'node:url'
+import { createLocalReq, getPayload } from 'payload'
+import { findConfig } from 'payload/node'
+
+import type { AuthorizedMCP, SanitizedMCPPluginConfig } from './types.js'
+
+import { buildMcpServer } from './mcp/buildMcpServer.js'
+import { sanitizeMCPConfig } from './mcp/sanitizeMCPConfig.js'
+import { getPluginConfig } from './utils/getPluginConfig.js'
+import { resolveProjectRoot } from './utils/resolveProjectRoot.js'
+
+/**
+ * Stdio adapter for the Payload MCP server.
+ *
+ * Do not use in production. There's no auth; whoever can spawn the process gets
+ * full access to your local data.
+ */
+export const runMcpStdio = async (): Promise<void> => {
+  /**
+   * If MCP clients spawn stdio servers from an arbitrary working directory,
+   * Payload's cwd-anchored `findConfig()` can't locate the project on
+   * its own. This module always lives inside the project's `node_modules`, so
+   * we derive the project root from its own path and `chdir` into it.
+   *
+   * An absolute `PAYLOAD_CONFIG_PATH` still overrides everything; unusual
+   * layouts (e.g. a monorepo where the package is hoisted above the app) can
+   * fall back to it. See the implementation of findConfig for details.
+   */
+  const projectRoot = resolveProjectRoot(fileURLToPath(import.meta.url))
+  if (projectRoot) {
+    process.chdir(projectRoot)
+  }
+
+  const configPath = findConfig()
+  const configModule = await import(pathToFileURL(configPath).toString())
+  const config = (await (configModule.default ?? configModule)) as MaybePromise<SanitizedConfig>
+
+  const payload = await getPayload({ config })
+
+  /**
+   * If the user added `mcpPlugin({...})` to their `plugins` array, read the
+   * sanitized config they registered. Otherwise fall back to the defaults. That way,
+   * the mcp works on any project that has `@payloadcms/plugin-mcp` installed,
+   * even if the plugin is not added to the plugins array.
+   */
+  let pluginConfig: SanitizedMCPPluginConfig
+  try {
+    pluginConfig = getPluginConfig({ config: payload.config })
+  } catch {
+    pluginConfig = sanitizeMCPConfig({ config: payload.config, pluginConfig: {} })
+
+    const fakePluginFn: Plugin = (config) => config
+    fakePluginFn.slug = '@payloadcms/plugin-mcp'
+    // @ts-expect-error
+    fakePluginFn.sanitizedOptions = pluginConfig
+
+    // Push to payload config, to ensure consecutive calls to `getPluginConfig()` work
+    ;(payload.config.plugins ??= []).push(fakePluginFn)
+  }
+
+  const authorizedMCP: AuthorizedMCP = {
+    items: pluginConfig.items,
+    overrideAccess: true,
+    user: null,
+  }
+
+  const req = await createLocalReq({}, payload)
+  req.payloadAPI = 'MCP' as const
+
+  const server = buildMcpServer({ authorizedMCP, pluginConfig, req })
+
+  const transport = new StdioServerTransport()
+  await server.connect(transport)
+
+  // Process now lives until stdin closes. Wire graceful shutdown so DB
+  // connections (and any other Payload-managed resources) get released.
+  const shutdown = async (code = 0) => {
+    try {
+      await server.close()
+    } catch (err) {
+      console.error('[payload-mcp] error closing server:', err)
+    }
+    try {
+      await payload.destroy()
+    } catch (err) {
+      console.error('[payload-mcp] error destroying payload:', err)
+    }
+    process.exit(code)
+  }
+
+  process.on('SIGINT', () => void shutdown(0))
+  process.on('SIGTERM', () => void shutdown(0))
+  process.stdin.on('close', () => void shutdown(0))
+}