add MCP support

2026-03-05 12:17:14 +02:00
parent 961f09e092
commit 62a969d8fe
20 changed files with 820 additions and 2 deletions
@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module Mcp
+  module HandleRequest
+    TOOLS = [
+      Mcp::Tools::SearchTemplates,
+      Mcp::Tools::CreateTemplate,
+      Mcp::Tools::SendDocuments,
+      Mcp::Tools::SearchDocuments
+    ].freeze
+
+    TOOLS_SCHEMA = TOOLS.map { |t| t::SCHEMA }
+
+    TOOLS_INDEX = TOOLS.index_by { |t| t::SCHEMA[:name] }
+
+    module_function
+
+    # rubocop:disable Metrics/MethodLength
+    def call(body, current_user, current_ability)
+      case body['method']
+      when 'initialize'
+        {
+          jsonrpc: '2.0',
+          id: body['id'],
+          result: {
+            protocolVersion: '2025-11-25',
+            serverInfo: {
+              name: 'DocuSeal',
+              version: Docuseal.version.to_s
+            },
+            capabilities: {
+              tools: {
+                listChanged: false
+              }
+            }
+          }
+        }
+      when 'notifications/initialized'
+        nil
+      when 'ping'
+        { jsonrpc: '2.0', id: body['id'], result: {} }
+      when 'tools/list'
+        { jsonrpc: '2.0', id: body['id'], result: { tools: TOOLS_SCHEMA } }
+      when 'tools/call'
+        tool = TOOLS_INDEX[body.dig('params', 'name')]
+
+        raise "Unknown tool: #{body.dig('params', 'name')}" unless tool
+
+        result = tool.call(body.dig('params', 'arguments') || {}, current_user, current_ability)
+
+        { jsonrpc: '2.0', id: body['id'], result: }
+      else
+        {
+          jsonrpc: '2.0',
+          id: body['id'],
+          error: {
+            code: -32_601,
+            message: "Method not found: #{body['method']}"
+          }
+        }
+      end
+    end
+    # rubocop:enable Metrics/MethodLength
+  end
+end