modelcontextprotocol/ruby-sdk
Watch
1
0
Fork
You've already forked ruby-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make StreamableHTTPTransport a Rack application

Browse source 
## Motivation and Context

Add `call(env)` to `StreamableHTTPTransport`, making it a Rack application
that works with `mount`, `run`, and Rack middleware.

Refactor examples to use Rack middleware classes for MCP logging instead of proc wrappers,
demonstrating idiomatic Rack composition with the new `run(transport)` pattern.

Update README.md with mount and controller integration patterns.

Closes #59, #60

## How Has This Been Tested?

Added tests for `call(env)` as a Rack app. All tests pass.

## Breaking Change

No breaking changes. All existing APIs are preserved:

- `StreamableHTTPTransport.new(server)` continues to work as before.
- `handle_request(request)` is unchanged. The new `call(env)` is a purely additive public method.
This commit is contained in:
Koichi ITO 2026-03-12 01:05:13 +09:00
commit af99a303e1
1 changed files with 234 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

102
test/mcp/server/transports/streamable_http_transport_test.rb
View file

@ -2860,6 +2860,108 @@ module MCP
assert session2_logging.should_notify?("debug")
end
test "call(env) works as a Rack app for POST requests" do
env = {
"REQUEST_METHOD" => "POST",
"PATH_INFO" => "/",
"rack.input" => StringIO.new({ jsonrpc: "2.0", method: "initialize", id: "init-1" }.to_json),
"CONTENT_TYPE" => "application/json",
"HTTP_ACCEPT" => "application/json, text/event-stream",
}
response = @transport.call(env)
assert_equal 200, response[0]
assert_equal "application/json", response[1]["Content-Type"]
body = JSON.parse(response[2][0])
assert_equal "2.0", body["jsonrpc"]
assert_equal "init-1", body["id"]
end
test "call(env) returns 405 for unsupported HTTP methods" do
env = {
"REQUEST_METHOD" => "PUT",
"PATH_INFO" => "/",
"rack.input" => StringIO.new(""),
}
response = @transport.call(env)
assert_equal 405, response[0]
end
test "call(env) handles GET SSE stream request" do
init_env = {
"REQUEST_METHOD" => "POST",
"PATH_INFO" => "/",
"rack.input" => StringIO.new({ jsonrpc: "2.0", method: "initialize", id: "init" }.to_json),
"CONTENT_TYPE" => "application/json",
"HTTP_ACCEPT" => "application/json, text/event-stream",
}
init_response = @transport.call(init_env)
session_id = init_response[1]["Mcp-Session-Id"]
get_env = {
"REQUEST_METHOD" => "GET",
"PATH_INFO" => "/",
"rack.input" => StringIO.new(""),
"HTTP_ACCEPT" => "text/event-stream",
"HTTP_MCP_SESSION_ID" => session_id,
}
response = @transport.call(get_env)
assert_equal 200, response[0]
assert_equal "text/event-stream", response[1]["Content-Type"]
assert response[2].is_a?(Proc)
end
test "call(env) handles DELETE session request" do
init_env = {
"REQUEST_METHOD" => "POST",
"PATH_INFO" => "/",
"rack.input" => StringIO.new({ jsonrpc: "2.0", method: "initialize", id: "init" }.to_json),
"CONTENT_TYPE" => "application/json",
"HTTP_ACCEPT" => "application/json, text/event-stream",
}
init_response = @transport.call(init_env)
session_id = init_response[1]["Mcp-Session-Id"]
delete_env = {
"REQUEST_METHOD" => "DELETE",
"PATH_INFO" => "/",
"rack.input" => StringIO.new(""),
"HTTP_MCP_SESSION_ID" => session_id,
}
response = @transport.call(delete_env)
assert_equal 200, response[0]
body = JSON.parse(response[2][0])
assert body["success"]
end
test "SSE response headers are not frozen so Rack middleware can modify them" do
init_request = create_rack_request(
"POST",
"/",
{ "CONTENT_TYPE" => "application/json" },
{ jsonrpc: "2.0", method: "initialize", id: "init-frozen" }.to_json,
)
init_response = @transport.handle_request(init_request)
session_id = init_response[1]["Mcp-Session-Id"]
request = create_rack_request(
"POST",
"/",
{ "CONTENT_TYPE" => "application/json", "HTTP_MCP_SESSION_ID" => session_id },
{ jsonrpc: "2.0", method: "tools/list", id: "sse-headers-1" }.to_json,
)
status, headers, = @transport.handle_request(request)
assert_equal 200, status
assert_equal "text/event-stream", headers["Content-Type"]
refute headers.frozen?, "SSE response headers should not be frozen"
end
private
def create_rack_request(method, path, headers, body = nil)