model-context-protocol-rb 0.4.0 → 0.5.1

data/tasks/templates/dev-http.erb

@@ -4,6 +4,8 @@ require "bundler/setup"
 require "rack"
 require 'rackup/handler/webrick'
 require "webrick"
+require "webrick/https"
+require "openssl"
 require "securerandom"
 require "redis"
 require "logger"
@@ -12,6 +14,14 @@ require 'stringio'
 
 require_relative "../lib/model_context_protocol"
 
+ModelContextProtocol::Server.configure_redis do |config|
+  config.redis_url = "redis://localhost:6379/0"
+  config.pool_size = 10
+  config.enable_reaper = true
+  config.reaper_interval = 10
+  config.idle_timeout = 15
+end
+
 Dir[File.join(__dir__, "../spec/support/**/*.rb")].each { |file| require file }
 
 logger = Logger.new(STDOUT)
@@ -73,6 +83,11 @@ class MCPHttpApp
       @logger.info("  Request: #{body_content}") unless body_content.empty?
     end
 
+    if ModelContextProtocol::Server::RedisConfig.configured?
+      pool_stats = ModelContextProtocol::Server::RedisConfig.stats
+      @logger.info("  Redis Pool: #{pool_stats}")
+    end
+
     env['rack.input'] = StringIO.new(body_content)
     request = Rack::Request.new(env)
 
@@ -89,30 +104,26 @@ class MCPHttpApp
       }, [""]]
     end
 
-    begin
-      redis_client = Redis.new(url: ENV.fetch("REDIS_URL", "redis://localhost:6379/0"))
-      @logger.debug("Testing Redis connection...")
-      redis_client.ping
-      @logger.info("Redis connected successfully")
-    end
-
     transport_config = {
       type: :streamable_http,
-      env: env,
-      redis_client: redis_client,
-      require_sessions: false,  # Optional sessions for easier testing
+      env:,
+      require_sessions: false,
       session_ttl: 3600,
-      validate_origin: false,   # Disable for testing
       allowed_origins: ["*"]
     }
 
     @logger.debug("Creating MCP server with transport config")
     server = create_mcp_server(transport_config)
+    transport = nil
 
     begin
       @logger.debug("Starting MCP server")
       result = server.start
 
+      if server.respond_to?(:transport)
+        transport = server.transport
+      end
+
       case result
       when Hash
         if result[:stream]
@@ -134,6 +145,10 @@ class MCPHttpApp
               @logger.info("← ERROR RESPONSE (code: #{response_json[:error][:code]})")
             elsif status == 202
               @logger.info("← NOTIFICATION ACCEPTED [NO RESPONSE REQUIRED]")
+            elsif response_json[:accepted] == true && status == 200
+              method = request_json['method'] rescue 'unknown'
+              id = request_json['id'] rescue 'unknown'
+              @logger.info("← #{method} RESPONSE (id: #{id}) [DELIVERED VIA SSE STREAM]")
             elsif response_json[:result]
               method = request_json['method'] rescue 'unknown'
               @logger.info("← #{method} RESPONSE (id: #{response_json[:id]})")
@@ -167,6 +182,7 @@ class MCPHttpApp
       @logger.debug("Full backtrace:\n#{e.backtrace.join("\n")}")
       [500, {"Content-Type" => "application/json"}, [%Q({"error": "Internal server error: #{e.message}"})]]
     ensure
+      transport&.cleanup if transport&.respond_to?(:cleanup)
       Thread.current[:request_id] = nil
     end
   end
@@ -217,13 +233,18 @@ class MCPHttpApp
           register TestToolWithMixedContentResponse
           register TestToolWithResourceResponse
           register TestToolWithToolErrorResponse
+          register TestToolWithCancellableSleep
         end
       end
     end
   end
 end
 
-logger.info("Starting MCP HTTP Development Server on http://localhost:9292/mcp")
+use_ssl = ENV['SSL'] == 'true'
+port = use_ssl ? 9293 : 9292
+protocol = use_ssl ? 'https' : 'http'
+
+logger.info("Starting MCP #{protocol.upcase} Development Server on #{protocol}://localhost:#{port}/mcp")
 
 app = Rack::Builder.new do
   map '/mcp' do
@@ -231,7 +252,30 @@ app = Rack::Builder.new do
   end
 end
 
-server = WEBrick::HTTPServer.new(Port: 9292, Host: '0.0.0.0')
+server_options = {
+  Port: port,
+  Host: '0.0.0.0'
+}
+
+if use_ssl
+  cert_path = File.join(__dir__, '..', 'tmp', 'ssl', 'server.crt')
+  key_path = File.join(__dir__, '..', 'tmp', 'ssl', 'server.key')
+
+  unless File.exist?(cert_path) && File.exist?(key_path)
+    logger.error("SSL certificates not found at tmp/ssl/")
+    logger.error("Generate them with: openssl req -x509 -newkey rsa:4096 -keyout tmp/ssl/server.key -out tmp/ssl/server.crt -days 365 -nodes -subj \"/C=US/ST=Dev/L=Dev/O=Dev/CN=localhost\"")
+    exit(1)
+  end
+
+  server_options.merge!(
+    SSLEnable: true,
+    SSLCertificate: OpenSSL::X509::Certificate.new(File.read(cert_path)),
+    SSLPrivateKey: OpenSSL::PKey::RSA.new(File.read(key_path)),
+    SSLVerifyClient: OpenSSL::SSL::VERIFY_NONE
+  )
+end
+
+server = WEBrick::HTTPServer.new(server_options)
 server.mount '/', Rackup::Handler::WEBrick, app
 
 ['INT', 'TERM'].each do |signal|
@@ -241,4 +285,4 @@ server.mount '/', Rackup::Handler::WEBrick, app
   end
 end
 
-server.start
+server.start

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: model-context-protocol-rb
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.1
 platform: ruby
 authors:
 - Dick Davis
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-09-08 00:00:00.000000000 Z
+date: 2025-09-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json-schema
@@ -38,6 +38,48 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: connection_pool
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.4'
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
 description:
 email:
 - dick@hey.com
@@ -56,20 +98,32 @@ files:
 - Rakefile
 - lib/model_context_protocol.rb
 - lib/model_context_protocol/server.rb
+- lib/model_context_protocol/server/cancellable.rb
 - lib/model_context_protocol/server/completion.rb
 - lib/model_context_protocol/server/configuration.rb
 - lib/model_context_protocol/server/content.rb
 - lib/model_context_protocol/server/content_helpers.rb
 - lib/model_context_protocol/server/mcp_logger.rb
 - lib/model_context_protocol/server/pagination.rb
+- lib/model_context_protocol/server/progressable.rb
 - lib/model_context_protocol/server/prompt.rb
+- lib/model_context_protocol/server/redis_client_proxy.rb
+- lib/model_context_protocol/server/redis_config.rb
+- lib/model_context_protocol/server/redis_pool_manager.rb
 - lib/model_context_protocol/server/registry.rb
 - lib/model_context_protocol/server/resource.rb
 - lib/model_context_protocol/server/resource_template.rb
 - lib/model_context_protocol/server/router.rb
-- lib/model_context_protocol/server/session_store.rb
 - lib/model_context_protocol/server/stdio_transport.rb
+- lib/model_context_protocol/server/stdio_transport/request_store.rb
 - lib/model_context_protocol/server/streamable_http_transport.rb
+- lib/model_context_protocol/server/streamable_http_transport/event_counter.rb
+- lib/model_context_protocol/server/streamable_http_transport/message_poller.rb
+- lib/model_context_protocol/server/streamable_http_transport/notification_queue.rb
+- lib/model_context_protocol/server/streamable_http_transport/request_store.rb
+- lib/model_context_protocol/server/streamable_http_transport/session_message_queue.rb
+- lib/model_context_protocol/server/streamable_http_transport/session_store.rb
+- lib/model_context_protocol/server/streamable_http_transport/stream_registry.rb
 - lib/model_context_protocol/server/tool.rb
 - lib/model_context_protocol/version.rb
 - tasks/mcp.rake