Refactor capability check

lib/mcp/methods.rb

@@ -47,38 +47,34 @@ module MCP
       def ensure_capability!(method, capabilities)
         case method
         when PROMPTS_GET, PROMPTS_LIST
-          unless capabilities[:prompts]
+          require_capability!(method, capabilities, :prompts)
-            raise MissingRequiredCapabilityError.new(method, :prompts)
-          end
         when RESOURCES_LIST, RESOURCES_TEMPLATES_LIST, RESOURCES_READ, RESOURCES_SUBSCRIBE, RESOURCES_UNSUBSCRIBE
-          unless capabilities[:resources]
+          require_capability!(method, capabilities, :resources)
-            raise MissingRequiredCapabilityError.new(method, :resources)
-          end
-
           if method == RESOURCES_SUBSCRIBE && !capabilities[:resources][:subscribe]
             raise MissingRequiredCapabilityError.new(method, :resources_subscribe)
           end
         when TOOLS_CALL, TOOLS_LIST
-          unless capabilities[:tools]
+          require_capability!(method, capabilities, :tools)
-            raise MissingRequiredCapabilityError.new(method, :tools)
-          end
         when SAMPLING_CREATE_MESSAGE
-          unless capabilities[:sampling]
+          require_capability!(method, capabilities, :sampling)
-            raise MissingRequiredCapabilityError.new(method, :sampling)
-          end
         when COMPLETION_COMPLETE
-          unless capabilities[:completions]
+          require_capability!(method, capabilities, :completions)
-            raise MissingRequiredCapabilityError.new(method, :completions)
-          end
         when LOGGING_SET_LEVEL
-          # Logging is unsupported by the Server
+          require_capability!(method, capabilities, :logging)
-          unless capabilities[:logging]
-            raise MissingRequiredCapabilityError.new(method, :logging)
-          end
         when INITIALIZE, PING
           # No specific capability required for initialize or ping
         end
       end
+
+      private
+
+      def require_capability!(method, capabilities, *keys)
+        name = keys.join(".") # :resources, :subscribe -> "resources.subscribe"
+        has_capability = capabilities.dig(*keys)
+        return if has_capability
+
+        raise MissingRequiredCapabilityError.new(method, name)
+      end
     end
   end
 end