Tạo cây định nghĩa phương thức từ biểu tượng phương thức và một nội dung

Question

Có cách nào thuận tiện để biến một số MethodSymbol sang bên trái của cây định nghĩa phương thức (ví dụ: DefDef) trong Scala 2.10 không?

Ví dụ: giả sử tôi muốn tạo macro sẽ lấy một cá thể của một đặc điểm và bao bọc tất cả các phương pháp của đặc điểm đó bằng một số chức năng gỡ lỗi. Tôi có thể viết như sau:

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

object WrapperExample { 
    def wrap[A](a: A): A = macro wrap_impl[A] 

    def wrap_impl[A: c.WeakTypeTag](c: Context)(a: c.Expr[A]) = { 
    import c.universe._ 

    val wrapped = weakTypeOf[A] 
    val f = Select(reify(Predef).tree, "println") 

    val methods = wrapped.declarations.collect { 
     case m: MethodSymbol if !m.isConstructor => DefDef(
     Modifiers(Flag.OVERRIDE), 
     m.name, 
     Nil, Nil, 
     TypeTree(), 
     Block(
      Apply(f, c.literal("Calling: " + m.name.decoded).tree :: Nil), 
      Select(a.tree, m.name) 
     ) 
    ) 
    }.toList 

    //... 
} 

Tôi đã elided kinh doanh nhàm chán của những phương pháp này gắn bó trong một lớp ẩn danh mới mà thực hiện các đặc điểm và sau đó instantiating rằng đẳng cấp bạn có thể tìm thấy trọn vẹn một ví dụ làm việc here nếu bạn' lại quan tâm.

Bây giờ tôi có thể viết những dòng này, ví dụ:

scala> trait X { def foo = 1; def bar = 'a } 
defined trait X 

scala> val x = new X {} 
x: X = $anon$1@15dd533 

scala> val w: X = WrapperExample.wrap[X](x) 
w: X = $1$$1@27c3a4a3 

scala> w.foo 
Calling: foo 
res0: Int = 1 

scala> w.bar 
Calling: bar 
res1: Symbol = 'a 

Vì vậy, nó hoạt động, nhưng chỉ trong trường hợp rất đơn giản-nó sẽ không nếu đặc điểm có phương pháp với danh sách tham số, với bổ truy cập, chú thích, v.v.

Điều tôi thực sự muốn là hàm sẽ lấy biểu tượng phương thức và cây cho phần nội dung mới và trả lại DefDef. Tôi đã bắt đầu viết một bằng tay, nhưng nó liên quan đến rất nhiều thứ khó sử dụng như thế này:

List(if (method.isImplicit) Some(Flag.IMPLICIT) else None, ...) 

Đó là gây phiền nhiễu, tiết và dễ bị lỗi. Tôi có thiếu một số cách hay hơn để thực hiện điều này trong API phản chiếu mới không?

Answer 1

Theo hiểu biết tốt nhất của tôi, không có cách nào tiêu chuẩn để đi từ biểu tượng đến cây xác định.

Đặt cược tốt nhất của bạn có thể sẽ lặp lại thông qua c.enclosingRun.units, đệ quy vào từng cây unit.body khi bạn di chuyển. Nếu bạn thấy một số DefDef, có số symbol bằng với ký hiệu của bạn, thì bạn đã đến đích. cập nhật. Đừng quên duplicate cây xác định trước khi sử dụng lại nó!

Kỹ thuật này không phải là điều thuận tiện nhất trên thế giới, nhưng nó sẽ hoạt động.

Answer 2

Bạn có thể thử cách sau. Điều này làm việc với nhiều tham số và chức năng được curried và các thông số loại;)

val methods = wrapped.declarations.collect { 
    case m: MethodSymbol if !m.isConstructor => DefDef(
    Modifiers(Flag.OVERRIDE), 
    m.name, 
    m.typeParams.map(TypeDef(_)), 
    m.paramss.map(_.map(ValDef(_))), 
    TypeTree(m.returnType), 
    Block(
     Apply(f, c.literal("Calling: " + m.name.decoded).tree :: Nil), 
     m.paramss.foldLeft(Select(a.tree.duplicate, m.name): Tree)((prev, params) => 
     Apply(prev, params.map(p => Ident(p.name))) 
    ) 
    ) 
) 
}.toList