stack-service-base 0.0.97 → 0.0.98

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a29686255562ea2b36806d5d50db270dda9c65d6b9d819f0f9c241a3a4f85049
-  data.tar.gz: 2e33c64c8f927a7ac034c39d2b2ea1b5118bedb3315d44d2265f80d463fafdad
+  metadata.gz: 4637e888970a11131f20239dd6b2e64c661d40daafe1942cdd2193d6b05f495e
+  data.tar.gz: a9d68103d79f1ab23a0c9cac1c2366b42238a1e5006b8c716b126e791966cf08
 SHA512:
-  metadata.gz: 0f8d15d2b3e8c33e8e321464df2068351da9e6fdb5b0439405635ed7876f4fb014d9525284a87c6237ee13c4731cc177ebfd0f5328f9d44fcc8a32b36f811ff9
-  data.tar.gz: ea9223839e6ccabe433121647be6a8c93ce62d6d807c482aad9c3a1fdf48b18a226a8d7eb1449b71c7ab93e41c4c390f7379d092e6403de8136d0249e3ce4647
+  metadata.gz: 747b9ae740e563c28057aaee05321183280a44152ac748fe949e2d04e9077b5bc8c66add36340daa79759cbce31a8f75dfb3da0811745aed259c35e3370a7059
+  data.tar.gz: a70bdbb1c4915a65db80745a470c905114add9e35df94c7ea4b9a3b057abfee7af29488c64b612cf9ed0fc609b0f98a3a402509583dda244d7d08ad5157dba9f

data/lib/stack-service-base/examples/mcp_config.ru

@@ -32,6 +32,35 @@ Tool :fetch do
   end
 end
 
+Tool :schema_echo do
+  description 'Echo a value using a direct JSON schema'
+  input_schema type: "object",
+               properties: {
+                 value: { type: "string", description: "Value to echo" }
+               },
+               required: ["value"]
+  annotations readOnlyHint: true
+  call do |inputs|
+    { value: inputs[:value] }
+  end
+end
+
+Tool :full_response_echo do
+  description 'Echo a value using a complete MCP tool response'
+  input value: { type: "string", description: "Value to echo", required: true }
+  call do |inputs|
+    {
+      content: [
+        { type: "text", text: inputs[:value] }
+      ],
+      structuredContent: {
+        value: inputs[:value]
+      },
+      isError: false
+    }
+  end
+end
+
 Tool :service_status do
   description 'Check current status of a service'
   input service_name: { type: "string", description: "Service name to inspect", required: true }

data/lib/stack-service-base/mcp/mcp_helper.rb

@@ -4,6 +4,21 @@ require_relative 'mcp_tool_registry'
 MCP_PROCESSOR = McpProcessor.new
 
 module McpHelper
+  VALID_TRANSPORTS = [:sse, :json].freeze
+
+  class << self
+    def transport
+      @transport ||= :sse
+    end
+
+    def transport=(value)
+      value = value.to_sym
+      raise ArgumentError, "Unknown MCP transport: #{value}" unless VALID_TRANSPORTS.include?(value)
+
+      @transport = value
+    end
+  end
+
   def self.included(base)
     base.class_eval do
 
@@ -18,28 +33,34 @@ module McpHelper
       end
 
       post '/mcp' do
-        content_type 'text/event-stream'
-        headers['Cache-Control'] = 'no-cache'
-        headers['X-Accel-Buffering'] = 'no'
-        headers['mcp-session-id'] = SecureRandom.uuid
         request.body&.rewind
-        body = request.body.read.to_s
 
         response_body =
           begin
-            MCP_PROCESSOR.rpc_endpoint(body)
+            MCP_PROCESSOR.rpc_endpoint(request.body.read.to_s)
           rescue McpProcessor::ParseError => e
             status e.status
             e.body
           end
 
-        LOGGER.debug "request body: #{body}"
-        LOGGER.debug "response body: #{response_body}"
+        if response_body.nil?
+          status 202
+          headers 'Content-Length' => '0'
+          ''
+        elsif McpHelper.transport == :json
+          content_type :json
+          response_body
+        else
+          content_type 'text/event-stream'
+          headers['Cache-Control'] = 'no-cache'
+          headers['X-Accel-Buffering'] = 'no'
+          headers['mcp-session-id'] = SecureRandom.uuid
 
-        stream true do |s|
-          s.callback { LOGGER.debug "stream closed: #{s}" }
-          s << "event: message\ndata: #{response_body}\n\n"
-          s.close
+          stream true do |s|
+            s.callback { LOGGER.debug "stream closed: #{s}" }
+            s << "event: message\ndata: #{response_body}\n\n"
+            s.close
+          end
         end
       end
     end

data/lib/stack-service-base/mcp/mcp_processor.rb

@@ -1,3 +1,5 @@
+require 'json'
+
 class JsonRpcError < StandardError
   attr_reader :code
 
@@ -15,6 +17,11 @@ end
 
 class McpProcessor
   PROTOCOL_VERSION = '2025-06-18'
+  DEFAULT_SERVER_INFO = {
+    name: 'mcp-server',
+    title: 'MCP Server',
+    version: '1.0.0'
+  }.freeze
 
   include RpcErrorHelpers
   ParseError = Class.new(StandardError) do
@@ -27,7 +34,9 @@ class McpProcessor
     end
   end
 
-  def initialize(logger: LOGGER)
+  def initialize(registry: nil, server_info: DEFAULT_SERVER_INFO, logger: (defined?(LOGGER) ? LOGGER : nil))
+    @registry = registry
+    @server_info = server_info
     @logger = logger
   end
 
@@ -37,14 +46,22 @@ class McpProcessor
 
   def rpc_endpoint(raw_body)
     req = JSON.parse(raw_body.to_s)
-    rpc_response(id: req["id"], method: req["method"], params: req["params"])
-  rescue JSON::ParserError
+    method = req["method"]
+    params = req["params"]
+
+    if req.key?("id")
+      rpc_response(id: req["id"], method: method, params: params)
+    else
+      notification_response(method: method, params: params)
+    end
+  rescue JSON::ParserError => e
+    @logger&.warn("MCP JSON parse failed: #{e.message}")
     body = error_response(id: nil, code: -32700, message: "Parse error")
     raise ParseError.new(body: body, status: 400)
   end
 
   def list_tools
-    { tools: ToolRegistry.list, nextCursor: 'no-more' }
+    { tools: registry.list, nextCursor: 'no-more' }
   end
 
   def root_response
@@ -56,32 +73,47 @@ class McpProcessor
   end
 
   def rpc_response(id:, method:, params:)
-    json_rpc_response(id: id) { |body| handle(method: method, params: params, body: body) }
+    json_rpc_response(id: id) { handle(method: method, params: params) }
+  end
+
+  def notification_response(method:, params:)
+    handle_notification(method: method, params: params)
+    nil
+  rescue => e
+    @logger&.error("Unhandled MCP notification error: #{e.class}: #{e.message}")
+    nil
   end
 
-  def handle(method:, params:, body: )
+  def handle(method:, params:)
     case method
     when "tools/list"  then list_tools
     # when "resources/list"  then {}
     # when "prompts/list"  then {}
     when "tools/call"  then call_tool(params || {})
-    when "initialize"  then initialize_(body)
-    when "notifications/initialized" then LOGGER.debug params; {}
-    when "logging/setLevel" then LOGGER.debug params; {}
+    when "initialize"  then initialize_response
+    when "notifications/initialized" then @logger&.debug(params); {}
+    when "logging/setLevel" then @logger&.debug(params); {}
     else
       rpc_error!(-32601, "Unknown method #{method}")
     end
   end
 
-  # https://gist.github.com/ruvnet/7b6843c457822cbcf42fc4aa635eadbb
+  def handle_notification(method:, params:)
+    case method
+    when "notifications/initialized", "notifications/cancelled"
+      @logger&.debug("MCP notification accepted: #{method}")
+    else
+      @logger&.debug("MCP notification ignored: #{method}")
+    end
+  end
 
-  def initialize_(body)
+  def initialize_(_body = nil)
+    initialize_response
+  end
+
+  def initialize_response
     {
-      serverInfo: {
-        name: 'mcp-server',
-        title: 'MCP Server',
-        version: '1.0.0'
-      },
+      serverInfo: @server_info,
       protocolVersion: PROTOCOL_VERSION,
       capabilities: {
         logging: {},
@@ -98,7 +130,7 @@ class McpProcessor
     body = { jsonrpc: "2.0", id: id }
 
     begin
-      result = yield(body)
+      result = yield
       body[:result] = result unless body[:error] || result.nil?
     rescue JsonRpcError => e
       body[:error] = { code: e.code, message: e.message }
@@ -114,11 +146,29 @@ class McpProcessor
   def call_tool(params)
     name      = params["name"]
     arguments = params["arguments"] || {}
-    tool      = ToolRegistry.fetch(name) || rpc_error!(-32601, "Unknown tool #{name}")
+    tool      = registry.fetch(name) || rpc_error!(-32601, "Unknown tool #{name}")
     response = tool.call_tool(arguments)
+    return response if mcp_tool_response?(response)
+
+    wrap_tool_response(response)
+  end
+
+  def registry
+    @registry || ToolRegistry.default
+  end
+
+  def mcp_tool_response?(response)
+    return false unless response.is_a?(Hash)
+
+    [:content, :structuredContent, :isError, "content", "structuredContent", "isError"].any? do |key|
+      response.key?(key)
+    end
+  end
+
+  def wrap_tool_response(response)
     {
       content: [
-        { "type": "text", "text": response.is_a?(String) ? response : response.to_json }
+        { "type": "text", "text": response.is_a?(String) ? response : JSON.dump(response) }
       ],
       isError: false
     }

data/lib/stack-service-base/mcp/mcp_tool_registry.rb

@@ -3,12 +3,35 @@ module ToolRegistry
     include RpcErrorHelpers
   end
 
+  class Registry
+    attr_reader :registry
+
+    def initialize
+      @registry = {}
+    end
+
+    def define(name, &block)
+      definition = Definition.new(name)
+      definition.instance_eval(&block)
+      @registry[definition.name] = definition
+    end
+
+    def list
+      registry.values.map(&:to_h)
+    end
+
+    def fetch(name)
+      registry[name.to_s]
+    end
+  end
+
   class Definition
     attr_reader :name
 
     def initialize(name)
       @name = name.to_s
       @inputs = {}
+      @annotations = {}
     end
 
     def description(text = nil)
@@ -20,62 +43,89 @@ module ToolRegistry
       @inputs.merge!(fields.transform_keys(&:to_s))
     end
 
-    def input_schema
-      properties = {}
-      required = []
+    def input_schema(schema = nil)
+      return @input_schema || build_input_schema if schema.nil?
 
-      @inputs.each do |field, config|
-        cfg = config.transform_keys { |key| key.is_a?(Symbol) ? key : key.to_sym }
-        required << field if cfg.delete(:required)
-        properties[field] = cfg.transform_keys(&:to_s)
-      end
+      @input_schema = stringify_keys(schema)
+    end
 
-      { type: 'object', properties: properties, required: required }
+    def annotations(value = nil)
+      return @annotations if value.nil?
+
+      @annotations = stringify_keys(value)
     end
 
     def call(&block) = @executor = block
 
     def to_h
-      {
+      payload = {
         name: name,
         description: description,
         inputSchema: input_schema
       }
+      payload[:annotations] = annotations unless annotations.empty?
+      payload
     end
 
     def call_tool(arguments)
-      raise JsonRpcError.new(code: 500, message: "Tool #{name} missing executor") unless @executor
+      raise JsonRpcError.new(code: -32603, message: "Tool #{name} missing executor") unless @executor
 
       ExecutionContext.new.instance_exec(symbolize_keys(arguments || {}), &@executor)
     end
 
     private
 
+    def build_input_schema
+      properties = {}
+      required = []
+
+      @inputs.each do |field, config|
+        cfg = stringify_keys(config)
+        required << field if cfg.delete("required")
+        properties[field] = cfg
+      end
+
+      { type: 'object', properties: properties, required: required }
+    end
+
     def symbolize_keys(hash)
       hash.each_with_object({}) do |(key, value), memo|
         memo[key.to_sym] = value
       end
     end
+
+    def stringify_keys(value)
+      case value
+      when Hash
+        value.each_with_object({}) { |(key, item), memo| memo[key.to_s] = stringify_keys(item) }
+      when Array
+        value.map { |item| stringify_keys(item) }
+      else
+        value
+      end
+    end
   end
 
   module_function
 
+  def default
+    @default ||= Registry.new
+  end
+
   def define(name, &block)
-    definition = Definition.new(name)
-    definition.instance_eval(&block)
-    registry[definition.name] = definition
+    default.define(name, &block)
   end
 
   def registry
-    @registry ||= {}
+    default.registry
   end
 
   def list
-    registry.values.map(&:to_h)
+    default.list
   end
 
   def fetch(name)
-    registry[name.to_s]
+    default.fetch(name)
   end
 end
 

data/lib/stack-service-base/stack_template/gitlab-c/.gitlab-ci.yml

@@ -1,3 +1,4 @@
+# Long ....
 stages:
   - deploy
 

data/lib/stack-service-base/version.rb

@@ -1,3 +1,3 @@
 module StackServiceBase
-  VERSION = '0.0.97'
+  VERSION = '0.0.98'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stack-service-base
 version: !ruby/object:Gem::Version
-  version: 0.0.97
+  version: 0.0.98
 platform: ruby
 authors:
 - Artyom B