fix endless method with proper indent

Author: flyerhzm

lib/ruby/prefer-endless-method.rb

@@ -19,9 +19,12 @@ def one_plus_one = 1 + 1
     ```
   EOS
 
+  if_ruby '3.0'
+
   within_files Synvert::ALL_RUBY_FILES + Synvert::ALL_RAKE_FILES do
     find_node '.def_node[body!=nil][body.body.length=1]' do
-      replace_with 'def {{name}} = {{body}}'
+      new_body = dedent(node.body.to_source)
+      replace_with "def {{name}}{{lparen}}{{parameters}}{{rparen}} = #{new_body}"
     end
   end
 end

spec/ruby/prefer-endless-method_spec.rb

@@ -5,14 +5,36 @@
 RSpec.describe 'Prefer endless method' do
   let(:rewriter_name) { 'ruby/prefer-endless-method' }
   let(:fake_file_path) { 'foobar.rb' }
-  let(:test_content) { <<~EOS }
-    def one_plus_one
-      1 + 1
-    end
-  EOS
-  let(:test_rewritten_content) { <<~EOS }
-    def one_plus_one = 1 + 1
-  EOS
 
-  include_examples 'convertable'
+  context 'expression' do
+    let(:test_content) { <<~EOS }
+      def one_plus_one
+        1 + 1
+      end
+    EOS
+    let(:test_rewritten_content) { <<~EOS }
+      def one_plus_one = 1 + 1
+    EOS
+
+    include_examples 'convertable'
+  end
+
+  context 'block call' do
+    let(:test_content) { <<~EOS }
+      def existing_key(device)
+        transaction do
+          key = device.one_time_keys.order(Arel.sql('random()')).first!
+          key.destroy!
+        end
+      end
+    EOS
+    let(:test_rewritten_content) { <<~EOS }
+      def existing_key(device) = transaction do
+        key = device.one_time_keys.order(Arel.sql('random()')).first!
+        key.destroy!
+      end
+    EOS
+
+    include_examples 'convertable'
+  end
 end