Tworzenie drzewa definicji metody z symbolu metody i obiektu

Question

Czy istnieje wygodny sposób przekształcania obiektu MethodSymbol w lewą stronę drzewa definicji metody (to jest, DefDef) w Scali 2.10?

Załóżmy na przykład, że chcę utworzyć makro, które zajmie instancję cechy i zawinie wszystkie te cechy, korzystając z niektórych funkcji debugowania. Mogę napisać następujące:

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 

    //... 
} 

Mam pomijana nudny biznes od przyklejania tych metod w nowej klasie anonimowej implementującej cechę, a następnie instancji klasy, które można znaleźć kompletny pracy przykład here jeśli was” jestem zainteresowany.

Teraz mogę napisać to na przykład:

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 

Tak to działa, ale tylko w bardzo prostych przypadkach, to nie będzie, jeśli cecha ma metod z listami parametrów, z modyfikatorów dostępu, adnotacje, itd.

To, czego naprawdę chcę, to funkcja, która pobiera symbol metody i drzewo dla nowego obiektu i zwraca wartość DefDef. Zacząłem pisać jedną ręką, ale to wymaga dużo fiddly rzeczy tak:

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

Która jest denerwujące, gadatliwe i podatne na błędy. Czy brakuje mi lepszego sposobu na to w nowym interfejsie API Reflection?

Answer 1

Zgodnie z moją wiedzą, nie ma standardowego sposobu przejścia z symbolu do drzewa definiującego.

Najlepszym rozwiązaniem byłoby prawdopodobnie powtórzenie przez c.enclosingRun.units, powtarzając w każdym z drzew unit.body. Jeśli zobaczysz DefDef, który ma symbol równy twojemu symbolowi, to osiągniesz cel. upd. Nie zapomnij o duplicate definiowaniu drzewa przed ponownym użyciem!

Ta technika jest daleka od bycia najwygodniejszą rzeczą na świecie, ale powinna działać.

Answer 2

Możesz wypróbować następujące rzeczy. Działa to z wieloma parametrami i funkcjami curry oraz parametrami typu;)

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