Provide the macro scala.tasty.Tasty context implicitly

Author: nicolasstucki

compiler/src/dotty/tools/dotc/core/Definitions.scala

@@ -686,6 +686,9 @@ class Definitions {
   def Unpickler_liftedExpr = ctx.requiredMethod("scala.runtime.quoted.Unpickler.liftedExpr")
   def Unpickler_unpickleType = ctx.requiredMethod("scala.runtime.quoted.Unpickler.unpickleType")
 
+  lazy val TastyTastyType = ctx.requiredClassRef("scala.tasty.Tasty")
+  def TastyTastyClass(implicit ctx: Context) = TastyTastyType.symbol.asClass
+
   lazy val TastyTopLevelSpliceModule = ctx.requiredModule("scala.tasty.TopLevelSplice")
     lazy val TastyTopLevelSplice_tastyContext = TastyTopLevelSpliceModule.requiredMethod("tastyContext")
 

compiler/src/dotty/tools/dotc/transform/Splicer.scala

@@ -286,6 +286,8 @@ object Splicer {
       case NamedArg(_, arg) => interpretTree(arg)
       case Ident(name) if env.contains(name) => env(name)
 
+      case Inlined(EmptyTree, Nil, expansion) => interpretTree(expansion)
+
       case _ => unexpectedTree(tree)
     }
 

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

@@ -626,6 +626,10 @@ trait Implicits { self: Typer =>
         EmptyTree
     }
 
+    def synthesizedTastyContext(formal: Type): Tree =
+      if (ctx.inTransparentMethod || enclosingInlineds.nonEmpty) ref(defn.TastyTopLevelSplice_tastyContext)
+      else EmptyTree
+
     /** If `formal` is of the form Eq[T, U], where no `Eq` instance exists for
      *  either `T` or `U`, synthesize `Eq.eqAny[T, U]` as solution.
      */
@@ -696,7 +700,8 @@ trait Implicits { self: Typer =>
         else
           trySpecialCase(defn.ClassTagClass, synthesizedClassTag,
             trySpecialCase(defn.QuotedTypeClass, synthesizedTypeTag,
-              trySpecialCase(defn.EqClass, synthesizedEq, failed)))
+              trySpecialCase(defn.TastyTastyClass, synthesizedTastyContext,
+                trySpecialCase(defn.EqClass, synthesizedEq, failed))))
     }
   }
 

library/src/scala/tasty/TopLevelSplice.scala

@@ -3,5 +3,5 @@ package scala.tasty
 /** Context in a top level ~ at inline site, intrinsically as `import TopLevelSplice._` */
 object TopLevelSplice {
   /** Compiler tasty context available in a top level ~ at inline site */
-  implicit def tastyContext: Tasty = throw new Exception("Not in transparent macro.")
+  def tastyContext: Tasty = throw new Exception("Not in transparent macro.")
 }

tests/neg/tasty-macro-assert/quoted_1.scala

@@ -12,7 +12,7 @@ object Asserts {
   object Ops
 
   transparent def macroAssert(cond: => Boolean): Unit =
-    ~impl('(cond))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(cond))
 
   def impl(cond: Expr[Boolean])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run-with-compiler/tasty-extractors-constants-2/quoted_1.scala

@@ -7,7 +7,7 @@ import scala.tasty.util._
 object Macros {
 
   transparent def testMacro: Unit =
-    ~impl(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl
 
   def impl(implicit tasty: Tasty): Expr[Unit] = {
     // 2 is a lifted constant

tests/run/i4515b/Macro_1.scala

@@ -0,0 +1,6 @@
+import scala.tasty.Tasty
+
+object Macro {
+  transparent def foo: Unit = ~fooImpl
+  def fooImpl(implicit tasty: Tasty): quoted.Expr[Unit] = '()
+}

tests/run/i4515b/Test_2.scala

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

tests/run/tasty-custom-show/quoted_1.scala

@@ -7,7 +7,7 @@ import scala.tasty.util.{TreeTraverser, Show}
 object Macros {
 
   implicit transparent def printOwners[T](x: => T): Unit =
-    ~impl('(x))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(x))
 
   def impl[T](x: Expr[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-eval/quoted_1.scala

@@ -6,7 +6,7 @@ import scala.tasty.util.TreeTraverser
 object Macros {
 
   implicit transparent def foo(i: Int): String =
-    ~impl('(i))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(i))
 
   def impl(i: Expr[Int])(implicit tasty: Tasty): Expr[String] = {
     value(i).toString.toExpr

tests/run/tasty-extractors-1/quoted_1.scala

@@ -5,7 +5,7 @@ import scala.tasty._
 object Macros {
 
   implicit transparent def printTree[T](x: => T): Unit =
-    ~impl('(x))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(x))
 
   def impl[T](x: Expr[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-extractors-2/quoted_1.scala

@@ -5,7 +5,7 @@ import scala.tasty._
 object Macros {
 
   implicit transparent def printTree[T](x: => T): Unit =
-    ~impl('(x))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(x))
 
   def impl[T](x: Expr[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-extractors-3/quoted_1.scala

@@ -7,7 +7,7 @@ import scala.tasty.util.TreeTraverser
 object Macros {
 
   implicit transparent def printTypes[T](x: => T): Unit =
-    ~impl('(x))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(x))
 
   def impl[T](x: Expr[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-extractors-constants-1/quoted_1.scala

@@ -6,7 +6,7 @@ import scala.tasty.util._
 object Macros {
 
   implicit transparent def testMacro: Unit =
-    ~impl(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl
 
   def impl(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-extractors-owners/quoted_1.scala

@@ -6,7 +6,7 @@ import scala.tasty.util.TreeTraverser
 object Macros {
 
   implicit transparent def printOwners[T](x: => T): Unit =
-    ~impl('(x))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(x))
 
   def impl[T](x: Expr[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-extractors-types/quoted_1.scala

@@ -6,7 +6,7 @@ import scala.tasty.util.TreeTraverser
 object Macros {
 
   implicit transparent def printType[T]: Unit =
-    ~impl('[T])(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('[T])
 
   def impl[T](x: Type[T])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-getfile/Macro_1.scala

@@ -4,7 +4,7 @@ import scala.tasty.{Tasty, TopLevelSplice}
 object SourceFiles {
 
   implicit transparent def getThisFile: String =
-    ~getThisFileImpl(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~getThisFileImpl
 
   private def getThisFileImpl(implicit tasty: Tasty): Expr[String] = {
     import tasty._

tests/run/tasty-linenumber-2/quoted_1.scala

@@ -9,7 +9,7 @@ class LineNumber(val value: Int) {
 object LineNumber {
 
   implicit transparent def line: LineNumber =
-    ~lineImpl(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~lineImpl
 
   def lineImpl(implicit tasty: Tasty): Expr[LineNumber] = {
     import tasty._

tests/run/tasty-linenumber/quoted_1.scala

@@ -9,7 +9,7 @@ class LineNumber(val value: Int) {
 object LineNumber {
 
   implicit transparent def line[T >: Unit <: Unit]: LineNumber =
-    ~lineImpl('[T])(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~lineImpl('[T])
 
   def lineImpl(x: Type[Unit])(implicit tasty: Tasty): Expr[LineNumber] = {
     import tasty._

tests/run/tasty-location/quoted_1.scala

@@ -7,7 +7,7 @@ case class Location(owners: List[String])
 object Location {
 
   implicit transparent def location: Location =
-    ~impl(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl
 
   def impl(implicit tasty: Tasty): Expr[Location] = {
     import tasty._

tests/run/tasty-macro-assert/quoted_1.scala

@@ -12,7 +12,7 @@ object Asserts {
   object Ops
 
   transparent def macroAssert(cond: => Boolean): Unit =
-    ~impl('(cond))(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~impl('(cond))
 
   def impl(cond: Expr[Boolean])(implicit tasty: Tasty): Expr[Unit] = {
     import tasty._

tests/run/tasty-subtyping/quoted_1.scala

@@ -5,10 +5,10 @@ import scala.tasty._
 object Macros {
 
   transparent def isTypeEqual[T, U]: Boolean =
-    ~isTypeEqualImpl('[T], '[U])(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~isTypeEqualImpl('[T], '[U])
 
   transparent def isSubTypeOf[T, U]: Boolean =
-    ~isSubTypeOfImpl('[T], '[U])(TopLevelSplice.tastyContext) // FIXME infer TopLevelSplice.tastyContext within top level ~
+    ~isSubTypeOfImpl('[T], '[U])
 
   def isTypeEqualImpl[T, U](t: Type[T], u: Type[U])(implicit tasty: Tasty): Expr[Boolean] = {
     import tasty._