Trees: improve detection of mod for ParamClause

scalameta · Nov 1, 2022 · 5a39e66 · 5a39e66
1 parent 814e251
commit 5a39e66
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 11 deletions.
diff --git a/scalameta/parsers/shared/src/main/scala/scala/meta/internal/parsers/ScalametaParser.scala b/scalameta/parsers/shared/src/main/scala/scala/meta/internal/parsers/ScalametaParser.scala
@@ -4475,8 +4475,9 @@ object ScalametaParser {
     if (ok) copyPos(t)(Type.AnonymousLambda(t)) else t
   }
 
-  private def toParamClause(mod: Option[Mod.ParamsType])(v: List[Term.Param]) =
-    Term.ParamClause(v, mod)
+  private def toParamClause(mod: Option[Mod.ParamsType]): List[Term.Param] => Term.ParamClause =
+    if (mod.isDefined) Term.ParamClause(_, mod)
+    else { v => Term.ParamClause(v, Term.ParamClause.getMod(v)) }
 
   private implicit class ImplicitTree[T <: Tree](private val tree: T) extends AnyVal {
 

diff --git a/scalameta/trees/shared/src/main/scala/scala/meta/Trees.scala b/scalameta/trees/shared/src/main/scala/scala/meta/Trees.scala
@@ -185,6 +185,17 @@ object Term {
       extends Member.Param
   @ast class ParamClause(values: List[Param], mod: Option[Mod.ParamsType] = None)
       extends Member.ParamClause
+  object ParamClause {
+    private[meta] def getMod(v: Seq[Param]): Option[Mod.ParamsType] =
+      v.filter(!_.is[Param.Quasi]) match {
+        case head :: tail =>
+          head.mods.collectFirst {
+            case x: Mod.Using => x
+            case x: Mod.Implicit if tail.forall(_.mods.exists(_.is[Mod.Implicit])) => x
+          }
+        case _ => None
+      }
+  }
   def fresh(): Term.Name = fresh("fresh")
   def fresh(prefix: String): Term.Name = Term.Name(prefix + Fresh.nextId())
 }

diff --git a/scalameta/trees/shared/src/main/scala/scala/meta/internal/prettyprinters/TreeSyntax.scala b/scalameta/trees/shared/src/main/scala/scala/meta/internal/prettyprinters/TreeSyntax.scala
@@ -556,7 +556,8 @@ object TreeSyntax {
             }
             m(Expr, param, " ", kw("=>"), " ", p(Expr, body))
           case Term.Function(params, body) =>
-            m(Expr, s(printParams(params), " ", kw("=>"), " ", p(Expr, body)))
+            val mod = Term.ParamClause.getMod(params)
+            m(Expr, s(printParams(params, mod), " ", kw("=>"), " ", p(Expr, body)))
         }
       case Term.QuotedMacroExpr(Term.Block(stats)) =>
         stats match {
@@ -606,7 +607,7 @@ object TreeSyntax {
         // NOTE: `implicit/using` in parameters is skipped as it applies to whole list
         printParam(t)
       case t: Term.ParamClause =>
-        printParams(t.values, needParens = !t.parent.exists(_.is[Term]))
+        printParams(t.values, t.mod, needParens = !t.parent.exists(_.is[Term]))
 
       // Type
       case t: Type.AnonymousName => m(Path, s(""))
@@ -1001,7 +1002,7 @@ object TreeSyntax {
       case t: Pkg.Object =>
         r(" ")(kw("package"), t.mods, kw("object"), t.name, t.templ)
       case t: Ctor.Primary =>
-        val paramClauses = r(t.paramClauses.map(x => printParams(x.values)))
+        val paramClauses = r(t.paramClauses.map(x => printParams(x.values, x.mod)))
         s(w(t.mods, " ", t.mods.nonEmpty && t.paramClauses.nonEmpty), paramClauses)
       case t: Ctor.Secondary =>
         if (t.stats.isEmpty)
@@ -1204,13 +1205,13 @@ object TreeSyntax {
     implicit def syntaxAnnots: Syntax[Seq[Mod.Annot]] = Syntax { annots =>
       r(annots, " ")
     }
-    private def printParams(t: Seq[Term.Param], needParens: Boolean = true): Show.Result = {
+    private def printParams(
+        t: Seq[Term.Param],
+        modOpt: Option[Mod.ParamsType],
+        needParens: Boolean = true
+    ): Show.Result = {
       val (useParens, mod) = t match {
         case head +: tail =>
-          val modOpt = head.mods.collectFirst {
-            case x: Mod.Using => x
-            case x: Mod.Implicit if tail.forall(_.mods.exists(_.is[Mod.Implicit])) => x
-          }
           val useParens = needParens || tail.nonEmpty ||
             modOpt.fold(head.decltpe.isDefined)(!_.is[Mod.Implicit])
           (useParens, o(modOpt, " "))

diff --git a/tests/shared/src/test/scala/scala/meta/tests/prettyprinters/SyntacticSuite.scala b/tests/shared/src/test/scala/scala/meta/tests/prettyprinters/SyntacticSuite.scala
@@ -706,7 +706,7 @@ class SyntacticSuite extends scala.meta.tests.parsers.ParseSuite {
               Some(Type.Name("Boolean")),
               None
             ) :: Nil,
-            None
+            Some(Mod.Implicit())
           ) :: Nil
         ),
         Template(Nil, Nil, Self(Name(""), None), Nil, Nil)