Splits: handle do-while conditions as well

Author: kitbellew

scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Splits.scala

@@ -2003,12 +2003,10 @@ object SplitsAfterLeftParen extends Splits {
         else alignIndents,
       ))
     }
-    def withCond(t: Tree.WithCond) =
-      if (isTokenHeadOrBefore(left, t)) None else impl(isEnclosedInBraces(t.cond))
     leftOwner match { // If/For/While/For with (
       case t: Term.EnumeratorsBlock if getHeadOpt(t).contains(ft) => impl(false)
-      case t: Term.If => withCond(t)
-      case t: Term.While => withCond(t)
+      case t: Tree.WithCond if !isTokenHeadOrBefore(left, t) =>
+        impl(isEnclosedInBraces(t.cond))
       case _ => None
     }
   }

scalafmt-tests/shared/src/test/resources/newlines/source_classic.stat

@@ -12093,7 +12093,9 @@ def foo = {
 def foo = {
   next.offset = begCharOffset
   do putCharAndNext()
-  while (ch != CR && ch != LF && ch != FF && ch != SU)
+  while (
+    ch != CR && ch != LF && ch != FF && ch != SU
+  )
 }
 <<< `do` starts a new statement, wide, danglingParentheses.ctrlSite
 maxColumn = 60

scalafmt-tests/shared/src/test/resources/newlines/source_fold.stat

@@ -11319,8 +11319,10 @@ def foo = {
 >>>
 def foo = {
   next.offset = begCharOffset
-  do putCharAndNext() while (ch != CR &&
-    ch != LF && ch != FF && ch != SU)
+  do putCharAndNext() while (
+    ch != CR && ch != LF && ch != FF &&
+      ch != SU
+  )
 }
 <<< `do` starts a new statement, wide, danglingParentheses.ctrlSite
 maxColumn = 60
@@ -11334,8 +11336,9 @@ def foo = {
 >>>
 def foo = {
   next.offset = begCharOffset
-  do putCharAndNext()
-  while (ch != CR && ch != LF && ch != FF && ch != SU)
+  do putCharAndNext() while (
+    ch != CR && ch != LF && ch != FF && ch != SU
+  )
 }
 <<< `do` starts a new statement, narrow, !danglingParentheses.ctrlSite
 danglingParentheses.ctrlSite = false
@@ -11348,8 +11351,9 @@ def foo = {
 >>>
 def foo = {
   next.offset = begCharOffset
-  do putCharAndNext() while (ch != CR &&
-    ch != LF && ch != FF && ch != SU)
+  do putCharAndNext()
+  while (ch != CR && ch != LF &&
+      ch != FF && ch != SU)
 }
 <<< `do` starts a new statement, wide, !danglingParentheses.ctrlSite
 maxColumn = 60

scalafmt-tests/shared/src/test/resources/newlines/source_keep.stat

@@ -11865,7 +11865,9 @@ def foo = {
 def foo = {
   next.offset = begCharOffset
   do putCharAndNext()
-  while (ch != CR && ch != LF && ch != FF && ch != SU)
+  while (
+    ch != CR && ch != LF && ch != FF && ch != SU
+  )
 }
 <<< `do` starts a new statement, wide, danglingParentheses.ctrlSite
 maxColumn = 60

scalafmt-tests/shared/src/test/resources/newlines/source_unfold.stat

@@ -12350,8 +12350,9 @@ def foo = {
 >>>
 def foo = {
   next.offset = begCharOffset
-  do putCharAndNext()
-  while (ch != CR && ch != LF && ch != FF && ch != SU)
+  do putCharAndNext() while (
+    ch != CR && ch != LF && ch != FF && ch != SU
+  )
 }
 <<< `do` starts a new statement, narrow, !danglingParentheses.ctrlSite
 danglingParentheses.ctrlSite = false
@@ -12364,10 +12365,9 @@ def foo = {
 >>>
 def foo = {
   next.offset = begCharOffset
-  do putCharAndNext() while (
-    ch != CR && ch != LF && ch != FF &&
-      ch != SU
-  )
+  do putCharAndNext()
+  while (ch != CR && ch != LF &&
+      ch != FF && ch != SU)
 }
 <<< `do` starts a new statement, wide, !danglingParentheses.ctrlSite
 maxColumn = 60

scalafmt-tests/shared/src/test/resources/rewrite/RedundantBraces.stat

@@ -2480,9 +2480,15 @@ object a {
 >>>
 object a {
   do foo
-  while (true)
+  while (
+    true
+  )
   do a + b
-  while (true)
+  while (
+    true
+  )
   do a + b
-  while (true)
+  while (
+    true
+  )
 }

scalafmt-tests/shared/src/test/scala/org/scalafmt/FormatTests.scala

@@ -148,7 +148,7 @@ class FormatTests extends FunSuite with CanRunTests with FormatAssertions {
     val explored = Debug.explored.get()
     logger.debug(s"Total explored: $explored")
     if (!onlyUnit && !onlyManual)
-      assertEquals(explored, 2551504, "total explored")
+      assertEquals(explored, 2552242, "total explored")
     // TODO(olafur) don't block printing out test results.
     TestPlatformCompat.executeAndWait(PlatformFileOps.writeFile(
       FileOps.getPath("target", "index.html"),