Set positions for constructed trees manually

Author: liufengyun

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

@@ -12,6 +12,7 @@ trait TreeOpsImpl extends scala.tasty.reflect.TreeOps with CoreImpl with Helpers
   def TreeDeco(tree: Tree): TreeAPI = new TreeAPI {
     def pos(implicit ctx: Context): Position = tree.sourcePos
     def symbol(implicit ctx: Context): Symbol = tree.symbol
+    def withPos(pos: Position)(implicit ctx: Context): Unit = tree.withSpan(pos.span)
   }
 
   def PackageClauseDeco(pack: PackageClause): PackageClauseAPI = new PackageClauseAPI {

library/src-bootstrapped/scala/tasty/reflect/utils/TreeUtils.scala

@@ -14,7 +14,11 @@ trait TreeUtils {
     val rhsExpr = rhs.seal[T]
     val expr = '{
       val x = ~rhsExpr
-      ~in(('(x)).unseal.asInstanceOf[Term.Ident]).seal[Any]
+      ~{
+        val id = ('(x)).unseal.asInstanceOf[Term.Ident]
+        id.withPos(rhs.pos)
+        in(id).seal[Any]
+      }
     }
     expr.unseal
   }

library/src/scala/tasty/reflect/TreeOps.scala

@@ -45,6 +45,13 @@ trait TreeOps extends Core {
     /** Position in the source code */
     def pos(implicit ctx: Context): Position
 
+    /** Set the position of the tree
+     *
+     *  @note It only changes positions, not source file of the tree. Source
+     *        file of trees are derived from context and cannot be changed.
+     */
+    def withPos(pos: Position)(implicit ctx: Context): Unit
+
     def symbol(implicit ctx: Context): Symbol
   }
 

tests/run/reflect-select-constructor/assert_1.scala renamed to tests/run-with-compiler/reflect-select-constructor/assert_1.scala

@@ -14,10 +14,12 @@ object scalatest {
       Type.IsMethodType.unapply(tp).flatMap(tp => if tp.isImplicit then Some(true) else None).nonEmpty
 
     cond.unseal.underlyingArgument match {
-      case Term.Apply(Term.Select(lhs, op), rhs :: Nil) =>
+      case t @ Term.Apply(Term.Select(lhs, op), rhs :: Nil) =>
         let(lhs) { left =>
           let(rhs) { right =>
-            let(Term.Select.overloaded(left, op, Nil, right :: Nil)) { result =>
+            val app = Term.Select.overloaded(left, op, Nil, right :: Nil)
+            app.withPos(t.pos)
+            let(app) { result =>
               val l = left.seal[Any]
               val r = right.seal[Any]
               val b = result.seal[Boolean]
@@ -30,7 +32,9 @@ object scalatest {
       if isImplicitMethodType(f.tpe) =>
         let(lhs) { left =>
           let(rhs) { right =>
-            let(Term.Apply(Term.Select.overloaded(Term.Apply(qual, left :: Nil), op, Nil, right :: Nil), implicits)) { result =>
+            val app = Term.Select.overloaded(Term.Apply(qual, left :: Nil), op, Nil, right :: Nil)
+            app.withPos(f.pos)
+            let(Term.Apply(app, implicits)) { result =>
               val l = left.seal[Any]
               val r = right.seal[Any]
               val b = result.seal[Boolean]