diff options
author | Danny van Bruggen <hexagonaal@gmail.com> | 2017-04-23 00:47:44 +0200 |
---|---|---|
committer | Danny van Bruggen <hexagonaal@gmail.com> | 2017-04-23 00:47:44 +0200 |
commit | a0950452a16fa9ed5e16ec8bd0750617f1dae35d (patch) | |
tree | a41f7fa16892c715c6679cce9b3690b5492bff3c /javaparser-core-generators/src | |
parent | 826d969210baf896f9a9ce785de9e64b58960604 (diff) | |
download | javaparser-a0950452a16fa9ed5e16ec8bd0750617f1dae35d.tar.gz |
Mark all generated getNodeLists methods generated
Diffstat (limited to 'javaparser-core-generators/src')
-rw-r--r-- | javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java | 24 |
1 files changed, 6 insertions, 18 deletions
diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java index 676d14cd9..85550ce99 100644 --- a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java +++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/node/GetNodeListsGenerator.java @@ -1,18 +1,13 @@ 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.ClassOrInterfaceDeclaration; import com.github.javaparser.ast.body.MethodDeclaration; -import com.github.javaparser.ast.stmt.BlockStmt; import com.github.javaparser.generator.NodeGenerator; -import com.github.javaparser.utils.SeparatedItemStringBuilder; -import com.github.javaparser.utils.SourceRoot; import com.github.javaparser.metamodel.BaseNodeMetaModel; import com.github.javaparser.metamodel.PropertyMetaModel; - -import java.util.List; +import com.github.javaparser.utils.SeparatedItemStringBuilder; +import com.github.javaparser.utils.SourceRoot; import static com.github.javaparser.JavaParser.parseClassBodyDeclaration; import static com.github.javaparser.utils.CodeGenerationUtils.f; @@ -38,20 +33,13 @@ public class GetNodeListsGenerator extends NodeGenerator { } } } - - List<MethodDeclaration> getNodeListsMethods = nodeCoid.getMethodsByName("getNodeLists"); + if (!statement.hasItems()) { - getNodeListsMethods.forEach(Node::remove); - return; - } - - if (getNodeListsMethods.isEmpty()) { - nodeCoid.addMember(parseClassBodyDeclaration(f("@Override public List<NodeList<?>> getNodeLists() {%s}", statement))); return; } - BlockStmt block = getNodeListsMethods.get(0).getBody().get(); - block.getStatements().clear(); - block.addStatement(statement.toString()); + final MethodDeclaration getNodeListsMethod = (MethodDeclaration) parseClassBodyDeclaration(f("@Override public List<NodeList<?>> getNodeLists() {%s}", statement)); + addOrReplaceWhenSameSignature(nodeCoid, getNodeListsMethod); + markGenerated(getNodeListsMethod); } } |