@@ -42,17 +42,17 @@ class CodeLens < ::RubyLsp::Listener
4242 sig { override . returns ( ResponseType ) }
4343 attr_reader :_response
4444
45- sig { params ( uri : URI ::Generic , emitter : EventEmitter , message_queue : Thread ::Queue ) . void }
46- def initialize ( uri , emitter , message_queue )
45+ sig { params ( uri : URI ::Generic , dispatcher : Prism :: Dispatcher , message_queue : Thread ::Queue ) . void }
46+ def initialize ( uri , dispatcher , message_queue )
4747 @_response = T . let ( [ ] , ResponseType )
4848 @path = T . let ( uri . to_standardized_path , T . nilable ( String ) )
49- emitter . register ( self , :on_call , :on_class , :on_def )
49+ dispatcher . register ( self , :on_call_node_enter , :on_class_node_enter , :on_def_node_enter )
5050
51- super ( emitter , message_queue )
51+ super ( dispatcher , message_queue )
5252 end
5353
54- sig { params ( node : YARP ::CallNode ) . void }
55- def on_call ( node )
54+ sig { params ( node : Prism ::CallNode ) . void }
55+ def on_call_node_enter ( node )
5656 message_value = node . message
5757 return unless message_value == "test"
5858
@@ -62,15 +62,15 @@ def on_call(node)
6262 first_argument = arguments . first
6363
6464 content = case first_argument
65- when YARP ::StringConcatNode
65+ when Prism ::StringConcatNode
6666 left = first_argument . left
6767 right = first_argument . right
6868 # We only support two lines of concatenation on test names
69- if left . is_a? ( YARP ::StringNode ) &&
70- right . is_a? ( YARP ::StringNode )
69+ if left . is_a? ( Prism ::StringNode ) &&
70+ right . is_a? ( Prism ::StringNode )
7171 left . content + right . content
7272 end
73- when YARP ::StringNode
73+ when Prism ::StringNode
7474 first_argument . content
7575 end
7676
@@ -82,8 +82,8 @@ def on_call(node)
8282 end
8383
8484 # Although uncommon, Rails tests can be written with the classic "def test_name" syntax.
85- sig { params ( node : YARP ::DefNode ) . void }
86- def on_def ( node )
85+ sig { params ( node : Prism ::DefNode ) . void }
86+ def on_def_node_enter ( node )
8787 method_name = node . name . to_s
8888 if method_name . start_with? ( "test_" )
8989 line_number = node . location . start_line
@@ -92,8 +92,8 @@ def on_def(node)
9292 end
9393 end
9494
95- sig { params ( node : YARP ::ClassNode ) . void }
96- def on_class ( node )
95+ sig { params ( node : Prism ::ClassNode ) . void }
96+ def on_class_node_enter ( node )
9797 class_name = node . constant_path . slice
9898 if class_name . end_with? ( "Test" )
9999 command = "#{ BASE_COMMAND } #{ @path } "
@@ -103,7 +103,7 @@ def on_class(node)
103103
104104 private
105105
106- sig { params ( node : YARP ::Node , name : String , command : String , kind : Symbol ) . void }
106+ sig { params ( node : Prism ::Node , name : String , command : String , kind : Symbol ) . void }
107107 def add_test_code_lens ( node , name :, command :, kind :)
108108 return unless @path
109109
0 commit comments