Fix parse error for delegate statements with dynamic arguments (#114)

Co-authored-by: Mislav Marohnić <[email protected]>
tmm1 · May 30, 2023 · aac79d8 · aac79d8
1 parent 2200945
commit aac79d8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/ripper-tags/parser.rb b/lib/ripper-tags/parser.rb
@@ -316,8 +316,8 @@ def on_assoc_new(key, value)
   end
 
   def on_delegate(*args)
-    method_names = args.select { |arg| arg.first.is_a? String }
-    options = args.select { |arg| arg.first.is_a?(Array) && arg.first.first == :assoc }.flatten(1)
+    method_names = args.select { |arg| arg.is_a?(Array) && arg.first.is_a?(String) }
+    options = args.select { |arg| arg.is_a?(Array) && arg.first.is_a?(Array) && arg.first.first == :assoc }.flatten(1)
     options = Hash[options.map { |_assoc, key, val| [key[0], val[0]] if key }.compact]
 
     target = options["to:"] || options["to"] # When using hashrocket syntax there is no ':'

diff --git a/test/test_ripper_tags.rb b/test/test_ripper_tags.rb
@@ -578,6 +578,9 @@ def test_invalid_delegate
       class C
         delegate
         delegate [1, 2]
+        [:test].each do |m|
+          delegate m, :"\#{m}=", :to => :object
+        end
       end
     EOC
     assert_equal 1, tags.count