Add custom initialization code option for constructors

javaparser · Apr 22, 2017 · ab12a2e · ab12a2e
1 parent f1e770a
commit ab12a2e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/...ors/src/main/java/com/github/javaparser/generator/core/node/MainConstructorGenerator.java b/...ors/src/main/java/com/github/javaparser/generator/core/node/MainConstructorGenerator.java
@@ -40,13 +40,15 @@ protected void generateNode(BaseNodeMetaModel nodeMetaModel, CompilationUnit nod
         for (PropertyMetaModel parameter : nodeMetaModel.getConstructorParameters()) {
             constructor.addParameter(parameter.getTypeNameForSetter(), parameter.getName());
             if (nodeMetaModel.getDeclaredPropertyMetaModels().contains(parameter)) {
-                body.addStatement(parseStatement(f("%s(%s);", parameter.getSetterMethodName(), parameter.getName())));
+                body.addStatement(f("%s(%s);", parameter.getSetterMethodName(), parameter.getName()));
             } else {
                 superCall.append(parameter.getName());
             }
         }
 
         body.getStatements().add(0, parseExplicitConstructorInvocationStmt(superCall.toString()));
+
+        body.addStatement("customInitialization();");
 
         replaceWhenSameSignature(nodeCoid, constructor);
         nodeCu.addImport(Range.class);

diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/Node.java b/javaparser-core/src/main/java/com/github/javaparser/ast/Node.java
@@ -150,6 +150,14 @@ public Node(Range range) {
         this.range = range;
     }
 
+    /**
+     * Called in every constructor for node specific code.
+     * It can't be written in the constructor itself because it will
+     * be overwritten during code generation.
+     */
+    protected void customInitialization() {
+    }
+
     /**
      * This is a comment associated with this node.
      *

diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/body/VariableDeclarator.java b/javaparser-core/src/main/java/com/github/javaparser/ast/body/VariableDeclarator.java
@@ -91,9 +91,10 @@ public VariableDeclarator(Range range, Type type, SimpleName name, Expression in
         setType(type);
         setName(name);
         setInitializer(initializer);
+        customInitialization();
     }
 
-    private void registerObserversForDerivedProperties() {
+    private void customInitialization() {
         // We register an observer on the type property. When it is changed the MaximumCommonType is changes as well,
         // because it is derived from the type of the variables it contains, for this reason we notify about the change
         this.register(new AstObserverAdapter() {