Add resource template

Author: codergeek121

Gemfile.lock

@@ -2,6 +2,7 @@ PATH
   remote: .
   specs:
     ruby_mcp (0.1.0)
+      addressable (~> 2.8, >= 2.8.7)
       zeitwerk (~> 2.7, >= 2.7.2)
 
 GEM
@@ -20,6 +21,8 @@ GEM
       securerandom (>= 0.3)
       tzinfo (~> 2.0, >= 2.0.5)
       uri (>= 0.13.1)
+    addressable (2.8.7)
+      public_suffix (>= 2.0.2, < 7.0)
     ast (2.4.3)
     base64 (0.2.0)
     benchmark (0.4.0)
@@ -53,6 +56,7 @@ GEM
     psych (5.2.3)
       date
       stringio
+    public_suffix (6.0.1)
     racc (1.8.1)
     rack (3.1.12)
     rainbow (3.1.1)

lib/ruby_mcp.rb

@@ -2,6 +2,7 @@
 require "securerandom"
 require "logger"
 require "zeitwerk"
+require "addressable"
 
 loader = Zeitwerk::Loader.for_gem
 loader.inflector.inflect("ruby_mcp" => "RubyMCP")

lib/ruby_mcp/capabilities/resources.rb

@@ -0,0 +1,9 @@
+module RubyMCP::Capabilities::Resources
+  def add_resource(...)
+    @resources.add(...)
+  end
+
+  def add_resource_template(uri_template:, name:, description:, mime_type:, reader:)
+    @resources.add_resource_template(uri_template:, name:, description:, mime_type:, reader:)
+  end
+end

lib/ruby_mcp/handlers/resources_read.rb

@@ -1,12 +1,15 @@
 class RubyMCP::Handlers::ResourcesRead
   def handle(server, request)
-    if resource = server.resources.find(request.uri)
+    resources = server.resources.find(request.uri)
+    if resources.any?
       server.answer(request,
-        contents: [ {
-          uri: resource.uri,
-          mimeType: resource.mime_type,
-          text: resource.reader.call(resource)
-        } ]
+        contents: resources.map do |resource|
+          {
+            uri: request.uri,
+            mimeType: resource.mime_type,
+            text: resource.reader.call(resource)
+          }
+        end
       )
     else
       server.error(

lib/ruby_mcp/resources.rb

@@ -1,17 +1,26 @@
 module RubyMCP
   Resource = Data.define(:uri, :name, :description, :mime_type, :reader)
+  ResourceTemplate = Data.define(:uri_template, :name, :description, :mime_type, :reader)
 
   class Resources
     def initialize
       @resources = {}
+      @resource_templates = {}
     end
 
     def add(uri:, name:, description: nil, mime_type: nil, reader: nil)
       @resources[uri] = Resource.new(uri, name, description, mime_type, reader)
     end
 
     def find(uri)
-      @resources[uri]
+      [ @resources[uri] ].concat(find_in_templates(uri).map(&:last)).compact
+    end
+
+    def add_resource_template(uri_template:, **args)
+      @resource_templates[uri_template] = ResourceTemplate.new(
+        uri_template: Addressable::Template.new(uri_template),
+        **args
+      )
     end
 
     def as_json
@@ -24,5 +33,14 @@ def as_json
         }
       end
     end
+
+    private
+
+    def find_in_templates(uri)
+      addressable_uri = Addressable::URI.parse(uri)
+      @resource_templates.find_all do |template, resource_template|
+        resource_template.uri_template.extract(addressable_uri)
+      end
+    end
   end
 end

lib/ruby_mcp/server.rb

@@ -1,6 +1,7 @@
 module RubyMCP
   class Server
     include Capabilities::Logging
+    include Capabilities::Resources
 
     attr_reader :lifecycle, :prompts, :resources
 

ruby_mcp.gemspec

@@ -28,5 +28,6 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = [ "lib" ]
 
-  spec.add_dependency "zeitwerk", '~> 2.7', '>= 2.7.2'
+  spec.add_dependency "zeitwerk", "~> 2.7", ">= 2.7.2"
+  spec.add_dependency "addressable", "~> 2.8", ">= 2.8.7"
 end

test/capabilities/test_resources_capability.rb

@@ -122,4 +122,41 @@ def test_read_resources_not_found
       }
     )
   end
+
+  def test_read_template
+    @server.add_resource_template(
+      uri_template: "https://{host}.de",
+      name: "german_website",
+      description: "Every german website",
+      mime_type: "text/html",
+      reader: ->(resource) {
+        "The first demo content of #{resource.name}"
+      }
+    )
+
+    @transport.client_message(
+      jsonrpc: "2.0",
+      id: 1,
+      method: "resources/read",
+      params: {
+        uri: "https://example.de"
+      }
+    )
+
+    @transport.process_message
+
+    assert_last_response(
+      jsonrpc: "2.0",
+      id: 1,
+      result: {
+        contents: [
+          {
+            uri: "https://example.de",
+            mimeType: "text/html",
+            text: "The first demo content of german_website"
+          }
+        ]
+      }
+    )
+  end
 end