Fix #5404: Add MatchType to TASTy reflect

Author: nicolasstucki

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

@@ -13,4 +13,12 @@ trait CaseDefOpsImpl extends scala.tasty.reflect.CaseDefOps with TastyCoreImpl w
     def unapply(x: CaseDef): Some[(Pattern, Option[Term], Term)] = Some(x.pat, optional(x.guard), x.body)
   }
 
+  def TypeCaseDefDeco(caseDef: TypeCaseDef): TypeCaseDefAPI = new TypeCaseDefAPI {
+    def pattern(implicit ctx: Context): Pattern = caseDef.pat
+    def rhs(implicit ctx: Context): Term = caseDef.body
+  }
+
+  object TypeCaseDef extends TypeCaseDefExtractor {
+    def unapply(x: TypeCaseDef): Some[(TypeTree, TypeTree)] = Some((x.pat, x.body))
+  }
 }

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

@@ -26,6 +26,7 @@ trait TastyCoreImpl extends scala.tasty.reflect.TastyCore {
       type Term = tpd.Tree
 
   type CaseDef = tpd.CaseDef
+  type TypeCaseDef = tpd.CaseDef
 
   type Pattern = tpd.Tree
     type Value = tpd.Tree

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

@@ -123,6 +123,13 @@ trait TypeOrBoundsOpsImpl extends scala.tasty.reflect.TypeOrBoundsOps with Tasty
       }
     }
 
+    object MatchType extends MatchTypeExtractor {
+      def unapply(x: TypeOrBounds)(implicit ctx: Context): Option[(Type, Type, List[Type])] = x match {
+        case Types.MatchType(bound, scrutinee, cases) => Some((bound, scrutinee, cases))
+        case _ => None
+      }
+    }
+
     object ByNameType extends ByNameTypeExtractor {
       def unapply(x: TypeOrBounds)(implicit ctx: Context): Option[Type] = x match {
         case Types.ExprType(resType) => Some(resType.stripTypeVar)

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

@@ -100,6 +100,13 @@ trait TypeOrBoundsTreesOpsImpl extends scala.tasty.reflect.TypeOrBoundsTreeOps w
       }
     }
 
+    object MatchType extends MatchTypeExtractor {
+      def unapply(x: TypeOrBoundsTree)(implicit ctx: Context): Option[(Option[TypeTree], TypeTree, List[CaseDef])] = x match {
+        case x: tpd.MatchTypeTree => Some((if (x.bound == tpd.EmptyTree) None else Some(x.bound), x.selector, x.cases))
+        case _ => None
+      }
+    }
+
     object ByName extends ByNameExtractor {
       def unapply(x: TypeTree)(implicit ctx: Context): Option[TypeTree] = x match {
         case x: tpd.ByNameTypeTree => Some(x.result)

compiler/test/dotc/pos-from-tasty.blacklist

@@ -16,9 +16,6 @@ repeatedArgs213.scala
 # Error printing parent constructors that are blocks
 default-super.scala
 
-# Need to implement printing of match types
-matchtype.scala
-
 # Did not survive addition of type ascriptions in decompiled tests
 i3050.scala
 i4006b.scala

compiler/test/dotc/pos-recompilation.whitelist

@@ -376,7 +376,6 @@ lookupswitch
 macro-deprecate-dont-touch-backquotedidents
 Map
 match
-matchtype
 matthias3
 matthias4
 matthias5

library/src/scala/tasty/reflect/CaseDefOps.scala

@@ -16,4 +16,15 @@ trait CaseDefOps extends TastyCore {
     def unapply(x: CaseDef): Option[(Pattern, Option[Term], Term)]
   }
 
+
+  trait TypeCaseDefAPI {
+    def pattern(implicit ctx: Context): TypeTree
+    def rhs(implicit ctx: Context): TypeTree
+  }
+  implicit def TypeCaseDefDeco(caseDef: TypeCaseDef): TypeCaseDefAPI
+
+  val TypeCaseDef: TypeCaseDefExtractor
+  abstract class TypeCaseDefExtractor {
+    def unapply(x: TypeCaseDef): Option[(TypeTree, TypeTree)]
+  }
 }

library/src/scala/tasty/reflect/TastyCore.scala

@@ -46,6 +46,7 @@ package scala.tasty.reflect
  *                         |               +- Annotated
  *                         |               +- And
  *                         |               +- Or
+ *                         |               +- MatchType
  *                         |               +- ByName
  *                         |               +- TypeLambdaTree
  *                         |               +- Bind
@@ -54,6 +55,7 @@ package scala.tasty.reflect
  *                         +- SyntheticBounds
  *
  *  +- CaseDef
+ *  +- TypeCaseDef
  *
  *  +- Pattern --+- Value
  *               +- Bind
@@ -75,6 +77,7 @@ package scala.tasty.reflect
  *                                  +- AnnotatedType
  *                                  +- AndType
  *                                  +- OrType
+ *                                  +- MatchType
  *                                  +- ByNameType
  *                                  +- ParamRef
  *                                  +- ThisType
@@ -137,6 +140,9 @@ trait TastyCore {
   /** Branch of a pattern match or catch clause */
   type CaseDef
 
+  /** Branch of a type pattern match */
+  type TypeCaseDef
+
   /** Pattern tree of the pattern part of a CaseDef */
   type Pattern
     type Value <: Pattern

library/src/scala/tasty/reflect/TypeOrBoundsOps.scala

@@ -92,6 +92,11 @@ trait TypeOrBoundsOps extends TastyCore {
       def unapply(typeOrBounds: TypeOrBounds)(implicit ctx: Context): Option[(Type, Type)]
     }
 
+    val MatchType: MatchTypeExtractor
+    abstract class MatchTypeExtractor {
+      def unapply(typeOrBounds: TypeOrBounds)(implicit ctx: Context): Option[(Type, Type, List[Type])]
+    }
+
     val ByNameType: ByNameTypeExtractor
     abstract class ByNameTypeExtractor {
       def unapply(typeOrBounds: TypeOrBounds)(implicit ctx: Context): Option[Type]

library/src/scala/tasty/reflect/TypeOrBoundsTreeOps.scala

@@ -77,6 +77,11 @@ trait TypeOrBoundsTreeOps extends TastyCore {
       def unapply(typeOrBoundsTree: TypeOrBoundsTree)(implicit ctx: Context): Option[(TypeTree, TypeTree)]
     }
 
+    val MatchType: MatchTypeExtractor
+    abstract class MatchTypeExtractor {
+      def unapply(typeOrBoundsTree: TypeOrBoundsTree)(implicit ctx: Context): Option[(Option[TypeTree], TypeTree, List[TypeCaseDef])]
+    }
+
     val ByName: ByNameExtractor
     abstract class ByNameExtractor {
       def unapply(typeOrBoundsTree: TypeOrBoundsTree)(implicit ctx: Context): Option[TypeTree]