Model call chains as an AbstractTokenTarget

fixtures/small/aref_in_call_actual.rb

@@ -9,4 +9,11 @@
   )
     .void
 end
-def foo; end
+def foo; end
+
+Array[
+  @root_fragment,
+  @lemon_tea_fragment,
+  @green_tea_fragment,
+  @cake_fragment
+].sort_by(&:name)

fixtures/small/aref_in_call_expected.rb

@@ -11,3 +11,10 @@
 end
 def foo
 end
+
+Array[
+  @root_fragment,
+  @lemon_tea_fragment,
+  @green_tea_fragment,
+  @cake_fragment
+].sort_by(&:name)

fixtures/small/begin_end_stack_actual.rb

@@ -2,3 +2,7 @@
   a =  begin
        end
 end
+
+begin
+end
+  .freeze

fixtures/small/begin_end_stack_expected.rb

@@ -2,3 +2,6 @@
   a = begin
   end
 end
+
+begin
+end.freeze

fixtures/small/brace_blocks_with_no_args_actual.rb

@@ -1,3 +1,12 @@
 def func
   brace_block_with_no_args { p 'hi' }
 end
+
+{
+  "/v1/transfers" => lambda {
+    foo
+    bar
+    # TODO: add baz here
+    # Oh and maybe quux if you have time
+  }
+}

fixtures/small/brace_blocks_with_no_args_expected.rb

@@ -1,3 +1,12 @@
 def func
   brace_block_with_no_args { p("hi") }
 end
+
+{
+  "/v1/transfers" => lambda {
+    foo
+    bar
+    # TODO: add baz here
+    # Oh and maybe quux if you have time
+  }
+}

fixtures/small/conditional_actual.rb

@@ -7,3 +7,7 @@ def foo
     puts("hi")
   end
 end
+
+do_stuff! if it_isnt_dangerous(
+  i_promise: true
+)

fixtures/small/conditional_expected.rb

@@ -7,3 +7,9 @@ def foo
     puts("hi")
   end
 end
+
+if it_isnt_dangerous(
+    i_promise: true
+  )
+  do_stuff!
+end

fixtures/small/heredoc_method_call_actual.rb

@@ -25,3 +25,14 @@ class William::Carlos::Williams
     Icarus drowning
   LANDSCAPE
 end
+
+optp
+  .on do |value|
+    <<~EOF
+      There's some lines here
+
+      But that one's a blank line!
+
+      There shouldn't be any whitespace on those
+    EOF
+  end

fixtures/small/heredoc_method_call_expected.rb

@@ -30,3 +30,14 @@ class William::Carlos::Williams
     Williams
   )
 end
+
+optp
+  .on do |value|
+    <<~EOF
+      There's some lines here
+
+      But that one's a blank line!
+
+      There shouldn't be any whitespace on those
+    EOF
+  end

fixtures/small/heredocs_actual.rb

@@ -68,6 +68,27 @@ class Foo
   rubocop.smash(bad_thing)
 end
 
+this_one_is
+  .in_a_call_chain {
+    # some stuff
+  }
+  .each do
+    <<~MYHEREDOC
+    Words are pale shadows of forgotten names.
+    As names have power, words have power.
+    Words can light fires in the minds of men.
+    Words can wring tears from the hardest hearts.
+    MYHEREDOC
+  end
+
+def foo
+  "#{stuff.each do
+  <<~MESSAGE.strip
+    #{message.text}
+  MESSAGE
+  end.join("\n\n")}"
+end
+
 
 puts a
 puts b

fixtures/small/heredocs_expected.rb

@@ -69,6 +69,28 @@ class Foo
   rubocop.smash(bad_thing)
 end
 
+this_one_is
+  .in_a_call_chain {
+    # some stuff
+  }
+  .each do
+    <<~MYHEREDOC
+      Words are pale shadows of forgotten names.
+      As names have power, words have power.
+      Words can light fires in the minds of men.
+      Words can wring tears from the hardest hearts.
+    MYHEREDOC
+  end
+
+def foo
+  "#{stuff.each do
+    <<~MESSAGE
+      #{message.text}
+    MESSAGE
+      .strip
+  end.join("\n\n")}"
+end
+
 puts(a)
 puts(b)
 foo

fixtures/small/method_annotation_expected.rb

@@ -32,7 +32,8 @@ def do_the_thing(a, b)
   params(
     a: String,
     b: String
-  ).void
+  )
+    .void
 end
 def example(a, b)
 end
@@ -95,7 +96,8 @@ class Bees
       first_param: MyClass,
       # This one is the second one, nice
       second_param: YourClass
-    ).void
+    )
+      .void
     # Please not the bees!
   }
   def not_the_bees!

fixtures/small/method_chains_actual.rb

@@ -18,8 +18,6 @@
 foo.bar
 .baz
 
-# If they're all on the same line but different from
-# the first receiver, consider that "on one line"
 foo
 .bar.baz
 
@@ -97,6 +95,24 @@ def example
     b: ""
   )
 
+ThisIs::OnlyOneCall
+  # but it's explicitly multilined with a
+  .comment!
+
+OnlyOneCall
+  # but it's explicitly multilined with a
+  .comment!
+
+  [
+    # foo
+  ]
+    # bar
+    .baz
+
+  []
+    # Please don't do this
+    .freeze
+
 Paul::Blart::Mall::Cop::PerformedByTheLegendaryKevinJamesWhoIsAnAbsoluteLegendInAllOfHisFilmsWhichAreAbsolutelyIncredible.consume_pixie_sticks(mall: "downtown").each do |punch_list_type|
 end
 
@@ -129,3 +145,19 @@ def gather_thanes!
   puts "h"
   end
   &.foo
+
+My::Error.soft(
+  "",
+  stuff: {
+    message_token: message.token,
+
+    # Some comments!
+    value: id_or_email.name
+  }
+)
+
+# rubocop:disable PrisonGuard/PrivateModule
+(foo.load_one
+  # rubocop:enable PrisonGuard/PrivateModule
+  .bar)
+  .thing

fixtures/small/method_chains_expected.rb

@@ -21,11 +21,12 @@
   .bar
   .baz
 
-# If they're all on the same line but different from
-# the first receiver, consider that "on one line"
-foo.bar.baz
+foo
+  .bar
+  .baz
 
-foo::bar&.nil?
+foo::bar
+  &.nil?
 
 foo::bar
   &.nil?::klass
@@ -120,6 +121,24 @@ def example
     b: ""
   )
 
+ThisIs::OnlyOneCall
+  # but it's explicitly multilined with a
+  .comment!
+
+OnlyOneCall
+  # but it's explicitly multilined with a
+  .comment!
+
+[
+  # foo
+]
+  # bar
+  .baz
+
+[]
+  # Please don't do this
+  .freeze
+
 Paul::Blart::Mall::Cop::PerformedByTheLegendaryKevinJamesWhoIsAnAbsoluteLegendInAllOfHisFilmsWhichAreAbsolutelyIncredible
   .consume_pixie_sticks(mall: "downtown")
   .each do |punch_list_type|
@@ -160,3 +179,20 @@ def gather_thanes!
     puts("h")
   end
   &.foo
+
+My::Error.soft(
+  "",
+  stuff: {
+    message_token: message.token,
+
+    # Some comments!
+    value: id_or_email.name
+  }
+)
+
+# rubocop:disable PrisonGuard/PrivateModule
+(foo
+  .load_one
+  # rubocop:enable PrisonGuard/PrivateModule
+  .bar)
+  .thing

fixtures/small/multiline_chain_in_block_expected.rb

@@ -1,7 +1,8 @@
 sig {
   params(
     route: String
-  ).void
+  )
+    .void
 }
 def ajax_get(route)
   super

fixtures/small/multiline_chained_call_expected.rb

@@ -2,5 +2,6 @@ def blorp
   method_call(
     arg_a,
     arg_b
-  ).chained_call
+  )
+    .chained_call
 end

fixtures/small/multiline_method_chain_with_arguments_actual.rb

@@ -5,3 +5,11 @@ def stub_server(path:, body: {})
     "https://example.com#{path}"
   ).to_return(body: body.to_json)
 end
+
+{
+  "original_fields" => foo,
+  "alternative_fields" => (thing_one(id, api) + thing_two(
+    id,
+    api
+  )).sort
+}

fixtures/small/multiline_method_chain_with_arguments_expected.rb

@@ -3,5 +3,15 @@ def stub_server(path:, body: {})
   stub_request(
     :get,
     "https://example.com#{path}"
-  ).to_return(body: body.to_json)
+  )
+    .to_return(body: body.to_json)
 end
+
+{
+  "original_fields" => foo,
+  "alternative_fields" => (thing_one(id, api) + thing_two(
+    id,
+    api
+  ))
+    .sort
+}

fixtures/small/paren_expr_calls_actual.rb

@@ -1,2 +1,16 @@
 a (1)
 other_cool_method (a + b).round(4)
+
+# rubocop:disable PrisonGuard/PrivateModule
+(foo(
+  foo # rubocop:enable PrisonGuard/PrivateModule
+)).flatten
+
+# rubocop:disable Style/Stuff
+(MyModel::InSomeNamespace
+  .load_one(
+    # rubocop:enable Style/Stuff
+    {name: "name"}
+  )
+  &.rules)
+  .freeze

fixtures/small/paren_expr_calls_expected.rb

@@ -1,2 +1,18 @@
 a(1)
 other_cool_method((a + b).round(4))
+
+# rubocop:disable PrisonGuard/PrivateModule
+(foo(
+  # rubocop:enable PrisonGuard/PrivateModule
+  foo
+))
+  .flatten
+
+# rubocop:disable Style/Stuff
+(MyModel::InSomeNamespace
+  .load_one(
+    # rubocop:enable Style/Stuff
+    {name: "name"}
+  )
+  &.rules)
+  .freeze