Объединение переданного тела функции в выражение, переписанное макросом

Если есть сомнения, resetAllAttrs:

import scala.language.experimental.macros
import scala.reflect.macros.BlackboxContext

def _forRange[A](c: BlackboxContext)(range: c.Expr[Range])(
  func: c.Expr[Int => A]
): c.Expr[Unit] = {
  import c.universe._

  val tree = func.tree match {
    case q"($i: $t) => $body" => q"""
        val iter = ${range}.iterator
        while (iter.hasNext) {
          val $i = iter.next
          ${c.resetAllAttrs(body)} // The only line I've changed.
        }
      """
    case _ => q""
  }

  c.Expr(tree)
}

А потом:

scala> def forRange[A](range: Range)(func: Int => A) = macro _forRange[A]
defined term macro forRange: [A](range: Range)(func: Int => A)Unit

scala> forRange(0 to 10) { i => println(i) }
0
1
2
3
4
5
6
7
8
9
10

В общем, когда вы берете дерево из одного места и шлепаете его куда-то еще, вероятно, потребуется использовать resetAllAttrs, чтобы получить все символы правильно.

Оскар Бойкин указал в Твиттере, что мой предыдущий ответ больше не работает, и в любом случае это был не очень полный ответ— он решает проблему, указанную в OP на Scala 2.10, но не заботится о гигиене - например, если вы написали iter => println(iter), вы получите сбой во время компиляции.

Лучшей реализацией для версии 2.11 было бы использование Transformer для перезаписи дерева после отмены проверки типов:

import scala.language.experimental.macros
import scala.reflect.macros.blackbox.Context

def _forRange[A](c: Context)(r: c.Expr[Range])(f: c.Expr[Int => A]): c.Tree = {
  import c.universe._

  f.tree match {
    case q"($i: $_) => $body" =>
      val newName = TermName(c.freshName())
      val transformer = new Transformer {
        override def transform(tree: Tree): Tree = tree match {
          case Ident(`i`) => Ident(newName)
          case other => super.transform(other)
        }
      }

      q"""
        val iter = ${r.tree}.iterator
        while (iter.hasNext) {
          val $newName = iter.next
          ${ transformer.transform(c.untypecheck(body)) }
        }
      """
  }
}

def forRange[A](r: Range)(f: Int => A): Unit = macro _forRange[A]

Что работает следующим образом:

scala> forRange(0 to 10)((i: Int) => println(i))
0
1
2
3
4
5
6
7
8
9
10

Теперь не имеет значения, какое имя переменной мы используем в нашем литерале функции, так как оно все равно будет заменено новой переменной.