Merge pull request #11156 from gsamokovarov/operator-method-chaining-…

…call

Fix `Style/OperatorMethodCall` autocorrection when operators are chained

changelog/fix_styleoperatormethodcall.md

@@ -0,0 +1 @@
+* [#11156](https://github.com/rubocop/rubocop/pull/11156): Fix `Style/OperatorMethodCall` autocorrection when operators are chained. ([@gsamokovarov][])

lib/rubocop/cop/style/operator_method_call.rb

@@ -31,9 +31,22 @@ def on_send(node)
           return if rhs.nil? || rhs.children.first
 
           add_offense(dot) do |corrector|
+            wrap_in_parentheses_if_chained(corrector, node)
             corrector.replace(dot, ' ')
           end
         end
+
+        private
+
+        def wrap_in_parentheses_if_chained(corrector, node)
+          return unless node.parent&.call_type?
+
+          operator = node.loc.selector
+
+          ParenthesesCorrector.correct(corrector, node)
+          corrector.insert_after(operator, ' ')
+          corrector.wrap(node, '(', ')')
+        end
       end
     end
   end

spec/rubocop/cop/style/operator_method_call_spec.rb

@@ -30,6 +30,17 @@
         foo #{operator_method}(bar)
       RUBY
     end
+
+    it "registers an offense when chaining `foo.bar.#{operator_method}(baz).round(2)`" do
+      expect_offense(<<~RUBY, operator_method: operator_method)
+        foo.bar.#{operator_method}(baz).quux(2)
+               ^ Redundant dot detected.
+      RUBY
+
+      expect_correction(<<~RUBY)
+        (foo.bar #{operator_method} baz).quux(2)
+      RUBY
+    end
   end
 
   it 'does not register an offense when using `foo.+@bar.to_s`' do