Load type information of required gems

Author: tompng

lib/repl_type_completor/types.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'rbs'
+require 'rubygems'
 require 'rbs/cli'
 require_relative 'methods'
 
@@ -25,7 +26,13 @@ def self.preload_rbs_builder
     def self.load_rbs_builder
       @load_started = true
       loader = RBS::CLI::LibraryOptions.new.loader
-      loader.add path: Pathname('sig')
+      sig_path = Pathname('sig')
+      loader.add path: sig_path
+      expanded_sig_path = sig_path.expand_path.to_s
+      Gem.loaded_specs.values.each do |spec|
+        gem_sig_path = File.expand_path("#{spec.gem_dir}/sig")
+        loader.add(library: spec.name, version: spec.version) if Dir.exist?(gem_sig_path) && expanded_sig_path != gem_sig_path
+      end
       @rbs_builder = RBS::DefinitionBuilder.new env: RBS::Environment.from_loader(loader).resolve_type_names
     rescue LoadError, StandardError => e
       @rbs_load_error = e

test/repl_type_completor/test_repl_type_completor.rb

@@ -236,6 +236,13 @@ def test_doc_namespace_error
       ReplTypeCompletor.instance_variable_set(:@last_completion_error, nil)
     end
 
+    def test_loaded_gem_types
+      result = ReplTypeCompletor.analyze 'Prism.parse("code").', binding: binding
+      candidtes = result.completion_candidates
+      assert_includes candidtes, 'success?'
+      assert_includes candidtes, 'failure?'
+    end
+
     def test_info
       assert_equal "ReplTypeCompletor: #{ReplTypeCompletor::VERSION}, Prism: #{Prism::VERSION}, RBS: #{RBS::VERSION}", ReplTypeCompletor.info
     end