如何使用 JDT 将方法源添加到现有 java 文件？

Question

我正在编写一个简单的 eclipse 插件，它是一个代码生成器。用户可以选择一个现有的方法，然后用JDT在相应的测试文件中生成一个新的测试方法。

假设测试文件已经存在，其内容为：

public class UserTest extends TestCase {
    public void setUp(){}
    public void tearDown(){}
    public void testCtor(){}
}

现在我已经生成了一些测试代码：

/** complex javadoc */
public void testSetName() {
    ....
    // complex logic
}

我想要做的是将其附加到现有的 UserTest 中。我必须编码：

String sourceContent = FileUtils.readFileToString("UserTest.java", "UTF-8");
ASTParser parser = ASTParser.newParser(AST.JLS3);
parser.setSource(content.toCharArray());
CompilationUnit testUnit = (CompilationUnit) parser.createAST(null);

String newTestCode = "public void testSetName() { .... }";

// get the first type
final List<TypeDeclaration> list = new ArrayList<TypeDeclaration>();
testUnit .accept(new ASTVisitor() {
    @Override
    public boolean visit(TypeDeclaration node) {
        list.add(node);
        return false;
    }
});
TypeDeclaration type = list.get(0);

type.insertMethod(newTestCode); // ! no this method: insertMethod 

但是没有这样的方法insertMethod。

我现在知道两个选择：

1. 不要使用jdt，只是将新代码插入到测试文件中，在最后一个}之前
2. 使用testUnit.getAST().newMethodDeclaration () 创建一个方法，然后更新它。

但我不喜欢这两个选项，我希望有像 insertMethod 这样的东西，它可以让我向测试编译单元附加一些文本，或者将测试代码转换为 MethodDeclaration，然后附加到测试编译单元。

---

更新

我看到nonty的回答，发现jdt中有两个CompilationUnit。一个是 org.eclipse.jdt.internal.core.CompilationUnit，另一个是 org.eclipse.jdt.core.dom.CompilationUnit。我用的是第二个，没有人用过第一个。

我需要补充一下我的问题：一开始我想创建一个eclipse插件，但后来我发现很难通过swt创建一个复杂的UI，所以我决定创建一个Web应用程序来生成代码。我从 eclipse 复制了这些 jdt jar，因此我可以使用 org.eclipse.jdt.core.dom.CompilationUnit。

有没有办法在 eclipse 之外使用 org.eclipse.jdt.internal.core.CompilationUnit ？

Answer 1

有什么问题

...
String newTestCode = "public void testSetName() { .... }";
IProgressMonitor monitor = new NullProgressMonitor();

IMethod method = testUnit.getTypes[0].createMethod(newTestCode, null, false, monitor);

您甚至可以指定在哪里添加新方法以及是否“覆盖”任何现有方法。

Answer 2

JavaCore 有一个方法可以从给定的 IFile 创建 CompilationUnit。

IFile file;    
JavaCore.createCompilationUnitFrom(file);