yardmcp 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a0b56fc156ab0bb9d90ec43b123ff1f04673beab24b2e99d08bcb10626c1246b
-  data.tar.gz: 7bdc59fcd90f8ff1e0ba894bbeef72f84db29b62e535aeb72d38247e2ac63b14
+  metadata.gz: 88e4b95aa833c7d4beb6cf789d921410f3ff5d0112a1a10456dff4943ec3561c
+  data.tar.gz: 751d12f0deedd21838b34ddf5cc1b4e36585888be3b5a899547d6bc4684462c4
 SHA512:
-  metadata.gz: 718150e2c6ea52f8dc5f7edadbc5f7109e70d19246927098011d2341f4d318425b72444dd62de52f48301dc2567a48904e18fbd20dd9eb45c096e2f782f70c94
-  data.tar.gz: 56c33f901052cdca43f904d48cec5e9f543da349170e0894d3915ff2e5a84b8f71bfa919158e81803e12be866afb1b134ef01e7ee30a6ccf8ccdba6069725b3e
+  metadata.gz: f9fbd30f73eaac7a36457ed89ffce04b87c6537641910c71c0f01aeac3bcfe700a8cb3c249c8c1f03f76d365028aec4e51fa355d1360ffaee18d1d0e404ee038
+  data.tar.gz: dea12928f18006a12bd44473f55f0e849b65530e9ee7b5dc0a79fab5a321e06368b20222958491080930857beb98e205bd90adee601a77140d9ea2a22571684c

data/lib/yardmcp/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module YardMCP
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

data/lib/yardmcp.rb

@@ -11,6 +11,7 @@ require_relative 'yardmcp/version'
 # Utility class for YARD operations
 class YardUtils
   include Singleton
+
   attr_reader :libraries, :logger, :object_to_gem
 
   def initialize
@@ -43,7 +44,7 @@ class YardUtils
   # Ensures the correct .yardoc is loaded for the given object path
   def ensure_yardoc_loaded_for_object!(object_path)
     # TODO: Handle multiple gems for the same object path, use some heuristic to determine the correct gem
-    gem_name = @object_to_gem[object_path].first
+    gem_name = @object_to_gem[object_path]&.first
     raise "No documentation found for #{object_path}" unless gem_name
 
     load_yardoc_for_gem(gem_name)
@@ -275,22 +276,50 @@ class YardUtils
       (libraries[spec.name] ||= []) << YARD::Server::LibraryVersion.new(spec.name, spec.version.to_s, nil, :gem)
     end
 
-    list_gems.each do |gem_name|
-      logger.debug "Loading #{gem_name}..."
-      begin
-        load_yardoc_for_gem(gem_name)
-      rescue StandardError => e
-        logger.error "Error loading #{gem_name}: #{e.message}"
-        next
-      end
-      YARD::Registry.all.each do |obj|
-        logger.debug "Adding #{obj.path} to #{gem_name}"
-        (@object_to_gem[obj.path.to_s] ||= []) << gem_name
-      end
+    begin
+      require 'parallel'
+      logger.info 'Using parallel gem for index building'
+
+      # Use processes to avoid YARD thread-safety issues
+      # Each process returns a hash of object_path => [gem_names]
+      results = Parallel.map(list_gems, in_processes: 8) { |gem| process_gem_for_index(gem) }
+      merge_gem_results(results)
+    rescue LoadError
+      logger.warn 'parallel gem not found, falling back to single-threaded processing'
+      results = list_gems.map { |gem| process_gem_for_index(gem) }
+      merge_gem_results(results)
     end
     logger.info "Index built: #{libraries.size} gems, #{@object_to_gem.size} objects"
   end
 
+  # Merge gem processing results into @object_to_gem
+  def merge_gem_results(results)
+    results.each do |gem_objects|
+      gem_objects.each do |obj_path, gem_names|
+        (@object_to_gem[obj_path] ||= []).concat(gem_names)
+      end
+    end
+  end
+
+  # Process a single gem and return its objects as a hash
+  def process_gem_for_index(gem_name)
+    logger.debug "Loading #{gem_name}..."
+    begin
+      load_yardoc_for_gem(gem_name)
+    rescue StandardError => e
+      logger.error "Error loading #{gem_name}: #{e.message}"
+      return {}
+    end
+
+    # Collect all objects for this gem
+    gem_objects = {}
+    YARD::Registry.all.each do |obj|
+      logger.debug "Adding #{obj.path} to #{gem_name}"
+      gem_objects[obj.path.to_s] = [gem_name]
+    end
+    gem_objects
+  end
+
   def build_docs(gem_name)
     logger.info "Building docs for #{gem_name}..."
     YARD::CLI::Gems.new.run(gem_name)
@@ -302,7 +331,8 @@ class ListGemsTool < FastMcp::Tool
   description 'List all installed gems that have a .yardoc file'
 
   def call
-    { content: YardUtils.instance.list_gems }
+    gems = YardUtils.instance.list_gems
+    { content: gems.map { |gem| { text: gem, type: 'gem' } } }
   end
 end
 
@@ -314,7 +344,8 @@ class ListClassesTool < FastMcp::Tool
   end
 
   def call(gem_name:)
-    { content: YardUtils.instance.list_classes(gem_name) }
+    classes = YardUtils.instance.list_classes(gem_name)
+    { content: classes.map { |cls| { text: cls, type: 'class' } } }
   end
 end
 
@@ -339,7 +370,8 @@ class ChildrenTool < FastMcp::Tool
   end
 
   def call(path:)
-    { content: YardUtils.instance.children(path) }
+    children = YardUtils.instance.children(path)
+    { content: children.map { |child| { text: child, type: 'child' } } }
   end
 end
 
@@ -351,7 +383,8 @@ class MethodsListTool < FastMcp::Tool
   end
 
   def call(path:)
-    { content: YardUtils.instance.methods_list(path) }
+    methods = YardUtils.instance.methods_list(path)
+    { content: methods.map { |method| { text: method, type: 'method' } } }
   end
 end
 
@@ -376,7 +409,8 @@ class SearchTool < FastMcp::Tool
 
   def call(query:)
     # Enhanced search: ranked, fuzzy, and full-text
-    { content: YardUtils.instance.search(query) }
+    results = YardUtils.instance.search(query)
+    { content: results.map { |result| { text: result[:path], score: result[:score], type: 'search_result' } } }
   end
 end
 
@@ -400,7 +434,8 @@ class CodeSnippetTool < FastMcp::Tool
   end
 
   def call(path:)
-    YardUtils.instance.code_snippet(path)
+    snippet = YardUtils.instance.code_snippet(path)
+    { content: { text: snippet, type: 'code_snippet' } }
   end
 end
 
@@ -412,7 +447,8 @@ class AncestorsTool < FastMcp::Tool
   end
 
   def call(path:)
-    { content: YardUtils.instance.ancestors(path) }
+    ancestors = YardUtils.instance.ancestors(path)
+    { content: ancestors.map { |ancestor| { text: ancestor, type: 'ancestor' } } }
   end
 end
 
@@ -429,7 +465,7 @@ class RelatedObjectsTool < FastMcp::Tool
 end
 
 module YardMCP
-  def self.start_server(preload: false)
+  def self.start_server(preload: true)
     YardUtils.instance if preload
     server = FastMcp::Server.new(name: 'yard-mcp-server', version: YardMCP::VERSION)
     server.register_tool(ListGemsTool)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yardmcp
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Roman Shterenzon