Router: inline, remove TreeOps.findTemplate

scalameta · Oct 11, 2024 · 1bc8b0d · 1bc8b0d
1 parent 8d096f1
commit 1bc8b0d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 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
@@ -483,7 +483,8 @@ class Router(formatOps: FormatOps) {
       case FormatToken(_: T.RightArrow | _: T.ContextArrow, right, _)
           if (leftOwner match {
             case _: Term.FunctionTerm | _: Term.PolyFunction => true
-            case t => findTemplate(t).parent.is[Term.NewAnonymous]
+            case t: Self => t.ancestor(2).is[Term.NewAnonymous]
+            case _ => false
           }) =>
         val (endOfFunction, expiresOn) = leftOwner match {
           case t: Term.FunctionTerm => functionExpire(t)

diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/util/TreeOps.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/util/TreeOps.scala
@@ -663,15 +663,6 @@ object TreeOps {
     if (groups.isEmpty) None else Some(groups)
   }
 
-  def findTemplate(t: Tree): Option[Template] = t match {
-    case t: Template => Some(t)
-    case _: Self | _: Template.Body => t.parent match {
-        case Some(p) => findTemplate(p)
-        case _ => None
-      }
-    case _ => None
-  }
-
   // Scala syntax allows commas before right braces in weird places,
   // like constructor bodies:
   // def this() = {