Support source-future (#910)

scalacenter · Jul 27, 2023 · ee167dc · ee167dc
1 parent 68c5910
commit ee167dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/instrumentation/src/main/scala/com.olegych.scastie.instrumentation/Instrument.scala b/instrumentation/src/main/scala/com.olegych.scastie.instrumentation/Instrument.scala
@@ -161,7 +161,10 @@ object Instrument {
   }
 
   def apply(code: String, target: ScalaTarget): Either[InstrumentationFailure, String] = {
-    val runtimeImport = "import _root_.com.olegych.scastie.api.runtime._"
+    val runtimeImport = target match {
+      case Scala3(scalaVersion) => "import _root_.com.olegych.scastie.api.runtime.*"
+      case _ => "import _root_.com.olegych.scastie.api.runtime._"
+    }
 
     val isScalaJs = target.targetType == ScalaTargetType.JS