wip

lib/ruby_indexer/lib/ruby_indexer/index.rb

@@ -614,6 +614,26 @@ def existing_or_new_singleton_class(name)
       singleton
     end
 
+    # TODO: fix typing
+    # sig do
+    #   type_parameters(:T).params(
+    #     path: String,
+    #     type: T.nilable(T::Class[T.all(T.type_parameter(:T), Entry)]),
+    #   ).returns(T.nilable(T::Array[T.type_parameter(:T)]))
+    # end
+    def entries_for(path, type = nil)
+      type = Entry if type.nil?
+
+      # TODO: Make this work
+      # return @files_to_entries[path]&.grep(type)
+
+      # old way:
+      @files_to_entries[path].select do |entry|
+        # entry.ancestors.include?(RubyIndexer::Entry::Namespace) # FAILS
+        entry.class == RubyIndexer::Entry::Class || entry.class == RubyIndexer::Entry::Module
+      end
+    end
+
     private
 
     # Runs the registered included hooks

lib/ruby_indexer/test/index_test.rb

@@ -1822,5 +1822,21 @@ class Child < Namespace::Parent
         @index.linearized_ancestors_of("Foo::Child::<Class:Child>"),
       )
     end
+
+    def test_entries_for
+      index(<<~RUBY)
+        class Foo; end
+
+        module Bar
+          def my_def; end
+        end
+      RUBY
+
+      entries = @index.entries_for("/fake/path/foo.rb")
+      assert_equal(["Foo", "Bar", "my_def"], entries.map(&:name))
+
+      entries = @index.entries_for("/fake/path/foo.rb", RubyIndexer::Entry::Namespace)
+      assert_equal(["Foo", "Bar"], entries.map(&:name))
+    end
   end
 end