Scala Compiler Pluginに新しいクラスを追加するには？

Question

Scala Compiler Pluginでは、既存の特性を実装する新しいクラスを作成しようとしています。これまでのところ、私のコードは次のようになります：

def trait2Impl(original: ClassDef, newName: String): ClassDef = { 
    val impl = original.impl 
    // Seems OK to have same self, but does not make sense to me ... 
    val self = impl.self 
    // TODO: implement methods ... 
    val body = impl.body 
    // We implement original 
    val parents = original :: impl.parents 
    val newImpl = treeCopy.Template(impl, parents, self, body) 
    val name = newTypeName(newName) 
    // We are a syntheic class, not a user-defined trait 
    val mods = (original.mods | SYNTHETIC) &~ TRAIT 
    val tp = original.tparams 
    val result = treeCopy.ClassDef(original, mods, name, tp, newImpl) 
    // Same Package? 
    val owner = original.symbol.owner 
    // New symbol. What's a Position good for? 
    val symbol = new TypeSymbol(owner, NoPosition, name) 
    result.setSymbol(symbol) 
    symbol.setFlag(SYNTHETIC) 
    symbol.setFlag(ABSTRACT) 
    symbol.resetFlag(INTERFACE) 
    symbol.resetFlag(TRAIT) 
    owner.info.decls.enter(symbol) 
    result 
} 

しかし、パッケージには追加されません。私はそれが実際にパッケージが生成を引き起こす特性の前に "横断"されたか、TypingTransformerの "def def transform（tree：Tree）：Tree"メソッドがという1つしか返さないためであると思われます Tree、for受け取ったすべてのツリー。新しいツリーを実際に作成することはできませんが、変更するのは1つだけです。

既存のパッケージに新しいクラスを追加するにはどうすればよいですか？もし私が "transform（Tree）"を取得したときにパッケージを変換すると動作するかもしれませんが、パッケージの内容はまだ分かりませんので、新しいClassを生成することはできません。 。あるいは、Symbolの "Position"パラメータに関連しているかもしれませんか？

これまでのところ、ツリーが変更されていても、コンパイラプラグインで完全に新しいクラスが作成された例はありませんでした。

Answer 1

完全なソースコードはここにある：https://gist.github.com/1794246

トリックは、新しく作成されたClassDef Sを保存し、新しいPackageDefを作成するときにそれらを使用することです。シンボルとツリーの両方を処理する必要があることに注意してください。パッケージシンボルは単なるハンドルです。コードを生成するには、ASTを生成する必要があります（クラスの場合と同様に、シンボルにクラス名と型が含まれていますが、コードはClassDefツリーにあります）。

あなたが指摘したように、パッケージ定義はクラスよりもツリーの上にあるので、最初に再帰する必要があります（既存のクラスから新しいクラスを生成すると仮定します）。次に、サブツリーが一度トラバースされると、新しいクラスを持つ新しいPackageDef（すべてのコンパイル単位にはパッケージ定義があり、デフォルトでは空のパッケージです）を準備することができます。ソースコードを想定した例で

は、

class Foo { 
    def foo { 
    "spring" 
    } 
} 

コンパイラは

package <empty> { 
    class Foo { 
    def foo { 
     "spring" 
    } 
    } 
} 

にそれをラップし、プラグインは、あなたが問題を解決しました

package <empty> { 
    class Foo { 
    def foo { 
     "spring" 
    } 
    } 
    package mypackage { 
    class MyClass extends AnyRef 
    } 
}