Làm thế nào để thêm một Class mới trong một Plugin Scala Compiler?

Question

Trong Trình biên dịch Scala Plugin, tôi đang cố gắng tạo một lớp mới thực hiện một đặc điểm đã có từ trước. Cho đến nay mã của tôi trông giống như sau:

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 
} 

Nhưng dường như mã này không được thêm vào gói. Tôi nghi ngờ đó là bởi vì thực sự các gói đã "đi ngang" trước khi các đặc điểm gây ra các thế hệ, và/hoặc vì "ghi đè def chuyển đổi (cây: Cây): Cây" phương pháp của TypingTransformer chỉ có thể trả lại một Tree, cho mọi cây mà nó nhận được, vì vậy nó không thể thực sự tạo ra một cây mới, mà chỉ thay đổi một cây.

Vì vậy, làm cách nào để bạn thêm một Lớp mới vào gói hiện có? Có lẽ nó sẽ làm việc nếu tôi chuyển đổi gói khi "transform (Tree)" nhận được nó, nhưng tôi điểm đó tôi không biết nội dung của gói, vì vậy tôi không thể tạo ra Class mới sớm (hoặc có thể tôi?) . Hoặc có thể nó liên quan đến tham số "Position" của Symbol?

Cho đến nay tôi đã tìm thấy một số ví dụ về nơi Cây được sửa đổi, nhưng không có nơi nào một Lớp hoàn toàn mới được tạo trong một Plugin Trình biên dịch.

Answer 1

Mã nguồn đầy đủ là ở đây: https://gist.github.com/1794246

Bí quyết là để lưu trữ các mới được tạo ra ClassDef s và sử dụng chúng khi tạo mới PackageDef. Lưu ý rằng bạn cần phải đối phó với cả Biểu tượng và cây: biểu tượng gói chỉ là một tay cầm. Để tạo mã, bạn cần tạo một AST (giống như một lớp, nơi biểu tượng giữ tên và kiểu của lớp, nhưng mã nằm trong các cây ClassDef). Như bạn đã lưu ý, định nghĩa gói cao hơn cây so với các lớp, vì vậy bạn cần phải recurse đầu tiên (giả sử bạn sẽ tạo lớp mới từ một lớp hiện có). Sau đó, khi các subtrees được duyệt, bạn có thể chuẩn bị một PackageDef mới (mỗi đơn vị biên dịch có một định nghĩa gói, theo mặc định là gói rỗng) với các lớp mới.

Trong ví dụ này, giả sử mã nguồn là

class Foo { 
    def foo { 
    "spring" 
    } 
} 

trình biên dịch kết thúc tốt đẹp vào

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

và các plugin biến nó thành

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