diff options
Diffstat (limited to 'javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/visitor/CloneVisitorGenerator.java')
-rw-r--r-- | javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/visitor/CloneVisitorGenerator.java | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/visitor/CloneVisitorGenerator.java b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/visitor/CloneVisitorGenerator.java new file mode 100644 index 000000000..44a7e9f6f --- /dev/null +++ b/javaparser-core-generators/src/main/java/com/github/javaparser/generator/core/visitor/CloneVisitorGenerator.java @@ -0,0 +1,59 @@ +package com.github.javaparser.generator.core.visitor; + +import com.github.javaparser.ast.CompilationUnit; +import com.github.javaparser.ast.body.MethodDeclaration; +import com.github.javaparser.ast.stmt.BlockStmt; +import com.github.javaparser.generator.VisitorGenerator; +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 static com.github.javaparser.utils.CodeGenerationUtils.f; + +/** + * Generates JavaParser's CloneVisitor. + */ +public class CloneVisitorGenerator extends VisitorGenerator { + public CloneVisitorGenerator(SourceRoot sourceRoot) { + super(sourceRoot, "com.github.javaparser.ast.visitor", "CloneVisitor", "Visitable", "Object", true); + } + + @Override + protected void generateVisitMethodBody(BaseNodeMetaModel node, MethodDeclaration visitMethod, CompilationUnit compilationUnit) { + visitMethod.getParameters().forEach(p -> p.setFinal(true)); + + BlockStmt body = visitMethod.getBody().get(); + body.getStatements().clear(); + + for (PropertyMetaModel field : node.getAllPropertyMetaModels()) { + final String getter = field.getGetterMethodName() + "()"; + if (field.getNodeReference().isPresent()) { + if (field.isOptional() && field.isNodeList()) { + body.addStatement(f("NodeList<%s> %s = cloneList(n.%s.orElse(null), arg);", field.getTypeNameGenerified(), field.getName(), getter)); + } else if (field.isNodeList()) { + body.addStatement(f("NodeList<%s> %s = cloneList(n.%s, arg);", field.getTypeNameGenerified(), field.getName(), getter)); + } else { + body.addStatement(f("%s %s = cloneNode(n.%s, arg);", field.getTypeNameGenerified(), field.getName(), getter)); + } + } + } + + SeparatedItemStringBuilder builder = new SeparatedItemStringBuilder(f("%s r = new %s(", node.getTypeNameGenerified(), node.getTypeNameGenerified()), ",", ");"); + builder.append("n.getTokenRange().orElse(null)"); + for (PropertyMetaModel field : node.getConstructorParameters()) { + if (field.getName().equals("comment")) { + continue; + } + if (field.getNodeReference().isPresent()) { + builder.append(field.getName()); + } else { + builder.append(f("n.%s()", field.getGetterMethodName())); + } + } + + body.addStatement(builder.toString()); + body.addStatement("r.setComment(comment);"); + body.addStatement("return r;"); + } +} |