Class: CompletionKit::McpController

Inherits:
Api::V1::BaseController show all
Defined in:
app/controllers/completion_kit/mcp_controller.rb

Instance Method Summary collapse

Instance Method Details

#destroy ⇒ Object 



41
42
43
44
45
 
# File 'app/controllers/completion_kit/mcp_controller.rb', line 41

def destroy
  session_id = request.headers["Mcp-Session-Id"]
  Rails.cache.delete("mcp_session:#{session_id}") if session_id
  head :ok
end

#handle ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'app/controllers/completion_kit/mcp_controller.rb', line 3

def handle
  request_body = JSON.parse(request.body.read)

  if request_body["method"] == "initialize"
    result = McpDispatcher.initialize_session
    session_id = result.delete(:session_id)
    response.headers["Mcp-Session-Id"] = session_id
    render json: jsonrpc_response(request_body["id"], result)
    return
  end

  if request_body["method"] == "notifications/initialized"
    head :ok
    return
  end

  session_id = request.headers["Mcp-Session-Id"]
  unless session_id && Rails.cache.exist?("mcp_session:#{session_id}")
    render json: jsonrpc_error(request_body["id"], -32000, "Session not initialized. Send initialize first."), status: :bad_request
    return
  end

  result = McpDispatcher.dispatch(request_body["method"], request_body["params"])
  render json: jsonrpc_response(request_body["id"], result)
rescue JSON::ParserError
  render json: jsonrpc_error(nil, -32700, "Parse error"), status: :bad_request
rescue McpDispatcher::MethodNotFound => e
  render json: jsonrpc_error(request_body.dig("id"), -32601, e.message), status: :ok
rescue McpDispatcher::InvalidParams => e
  render json: jsonrpc_error(request_body.dig("id"), -32602, e.message), status: :ok
rescue ActiveRecord::RecordNotFound => e
  render json: jsonrpc_error(request_body.dig("id"), -32602, e.message), status: :ok
rescue ActiveRecord::RecordInvalid, ActiveRecord::InvalidForeignKey => e
  render json: jsonrpc_error(request_body.dig("id"), -32602, e.message), status: :ok
rescue StandardError => e
  render json: jsonrpc_error(request_body.dig("id"), -32603, e.message), status: :ok
end