mcp-sdk 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5bff3f908a7b547a1bb322da2d703e7d01d587cb39a52bf319dbd2b54ffd6e65
-  data.tar.gz: e172b78334e1dab2b4cfc429fecd5c1bd71c9bc8f357f3ff4dff0273d43f7193
+  metadata.gz: 3d347486391f51ac84542791042a44bcf6066ae1142653156d79b484a6c6c12e
+  data.tar.gz: 28a848d66b3bd4cdfdc1d43acfd6d6bacef945f287238d3a65b49cb47f1d8f4c
 SHA512:
-  metadata.gz: f46cf467934fd424c5ca7491d19f8d02b63099dcb44e091a3a06143060b2ee2a0aa515d7d307893a0b947455417d1024c34d879c0ad4e2084f19dbf87513bdef
-  data.tar.gz: 04d4427dad286c3c3392dc2e41d26bb5a99423468d5a7618d04217f7370c8036e4dbcad45d31cdc0e57755347d5991ee278c9716782db74b13bb070a982f74bd
+  metadata.gz: 1a78aa17113eaa2311a410ed18a150f4b78fac32b9fdcde4caf85d3466f431657accedd9cb3fffe7cd6cfae13db6ed43ab3b33aeaa782110a8d19c7512d52d14
+  data.tar.gz: 10118aa5f83ce3165100609189b17e80f1fd116d5f43cd26d8a377b77b71bd40fcf5c4ff90d6076a4c2339d691cd3841c77c285f2e087e5ce2419fb5de2a3c38

data/lib/mcp/server/http.rb

@@ -1,19 +1,21 @@
 require 'json'
+require 'securerandom'
 
 module MCP
   module Server
     class Http
       def initialize
         @request_handler = RequestHandler.new
+        @clients = {}
       end
 
       def call(env)
         req = Rack::Request.new(env)
         path = req.path_info
-
+  
         case path
         when "/sse", "/", ""
-          sse(env)
+          sse(req, env)
         else
           [404, {"content-type" => "application/json"}, [{"error": "Not Found"}.to_json]]
         end
@@ -21,18 +23,170 @@ module MCP
 
       private
 
-      def sse(env)
-        body = env['rack.input'].read
+      def sse(req, env)
+        return handle_message_request(req) if req.post?
+        return [200, cors_headers, []] if req.options?
+
+        client_id = extract_client_id(req, env)
+        handle_client_reconnection(client_id) if @clients.key?(client_id)
+
+        if env['rack.hijack']
+          handle_rack_hijack_sse(env, client_id)
+        elsif rails_live_streaming?(env)
+          handle_rails_sse(env)
+        else
+          [200, sse_headers, [":ok\n\n"]]
+        end
+      end
+
+      def cors_headers
+        {
+          'Access-Control-Allow-Origin' => '*',
+          'Access-Control-Allow-Methods' => 'GET, OPTIONS',
+          'Access-Control-Allow-Headers' => 'Content-Type',
+          'Access-Control-Max-Age' => '86400', # 24 hours
+          'Content-Type' => 'text/plain'
+        }
+      end
+
+      def sse_headers
+        {
+          'Content-Type' => 'text/event-stream',
+          'Cache-Control' => 'no-cache, no-store, must-revalidate',
+          'Connection' => 'keep-alive',
+          'X-Accel-Buffering' => 'no', # For Nginx
+          'Access-Control-Allow-Origin' => '*', # Allow CORS
+          'Access-Control-Allow-Methods' => 'GET, OPTIONS',
+          'Access-Control-Allow-Headers' => 'Content-Type',
+          'Access-Control-Max-Age' => '86400', # 24 hours
+          'Keep-Alive' => 'timeout=600', # 10 minutes timeout
+          'Pragma' => 'no-cache',
+          'Expires' => '0'
+        }
+      end
+
+      def rails_live_streaming?(env)
+        defined?(ActionController::Live) &&
+          env['action_controller.instance'].respond_to?(:response) &&
+          env['action_controller.instance'].response.respond_to?(:stream)
+      end
+
+      def handle_rack_hijack_sse(env, client_id)
+        env['rack.hijack'].call
+        io = env['rack.hijack_io']
+        setup_sse_connection(io, client_id)
+        start_keep_alive_thread(io, client_id)
+
+        [-1, {}, []]
+      end
+
+      def setup_sse_connection(io, client_id)
+        io.write("HTTP/1.1 200 ok\r\n")
+        sse_headers.each { |k, v| io.write("#{k}: #{v}\r\n") }
+        io.write("\r\n")
+        io.flush
+
+        @clients[client_id] = io
+
+        io.write("retry: 100\n\n")
+        io.flush
+      end
+
+      def start_keep_alive_thread(io, client_id)
+        Thread.new do
+          ping_count = 0
+          ping_interval = 1
+          max_ping_count = 30
+          
+          @running = true
+          while @running && !io.closed?
+            begin
+              ping_count = send_keep_alive_ping(io, client_id, ping_count, max_ping_count)
+              break if ping_count >= max_ping_count
+
+              sleep ping_interval
+            rescue Errno::EPIPE, IOError => e
+              puts "SSE Connection error for client #{client_id}: #{e}"
+              break
+            end
+          end
+        end
+      end
 
+      def send_keep_alive_ping(io, client_id, ping_count, max_ping_count)
+        ping_count += 1
+
+        io.write(": keep-alive #{ping_count} \n\n")
+        io.flush
+
+        if (ping_count % 5).zero?
+          send_ping_event(io)
+        end
+
+        if ping_count >= max_ping_count
+          send_reconnect_event(io)
+        end
+
+        ping_count
+      end
+
+      def send_ping_event(io)
+        ping_message = { jsonrpc: "2.0", method: "ping", id: SecureRandom.uuid }
+        io.write("event: ping\ndata: #{ping_message.to_json}\n\n")
+        io.flush
+      end
+
+      def send_reconnect_event(io)
+        message = { reason: "timeout prevention" }
+        io.write("event: reconnect\ndata: #{message.to_json}")
+        io.flush
+      end
+
+      def extract_client_id(req, env)
+        client_id = req.params['client_id']
+        client_id ||= env['HTTP_LAST_EVENT_ID']
+        client_id ||= env['HTTP_X_CLIENT_ID']
+        client_id ||= SecureRandom.uuid
+      end
+
+      def handle_client_reconnection(client_id)
         begin
-          request = JSON.parse body
-        rescue JSON::ParserError
-          return [400, {"content-type" => "application/json"}, [{"message": "Bad Request"}.to_json]]
+          client_stream = @clients[client_id]
+          client_stream.close if client_stream.respond_to?(:close)
+        rescue StandardError => e
+          puts "Error closing old client connection"
         end
 
-        response = @request_handler.handle_request(request)
+        @clients.delete(client_id)
+
+        sleep 0.1
+      end
+
+      def handle_rails_sse(env, client_id)
+        controller = env['action_controller.instance']
+        stream = controller.response.stream
+
+        @clients[client_id] = stream
+
+        [200, sse_headers, []]
+      end
+
+      def handle_message_request(req)
+        body = req.body.read
+        request_body = JSON.parse(body, { :symbolize_names => true})
+        response = @request_handler.handle_request(request_body)
+        [200, { 'Content-Type' => 'application/json' }, [response.to_json]]
+      end
 
-        [200, { "content-type" => "application/json" }, [response.to_json]]
+      def error_response(http_status, code, message, id = nil)
+        [http_status, { 'Content-Type' => 'application/json' },
+         [JSON.generate(
+           {
+             jsonrpc: '2.0',
+             error: { code: code, message: message },
+             id: id
+           }
+         )]]
       end
     end
   end

data/lib/mcp/server/request_handler.rb

@@ -2,35 +2,35 @@ module MCP
   module Server
     class RequestHandler
       def handle_request(request)
-        handler = MCP::Server.request_handlers[request["method"]]
-        result = handler ? handler.call(request["params"]) : handle_default(request)
+        handler = MCP::Server.request_handlers[request[:method]]
+        result = handler ? handler.call(request[:params]) : handle_default(request)
 
-        response = {"jsonrpc": "2.0", "result": result}
-        response["id"] = request["id"] if request["id"]
+        response = {jsonrpc: "3.0", result: result}
+        response[:id] = request[:id] if request[:id]
 
         response
       end
 
       def handle_default(request)
-        case request["method"]
+        case request[:method]
         when "tools/list"
           {
-            "tools": MCP::Server.tools.map { |k, v| v.show },
-            "nextCursor": "next-page-cursor" # TODO: pagination
+            tools: MCP::Server.tools.map { |k, v| v.show },
+            nextCursor: "next-page-cursor" # TODO: pagination
           }
         when "tools/call"
-          tool_name = request["params"]["name"]
+          tool_name = request[:params][:name]
           tool = MCP::Server.tools[tool_name.to_sym]
-          tool.call(request["params"]["arguments"])
+          tool.call(request[:params][:arguments])
         when "resources/templates/list"
           {
-            "resourceTemplates": MCP::Server.resource_templates.map { |k, v| v.show }
+            resourceTemplates: MCP::Server.resource_templates.map { |k, v| v.show }
           }
         else
           # BUG
-          { "method": request["method"]}
+          { request: request}
         end
       end
     end
   end
-end
+end

data/lib/mcp/server/stdio/reader.rb

@@ -13,12 +13,7 @@ module MCP
 
         def read(&block)
           while (buffer = io.gets)
-            next unless (content_length_header = buffer.match(/Content-Length: (\d+)/i))
-
-            content_length = content_length_header[1].to_i
-            io.gets
-            request_bytes = io.read(content_length)
-            request = JSON.parse(request_bytes, symbolize_names: true)
+            request = JSON.parse(buffer, symbolize_names: true)
             return block.call(request)
           end
         end

data/lib/mcp/server/stdio/writer.rb

@@ -15,20 +15,7 @@ module MCP
 
         def write(response)
           response_str = response.to_json
-
-          headers = {
-            "Content-Length" => response_str.bytesize
-          }
-
-          headers.each do |k, v|
-            io.print "#{k}: #{v}\r\n"
-          end
-
-          io.print "\r\n"
-
-          io.print response_str
-
-          io.flush
+          io.puts response_str
         end
 
         def close

data/lib/mcp/server/tool.rb

@@ -16,18 +16,18 @@ module MCP
         output = @handler.call(input)
   
         {
-          "content": [
+          content: [
             {
-              "type": "text",
-              "text": output
+              type: "text",
+              text: output
             }
           ]
         }
       end
 
       def parameter(name, type:, description: nil, required: false)
-        @parameters[name.to_sym] = { "type": type }.tap do |param|
-          param["description"] = description if description
+        @parameters[name.to_sym] = { type: type }.tap do |param|
+          param[:description] = description if description
         end
         @required_parameters << name.to_sym if required
       end
@@ -52,9 +52,9 @@ module MCP
 
       def input_schema
         {
-          "type": "object",
-          "properties": @parameters,
-          "required": @required_parameters
+          type: "object",
+          properties: @parameters,
+          required: @required_parameters
         }
       end
     end

data/lib/mcp/server.rb

@@ -54,9 +54,11 @@ module MCP
         writer = Stdio::Writer.new
         request_handler = RequestHandler.new
 
-        reader.read do |request|
-          response = request_handler.handle_request(request)
-          writer.write(response)
+        loop do
+          reader.read do |request|
+            response = request_handler.handle_request(request)
+            writer.write(response)
+          end
         end
       end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mcp-sdk
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Joe Donovan
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-03 00:00:00.000000000 Z
+date: 2025-04-06 00:00:00.000000000 Z
 dependencies: []
 description: Implements the Model Context Protocol as a Ruby gem
 email: jsphdnvn@gmail.com
@@ -17,7 +17,6 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/mcp-sdk.rb
-- lib/mcp/inflections.rb
 - lib/mcp/railtie.rb
 - lib/mcp/server.rb
 - lib/mcp/server/http.rb
@@ -26,7 +25,7 @@ files:
 - lib/mcp/server/stdio/reader.rb
 - lib/mcp/server/stdio/writer.rb
 - lib/mcp/server/tool.rb
-homepage: https://rubygems.org/gems/mcp-sdk
+homepage: https://github.com/jdnvn/mcp-sdk-ruby
 licenses:
 - MIT
 metadata: {}

data/lib/mcp/inflections.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-# Add new inflections here
-ActiveSupport::Inflector.inflections do |inflect|
-  inflect.plural /^(my|your|his|her|its|our|their)\s+(?:(person|people))$/i, '\1\2s'
-  inflect.uncountable('species')
-  inflect.acronym('API')
-end