Add shim for @index

Shopify · Oct 31, 2024 · 6698760 · 6698760
1 parent 144a687
commit 6698760
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/lib/ruby_lsp/ruby_lsp_rails/indexing_enhancement.rb b/lib/ruby_lsp/ruby_lsp_rails/indexing_enhancement.rb
@@ -6,12 +6,6 @@ module Rails
     class IndexingEnhancement < RubyIndexer::Enhancement
       extend T::Sig
 
-      sig { params(index: RubyIndexer::Index).void }
-      def initialize(index)
-        super
-        @index = index
-      end
-
       sig do
         override.params(
           owner: T.nilable(RubyIndexer::Entry::Namespace),

diff --git a/sorbet/rbi/shims/ruby_lsp.rbi b/sorbet/rbi/shims/ruby_lsp.rbi
@@ -0,0 +1,11 @@
+# typed: strict
+# frozen_string_literal: true
+
+module RubyIndexer
+  class Enhancement
+    # If we change ruby-lsp to use a `T.let` then this can be removed
+    def initialize(index)
+      @index = T.let(index, RubyIndexer::Index)
+    end
+  end
+end