Add Tasty compilation unit source file

Author: nicolasstucki

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

@@ -41,6 +41,12 @@ object TastyImpl extends scala.tasty.Tasty {
 
   def ContextDeco(ctx: Context): AbstractContext = new AbstractContext {
     def owner: Definition = FromSymbol.definition(ctx.owner)(ctx)
+
+    def source: String = {
+      val thisPath = java.nio.file.Paths.get("").toAbsolutePath
+      val path = ctx.compilationUnit.source.file.jpath.toAbsolutePath
+      thisPath.relativize(path).toString
+    }
   }
 
   // ===== Id =======================================================

library/src/scala/tasty/Tasty.scala

@@ -31,6 +31,9 @@ abstract class Tasty { tasty =>
 
   trait AbstractContext {
     def owner: Definition
+
+    /** Returns the source file being compiled. The path is relative to the current working directory. */
+    def source: String
   }
   implicit def ContextDeco(ctx: Context): AbstractContext
 

tests/run/tasty-getfile.check

@@ -0,0 +1 @@
+tests/run/tasty-getfile/App_2.scala

tests/run/tasty-getfile/App_2.scala

@@ -0,0 +1,6 @@
+
+object Test {
+  def main(args: Array[String]): Unit = {
+    println(SourceFiles.getThisFile)
+  }
+}

tests/run/tasty-getfile/Macro_1.scala

@@ -0,0 +1,15 @@
+import scala.quoted._
+
+import scala.tasty.Universe
+
+object SourceFiles {
+
+  implicit inline def getThisFile: String =
+    ~getThisFileImpl(Universe.compilationUniverse) // FIXME infer Universe.compilationUniverse within top level ~
+
+  private def getThisFileImpl(implicit u: Universe): Expr[String] = {
+    import u.tasty._
+    u.context.source.toExpr
+  }
+
+}