mcp-sdk 0.0.0 → 0.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6c9a0c7f3108df8bf65658de1f0882d8eea74acdd09ed41570ef300378c81d9f
-  data.tar.gz: 6447dd877f10b0954faaa6b69bdd1eb4e50db0c7dfe2dea856eac23a9d820cac
+  metadata.gz: 5bfccbc3dbbb652bb24f57517c8b28aa4d517d3d7635df0db7611fdad12c3e63
+  data.tar.gz: 747f6bad3ba4342cfc531b07df8e705accebc62a8cdded803086840aecd88b1e
 SHA512:
-  metadata.gz: 070a30c38b06d5c70e8502d64d483a6df0db426fb7defaa8897abe8f0d7bfc9a36c08f5121936c11a6dd7458e3038101ad9066c2f84262511db884d206d6e517
-  data.tar.gz: 0c9f80d3614241faac6be2645f629cc5859b7d8920880968e608166dae2a2e2f8097bc0dc62b5d48c11eef47b49005da1fcc5d19097ba41412ed5d74abd17082
+  metadata.gz: 4d8c69bdd6f0ef59abd6d6db7e5d53fd5d6348c9ce82aaef590cc1438671c0f8bfd7fea129b116bbb6decde0591904cdbce211e7170b2274095f20fb34601002
+  data.tar.gz: 24547b56808d1cc94652d00c66d7d2671a0cf54b31ace47ed1edd346118113974e32bbddd04289c6a3a4f220f1daa0ff37e14e748e81ad037a45531f5c7530db

data/lib/mcp/mcp.rb

@@ -0,0 +1,3 @@
+module Mcp
+  VERSION = "0.0.1"
+end

data/lib/mcp/server/http.rb

@@ -0,0 +1,37 @@
+require 'json'
+
+module Mcp
+  module Server
+    class Http
+      def initialize
+        @request_handler = RequestHandler.new
+      end
+
+      def call(env)
+        req = Rack::Request.new(env)
+        path = req.path_info
+
+        case path
+        when "/sse"
+          sse(env)
+        end
+      end
+
+      private
+      
+      def sse(env)
+        body = env['rack.input'].read
+
+        begin
+          request = JSON.parse body
+        rescue JSON::ParserError
+          return [400, {"content-type" => "application/json"}, [{"message": "Bad Request"}.to_json]]
+        end
+
+        response = @request_handler.handle_request(request)
+
+        [200, { "content-type" => "application/json" }, [response.to_json]]
+      end
+    end
+  end
+end

data/lib/mcp/server/request_handler.rb

@@ -0,0 +1,36 @@
+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)
+
+        response = {"jsonrpc": "2.0", "result": result}
+        response["id"] = request["id"] if request["id"]
+
+        response
+      end
+
+      def handle_default(request)
+        case request["method"]
+        when "tools/list"
+          {
+            "tools": Mcp::Server.tools.map { |k, v| v.show },
+            "nextCursor": "next-page-cursor" # TODO: pagination
+          }
+        when "tools/call"
+          tool_name = request["params"]["name"]
+          tool = Mcp::Server.tools[tool_name.to_sym]
+          tool.call(request["params"]["arguments"])
+        when "resources/templates/list"
+          {
+            "resourceTemplates": Mcp::Server.resource_templates.map { |k, v| v.show }
+          }
+        else
+          # BUG
+          { "method": request["method"]}
+        end
+      end
+    end
+  end
+end

data/lib/mcp/server/resource_template.rb

@@ -0,0 +1,40 @@
+module Mcp
+  module Server
+    class ResourceTemplate
+      attr_reader :uri_template, :name, :description, :mime_type, :handler
+
+      def initialize(uri_template, name = nil, description = nil, mime_type = nil, &handler)
+        @uri_template = uri_template
+        @name = name
+        @description = description
+        @mime_type = mime_type
+        @handler = handler
+      end
+
+      def handler(&block)
+        @handler = block
+      end
+
+      def name(value)
+        @name = value
+      end
+
+      def description(value)
+        @description = value
+      end
+
+      def mime_type(value)
+        @mime_type = value
+      end
+
+      def show
+        {
+          "uriTemplate": uri_template,
+          "name": @name,
+          "description": @description,
+          "mimeType": @mime_type
+        }
+      end
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Mcp
+  module Server
+    module Stdio
+      class Reader
+        def initialize(io = STDIN)
+          @io = io
+          io.binmode
+        end
+
+        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)
+            return block.call(request)
+          end
+        end
+
+        def close
+          io.close
+        end
+
+        private
+
+        attr_reader :io
+      end
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require 'json'
+
+module Mcp
+  module Server
+    module Stdio
+      class Writer
+        attr_reader :io
+
+        def initialize(io = STDOUT)
+          @io = io
+          io.binmode
+        end
+
+        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
+        end
+
+        def close
+          io.close
+        end
+      end
+    end
+  end
+end

data/lib/mcp/server/tool.rb

@@ -0,0 +1,62 @@
+module Mcp
+  module Server
+    class Tool
+      attr_accessor :name, :description, :parameters
+
+      def initialize(name, description = nil, parameters = {}, &handler)
+        @name = name
+        @description = description
+        @parameters = parameters
+        @required_parameters = []
+        @handler = handler
+      end
+
+      def call(input)
+        # TODO: validate and handle errors
+        output = @handler.call(input)
+  
+        {
+          "content": [
+            {
+              "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
+        end
+        @required_parameters << name.to_sym if required
+      end
+
+      def description(value)
+        @description = value
+      end
+
+      def handler(&block)
+        @handler = block
+      end
+
+      def show
+        {
+          name: @name,
+          description: @description,
+          inputSchema: input_schema
+        }
+      end
+
+      private
+
+      def input_schema
+        {
+          "type": "object",
+          "properties": @parameters,
+          "required": @required_parameters
+        }
+      end
+    end
+  end
+end

data/lib/mcp/server.rb

@@ -0,0 +1,55 @@
+module Mcp
+  module Server
+    autoload :Configuration, "mcp/server/configuration"
+    autoload :Http, "mcp/server/http"
+
+    class << self
+      def configure(&block)
+        self.instance_eval(&block) if block_given?
+      end
+
+      def tools
+        @tools ||= {}
+      end
+
+      def resource_templates
+        @resource_templates ||= {}
+      end
+
+      def request_handlers
+        @request_handlers ||= {}
+      end
+
+      def tool(name, &block)
+        tool = Tool.new(name)
+        tool.instance_eval(&block)
+        tools[name] = tool
+      end
+
+      def resource(uri, &block)
+        resource_template = ResourceTemplate.new(uri)
+        resource_template.instance_eval(&block)
+        resource_templates[uri] = resource_template
+      end
+
+      def list_tools(&block)
+        request_handlers["tools/list"] = block
+      end
+
+      def http
+        Http.new
+      end
+
+      def stdio
+        reader = Stdio::Reader.new
+        writer = Stdio::Writer.new
+        request_handler = RequestHandler.new
+
+        reader.read do |request|
+          response = request_handler.handle_request(request)
+          writer.write(response)
+        end
+      end
+    end
+  end
+end

data/lib/mcp-sdk.rb

@@ -0,0 +1,8 @@
+require 'mcp/mcp'
+require 'mcp/server'
+require 'mcp/server/http'
+require 'mcp/server/request_handler'
+require 'mcp/server/tool'
+require 'mcp/server/resource_template'
+require 'mcp/server/stdio/reader'
+require 'mcp/server/stdio/writer'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mcp-sdk
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.1
 platform: ruby
 authors:
 - Joe Donovan
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-31 00:00:00.000000000 Z
+date: 2025-04-01 00:00:00.000000000 Z
 dependencies: []
 description: Implements the Model Context Protocol as a Ruby gem
 email: jsphdnvn@gmail.com
@@ -16,7 +16,15 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/mcp_sdk.rb
+- lib/mcp-sdk.rb
+- lib/mcp/mcp.rb
+- lib/mcp/server.rb
+- lib/mcp/server/http.rb
+- lib/mcp/server/request_handler.rb
+- lib/mcp/server/resource_template.rb
+- lib/mcp/server/stdio/reader.rb
+- lib/mcp/server/stdio/writer.rb
+- lib/mcp/server/tool.rb
 homepage: https://rubygems.org/gems/mcp-sdk
 licenses:
 - MIT

data/lib/mcp_sdk.rb

@@ -1,5 +0,0 @@
-class Mcp
-  def self.hi
-    puts "Hello world!"
-  end
-end