Router: move up check for no NL after <- enums

It might have been overridden by `newlines.forceBeforeAssign` param.
scalameta · Oct 9, 2024 · f1f53ed · f1f53ed
1 parent 5d3aa3d
commit f1f53ed
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 30 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
@@ -2615,6 +2615,16 @@ class Router(formatOps: FormatOps) {
     else if (isRightCommentWithBreak(ft))
       Seq(CtrlBodySplits.withIndent(Split(Space.orNL(ft), 0), body, endFt))
     else if (isJsNative(body)) Seq(Split(Space, 0).withSingleLine(endFt.left))
+    else if (
+      style.dialect.allowSignificantIndentation &&
+      (style.newlines.sourceIgnored || ft.noBreak) && body.parent.exists {
+        case p: Enumerator.Assign => (p.body eq body) && p.parent.exists {
+            case pp: Term.EnumeratorsBlock => isEnclosedInParens(pp)
+            case _ => false
+          }
+        case _ => false
+      }
+    ) Seq(Split(Space, 0).withIndents(spaceIndents))
     else if (style.newlines.forceBeforeAssign(ft.meta.leftOwner))
       Seq(CtrlBodySplits.withIndent(Split(Newline2x(ft), 0), body, endFt))
     else if (style.newlines.shouldForceBeforeMultilineAssign(ft.meta.leftOwner))
@@ -2717,7 +2727,7 @@ class Router(formatOps: FormatOps) {
       if (!style.align.arrowEnumeratorGenerator) Seq.empty
       else Seq(Indent(StateColumn, expire, After))
     getSplitsDefValEquals(body, endFt, spaceIndents) {
-      def splits = CtrlBodySplits.get(body, spaceIndents) {
+      CtrlBodySplits.get(body, spaceIndents) {
         if (spaceIndents.nonEmpty) Split(Space, 0).withIndents(spaceIndents)
         else {
           val noSlb = body match {
@@ -2729,13 +2739,6 @@ class Router(formatOps: FormatOps) {
           else Split(Space, 0).withSingleLine(expire)
         }
       }(cost => CtrlBodySplits.withIndent(Split(Newline2x(ft), cost), endFt))
-      body.parent.parent match {
-        case Some(pp: Term.EnumeratorsBlock)
-            if style.dialect.allowSignificantIndentation &&
-              isEnclosedInParens(pp) =>
-          Seq(Split(Space, 0).withIndents(spaceIndents))
-        case _ => splits
-      }
     }
   }
 

diff --git a/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces.stat b/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces.stat
@@ -7355,7 +7355,8 @@ object a:
      for (
        x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
        y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       z <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
+       z <-
+         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
      ) yield x + y
 <<< #4219 forceBeforeAssign
 newlines.forceBeforeAssign = any
@@ -7371,10 +7372,8 @@ object a:
 object a:
    val expression =
      for (
-       x <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       y <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
        z <-
          loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
      ) yield x + y
diff --git a/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_fold.stat b/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_fold.stat
@@ -7091,10 +7091,7 @@ object a:
 object a:
    val expression =
      for (
-       x <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       y <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       z <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
+       x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       z <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
      ) yield x + y
diff --git a/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_keep.stat b/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_keep.stat
@@ -7384,7 +7384,8 @@ object a:
    val expression = for (
      x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
      y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-     z <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
+     z <-
+       loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
    ) yield x + y
 <<< #4219 forceBeforeAssign
 newlines.forceBeforeAssign = any
@@ -7400,10 +7401,8 @@ object a:
 object a:
    val expression =
      for (
-       x <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       y <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
        z <-
          loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
      ) yield x + y
diff --git a/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_unfold.stat b/scalafmt-tests/shared/src/test/resources/scala3/OptionalBraces_unfold.stat
@@ -7665,11 +7665,8 @@ object a:
 object a:
    val expression =
      for (
-       x <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       y <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
-       z <-
-         loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
+       x <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       y <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu;
+       z <- loremipsumdolorsitametconsecteturadipiscingelitseddoeiusmodtemporincididu
      )
        yield x + y