ask-mcp 0.1.0

data/lib/ask/mcp/transport/stdio.rb

@@ -0,0 +1,122 @@
+# frozen_string_literal: true
+
+require "open3"
+
+module Ask
+  module MCP
+    module Transport
+      class Stdio
+        attr_reader :command, :args, :pid
+
+        def initialize(command, args = [], options = {})
+          @command = command
+          @args = args
+          @options = options
+          @pid = nil
+          @stdin = nil
+          @stdout = nil
+          @stderr = nil
+          @wait_thr = nil
+          @buffer = +""
+          @message_handlers = []
+          @running = false
+          @mutex = Mutex.new
+        end
+
+        def on_message(&block)
+          @message_handlers << block
+        end
+
+        def start
+          env = @options[:env] || {}
+          workdir = @options[:workdir]
+
+          cmd = build_command
+          @stdin, @stdout, @stderr, @wait_thr = Open3.popen3(env, *cmd, chdir: workdir || Dir.pwd)
+          @pid = @wait_thr.pid
+          @running = true
+
+          start_reader
+          self
+        end
+
+        def stop
+          @running = false
+          @stdin&.close unless @stdin&.closed?
+          @stdout&.close unless @stdout&.closed?
+          @stderr&.close unless @stderr&.closed?
+          @wait_thr&.value
+        rescue Errno::EPIPE, Errno::ECHILD
+          # Process already exited
+        end
+
+        def send(message)
+          data = message.is_a?(String) ? message : message.to_json
+          @mutex.synchronize do
+            @stdin&.puts(data)
+            @stdin&.flush
+          end
+        rescue Errno::EPIPE, IOError => e
+          raise ConnectionError, "Failed to send message: #{e.message}"
+        end
+
+        def running?
+          @running && @wait_thr&.alive?
+        end
+
+        def shutdown
+          stop
+        end
+
+        private
+
+        def build_command
+          if @command.is_a?(Array)
+            @command
+          else
+            [@command] + @args
+          end
+        end
+
+        def start_reader
+          @reader_thread = Thread.new do
+            read_partial_line(@stdout)
+          rescue IOError, Errno::EPIPE, Errno::EBADF => e
+            @running = false
+            notify_error(e)
+          end
+        end
+
+        def read_partial_line(io)
+          while @running && (char = io.getc)
+            @buffer << char
+            if @buffer.end_with?("\n")
+              line = @buffer.strip
+              @buffer = +""
+              next if line.empty?
+
+              process_line(line)
+            end
+          end
+        rescue IOError, Errno::EPIPE, Errno::EBADF
+          @running = false
+        end
+
+        def process_line(line)
+          message = Native::Messages::Parser.parse(line)
+          @message_handlers.each { |handler| handler.call(message) }
+        rescue JSON::ParserError => e
+          # Ignore non-JSON output (e.g., stderr mixed in)
+        rescue ProtocolError => e
+          notify_error(e)
+        end
+
+        def notify_error(error)
+          @message_handlers.each do |handler|
+            handler.call(error) if error.is_a?(Exception)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ask/mcp/transport/streamable_http.rb

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+
+module Ask
+  module MCP
+    module Transport
+      class StreamableHTTP
+        attr_reader :url
+
+        def initialize(url, options = {})
+          @url = url
+          @options = options
+          @running = false
+          @message_handlers = []
+          @http = nil
+          @session_id = nil
+        end
+
+        def on_message(&block)
+          @message_handlers << block
+        end
+
+        def start
+          require "httpx"
+
+          headers = { "Content-Type" => "application/json" }
+          headers["Accept"] = "text/event-stream" if @options[:stream]
+          headers.merge!(@options[:headers]) if @options[:headers]
+
+          @http = HTTPX.with(
+            headers:,
+            timeout: { request_timeout: @options[:timeout] || 30 }
+          )
+          @running = true
+          self
+        end
+
+        def stop
+          @running = false
+          @http&.close
+        end
+
+        def send(message)
+          data = message.is_a?(String) ? message : message.to_json
+
+          if @options[:stream]
+            send_streaming(data)
+          else
+            send_request_response(data)
+          end
+        rescue HTTPX::Error => e
+          raise ConnectionError, "HTTP error: #{e.message}"
+        end
+
+        def running?
+          @running
+        end
+
+        def shutdown
+          stop
+        end
+
+        private
+
+        def send_request_response(data)
+          response = @http.post(@url, body: data)
+          status = response.status
+
+          if status == 200 || status == 202
+            body = response.body.to_s
+            if body && !body.empty?
+              message = Native::Messages::Parser.parse(body)
+              @message_handlers.each { |handler| handler.call(message) }
+            end
+          elsif status == 204
+            # No content — nothing to process
+          else
+            raise ConnectionError, "HTTP #{status}: #{response.body.to_s[0..200]}"
+          end
+
+          response
+        end
+
+        def send_streaming(data)
+          response = @http.post(@url, body: data)
+
+          unless response.status == 200
+            raise ConnectionError, "HTTP #{response.status}: #{response.body.to_s[0..200]}"
+          end
+
+          buffer = +""
+          response.body.each do |chunk|
+            buffer << chunk
+            while (line = buffer.slice!(/\A.*\n/))
+              line = line.strip
+              next if line.empty?
+
+              if line.start_with?("data: ")
+                data_line = line[6..]
+                begin
+                  message = Native::Messages::Parser.parse(data_line)
+                  @message_handlers.each { |handler| handler.call(message) }
+                rescue JSON::ParserError
+                  # Skip non-JSON data lines
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ask/mcp/validator.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Ask
+  module MCP
+    # Validates tool call arguments against JSON Schema input schemas.
+    # Uses the json-schema gem to validate arguments before sending to a server.
+    class Validator
+      class ValidationError < Error; end
+
+      def initialize(schema)
+        @schema = schema
+      end
+
+      def validate!(arguments)
+        return true if @schema.nil? || @schema.empty?
+
+        require "json-schema"
+
+        string_schema = deep_stringify_keys(@schema)
+        data = arguments.is_a?(Hash) ? deep_stringify_keys(arguments) : arguments
+
+        errors = JSON::Validator.fully_validate(string_schema, data)
+        if errors.any?
+          raise ValidationError, "Validation failed: #{errors.join(", ")}"
+        end
+
+        true
+      end
+
+      def valid?(arguments)
+        validate!(arguments)
+        true
+      rescue ValidationError
+        false
+      end
+
+      private
+
+      def deep_stringify_keys(obj)
+        case obj
+        when Hash
+          obj.each_with_object({}) { |(k, v), h| h[k.to_s] = deep_stringify_keys(v) }
+        when Array
+          obj.map { |v| deep_stringify_keys(v) }
+        else
+          obj
+        end
+      end
+    end
+  end
+end

data/lib/ask/mcp/version.rb

@@ -0,0 +1,5 @@
+module Ask
+  module MCP
+    VERSION = "0.1.0"
+  end
+end

data/lib/ask/mcp.rb

@@ -0,0 +1,65 @@
+require "json"
+require "securerandom"
+require_relative "mcp/version"
+
+module Ask
+  module MCP
+    autoload :Client, "ask/mcp/client"
+    autoload :Server, "ask/mcp/server"
+    autoload :Tool, "ask/mcp/tool"
+    autoload :Resource, "ask/mcp/resource"
+    autoload :Prompt, "ask/mcp/prompt"
+    autoload :Validator, "ask/mcp/validator"
+
+    module Native
+      autoload :Messages, "ask/mcp/native/messages"
+    end
+
+    module Transport
+      autoload :Stdio, "ask/mcp/transport/stdio"
+      autoload :SSE, "ask/mcp/transport/sse"
+      autoload :StreamableHTTP, "ask/mcp/transport/streamable_http"
+    end
+
+    module Auth
+      autoload :OAuth, "ask/mcp/auth/oauth"
+      autoload :Token, "ask/mcp/auth/token"
+    end
+
+    module Adapters
+      autoload :AskTool, "ask/mcp/adapters/ask_tool"
+    end
+
+    class Error < StandardError; end
+    class ConnectionError < Error; end
+    class ProtocolError < Error; end
+    class AuthError < Error; end
+    class ValidationError < Error; end
+
+    class << self
+      def connect(transport, options = {})
+        Client.new(transport, options)
+      end
+
+      def from_stdio(command, args = [], options = {})
+        transport = Transport::Stdio.new(command, args, options)
+        Client.new(transport)
+      end
+
+      def from_sse(url, options = {})
+        transport = Transport::SSE.new(url, options)
+        Client.new(transport)
+      end
+
+      def from_http(url, options = {})
+        transport = Transport::StreamableHTTP.new(url, options)
+        Client.new(transport)
+      end
+
+      # Validate tool arguments against a JSON Schema input schema
+      def validate!(schema, arguments)
+        Validator.new(schema).validate!(arguments)
+      end
+    end
+  end
+end

data/lib/ask-mcp.rb

@@ -0,0 +1 @@
+require_relative "ask/mcp"

metadata

@@ -0,0 +1,135 @@
+--- !ruby/object:Gem::Specification
+name: ask-mcp
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kaka Ruto
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httpx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: json-schema
+  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: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.25'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.25'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.1'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+description: Connect to MCP servers via stdio, SSE, and Streamable HTTP transports.
+  Discover tools, resources, and prompts. OAuth 2.1 authentication. Convert MCP tools
+  to Ask::Tool instances. Supports the full MCP spec.
+email:
+- kaka@myrrlabs.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE
+- README.md
+- docs/auth-setup.md
+- lib/ask-mcp.rb
+- lib/ask/mcp.rb
+- lib/ask/mcp/adapters/ask_tool.rb
+- lib/ask/mcp/auth/oauth.rb
+- lib/ask/mcp/auth/token.rb
+- lib/ask/mcp/client.rb
+- lib/ask/mcp/native/messages.rb
+- lib/ask/mcp/prompt.rb
+- lib/ask/mcp/resource.rb
+- lib/ask/mcp/server.rb
+- lib/ask/mcp/tool.rb
+- lib/ask/mcp/transport/sse.rb
+- lib/ask/mcp/transport/stdio.rb
+- lib/ask/mcp/transport/streamable_http.rb
+- lib/ask/mcp/validator.rb
+- lib/ask/mcp/version.rb
+homepage: https://github.com/ask-rb/ask-mcp
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/ask-rb/ask-mcp
+  source_code_uri: https://github.com/ask-rb/ask-mcp
+  changelog_uri: https://github.com/ask-rb/ask-mcp/blob/master/CHANGELOG.md
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.2'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.3
+specification_version: 4
+summary: Model Context Protocol (MCP) client for Ruby
+test_files: []