Add .show to Tasty.CaseDef

Author: nicolasstucki

compiler/src/dotty/tools/dotc/tastyreflect/TastyImpl.scala

@@ -436,6 +436,10 @@ object TastyImpl extends scala.tasty.Tasty {
 
   def caseDefClassTag: ClassTag[CaseDef] = implicitly[ClassTag[CaseDef]]
 
+  def CaseDefDeco(caseDef: CaseDef): AbstractCaseDef = new AbstractCaseDef {
+    def show(implicit ctx: Context, s: Show[TastyImpl.this.type]): String = s.showCaseDef(caseDef)
+  }
+
   object CaseDef extends CaseDefExtractor {
     def unapply(x: CaseDef): Option[(Pattern, Option[Term], Term)] = x match {
       case x: tpd.CaseDef @unchecked =>

library/src/scala/tasty/Tasty.scala

@@ -305,6 +305,11 @@ abstract class Tasty { tasty =>
 
   implicit def caseDefClassTag: ClassTag[CaseDef]
 
+  trait AbstractCaseDef {
+    def show(implicit ctx: Context, s: Show[tasty.type]): String
+  }
+  implicit def CaseDefDeco(caseDef: CaseDef): AbstractCaseDef
+
   val CaseDef: CaseDefExtractor
   abstract class CaseDefExtractor {
     def unapply(x: CaseDef): Option[(Pattern, Option[Term], Term)]

library/src/scala/tasty/util/Show.scala

@@ -6,6 +6,8 @@ abstract class Show[T <: Tasty with Singleton](val tasty: T) {
 
   def showTree(tree: tasty.Tree)(implicit ctx: tasty.Context): String
 
+  def showCaseDef(caseDef: tasty.CaseDef)(implicit ctx: tasty.Context): String
+
   def showTypeOrBoundsTree(tpt: tasty.TypeOrBoundsTree)(implicit ctx: tasty.Context): String
 
   def showTypeOrBounds(tpe: tasty.TypeOrBounds)(implicit ctx: tasty.Context): String

library/src/scala/tasty/util/ShowExtractors.scala

@@ -8,6 +8,9 @@ class ShowExtractors[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty
   def showTree(tree: Tree)(implicit ctx: Context): String =
     new Buffer().visitTree(tree).result()
 
+  def showCaseDef(caseDef: CaseDef)(implicit ctx: Context): String =
+    new Buffer().visitCaseDef(caseDef).result()
+
   def showTypeOrBoundsTree(tpt: TypeOrBoundsTree)(implicit ctx: Context): String =
     new Buffer().visitTypeTree(tpt).result()
 

library/src/scala/tasty/util/ShowSourceCode.scala

@@ -7,6 +7,9 @@ class ShowSourceCode[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty
   def showTree(tree: Tree)(implicit ctx: Context): String =
     (new Buffer).printTree(tree).result()
 
+  def showCaseDef(caseDef: CaseDef)(implicit ctx: Context): String =
+    (new Buffer).printCaseDef(caseDef).result()
+
   def showTypeOrBoundsTree(tpt: TypeOrBoundsTree)(implicit ctx: Context): String =
     (new Buffer).printTypeOrBoundsTree(tpt).result()
 
@@ -351,9 +354,9 @@ class ShowSourceCode[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty
     def printCases(cases: List[CaseDef], sep: String): Buffer = {
       def printSeparated(list: List[CaseDef]): Unit = list match {
         case Nil =>
-        case x :: Nil => printCase(x)
+        case x :: Nil => printCaseDef(x)
         case x :: xs =>
-          printCase(x)
+          printCaseDef(x)
           this += sep
           printSeparated(xs)
       }
@@ -468,7 +471,7 @@ class ShowSourceCode[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty
       printTypeTree(tpt)
     }
 
-    def printCase(caseDef: CaseDef): Buffer = {
+    def printCaseDef(caseDef: CaseDef): Buffer = {
       val CaseDef(pat, guard, body) = caseDef
       this += "case "
       printPattern(pat)

tests/run/tasty-custom-show/quoted_1.scala

@@ -48,6 +48,7 @@ object Macros {
 class DummyShow[T <: Tasty with Singleton](tasty0: T) extends Show[T](tasty0) {
   import tasty._
   def showTree(tree: Tree)(implicit ctx: Context): String = "Tree"
+  def showCaseDef(caseDef: CaseDef)(implicit ctx: Context): String = "CaseDef"
   def showTypeOrBoundsTree(tpt: TypeOrBoundsTree)(implicit ctx: Context): String = "TypeOrBoundsTree"
   def showTypeOrBounds(tpe: TypeOrBounds)(implicit ctx: Context): String = "TypeOrBounds"
   def showConstant(const: Constant)(implicit ctx: Context): String = "Constant"