diff options
author | Danny van Bruggen <hexagonaal@gmail.com> | 2017-04-23 00:42:40 +0200 |
---|---|---|
committer | Danny van Bruggen <hexagonaal@gmail.com> | 2017-04-23 00:42:40 +0200 |
commit | 826d969210baf896f9a9ce785de9e64b58960604 (patch) | |
tree | 13be70a0d28c9d1a01f7565534438e3fa46f7c61 /javaparser-core-generators/src/main/java/com/github/javaparser | |
parent | 2769f2497af191da82e8718bfd0db65d64813e61 (diff) | |
download | javaparser-826d969210baf896f9a9ce785de9e64b58960604.tar.gz |
Mark all generated getMetaModel methods generated
Diffstat (limited to 'javaparser-core-generators/src/main/java/com/github/javaparser')
-rw-r--r-- | javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java index 62ff9e06a..48ab86b1a 100644 --- a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java +++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetMetaModelGenerator.java @@ -3,7 +3,9 @@ package com.github.javaparser.generator.core.node; import com.github.javaparser.JavaParser; import com.github.javaparser.ast.CompilationUnit; import com.github.javaparser.ast.Node; +import com.github.javaparser.ast.body.BodyDeclaration; import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration; +import com.github.javaparser.ast.body.MethodDeclaration; import com.github.javaparser.generator.NodeGenerator; import com.github.javaparser.metamodel.BaseNodeMetaModel; import com.github.javaparser.metamodel.JavaParserMetaModel; @@ -19,15 +21,14 @@ public class GetMetaModelGenerator extends NodeGenerator { @Override protected void generateNode(BaseNodeMetaModel nodeMetaModel, CompilationUnit nodeCu, ClassOrInterfaceDeclaration nodeCoid) { - nodeCoid.getMethodsByName("getMetaModel").forEach(Node::remove); - - nodeCu.addImport(nodeMetaModel.getClass().getName()); - nodeCu.addImport(JavaParserMetaModel.class); - final String method = f("%s public %s getMetaModel() { return JavaParserMetaModel.%s; }", + final MethodDeclaration getMetaModelMethod = (MethodDeclaration)parseClassBodyDeclaration(f("%s public %s getMetaModel() { return JavaParserMetaModel.%s; }", nodeMetaModel.isRootNode() ? "" : "@Override", nodeMetaModel.getClass().getSimpleName(), - nodeMetaModel.getMetaModelFieldName()); + nodeMetaModel.getMetaModelFieldName())); - nodeCoid.addMember(parseClassBodyDeclaration(method)); + addOrReplaceWhenSameSignature(nodeCoid, getMetaModelMethod); + nodeCu.addImport(nodeMetaModel.getClass().getName()); + nodeCu.addImport(JavaParserMetaModel.class); + markGenerated(getMetaModelMethod); } } |