State: optionally apply indents to alt mod

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

@@ -13,6 +13,7 @@ case class ModExt(
     mod: Modification,
     indents: Seq[Indent] = Nil,
     altOpt: Option[ModExt] = None,
+    noAltIndent: Boolean = false,
 ) {
   @inline
   def isNL: Boolean = mod.isNL
@@ -27,18 +28,20 @@ case class ModExt(
 
     altOpt match {
       case None => res("")
-      case Some(x) => x.toString(res("|"), "+")
+      case Some(x) => x.toString(res("|"), if (noAltIndent) "" else "+")
     }
   }
 
   override def toString: String = toString("", "")
 
-  def withAlt(alt: ModExt): ModExt =
-    if (altOpt.contains(alt)) this else copy(altOpt = Some(alt))
+  def withAlt(alt: => ModExt, noAltIndent: Boolean = false): ModExt =
+    copy(altOpt = Some(alt), noAltIndent = noAltIndent)
 
   @inline
-  def withAltIf(ok: Boolean)(alt: => ModExt): ModExt =
-    if (ok) withAlt(alt) else this
+  def withAltIf(
+      ok: Boolean,
+  )(alt: => ModExt, noAltIndent: Boolean = false): ModExt =
+    if (ok) withAlt(alt, noAltIndent = noAltIndent) else this
 
   def orMod(flag: Boolean, mod: => ModExt): ModExt = if (flag) this else mod
 

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

@@ -1951,7 +1951,10 @@ class Router(formatOps: FormatOps) {
           case Newlines.fold =>
             def nlSplitBase(cost: Int, policy: Policy = NoPolicy)(implicit
                 fileLine: FileLine,
-            ) = Split(Newline.withAlt(modSpace), cost, policy = policy)
+            ) = {
+              val nlMod = Newline.withAlt(modSpace, noAltIndent = true)
+              Split(nlMod, cost, policy = policy)
+            }
             if (nextDotIfSig.isEmpty)
               if (nlOnly) Seq(nlSplitBase(0))
               else {

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

@@ -85,6 +85,7 @@ final class State(
         initialModExt.altOpt.flatMap { alt =>
           if (tok.left.is[T.Comment]) None
           else if (nextIndent < alt.mod.length + column) None
+          else if (initialModExt.noAltIndent) Some(alt)
           else Some(alt.withIndents(initialModExt.indents))
         }.fold((initialNextSplit, nextIndent, nextPushes)) { alt =>
           val altPushes = getUnexpired(alt, pushes)

scalafmt-tests-community/intellij/src/test/scala/org/scalafmt/community/intellij/CommunityIntellijScalaSuite.scala

@@ -13,7 +13,7 @@ abstract class CommunityIntellijScalaSuite(name: String)
 class CommunityIntellijScala_2024_2_Suite
     extends CommunityIntellijScalaSuite("intellij-scala-2024.2") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(57957808)
+  override protected def totalStatesVisited: Option[Int] = Some(59338368)
 
   override protected def builds = Seq(getBuild(
     "2024.2.28",
@@ -51,7 +51,7 @@ class CommunityIntellijScala_2024_2_Suite
 class CommunityIntellijScala_2024_3_Suite
     extends CommunityIntellijScalaSuite("intellij-scala-2024.3") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(58174471)
+  override protected def totalStatesVisited: Option[Int] = Some(59555031)
 
   override protected def builds = Seq(getBuild(
     "2024.3.4",

scalafmt-tests-community/scala2/src/test/scala/org/scalafmt/community/scala2/CommunityScala2Suite.scala

@@ -9,7 +9,7 @@ abstract class CommunityScala2Suite(name: String)
 
 class CommunityScala2_12Suite extends CommunityScala2Suite("scala-2.12") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(42560182)
+  override protected def totalStatesVisited: Option[Int] = Some(42554938)
 
   override protected def builds =
     Seq(getBuild("v2.12.20", dialects.Scala212, 1277))
@@ -18,7 +18,7 @@ class CommunityScala2_12Suite extends CommunityScala2Suite("scala-2.12") {
 
 class CommunityScala2_13Suite extends CommunityScala2Suite("scala-2.13") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(53060139)
+  override protected def totalStatesVisited: Option[Int] = Some(53041273)
 
   override protected def builds =
     Seq(getBuild("v2.13.14", dialects.Scala213, 1287))

scalafmt-tests-community/scala3/src/test/scala/org/scalafmt/community/scala3/CommunityScala3Suite.scala

@@ -9,15 +9,15 @@ abstract class CommunityScala3Suite(name: String)
 
 class CommunityScala3_2Suite extends CommunityScala3Suite("scala-3.2") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(39246875)
+  override protected def totalStatesVisited: Option[Int] = Some(39229069)
 
   override protected def builds = Seq(getBuild("3.2.2", dialects.Scala32, 791))
 
 }
 
 class CommunityScala3_3Suite extends CommunityScala3Suite("scala-3.3") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(42432863)
+  override protected def totalStatesVisited: Option[Int] = Some(42414755)
 
   override protected def builds = Seq(getBuild("3.3.3", dialects.Scala33, 861))
 

scalafmt-tests-community/spark/src/test/scala/org/scalafmt/community/spark/CommunitySparkSuite.scala

@@ -9,7 +9,7 @@ abstract class CommunitySparkSuite(name: String)
 
 class CommunitySpark3_4Suite extends CommunitySparkSuite("spark-3.4") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(86399848)
+  override protected def totalStatesVisited: Option[Int] = Some(86438964)
 
   override protected def builds =
     Seq(getBuild("v3.4.1", dialects.Scala213, 2585))
@@ -18,7 +18,7 @@ class CommunitySpark3_4Suite extends CommunitySparkSuite("spark-3.4") {
 
 class CommunitySpark3_5Suite extends CommunitySparkSuite("spark-3.5") {
 
-  override protected def totalStatesVisited: Option[Int] = Some(91404596)
+  override protected def totalStatesVisited: Option[Int] = Some(91445316)
 
   override protected def builds =
     Seq(getBuild("v3.5.3", dialects.Scala213, 2756))

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

@@ -11249,3 +11249,45 @@ object a {
 
   if (s.costWithPenalty <= 0) true else { preFork = false; false }
 }
+<<< chained maps with multi-line blocks, with braces-to-parens OK
+rewrite.rules = [RedundantBraces]
+===
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+<<< chained maps with multi-line blocks, with braces-to-parens not OK
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+      // c2
+      quux(x)
+    }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}

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

@@ -8430,7 +8430,7 @@ object a {
     )
   )
 }
->>> { stateVisits = 4769, stateVisits2 = 4769 }
+>>> { stateVisits = 3882, stateVisits2 = 3882 }
 object a {
   div(cls := "cover")(
     div(cls := "doc")(bodyContents),
@@ -9558,61 +9558,57 @@ object a {
       .map(_.filterNot(_.getCanonicalPath.contains("SSLOptions")))
   }
 }
->>> { stateVisits = 3668, stateVisits2 = 3668 }
+>>> { stateVisits = 52849, stateVisits2 = 52849 }
 object a {
   private def ignoreUndocumentedPackages(
       packages: Seq[Seq[File]]
   ): Seq[Seq[File]] = {
-    packages.map(_.filterNot(_.getName.contains("$")))
-      .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/deploy")))
-      .map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/examples")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/internal")
-      ))
-      .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/memory")))
-      .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/network")))
-      .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/rpc")))
-      .map(_.filterNot(f =>
-        f.getCanonicalPath.contains("org/apache/spark/shuffle") &&
-          !f.getCanonicalPath.contains("org/apache/spark/shuffle/api")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/executor")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/ExecutorAllocationClient")
-      )).map(_.filterNot(_.getCanonicalPath.contains(
-        "org/apache/spark/scheduler/cluster/CoarseGrainedSchedulerBackend"
-      ))).map(_.filterNot(f =>
-        f.getCanonicalPath.contains("org/apache/spark/unsafe") &&
-          !f.getCanonicalPath
-            .contains("org/apache/spark/unsafe/types/CalendarInterval")
-      )).map(_.filterNot(_.getCanonicalPath.contains("python")))
-      .map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/util/collection")
-      ))
-      .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/util/io")))
-      .map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/util/kvstore")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/catalyst")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/connect/")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/execution")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/internal")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/hive")
-      )).map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/catalog/v2/utils")
-      ))
-      .map(_.filterNot(_.getCanonicalPath.contains("org.apache.spark.errors")))
-      .map(_.filterNot(
-        _.getCanonicalPath.contains("org.apache.spark.sql.errors")
-      )).map(_.filterNot(_.getCanonicalPath.contains("org/apache/hive")))
-      .map(_.filterNot(
-        _.getCanonicalPath.contains("org/apache/spark/sql/v2/avro")
-      )).map(_.filterNot(_.getCanonicalPath.contains("SSLOptions")))
+    packages.map(_.filterNot(_.getName.contains("$"))).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/deploy")
+    )).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/examples")
+    ))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/internal")))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/memory")))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/network")))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/rpc")))
+    .map(_.filterNot(f =>
+      f.getCanonicalPath.contains("org/apache/spark/shuffle") &&
+        !f.getCanonicalPath.contains("org/apache/spark/shuffle/api")
+    ))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/executor")))
+    .map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/ExecutorAllocationClient")
+    )).map(_.filterNot(_.getCanonicalPath.contains(
+      "org/apache/spark/scheduler/cluster/CoarseGrainedSchedulerBackend"
+    ))).map(_.filterNot(f =>
+      f.getCanonicalPath.contains("org/apache/spark/unsafe") &&
+        !f.getCanonicalPath
+          .contains("org/apache/spark/unsafe/types/CalendarInterval")
+    )).map(_.filterNot(_.getCanonicalPath.contains("python"))).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/util/collection")
+    )).map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/util/io")))
+    .map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/util/kvstore")
+    )).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/catalyst")
+    )).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/connect/")
+    )).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/execution")
+    )).map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/internal")
+    ))
+    .map(_.filterNot(_.getCanonicalPath.contains("org/apache/spark/sql/hive")))
+    .map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/catalog/v2/utils")
+    )).map(_.filterNot(_.getCanonicalPath.contains("org.apache.spark.errors")))
+    .map(_.filterNot(
+      _.getCanonicalPath.contains("org.apache.spark.sql.errors")
+    )).map(_.filterNot(_.getCanonicalPath.contains("org/apache/hive")))
+    .map(_.filterNot(
+      _.getCanonicalPath.contains("org/apache/spark/sql/v2/avro")
+    )).map(_.filterNot(_.getCanonicalPath.contains("SSLOptions")))
   }
 }
 <<< #4133 #4133 spark: massive chain of applies with function calls
@@ -10069,7 +10065,7 @@ system.dynamicAccess.createInstanceFor[Serializer](fqn, Nil).recoverWith {
             }
       }
 }
->>> { stateVisits = 4820, stateVisits2 = 4820 }
+>>> { stateVisits = 3689, stateVisits2 = 3689 }
 system.dynamicAccess.createInstanceFor[Serializer](fqn, Nil).recoverWith {
   case _: NoSuchMethodException => system.dynamicAccess
       .createInstanceFor[Serializer](
@@ -10494,3 +10490,45 @@ object a {
 
   if (s.costWithPenalty <= 0) true else { preFork = false; false }
 }
+<<< chained maps with multi-line blocks, with braces-to-parens OK
+rewrite.rules = [RedundantBraces]
+===
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+<<< chained maps with multi-line blocks, with braces-to-parens not OK
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+      // c2
+      quux(x)
+    }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}

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

@@ -11001,3 +11001,45 @@ object a {
 
   if (s.costWithPenalty <= 0) true else { preFork = false; false }
 }
+<<< chained maps with multi-line blocks, with braces-to-parens OK
+rewrite.rules = [RedundantBraces]
+===
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}
+<<< chained maps with multi-line blocks, with braces-to-parens not OK
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+      // c2
+      quux(x)
+    }
+}
+>>>
+object a {
+  def foo = bar {
+    // c1
+    baz
+  }.qux { x =>
+    // c2
+    quux(x)
+  }
+}