Merge pull request #2962 from densh/topic/syntactic-assign

Quasiquotes: add syntactic extractor for assignment-like trees

1 files changed, 8 insertions, 0 deletions

diff --git a/src/reflect/scala/reflect/internal/TreeGen.scala b/src/reflect/scala/reflect/internal/TreeGen.scala
index a8af3e0f0e..a9f2807e9f 100644
--- a/src/reflect/scala/reflect/internal/TreeGen.scala
+++ b/src/reflect/scala/reflect/internal/TreeGen.scala
@@ -434,4 +434,12 @@ abstract class TreeGen extends macros.TreeBuilder {
     case head :: Nil => head
     case _           => gen.mkBlock(stats)
   }
+
+  /** Create a tree representing an assignment <lhs = rhs> */
+  def mkAssign(lhs: Tree, rhs: Tree): Tree = lhs match {
+    case Apply(fn, args) =>
+      Apply(atPos(fn.pos)(Select(fn, nme.update)), args :+ rhs)
+    case _ =>
+      Assign(lhs, rhs)
+  }
 }