add CompilationUnitInfo to the CompilationUnit

Author: bishabosha

compiler/src/dotty/tools/dotc/CompilationUnit.scala

@@ -19,7 +19,7 @@ import scala.annotation.internal.sharable
 import scala.util.control.NoStackTrace
 import transform.MacroAnnotations
 
-class CompilationUnit protected (val source: SourceFile) {
+class CompilationUnit protected (val source: SourceFile, val info: CompilationUnitInfo | Null) {
 
   override def toString: String = source.toString
 
@@ -106,7 +106,7 @@ class CompilationUnit protected (val source: SourceFile) {
     myAssignmentSpans.nn
 }
 
-@sharable object NoCompilationUnit extends CompilationUnit(NoSource) {
+@sharable object NoCompilationUnit extends CompilationUnit(NoSource, info = null) {
 
   override def isJava: Boolean = false
 
@@ -122,13 +122,14 @@ object CompilationUnit {
 
   /** Make a compilation unit for top class `clsd` with the contents of the `unpickled` tree */
   def apply(clsd: ClassDenotation, unpickled: Tree, forceTrees: Boolean)(using Context): CompilationUnit =
-    val file = clsd.symbol.associatedFile.nn
-    apply(SourceFile(file, Array.empty[Char]), unpickled, forceTrees)
+    val compilationUnitInfo = clsd.symbol.compilationUnitInfo.nn
+    val file = compilationUnitInfo.associatedFile
+    apply(SourceFile(file, Array.empty[Char]), unpickled, forceTrees, compilationUnitInfo)
 
   /** Make a compilation unit, given picked bytes and unpickled tree */
-  def apply(source: SourceFile, unpickled: Tree, forceTrees: Boolean)(using Context): CompilationUnit = {
+  def apply(source: SourceFile, unpickled: Tree, forceTrees: Boolean, info: CompilationUnitInfo)(using Context): CompilationUnit = {
     assert(!unpickled.isEmpty, unpickled)
-    val unit1 = new CompilationUnit(source)
+    val unit1 = new CompilationUnit(source, info)
     unit1.tpdTree = unpickled
     if (forceTrees) {
       val force = new Force
@@ -156,7 +157,8 @@ object CompilationUnit {
         NoSource
       }
       else source
-    new CompilationUnit(src)
+    val info = if src.exists then CompilationUnitInfo(src.file) else null
+    new CompilationUnit(src, info)
   }
 
   /** Force the tree to be loaded */

compiler/src/dotty/tools/dotc/fromtasty/AlreadyLoadedCompilationUnit.scala

@@ -7,4 +7,4 @@ import dotty.tools.dotc.util.NoSource
  *  encountered, and attempted to inspect, something that has already been loaded, for example a Scala primitive or a
  *  library class like Option.
  */
-class AlreadyLoadedCompilationUnit(val className: String) extends CompilationUnit(NoSource)
+class AlreadyLoadedCompilationUnit(val className: String) extends CompilationUnit(NoSource, null)

compiler/src/dotty/tools/dotc/fromtasty/TASTYCompilationUnit.scala

@@ -3,6 +3,6 @@ package dotty.tools.dotc.fromtasty
 import dotty.tools.dotc.CompilationUnit
 import dotty.tools.dotc.util.NoSource
 
-class TASTYCompilationUnit(val className: String) extends CompilationUnit(NoSource) {
+class TASTYCompilationUnit(val className: String) extends CompilationUnit(NoSource, null) {
   override def toString: String = s"class file $className"
 }

compiler/src/dotty/tools/dotc/typer/Namer.scala

@@ -248,7 +248,7 @@ class Namer { typer: Typer =>
         val cls =
           createOrRefine[ClassSymbol](tree, name, flags, ctx.owner,
             cls => adjustIfModule(new ClassCompleter(cls, tree)(ctx), tree),
-            newClassSymbol(ctx.owner, name, _, _, _, tree.nameSpan, CompilationUnitInfo(ctx.source.file)))
+            newClassSymbol(ctx.owner, name, _, _, _, tree.nameSpan, ctx.compilationUnit.info))
         cls.completer.asInstanceOf[ClassCompleter].init()
         cls
       case tree: MemberDef =>

compiler/src/dotty/tools/repl/ReplCompiler.scala

@@ -4,6 +4,7 @@ import dotty.tools.dotc.ast.Trees.*
 import dotty.tools.dotc.ast.{tpd, untpd}
 import dotty.tools.dotc.ast.tpd.TreeOps
 import dotty.tools.dotc.core.Contexts.*
+import dotty.tools.dotc.core.CompilationUnitInfo
 import dotty.tools.dotc.core.Decorators.*
 import dotty.tools.dotc.core.Flags.*
 import dotty.tools.dotc.core.Names.*
@@ -233,7 +234,7 @@ object ReplCompiler:
   val objectNames = mutable.Map.empty[Int, TermName]
 end ReplCompiler
 
-class ReplCompilationUnit(source: SourceFile) extends CompilationUnit(source):
+class ReplCompilationUnit(source: SourceFile) extends CompilationUnit(source, CompilationUnitInfo(source.file)):
   override def isSuspendable: Boolean = false
 
 /** A placeholder phase that receives parse trees..

staging/src/scala/quoted/staging/ExprCompilationUnit.scala

@@ -5,4 +5,4 @@ import dotty.tools.dotc.CompilationUnit
 import dotty.tools.dotc.util.NoSource
 
 /** Compilation unit containing the contents of a quoted expression */
-private class ExprCompilationUnit(val exprBuilder: Quotes => Expr[?]) extends CompilationUnit(NoSource)
+private class ExprCompilationUnit(val exprBuilder: Quotes => Expr[?]) extends CompilationUnit(NoSource, null)

staging/src/scala/quoted/staging/QuoteCompiler.scala

@@ -95,8 +95,9 @@ private class QuoteCompiler extends Compiler:
               val classTree = ClassDef(cls, DefDef(cls.primaryConstructor.asTerm), run :: Nil)
               val tree = PackageDef(ref(defn.RootPackage).asInstanceOf[Ident], classTree :: Nil).withSpan(pos)
               val source = SourceFile.virtual("<quoted.Expr>", "")
+              val unitInfo = CompilationUnitInfo(source.file, tastyInfo = None)
               result = Left(outputClassName.toString)
-              Some(CompilationUnit(source, tree, forceTrees = true))
+              Some(CompilationUnit(source, tree, forceTrees = true, unitInfo))
       }
 
     /** Get the literal value if this tree only contains a literal tree */