legion-mcp 0.4.5 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: de51bbc257bd7eb0c3b573dc4ba04b908656b360c03e92b76d9c1a2aa6fdceeb
-  data.tar.gz: 68d9bdb57318c9d5fb3dc19d720be5b7f490f4c589bda8bfd3c77fd08850b964
+  metadata.gz: 2eff77018e5f5162f17ce2582bee76bff4e78d73e3129c9cf3adac4f40922bd1
+  data.tar.gz: fbc6a1ad1463750b1381a8d48e56b36c78a2166fd7514db74e0e1cba7a904df0
 SHA512:
-  metadata.gz: 9f76d6e46d90eef89f1c4e2536126501aba765bd8f9bad9436a2a90c77900a3116199c49b7c7b68cba701b74a25c81d7fb58b4fd2fb179a43a55bd4d8b767e4b
-  data.tar.gz: 1072848b1f4d835d0a9c15dc992435bd6a38aa6cac6d004d6760983c2e66565530026019b69ca613f2dac39eef6316f364421cd809aad2bbf3524f17294ef756
+  metadata.gz: 3e098a68af07e9af9c554fa77b116c2de2ed36a076ee332ff5bbb040d96957ec7f5afc390a4b4d743244b3b524a045b9390a19c4ed8ca2cddb279b17006e4a55
+  data.tar.gz: 051acdc2bc2c3164400932f98fdc5a79a52f283766c8832d672cfdc44bdeed79c12a38745b450dbb501afe3ced38361418d0bbd279fc6d63ff28b25600ba1d11

data/CHANGELOG.md

@@ -1,5 +1,15 @@
 # legion-mcp Changelog
 
+## [0.5.0] - 2026-03-23
+
+### Added
+- MCP client: `ServerRegistry` for static (settings) and dynamic (runtime) server registration with health tracking and cooldown-based recovery
+- MCP client: `Connection` class for stdio and HTTP transport connections with TTL-cached tool lists
+- MCP client: `Pool` for long-lived connection management, aggregates tools across all healthy servers
+- MCP client: `Client.boot` loads server registry from `Legion::Settings[:mcp][:servers]` at startup
+- MCP client: `Client.register` / `Client.deregister` for runtime server management
+- `Settings` module with defaults for `servers`, `overrides`, `tool_cache_ttl`, `connect_timeout`, `call_timeout`
+
 ## [0.4.5] - 2026-03-22
 
 ### Fixed

data/lib/legion/mcp/client/connection.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module Legion
+  module MCP
+    module Client
+      class Connection
+        attr_reader :name, :transport_type, :config
+
+        TOOL_CACHE_TTL = 300 # seconds
+
+        def initialize(name:, transport:, **config)
+          @name = name
+          @transport_type = transport.to_sym
+          @config = config
+          @tools_cache = nil
+          @tools_cached_at = nil
+          @connected = false
+          @mutex = Mutex.new
+        end
+
+        def connected?
+          @connected
+        end
+
+        def connect
+          @mutex.synchronize do
+            return if @connected
+
+            case @transport_type
+            when :stdio
+              connect_stdio
+            when :http, :streamable_http
+              connect_http
+            else
+              raise ArgumentError, "Unknown transport: #{@transport_type}"
+            end
+            @connected = true
+          end
+        rescue StandardError
+          @connected = false
+          raise
+        end
+
+        def disconnect
+          @mutex.synchronize do
+            @transport&.close if @transport.respond_to?(:close)
+            @connected = false
+            @tools_cache = nil
+          end
+        end
+
+        def tools(force_refresh: false)
+          @mutex.synchronize do
+            if !force_refresh && @tools_cache && @tools_cached_at &&
+               (Time.now - @tools_cached_at) < TOOL_CACHE_TTL
+              return @tools_cache
+            end
+
+            @tools_cache = fetch_tools
+            @tools_cached_at = Time.now
+            @tools_cache
+          end
+        end
+
+        def call_tool(name:, arguments: {})
+          connect unless connected?
+          execute_tool_call(name: name, arguments: arguments)
+        end
+
+        private
+
+        def connect_stdio
+          @transport = { type: :stdio, command: @config[:command], pid: nil }
+        end
+
+        def connect_http
+          @transport = { type: :http, url: @config[:url], auth: @config[:auth] }
+        end
+
+        def fetch_tools
+          connect unless connected?
+          []
+        end
+
+        def execute_tool_call(_name:, _arguments:)
+          connect unless connected?
+          { content: [], error: false }
+        end
+      end
+    end
+  end
+end

data/lib/legion/mcp/client/pool.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Legion
+  module MCP
+    module Client
+      module Pool
+        @connections = {}
+        @mutex = Mutex.new
+
+        module_function
+
+        def connection_for(server_name)
+          @mutex.synchronize do
+            return @connections[server_name] if @connections.key?(server_name)
+
+            config = ServerRegistry.servers[server_name]
+            return nil unless config
+
+            conn = Connection.new(name: server_name, **config.except(:registered_at, :source))
+            @connections[server_name] = conn
+          end
+        end
+
+        def all_tools
+          ServerRegistry.healthy_servers.flat_map do |name, _config|
+            conn = connection_for(name)
+            next [] unless conn
+
+            conn.tools.map do |tool|
+              tool.merge(source: { type: :mcp, server: name })
+            end
+          rescue StandardError => e
+            Legion::Logging.warn("MCP tool discovery failed for #{name}: #{e.message}") if defined?(Legion::Logging)
+            ServerRegistry.mark_unhealthy(name)
+            []
+          end
+        end
+
+        def reset!
+          @mutex.synchronize do
+            @connections.each_value { |c| c.disconnect rescue nil } # rubocop:disable Style/RescueModifier
+            @connections.clear
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/mcp/client/server_registry.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Legion
+  module MCP
+    module Client
+      module ServerRegistry
+        @servers = {}
+        @health = {}
+        @mutex = Mutex.new
+
+        module_function
+
+        def load_from_settings(settings_hash)
+          @mutex.synchronize do
+            settings_hash.each do |name, config|
+              @servers[name] = config.merge(registered_at: Time.now, source: :settings)
+              @health[name] = { healthy: true, last_check: Time.now }
+            end
+          end
+        end
+
+        def register(name, **config)
+          @mutex.synchronize do
+            @servers[name] = config.merge(registered_at: Time.now, source: :dynamic)
+            @health[name] = { healthy: true, last_check: Time.now }
+          end
+        end
+
+        def deregister(name)
+          @mutex.synchronize do
+            @servers.delete(name)
+            @health.delete(name)
+          end
+        end
+
+        def servers
+          @mutex.synchronize { @servers.dup }
+        end
+
+        def healthy_servers
+          @mutex.synchronize do
+            @servers.select do |name, _|
+              h = @health[name]
+              next true if h.nil? || h[:healthy]
+
+              cooldown = h[:cooldown] || 60
+              if Time.now - (h[:marked_at] || Time.now) >= cooldown
+                h[:healthy] = true
+                true
+              else
+                false
+              end
+            end
+          end
+        end
+
+        def mark_unhealthy(name, cooldown: 60)
+          @mutex.synchronize do
+            @health[name] = {
+              healthy:    false,
+              marked_at:  Time.now,
+              cooldown:   cooldown,
+              last_check: Time.now
+            }
+          end
+        end
+
+        def mark_healthy(name)
+          @mutex.synchronize do
+            @health[name] = { healthy: true, last_check: Time.now }
+          end
+        end
+
+        def reset!
+          @mutex.synchronize do
+            @servers.clear
+            @health.clear
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/mcp/client.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Legion
+  module MCP
+    module Client
+      module_function
+
+      def boot
+        servers = Legion::Settings.dig(:mcp, :servers) rescue nil # rubocop:disable Style/RescueModifier
+        return unless servers.is_a?(Hash) && servers.any?
+
+        ServerRegistry.load_from_settings(servers)
+        Legion::Logging.info("MCP Client: #{servers.length} servers registered") if defined?(Legion::Logging)
+      end
+
+      def shutdown
+        Pool.reset!
+        ServerRegistry.reset!
+      end
+
+      def register(name, **config)
+        ServerRegistry.register(name, **config)
+      end
+
+      def deregister(name)
+        Pool.reset!
+        ServerRegistry.deregister(name)
+      end
+    end
+  end
+end
+
+require_relative 'client/server_registry'
+require_relative 'client/connection'
+require_relative 'client/pool'

data/lib/legion/mcp/settings.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Legion
+  module MCP
+    module Settings
+      module_function
+
+      def defaults
+        {
+          servers:         {},
+          overrides:       {},
+          tool_cache_ttl:  300,
+          connect_timeout: 10,
+          call_timeout:    30
+        }
+      end
+    end
+  end
+end

data/lib/legion/mcp/version.rb

@@ -2,6 +2,6 @@
 
 module Legion
   module MCP
-    VERSION = '0.4.5'
+    VERSION = '0.5.0'
   end
 end

data/lib/legion/mcp.rb

@@ -4,9 +4,11 @@ require 'mcp'
 require 'legion/json'
 require_relative 'mcp/version'
 
+require_relative 'mcp/settings'
 require_relative 'mcp/auth'
 require_relative 'mcp/tool_governance'
 require_relative 'mcp/server'
+require_relative 'mcp/client'
 
 module Legion
   module MCP

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legion-mcp
 version: !ruby/object:Gem::Version
-  version: 0.4.5
+  version: 0.5.0
 platform: ruby
 authors:
 - Esity
@@ -119,6 +119,10 @@ files:
 - lib/legion/mcp.rb
 - lib/legion/mcp/auth.rb
 - lib/legion/mcp/capability_generator.rb
+- lib/legion/mcp/client.rb
+- lib/legion/mcp/client/connection.rb
+- lib/legion/mcp/client/pool.rb
+- lib/legion/mcp/client/server_registry.rb
 - lib/legion/mcp/cold_start.rb
 - lib/legion/mcp/context_compiler.rb
 - lib/legion/mcp/context_guard.rb
@@ -133,6 +137,7 @@ files:
 - lib/legion/mcp/resources/extension_info.rb
 - lib/legion/mcp/resources/runner_catalog.rb
 - lib/legion/mcp/server.rb
+- lib/legion/mcp/settings.rb
 - lib/legion/mcp/tier_router.rb
 - lib/legion/mcp/tool_governance.rb
 - lib/legion/mcp/tools/ask_peer.rb