Do not compute PackageDef members in extractor

Author: nicolasstucki

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

@@ -194,17 +194,19 @@ object TastyImpl extends scala.tasty.Tasty {
 
   type PackageDef = PackageDefinition
 
+  def PackageDefDeco(t: PackageDef): AbstractPackageDef = new AbstractPackageDef {
+    def members(implicit ctx: Context): List[Statement] = {
+      if (t.symbol.is(core.Flags.JavaDefined)) Nil // FIXME should also support java packages
+      else t.symbol.info.decls.iterator.map(FromSymbol.definition).toList
+    }
+  }
+
   def packageDefClassTag: ClassTag[PackageDef] = implicitly[ClassTag[PackageDef]]
 
   object PackageDef extends PackageDefExtractor {
-    def unapply(x: PackageDef)(implicit ctx: Context): Option[(String, List[Statement])] = x match {
+    def unapply(x: PackageDef)(implicit ctx: Context): Option[(String, PackageDef)] = x match {
       case x: PackageDefinition =>
-        // FIXME Do not do this eagerly as it forces everithing in the package to be loaded.
-        //       An alternative would be to add it as an extension method instead.
-        val definitions =
-          if (x.symbol.is(core.Flags.JavaDefined)) Nil // FIXME should also support java packages
-          else x.symbol.info.decls.iterator.map(FromSymbol.definition).toList
-        Some(x.symbol.name.toString, definitions)
+        Some((x.symbol.name.toString, FromSymbol.packageDef(x.symbol.owner)))
       case _ => None
     }
   }

library/src/scala/tasty/Tasty.scala

@@ -153,11 +153,16 @@ abstract class Tasty {
 
   type PackageDef <: Definition
 
+  trait AbstractPackageDef {
+    def members(implicit ctx: Context): List[Statement]
+  }
+  implicit def PackageDefDeco(t: PackageDef): AbstractPackageDef
+
   implicit def packageDefClassTag: ClassTag[PackageDef]
 
   val PackageDef: PackageDefExtractor
   abstract class PackageDefExtractor {
-    def unapply(x: PackageDef)(implicit ctx: Context): Option[(String, List[Statement])]
+    def unapply(x: PackageDef)(implicit ctx: Context): Option[(String, PackageDef)]
   }
 
   // ----- Parents --------------------------------------------------

library/src/scala/tasty/util/TastyPrinter.scala

@@ -78,8 +78,8 @@ class TastyPrinter[T <: Tasty with Singleton](val tasty: T) {
           case parent @ TypeTree() => this += parent
         })
         this += ", " += self += ", " ++= body += ")"
-      case PackageDef(name, members) =>
-        this += "PackageDef(" += name += ", " ++= members += ")"
+      case PackageDef(name, owner) =>
+        this += "PackageDef(" += name += ", " += owner += ")"
       case Import(expr, selectors) =>
         this += "Import(" += expr += ", " ++= selectors += ")"
       case PackageClause(pid, stats) =>

tests/pos/tasty/definitions.scala

@@ -43,7 +43,9 @@ object definitions {
                       self: Option[ValDef], body: List[Statement]) extends Definition {
     def mods: List[Modifier] = ???
   }
-  case class PackageDef(name: String, members: List[Statement]) extends Definition
+  case class PackageDef(name: String, override val owner: PackageDef) extends Definition {
+    def members: List[Statement] = ???
+  }
 
 // ------ Terms ---------------------------------