Router: handle then and do similar to else/yield

scalameta · Jan 10, 2022 · 2c3a944 · 2c3a944
1 parent e5e2b1a
commit 2c3a944
Show file tree

Hide file tree

Showing 4 changed files with 97 additions and 56 deletions.
diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala
@@ -1932,6 +1932,15 @@ class Router(formatOps: FormatOps) {
             .withSingleLineNoOptimal(expire, exclude, noSyntaxNL = noSyntaxNL),
           Split(Newline, 1)
         )
+      case ft @ FormatToken(_, _: T.KwThen | _: T.KwDo, _) =>
+        if (style.newlines.sourceIgnored || newlines == 0)
+          Seq(
+            Split(Space, 0)
+              .withOptimalToken(nextNonCommentSameLine(next(ft)).left),
+            Split(Newline, 1)
+          )
+        else
+          Seq(Split(Newline, 0))
       // Last else branch
       case FormatToken(_: T.KwElse, _, _) if (leftOwner match {
             case t: Term.If => !t.elsep.is[Term.If]

diff --git a/scalafmt-tests/src/test/resources/scala3/OptionalBraces.stat b/scalafmt-tests/src/test/resources/scala3/OptionalBraces.stat
@@ -3115,14 +3115,14 @@ class C:
   else 4
 >>>
 class C:
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
 <<< #3033 if-then !danglingParentheses
 maxColumn = 15
@@ -3143,14 +3143,14 @@ class C:
   else 4
 >>>
 class C:
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
 <<< #3034 for-yield danglingParentheses
 danglingParentheses.ctrlSite = true
@@ -3199,10 +3199,13 @@ object a:
    matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
    for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
 <<< #3034 for-do !danglingParentheses
 danglingParentheses.ctrlSite = false
 ===
@@ -3220,10 +3223,13 @@ object a:
    matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
    for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
 <<< #3034 while-do danglingParentheses
 danglingParentheses.ctrlSite = true
 ===
@@ -3244,11 +3250,15 @@ object a:
     matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
    while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
 <<< #3034 while-do !danglingParentheses
 danglingParentheses.ctrlSite = false
 ===
@@ -3269,8 +3279,12 @@ object a:
     matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
    while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
diff --git a/scalafmt-tests/src/test/resources/scala3/OptionalBraces_fold.stat b/scalafmt-tests/src/test/resources/scala3/OptionalBraces_fold.stat
@@ -2948,14 +2948,14 @@ class C:
   else 4
 >>>
 class C:
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
 <<< #3033 if-then short !danglingParentheses
 maxColumn = 13
@@ -2976,14 +2976,14 @@ class C:
   else 4
 >>>
 class C:
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
-   if true then
-      3
+   if true
+   then 3
    else 4
 <<< #3034 for-yield danglingParentheses
 maxColumn = 50

diff --git a/scalafmt-tests/src/test/resources/scala3/OptionalBraces_keep.stat b/scalafmt-tests/src/test/resources/scala3/OptionalBraces_keep.stat
@@ -3081,12 +3081,15 @@ class C:
   else 4
 >>>
 class C:
-   if true then 3
+   if true
+   then 3
    else 4
-   if true then
+   if true
+   then
       3
    else 4
-   if true then
+   if true
+   then
       3
    else 4
 <<< #3033 if-then !danglingParentheses
@@ -3107,12 +3110,15 @@ class C:
   else 4
 >>>
 class C:
-   if true then 3
+   if true
+   then 3
    else 4
-   if true then
+   if true
+   then
       3
    else 4
-   if true then
+   if true
+   then
       3
    else 4
 <<< #3034 for-yield danglingParentheses
@@ -3162,10 +3168,12 @@ object a:
    matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
    for x <- range do
       matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do
+   for x <- range
+   do
       matrix(x)(0) += matrix(x - 1)(0)
    for
       x <- range
@@ -3188,10 +3196,12 @@ object a:
    matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
-   for x <- range do matrix(x)(0) += matrix(x - 1)(0)
+   for x <- range
+   do matrix(x)(0) += matrix(x - 1)(0)
    for x <- range do
       matrix(x)(0) += matrix(x - 1)(0)
-   for x <- range do
+   for x <- range
+   do
       matrix(x)(0) += matrix(x - 1)(0)
    for
       x <- range
@@ -3217,13 +3227,17 @@ object a:
     matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
    while x < range do
       matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do
       matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do
+   while x < range
+   do
       matrix(x)(0) += matrix(x - 1)(0)
 <<< #3034 while-do !danglingParentheses
 danglingParentheses.ctrlSite = false
@@ -3245,11 +3259,15 @@ object a:
     matrix(x)(0) += matrix(x - 1)(0)
 >>>
 object a:
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
    while x < range do
       matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do
+   while x < range
+   do matrix(x)(0) += matrix(x - 1)(0)
+   while x < range
+   do
       matrix(x)(0) += matrix(x - 1)(0)
-   while x < range do
+   while x < range
+   do
       matrix(x)(0) += matrix(x - 1)(0)