RedundantBraces: cosmetic, minor refactor

scalameta · Oct 15, 2024 · f4bf129 · f4bf129
1 parent 2559975
commit f4bf129
Showing 1 changed file with 6 additions and 15 deletions.
diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/RedundantBraces.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/RedundantBraces.scala
@@ -161,7 +161,7 @@ class RedundantBraces(implicit val ftoks: FormatTokens)
       style: ScalafmtConfig,
   ): (Replacement, Replacement) = left.how match {
     case ReplacementType.Remove =>
-      val resOpt = getRightBraceBeforeRightParen(false).map { rb =>
+      getRightBraceBeforeRightParen(shouldBeRemoved = false).map { rb =>
         ft.meta.rightOwner match {
           case ac: Term.ArgClause => ftoks.matchingOpt(rb.left).map(ftoks.prev)
               .foreach { lb =>
@@ -178,15 +178,14 @@ class RedundantBraces(implicit val ftoks: FormatTokens)
           case _ =>
         }
         (left, removeToken)
-      }
-      resOpt.orNull
+      }.orNull
 
     case ReplacementType.Replace if {
           val lft = left.ft
           val ro = ft.meta.rightOwner
           (lft.meta.rightOwner eq ro) && lft.right.is[Token.LeftBrace]
         } =>
-      val pftOpt = getRightBraceBeforeRightParen(true)
+      val pftOpt = getRightBraceBeforeRightParen(shouldBeRemoved = true)
       def replaceIfAfterRightBrace = pftOpt.map { pft =>
         val rb = pft.left
         // move right to the end of the function
@@ -456,7 +455,7 @@ class RedundantBraces(implicit val ftoks: FormatTokens)
         checkBlockAsBody(b, d.body, noParams = d.paramClauseGroup.isEmpty)
 
       case p: Term.FunctionTerm if isFunctionWithBraces(p) =>
-        okToRemoveAroundFunctionBody(b, true)
+        okToRemoveAroundFunctionBody(b, okIfMultipleStats = true)
 
       case _: Term.If => settings.ifElseExpressions &&
         shouldRemoveSingleStatBlock(b)
@@ -607,20 +606,12 @@ class RedundantBraces(implicit val ftoks: FormatTokens)
       case _ => okIfMultipleStats
     })
 
+  @tailrec
   private def getBlockNestedPartialFunction(
       tree: Tree,
   ): Option[Term.PartialFunction] = tree match {
     case x: Term.PartialFunction => Some(x)
-    case x: Term.Block => getBlockNestedPartialFunction(x)
-    case _ => None
-  }
-
-  @tailrec
-  private def getBlockNestedPartialFunction(
-      tree: Term.Block,
-  ): Option[Term.PartialFunction] = getBlockSingleStat(tree) match {
-    case Some(x: Term.PartialFunction) => Some(x)
-    case Some(x: Term.Block) => getBlockNestedPartialFunction(x)
+    case Term.Block(x :: Nil) => getBlockNestedPartialFunction(x)
     case _ => None
   }