lib/tapioca/gemfile.rb in tapioca-0.9.4 vs lib/tapioca/gemfile.rb in tapioca-0.10.0

@@ -134,10 +134,26 @@
 
     class GemSpec
       extend(T::Sig)
       include GemHelper
 
+      class << self
+        extend T::Sig
+
+        sig { returns(T::Hash[String, Gemfile::GemSpec]) }
+        def spec_lookup_by_file_path
+          @lookup ||= T.let(
+            [*::Gem::Specification.default_stubs, *::Gem::Specification.stubs]
+              .map! { |spec| new(spec.to_spec) }
+              .flat_map do |spec|
+                spec.files.filter_map { |file| [file.realpath.to_s, spec] if file.exist? }
+              end.to_h,
+            T.nilable(T::Hash[String, Gemfile::GemSpec])
+          )
+        end
+      end
+
       IGNORED_GEMS = T.let(
         [
           "sorbet", "sorbet-static", "sorbet-runtime", "sorbet-static-and-runtime",
           "debug", "fakefs",
         ].freeze,
@@ -158,10 +174,15 @@
         @version = T.let(version_string, String)
         @exported_rbi_files = T.let(nil, T.nilable(T::Array[String]))
         @files = T.let(collect_files, T::Array[Pathname])
       end
 
+      sig { params(other: BasicObject).returns(T::Boolean) }
+      def ==(other)
+        GemSpec === other && other.name == name && other.version == version
+      end
+
       sig { params(gemfile_dir: String).returns(T::Boolean) }
       def ignore?(gemfile_dir)
         gem_ignored? || gem_in_app_dir?(gemfile_dir, full_gem_path)
       end
 
@@ -207,9 +228,18 @@
           rbi = RBI::Parser.parse_file(file)
           rewriter.merge(rbi)
         end
 
         rewriter.tree
+      end
+
+      sig { params(file: Pathname).returns(Pathname) }
+      def relative_path_for(file)
+        if default_gem?
+          file.realpath.relative_path_from(RbConfig::CONFIG["rubylibdir"])
+        else
+          file.realpath.relative_path_from(full_gem_path)
+        end
       end
 
       private
 
       sig { returns(T::Array[Pathname]) }