更改TypeDeclaration

时间:2017-02-16 16:09:59

标签: java eclipse eclipse-plugin eclipse-jdt eclipse-pde

我正在尝试使用Java AST编辑几个Java类。但我的更改不会显示在Java类文件中。

我具体想做什么?我想要IPackageFragment并访问所有ICompilationUnit。对于每个声明的类,我想将超类设置为特定的类(使用超类的限定名称because it is an Xtend class)。我还尝试通过Document类应用编辑。

例如:main.model.someClass应继承自wrappers.main.model.someClassWrapper

我是JDT API的新手,所以我找不到更改类文件的原因。我已经检查了this post,但它没有帮助我。我试图尽可能接近来自Stackoverflow的How To Train the JDT Dragon其他提示/示例中的示例。但它不起作用。

我就是这样做的:

private void editTypesIn(IPackageFragment myPackage) throws JavaModelException {
    for (ICompilationUnit unit : myPackage.getCompilationUnits()) {
        TypeVisitor visitor = new TypeVisitor(myPackage.getElementName(), unit);
        unit.becomeWorkingCopy(new NullProgressMonitor());
        CompilationUnit parse = parse(unit);
        parse.recordModifications();
        parse.accept(visitor);
    }
}

private static CompilationUnit parse(ICompilationUnit unit) {
    ASTParser parser = ASTParser.newParser(AST.JLS8);
    parser.setKind(ASTParser.K_COMPILATION_UNIT);
    parser.setSource(unit);
    parser.setResolveBindings(true);
    return (CompilationUnit) parser.createAST(null); // parse
}

这是访问者类:

public class TypeVisitor extends ASTVisitor {
    private final String currentPackage;
    private final ICompilationUnit compilationUnit;

    public TypeVisitor(String currentPackage, ICompilationUnit compilationUnit) {
        this.currentPackage = currentPackage;
        this.compilationUnit = compilationUnit;
    }

    @Override
    public boolean visit(TypeDeclaration node) {
        if (!node.isInterface()) { // is class
            setSuperClass(node, "wrappers." + currentPackage + "." + node.getName().toString() + "Wrapper");
        }
        return super.visit(node);
    }

    public void setSuperClass(TypeDeclaration declaration, String qualifiedName) {
        try {
            // create ast and rewrite:
            AST ast = declaration.getAST();
            ASTRewrite astRewrite = ASTRewrite.create(ast);
            // set super:
            Name name = ast.newName(qualifiedName);
            Type type = ast.newSimpleType(name);
            declaration.setSuperclassType(type);
            // apply changes
            TextEdit edits = astRewrite.rewriteAST();
            compilationUnit.applyTextEdit(edits, new NullProgressMonitor());
            compilationUnit.commitWorkingCopy(true, new NullProgressMonitor());
        } catch (JavaModelException exception) {
            exception.printStackTrace();
        } catch (IllegalArgumentException exception) {
            exception.printStackTrace();
        } catch (MalformedTreeException exception) {
            exception.printStackTrace();
        }
    }
}

提前感谢您的帮助!

1 个答案:

答案 0 :(得分:1)

我猜你的修改会是空的,对吧?

通过调用CompilationUnit.recordModifications(),您告诉AST在内部创建一个ASTRewrite,其中将记录修改。

通过ASTRewrite.create(..),您创建了一个 second ASTRewrite,它不知道正在录制的修改。

消费记录修改的唯一API是CompilationUnit.rewrite(IDocument,Map)(请参阅对字段AST.rewriter的引用)。

如果您需要使用自己的ASTRewrite,请使用"描述性方法",使用例如ASTRewrite.set(..)