[Refactor] Drop dead code + centralize tool response helpers

Author: djalmaaraujo

mcp/lib/ruby_ui/mcp/rack_app.rb

@@ -6,18 +6,8 @@
 module RubyUI
   module MCP
     class RackApp
-      class << self
-        def call(env)
-          instance.call(env)
-        end
-
-        def instance
-          @instance ||= new
-        end
-
-        def reset!
-          @instance = nil
-        end
+      def self.call(env)
+        (@instance ||= new).call(env)
       end
 
       def initialize(registry: RubyUI::MCP.registry)

mcp/lib/ruby_ui/mcp/registry.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require "json"
-require "set"
 
 module RubyUI
   module MCP
@@ -22,22 +21,17 @@ def self.load(path)
         new(raw)
       end
 
-      attr_reader :version, :generated_at
+      attr_reader :version
 
       def initialize(raw)
         @version = raw[:version]
-        @generated_at = raw[:generated_at]
         @components = raw[:components] || {}
       end
 
       def list
         @components.values.map { |c| {name: c[:name], description: c[:description]} }
       end
 
-      def all
-        @components.values
-      end
-
       def find(name)
         key = name.to_s.downcase.to_sym
         @components[key]

mcp/lib/ruby_ui/mcp/server.rb

@@ -1,6 +1,5 @@
 # frozen_string_literal: true
 
-require "json"
 require "mcp"
 require "ruby_ui/mcp/registry"
 require "ruby_ui/mcp/tools/get_project_registries"
@@ -109,15 +108,7 @@ def build_tool_class(definition)
           description: definition[:description],
           input_schema: definition[:input_schema]
         ) do |server_context: nil, **args|
-          payload = impl.call(**args)
-          ::MCP::Tool::Response.new([
-            {type: "text", text: JSON.pretty_generate(payload)}
-          ])
-        rescue => e
-          ::MCP::Tool::Response.new(
-            [{type: "text", text: "error: #{e.class}: #{e.message}"}],
-            error: true
-          )
+          impl.respond { impl.call(**args) }
         end
       end
     end

mcp/lib/ruby_ui/mcp/tools/base.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require "json"
+
 module RubyUI
   module MCP
     module Tools
@@ -11,6 +13,17 @@ def initialize(registry:)
         def call(**args)
           raise NotImplementedError
         end
+
+        def gem_version
+          @registry&.version
+        end
+
+        def respond
+          payload = yield
+          ::MCP::Tool::Response.new([{type: "text", text: ::JSON.pretty_generate(payload)}])
+        rescue => e
+          ::MCP::Tool::Response.new([{type: "text", text: "error: #{e.class}: #{e.message}"}], error: true)
+        end
       end
     end
   end

mcp/lib/ruby_ui/mcp/tools/get_add_command_for_items.rb

@@ -15,7 +15,7 @@ def call(items:, **)
             components: known,
             unresolved: unresolved,
             command_string: known.empty? ? "" : "rails g #{GENERATOR} #{known.join(" ")}",
-            gem_version: @registry.version
+            gem_version: gem_version
           }
         end
       end

mcp/lib/ruby_ui/mcp/tools/get_item_examples_from_registries.rb

@@ -11,7 +11,7 @@ def call(items:, **)
             c = @registry.find(n)
             c ? {name: c[:name], examples: c[:examples] || []} : nil
           end.compact
-          {items: resolved, gem_version: @registry.version}
+          {items: resolved, gem_version: gem_version}
         end
       end
     end

mcp/lib/ruby_ui/mcp/tools/list_items_in_registries.rb

@@ -7,7 +7,7 @@ module MCP
     module Tools
       class ListItemsInRegistries < Base
         def call(**)
-          {items: @registry.list, gem_version: @registry.version}
+          {items: @registry.list, gem_version: gem_version}
         end
       end
     end

mcp/lib/ruby_ui/mcp/tools/search_items_in_registries.rb

@@ -7,7 +7,7 @@ module MCP
     module Tools
       class SearchItemsInRegistries < Base
         def call(query:, limit: 10, **)
-          {items: @registry.search(query, limit: limit), gem_version: @registry.version}
+          {items: @registry.search(query, limit: limit), gem_version: gem_version}
         end
       end
     end

mcp/lib/ruby_ui/mcp/tools/view_items_in_registries.rb

@@ -13,7 +13,7 @@ def call(items:, **)
             comp = @registry.find(name)
             comp ? resolved << comp : unresolved << name
           end
-          {items: resolved, unresolved: unresolved, gem_version: @registry.version}
+          {items: resolved, unresolved: unresolved, gem_version: gem_version}
         end
       end
     end

mcp/test/fixtures/registry.json

@@ -1,6 +1,5 @@
 {
   "version": "1.2.0",
-  "generated_at": "2026-05-09T00:00:00Z",
   "components": {
     "button": {
       "name": "Button",