Formatter: properly indent comments at the end of a proc literal (#8778)

crystal-lang · Feb 10, 2020 · 39ce85d · 39ce85d
1 parent fc9e15c
commit 39ce85d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/spec/compiler/formatter/formatter_spec.cr b/spec/compiler/formatter/formatter_spec.cr
@@ -1602,4 +1602,12 @@ describe Crystal::Formatter do
     CODE
 
   assert_format "a.!"
+
+  assert_format <<-CODE
+    ->{
+      # first comment
+      puts "hi"
+      # second comment
+    }
+    CODE
 end
diff --git a/src/compiler/crystal/tools/formatter.cr b/src/compiler/crystal/tools/formatter.cr
@@ -4060,7 +4060,9 @@ module Crystal
         end
       end
 
-      skip_space_or_newline
+      indent do
+        skip_space_or_newline
+      end
 
       if is_do
         check_end