Add ReplCompilationUnit

Author: nicolasstucki

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

@@ -14,7 +14,6 @@ import typer.Nullables
 import transform.SymUtils._
 import core.Decorators.{given _}
 import config.SourceVersion
-import StdNames.str
 
 class CompilationUnit protected (val source: SourceFile) {
 
@@ -27,9 +26,6 @@ class CompilationUnit protected (val source: SourceFile) {
   /** Is this the compilation unit of a Java file */
   def isJava: Boolean = source.file.name.endsWith(".java")
 
-  /** Is this the compilation unit of a REPL input */
-  def isREPL: Boolean = source.file.name.startsWith(str.REPL_SESSION_LINE)
-
   /** The source version for this unit, as determined by a language import */
   var sourceVersion: Option[SourceVersion] = None
 
@@ -53,7 +49,7 @@ class CompilationUnit protected (val source: SourceFile) {
   var suspended: Boolean = false
 
   /** Can this compilation unit be suspended */
-  def isSuspendable: Boolean = !isREPL
+  def isSuspendable: Boolean = true
 
   /** Suspends the compilation unit by thowing a SuspendException
    *  and recoring the suspended compilation unit

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

@@ -146,7 +146,7 @@ class ReplCompiler extends Compiler {
     val objectTermName = ctx.source.file.toString.toTermName
     objectNames.update(defs.state.objectIndex, objectTermName)
 
-    val unit = CompilationUnit(ctx.source)
+    val unit = new ReplCompilationUnit(ctx.source)
     unit.untpdTree = wrapped(defs, objectTermName, span)
     unit
   }

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

@@ -0,0 +1,8 @@
+package dotty.tools.repl
+
+import dotty.tools.dotc.CompilationUnit
+import dotty.tools.dotc.util.SourceFile
+
+
+class ReplCompilationUnit(source: SourceFile) extends CompilationUnit(source):
+  override def isSuspendable: Boolean = false