Make module tracking lazy

Shopify · Apr 25, 2024 · c3b0752 · c3b0752
1 parent 1639dd4
commit c3b0752
Showing 1 changed file with 9 additions and 19 deletions.
diff --git a/lib/ruby_indexer/lib/ruby_indexer/entry.rb b/lib/ruby_indexer/lib/ruby_indexer/entry.rb
@@ -60,29 +60,19 @@ class Namespace < Entry
 
       abstract!
 
-      sig { returns(T::Array[String]) }
-      attr_accessor :included_modules
+      sig { returns(String) }
+      def short_name
+        T.must(@name.split("::").last)
+      end
 
       sig { returns(T::Array[String]) }
-      attr_accessor :prepended_modules
-
-      sig do
-        params(
-          name: String,
-          file_path: String,
-          location: T.any(Prism::Location, RubyIndexer::Location),
-          comments: T::Array[String],
-        ).void
-      end
-      def initialize(name, file_path, location, comments)
-        super(name, file_path, location, comments)
-        @included_modules = T.let([], T::Array[String])
-        @prepended_modules = T.let([], T::Array[String])
+      def included_modules
+        @included_modules ||= T.let([], T.nilable(T::Array[String]))
       end
 
-      sig { returns(String) }
-      def short_name
-        T.must(@name.split("::").last)
+      sig { returns(T::Array[String]) }
+      def prepended_modules
+        @prepended_modules ||= T.let([], T.nilable(T::Array[String]))
       end
     end