Merge pull request #4627 from dotty-staging/add-tasty-ctx-file

Add Tasty compilation unit source file

Author: nicolasstucki

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

@@ -41,6 +41,8 @@ class TastyImpl(val rootContext: Contexts.Context) extends scala.tasty.Tasty { s
 
   def ContextDeco(ctx: Context): ContextAPI = new ContextAPI {
     def owner: Definition = FromSymbol.definition(ctx.owner)(ctx)
+
+    def source: java.nio.file.Path = ctx.compilationUnit.source.file.jpath
   }
 
   // ===== Id =======================================================

library/src/scala/tasty/Tasty.scala

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

tests/run/tasty-getfile.check

@@ -0,0 +1 @@
+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.getFileName.toString.toExpr
+  }
+
+}