Router: fix match/infix indent after select chain

scalameta · Oct 1, 2022 · 38df6bd · 38df6bd
1 parent c37a580
commit 38df6bd
Show file tree

Hide file tree

Showing 7 changed files with 55 additions and 41 deletions.
diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/FormatOps.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/FormatOps.scala
@@ -570,7 +570,7 @@ class FormatOps(
       }
 
     @tailrec
-    private[FormatOps] def findEnclosingInfix(child: InfixApp): InfixApp = {
+    private[internal] def findEnclosingInfix(child: InfixApp): InfixApp = {
       val childTree = child.all
       if (isEnclosedInParens(childTree)) child
       else

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
@@ -1688,6 +1688,15 @@ class Router(formatOps: FormatOps) {
         }
         val prevSelect = findPrevSelect(thisSelect, enclosed)
         val expire = tokens.getLastExceptParen(expireTree.tokens).left
+        val indentExpire = expireTree.parent match {
+          case _ if style.indentOperator.relativeToLhsFirstLine => expire
+          case _ if enclosed && tokens.isEnclosedInParens(expireTree) => expire
+          case Some(p: Term.Match) if p.expr eq expireTree =>
+            getLastToken(p)
+          case Some(p: Term.ApplyInfix) if p.lhs eq expireTree =>
+            getLastToken(InfixSplits.findEnclosingInfix(InfixApp(p)).all)
+          case _ => expire
+        }
         val indentLen = style.indent.main
 
         def breakOnNextDot: Policy =
@@ -1714,8 +1723,9 @@ class Router(formatOps: FormatOps) {
         val baseSplits = style.newlines.getSelectChains match {
           case Newlines.classic =>
             def getNlMod = {
-              val endSelect =
-                nextSelect.fold(expire)(x => getLastNonTrivialToken(x.qual))
+              val endSelect = nextSelect.fold(indentExpire)(x =>
+                getLastNonTrivialToken(x.qual)
+              )
               val altIndent = Indent(-indentLen, endSelect, After)
               NewlineT(alt = Some(ModExt(NoSplit).withIndent(altIndent)))
             }
@@ -1841,7 +1851,7 @@ class Router(formatOps: FormatOps) {
         }
 
         // trigger indent only on the first newline
-        val indent = Indent(indentLen, expire, After)
+        val indent = Indent(indentLen, indentExpire, After)
         val willBreak = nextNonCommentSameLine(tokens(t, 2)).right.is[T.Comment]
         val splits = baseSplits.map { s =>
           if (willBreak || s.isNL) s.withIndent(indent)

diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/util/TreeExtractors.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/util/TreeExtractors.scala
@@ -22,11 +22,15 @@ case class InfixApp(lhs: Tree, op: Name, rhs: Seq[Tree], all: Tree) {
 
 object InfixApp {
 
+  def apply(t: Type.ApplyInfix): InfixApp = InfixApp(t.lhs, t.op, Seq(t.rhs), t)
+  def apply(t: Term.ApplyInfix): InfixApp = InfixApp(t.lhs, t.op, t.args, t)
+  def apply(t: Pat.ExtractInfix): InfixApp = InfixApp(t.lhs, t.op, t.rhs, t)
+
   def unapply(tree: Tree): Option[InfixApp] =
     tree match {
-      case t: Type.ApplyInfix => Some(InfixApp(t.lhs, t.op, Seq(t.rhs), t))
-      case t: Term.ApplyInfix => Some(InfixApp(t.lhs, t.op, t.args, t))
-      case t: Pat.ExtractInfix => Some(InfixApp(t.lhs, t.op, t.rhs, t))
+      case t: Type.ApplyInfix => Some(apply(t))
+      case t: Term.ApplyInfix => Some(apply(t))
+      case t: Pat.ExtractInfix => Some(apply(t))
       case _ => None
     }
 

diff --git a/scalafmt-tests/src/test/resources/newlines/source_classic.stat b/scalafmt-tests/src/test/resources/newlines/source_classic.stat
@@ -6024,10 +6024,10 @@ object A {
     IO.delay {
       ds // c1
         .headOption match {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
 <<< #3327 infix
@@ -6049,9 +6049,9 @@ object A {
     IO.delay {
       ds // c1
         .headOption infix {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
diff --git a/scalafmt-tests/src/test/resources/newlines/source_fold.stat b/scalafmt-tests/src/test/resources/newlines/source_fold.stat
@@ -5733,10 +5733,10 @@ object A {
     IO.delay {
       ds // c1
         .headOption match {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
 <<< #3327 infix
@@ -5758,9 +5758,9 @@ object A {
     IO.delay {
       ds // c1
         .headOption infix {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
diff --git a/scalafmt-tests/src/test/resources/newlines/source_keep.stat b/scalafmt-tests/src/test/resources/newlines/source_keep.stat
@@ -6030,10 +6030,10 @@ object A {
     IO.delay {
       ds // c1
         .headOption match {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
 <<< #3327 infix
@@ -6055,9 +6055,9 @@ object A {
     IO.delay {
       ds // c1
         .headOption infix {
-        case Some(Row(value: Long)) =>
-          value
-        case _ => 0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ => 0
+        }
     }
 }
diff --git a/scalafmt-tests/src/test/resources/newlines/source_unfold.stat b/scalafmt-tests/src/test/resources/newlines/source_unfold.stat
@@ -6182,11 +6182,11 @@ object A {
     IO.delay {
       ds // c1
         .headOption match {
-        case Some(Row(value: Long)) =>
-          value
-        case _ =>
-          0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ =>
+            0
+        }
     }
 }
 <<< #3327 infix
@@ -6208,10 +6208,10 @@ object A {
     IO.delay {
       ds // c1
         .headOption infix {
-        case Some(Row(value: Long)) =>
-          value
-        case _ =>
-          0
-      }
+          case Some(Row(value: Long)) =>
+            value
+          case _ =>
+            0
+        }
     }
 }