Remove quoted.Const.unapply

nicolasstucki · nicolasstucki · commit c307589f0be2 · 2020-12-02T09:13:17.000+01:00
This extractor was replaced with the more versatil extractor `quoted.Unlifted.unapply`.

* Remove definition of `quoted.Const.unapply`
* Replace uses of `Const` with `Unlifted`
* Allow unlifting precise types for primitives

**Migration**

* Replace `Const(x)` with `Unlifted(x)` if the type is known
* Replace `Const(x: Int)` with `'{ ${Unlifted(x)}: Int }` if type of expression must be checked at runtime
* Consider using `expr.unlift` and them match on `Option`
diff --git a/library/src-bootstrapped/scala/quoted/Unliftable.scala b/library/src-bootstrapped/scala/quoted/Unliftable.scala
@@ -28,58 +28,58 @@ object Unliftable {
    *  - Unlifts `'{false}` into `Some(false)`
    *  - Otherwise unlifts to `None`
    */
-  given BooleanUnliftable as Unliftable[Boolean] = new PrimitiveUnliftable
+  given BooleanUnliftable[T <: Boolean] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Byte
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Byte`
    *  - Otherwise unlifts to `None`
    */
-  given ByteUnliftable as Unliftable[Byte] = new PrimitiveUnliftable
+  given ByteUnliftable[T <: Byte] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Short
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Short`
    *  - Otherwise unlifts to `None`
    */
-  given ShortUnliftable as Unliftable[Short] = new PrimitiveUnliftable
+  given ShortUnliftable[T <: Short] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Int
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Int`
    *  - Otherwise unlifts to `None`
    */
-  given IntUnliftable as Unliftable[Int] = new PrimitiveUnliftable
+  given IntUnliftable[T <: Int] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Long
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Long`
    *  - Otherwise unlifts to `None`
    */
-  given LongUnliftable as Unliftable[Long] = new PrimitiveUnliftable
+  given LongUnliftable[T <: Long] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Float
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Float`
    *  - Otherwise unlifts to `None`
    */
-  given FloatUnliftable as Unliftable[Float] = new PrimitiveUnliftable
+  given FloatUnliftable[T <: Float] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Double
    *  - Unlifts `'{n}` into `Some(n)` for a literal `n` of type `Double`
    *  - Otherwise unlifts to `None`
    */
-  given DoubleUnliftable as Unliftable[Double] = new PrimitiveUnliftable
+  given DoubleUnliftable[T <: Double] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for Char
    *  - Unlifts `'{c}` into `Some(c)` for a literal `c` of type `Char`
    *  - Otherwise unlifts to `None`
    */
-  given CharUnliftable as Unliftable[Char] = new PrimitiveUnliftable
+  given CharUnliftable[T <: Char] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Default unliftable for String
    *  - Unlifts `'{str}` into `Some(str)` for a literal `str` of type `String`
    *  - Otherwise unlifts to `None`
    */
-  given StringUnliftable as Unliftable[String] = new PrimitiveUnliftable
+  given StringUnliftable[T <: String] as Unliftable[T] = new PrimitiveUnliftable
 
   /** Lift a quoted primitive value `'{ x }` into `x` */
-  private class PrimitiveUnliftable[T <: Int | Boolean | Byte | Short | Int | Long | Float | Double | Char | String] extends Unliftable[T] {
+  private class PrimitiveUnliftable[T <: Boolean | Byte | Short | Int | Long | Float | Double | Char | String] extends Unliftable[T] {
     def fromExpr(expr: Expr[T]) =
       import quotes.reflect._
       def rec(tree: Term): Option[T] = tree match {
@@ -135,8 +135,8 @@ object Unliftable {
    */
   given StringContextUnliftable as Unliftable[StringContext] = new {
     def fromExpr(x: Expr[StringContext]) = x match {
-      case '{ new StringContext(${Varargs(Consts(args))}: _*) } => Some(StringContext(args: _*))
-      case '{     StringContext(${Varargs(Consts(args))}: _*) } => Some(StringContext(args: _*))
+      case '{ new StringContext(${Varargs(Unlifted(args))}: _*) } => Some(StringContext(args: _*))
+      case '{     StringContext(${Varargs(Unlifted(args))}: _*) } => Some(StringContext(args: _*))
       case _ => None
     }
   }
diff --git a/library/src/scala/quoted/Const.scala b/library/src/scala/quoted/Const.scala
diff --git a/library/src/scala/quoted/Consts.scala b/library/src/scala/quoted/Consts.scala
diff --git a/tests/bench/string-interpolation-macro/Macro.scala b/tests/bench/string-interpolation-macro/Macro.scala
@@ -9,7 +9,7 @@ object Macro {
     var res: Expr[String] = null
     for _ <- 0 to 5_000 do
       (strCtxExpr, argsExpr) match {
-        case ('{ StringContext(${Varargs(Consts(parts))}: _*) }, Varargs(Consts(args))) =>
+        case ('{ StringContext(${Varargs(Unlifted(parts))}: _*) }, Varargs(Unlifted(args))) =>
           res = Expr(StringContext(parts: _*).s(args: _*))
         case _ => ???
       }
diff --git a/tests/neg-macros/i6432/Macro_1.scala b/tests/neg-macros/i6432/Macro_1.scala
@@ -9,7 +9,7 @@ object Macro {
     import quotes.reflect._
     sc match {
       case '{ StringContext(${Varargs(parts)}: _*) } =>
-        for (part @ Const(s) <- parts)
+        for (part @ Unlifted(s) <- parts)
           report.error(s, Term.of(part).pos)
     }
     '{}
diff --git a/tests/neg-macros/i6432b/Macro_1.scala b/tests/neg-macros/i6432b/Macro_1.scala
@@ -9,7 +9,7 @@ object Macro {
     import quotes.reflect._
     sc match {
       case '{ StringContext(${Varargs(parts)}: _*) } =>
-        for (part @ Const(s) <- parts)
+        for (part @ Unlifted(s) <- parts)
           report.error(s, Term.of(part).pos)
     }
     '{}
diff --git a/tests/neg-macros/inline-macro-staged-interpreter/Macro_1.scala b/tests/neg-macros/inline-macro-staged-interpreter/Macro_1.scala
@@ -10,7 +10,7 @@ object E {
 
   implicit def ev1[T: Type]: Unliftable[E[T]] = new Unliftable {
     def fromExpr(x: Expr[E[T]]) = x match {
-      case '{ I(${Const(n)}) } => Some(I(n).asInstanceOf[E[T]])
+      case '{ I(${Unlifted(n)}) } => Some(I(n).asInstanceOf[E[T]])
       case '{ Plus[T](${Value(x)}, ${Value(y)})(using $op) } if op.matches('{Plus2.IPlus}) => Some(Plus(x, y)(using Plus2.IPlus.asInstanceOf[Plus2[T]]).asInstanceOf[E[T]])
       case _ => None
     }
diff --git a/tests/neg-with-compiler/GenericNumLits/EvenFromDigitsImpl_1.scala b/tests/neg-with-compiler/GenericNumLits/EvenFromDigitsImpl_1.scala
@@ -4,8 +4,8 @@ import scala.quoted._
 import Even._
 
 object EvenFromDigitsImpl:
-  def apply(digits: Expr[String])(using Quotes): Expr[Even] = digits match {
-    case Const(ds) =>
+  def apply(digits: Expr[String])(using Quotes): Expr[Even] = digits.unlift match {
+    case Some(ds) =>
       val ev =
         try evenFromDigits(ds)
         catch {
diff --git a/tests/run-macros/expr-map-1/Macro_1.scala b/tests/run-macros/expr-map-1/Macro_1.scala
@@ -9,7 +9,7 @@ private def stringRewriter(e: Expr[Any])(using Quotes): Expr[Any] =
 private object StringRewriter extends ExprMap {
 
   def transform[T](e: Expr[T])(using Type[T])(using Quotes): Expr[T] = e match
-    case Const(s: String) =>
+    case '{ ${Unlifted(s)}: String } =>
       Expr(s.reverse) match
         case '{ $x: T } => x
         case _ => e // e had a singlton String type
diff --git a/tests/run-macros/flops-rewrite-2/Macro_1.scala b/tests/run-macros/flops-rewrite-2/Macro_1.scala
@@ -26,9 +26,9 @@ private def rewriteMacro[T: Type](x: Expr[T])(using Quotes): Expr[T] = {
             case (_, Some(_)) => '{ $y * $x }
             case _ => '{ $x * $y }
           }
-        case '{ power(${Const(x)}, ${Const(y)}) } =>
+        case '{ power(${Unlifted(x)}, ${Unlifted(y)}) } =>
           Expr(power(x, y))
-        case '{ power($x, ${Const(y)}) } =>
+        case '{ power($x, ${Unlifted(y)}) } =>
           if y == 0 then '{1}
           else '{ times($x, power($x, ${Expr(y-1)})) }
       }),
diff --git a/tests/run-macros/flops-rewrite-3/Macro_1.scala b/tests/run-macros/flops-rewrite-3/Macro_1.scala
@@ -25,9 +25,9 @@ private def rewriteMacro[T: Type](x: Expr[T])(using Quotes): Expr[T] = {
           case (_, Some(_)) => '{ $y * $x }
           case _ => '{ $x * $y }
         }
-      case '{ power(${Const(x)}, ${Const(y)}) } =>
+      case '{ power(${Unlifted(x)}, ${Unlifted(y)}) } =>
         Expr(power(x, y))
-      case '{ power($x, ${Const(y)}) } =>
+      case '{ power($x, ${Unlifted(y)}) } =>
         if y == 0 then '{1}
         else '{ times($x, power($x, ${Expr(y-1)})) }
     }
diff --git a/tests/run-macros/inline-macro-staged-interpreter/Macro_1.scala b/tests/run-macros/inline-macro-staged-interpreter/Macro_1.scala
@@ -11,8 +11,8 @@ object E {
 
   implicit def ev1[T: Type]: Unliftable[E[T]] = new Unliftable { // TODO use type class derivation
     def fromExpr(x: Expr[E[T]]) = (x match {
-      case '{ I(${Const(n)}) } => Some(I(n))
-      case '{ D(${Const(n)}) } => Some(D(n))
+      case '{ I(${Unlifted(n)}) } => Some(I(n))
+      case '{ D(${Unlifted(n)}) } => Some(D(n))
       case '{ Plus[Int](${Value(x)}, ${Value(y)})(using $op) } => Some(Plus(x, y)(using Plus2.IPlus))
       case '{ Plus[Double](${Value(x)}, ${Value(y)})(using $op) } => Some(Plus(x, y)(using Plus2.DPlus))
       case '{ Times[Int](${Value(x)}, ${Value(y)})(using $op) } => Some(Times(x, y)(using Times2.ITimes))
diff --git a/tests/run-macros/quote-matcher-string-interpolator-3/quoted_1.scala b/tests/run-macros/quote-matcher-string-interpolator-3/quoted_1.scala
@@ -8,7 +8,7 @@ object Macros {
 
   private def impl(self: Expr[StringContext], args: Expr[Seq[String]])(using Quotes): Expr[String] = {
     self match {
-      case '{ StringContext(${Varargs(Consts(parts))}: _*) } =>
+      case '{ StringContext(${Varargs(Unlifted(parts))}: _*) } =>
         val upprerParts: List[String] = parts.toList.map(_.toUpperCase)
         val upprerPartsExpr: Expr[List[String]] = Expr.ofList(upprerParts.map(Expr(_)))
         '{ StringContext($upprerPartsExpr: _*).s($args: _*) }
diff --git a/tests/run-macros/quote-matcher-symantics-1/quoted_1.scala b/tests/run-macros/quote-matcher-symantics-1/quoted_1.scala
@@ -10,7 +10,7 @@ object Macros {
 
     def lift(e: Expr[DSL]): Expr[T] = e match {
 
-      case '{ LitDSL(${ Const(c) }) } =>
+      case '{ LitDSL(${ Unlifted(c) }) } =>
         '{ $sym.value(${Expr(c)}) }
 
       case '{ ($x: DSL) + ($y: DSL) } =>
diff --git a/tests/run-macros/quote-matcher-symantics-2/quoted_1.scala b/tests/run-macros/quote-matcher-symantics-2/quoted_1.scala
@@ -21,7 +21,7 @@ object Macros {
 
     def lift(e: Expr[DSL])(implicit env: Map[Int, Expr[T]]): Expr[T] = e match {
 
-      case '{ LitDSL(${Const(c)}) } => sym.value(c)
+      case '{ LitDSL(${Unlifted(c)}) } => sym.value(c)
 
       case '{ ($x: DSL) + ($y: DSL) } => sym.plus(lift(x), lift(y))
 
@@ -35,7 +35,7 @@ object Macros {
           lift(close(body1)(nEnvVar))(env + (i -> lift(value)))
         }
 
-      case '{ envVar(${Const(i)}) } => env(i)
+      case '{ envVar(${Unlifted(i)}) } => env(i)
 
       case _ =>
         import quotes.reflect._
diff --git a/tests/run-macros/quote-matcher-symantics-3/quoted_1.scala b/tests/run-macros/quote-matcher-symantics-3/quoted_1.scala
@@ -18,15 +18,15 @@ object Macros {
     object FromEnv {
       def unapply[T](e: Expr[Any])(using env: Env): Option[Expr[R[T]]] =
         e match
-          case '{envVar[t](${Const(id)})} =>
+          case '{envVar[t](${Unlifted(id)})} =>
             env.get(id).asInstanceOf[Option[Expr[R[T]]]] // We can only add binds that have the same type as the refs
           case _ =>
             None
     }
 
     def lift[T: Type](e: Expr[T])(using env: Env): Expr[R[T]] = ((e: Expr[Any]) match {
-      case Const(e: Int) => '{ $sym.int(${Expr(e)}).asInstanceOf[R[T]] }
-      case Const(e: Boolean) => '{ $sym.bool(${Expr(e)}).asInstanceOf[R[T]] }
+      case '{ ${Unlifted(e)}: Int } => '{ $sym.int(${Expr(e)}).asInstanceOf[R[T]] }
+      case '{ ${Unlifted(e)}: Boolean } => '{ $sym.bool(${Expr(e)}).asInstanceOf[R[T]] }
 
       case '{ ($x: Int) + ($y: Int) } =>
         '{ $sym.add(${lift(x)}, ${lift(y)}).asInstanceOf[R[T]] }
diff --git a/tests/run-macros/quoted-matching-docs/Macro_1.scala b/tests/run-macros/quoted-matching-docs/Macro_1.scala
@@ -9,7 +9,7 @@ private def sumExprShow(argsExpr: Expr[Seq[Int]]) (using Quotes): Expr[String] =
 
 private def sumExpr(argsExpr: Expr[Seq[Int]])(using Quotes) : Expr[Int] = {
   UnsafeExpr.underlyingArgument(argsExpr) match {
-    case Varargs(Consts(args)) => // args is of type Seq[Int]
+    case Varargs(Unlifted(args)) => // args is of type Seq[Int]
       Expr(args.sum) // precompute result of sum
     case Varargs(argExprs) => // argExprs is of type Seq[Expr[Int]]
       val staticSum: Int = argExprs.map(_.unlift.getOrElse(0)).sum
diff --git a/tests/run-macros/tasty-extractors-constants-1.check b/tests/run-macros/tasty-extractors-constants-1.check
@@ -2,4 +2,3 @@
 4
 abc
 null
-OK
diff --git a/tests/run-macros/tasty-extractors-constants-1/quoted_1.scala b/tests/run-macros/tasty-extractors-constants-1/quoted_1.scala
@@ -11,12 +11,12 @@ object Macros {
     val buff = new StringBuilder
     def stagedPrintln(x: Any): Unit = buff append java.util.Objects.toString(x) append "\n"
 
-    Expr(3) match { case Const(n) => stagedPrintln(n) }
-    '{4} match { case Const(n) => stagedPrintln(n) }
-    '{"abc"} match { case Const(n) => stagedPrintln(n) }
+    Expr(3) match { case Unlifted(n) => stagedPrintln(n) }
+    '{4} match { case Unlifted(n) => stagedPrintln(n) }
+    '{"abc"} match { case Unlifted(n) => stagedPrintln(n) }
     '{null} match { case '{null} => stagedPrintln(null) }
 
-    '{new Object} match { case Const(n) => println(n); case _ => stagedPrintln("OK") }
+    // '{new Object} match { case Unlifted(n) => println(n); case _ => stagedPrintln("OK") }
 
     '{print(${Expr(buff.result())})}
   }

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ object Macro {`
`9`	`9`	`var res: Expr[String] = null`
`10`	`10`	`for _ <- 0 to 5_000 do`
`11`	`11`	`(strCtxExpr, argsExpr) match {`
`12`		`- case ('{ StringContext(${Varargs(Consts(parts))}: _*) }, Varargs(Consts(args))) =>`
	`12`	`+ case ('{ StringContext(${Varargs(Unlifted(parts))}: _*) }, Varargs(Unlifted(args))) =>`
`13`	`13`	`res = Expr(StringContext(parts: _).s(args: _))`
`14`	`14`	`case _ => ???`
`15`	`15`	`}`
Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ object Macro {`
`9`	`9`	`import quotes.reflect._`
`10`	`10`	`sc match {`
`11`	`11`	`case '{ StringContext(${Varargs(parts)}: _*) } =>`
`12`		`- for (part @ Const(s) <- parts)`
	`12`	`+ for (part @ Unlifted(s) <- parts)`
`13`	`13`	`report.error(s, Term.of(part).pos)`
`14`	`14`	`}`
`15`	`15`	`'{}`
Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ object E {`
`10`	`10`
`11`	`11`	`implicit def ev1[T: Type]: Unliftable[E[T]] = new Unliftable {`
`12`	`12`	`def fromExpr(x: Expr[E[T]]) = x match {`
`13`		`- case '{ I(${Const(n)}) } => Some(I(n).asInstanceOf[E[T]])`
	`13`	`+ case '{ I(${Unlifted(n)}) } => Some(I(n).asInstanceOf[E[T]])`
`14`	`14`	`case '{ Plus[T](${Value(x)}, ${Value(y)})(using $op) } if op.matches('{Plus2.IPlus}) => Some(Plus(x, y)(using Plus2.IPlus.asInstanceOf[Plus2[T]]).asInstanceOf[E[T]])`
`15`	`15`	`case _ => None`
`16`	`16`	`}`