added codegen model items

speedment · Jan 20, 2015 · 9330c3e · 9330c3e
1 parent c532e6e
commit 9330c3e
Show file tree

Hide file tree

Showing 8 changed files with 106 additions and 41 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,4 @@
+/target/
+/nb-configuration.xml
+
+
diff --git a/src/main/java/com/speedment/codegen/control/AccessorImplementer.java b/src/main/java/com/speedment/codegen/control/AccessorImplementer.java
@@ -1,9 +1,46 @@
 package com.speedment.codegen.control;
 
+import com.speedment.codegen.model.Class_;
+import com.speedment.codegen.model.Field_;
+import com.speedment.codegen.model.Method_;
+import com.speedment.codegen.model.Statement_;
+
 /**
  *
  * @author pemi
  */
-public class AccessorImplementer {
+public class AccessorImplementer implements Controller {
+
+    private final Class_ class_;
+
+    public AccessorImplementer(final Class_ class_) {
+        this.class_ = class_;
+    }
+
+    @Override
+    public void apply() {
+        class_.getFields().forEach((f) -> generateAccessors(f));
+    }
+
+    protected void generateAccessors(final Field_ field_) {
+        generateGetter(field_);
+        generateSetter(field_);
+    }
+
+    protected void generateGetter(final Field_ field_) {
+        final Method_ method_ = new Method_(field_.getType_(), "get" + firstCharCapital(field_.getName_()));
+        method_.add(new Statement_("return " + field_.getName_()));
+        class_.add(method_);
+    }
+
+    protected void generateSetter(final Field_ field_) {
+        final Method_ method_ = new Method_(field_.getType_(), "set" + firstCharCapital(field_.getName_())).add(new Field_(field_.getType_(), field_.getName_()));
+        method_.add(new Statement_("this." + field_.getName_() + " = "));
+        class_.add(method_);
+    }
+
+    private String firstCharCapital(CharSequence s) {
+        return new StringBuilder().append(Character.toUpperCase(s.charAt(0))).append(s.subSequence(1, s.length())).toString();
+    }
 
 }
diff --git a/src/main/java/com/speedment/codegen/control/Controller.java b/src/main/java/com/speedment/codegen/control/Controller.java
@@ -0,0 +1,7 @@
+package com.speedment.codegen.control;
+
+public interface Controller {
+
+    void apply();
+
+}
diff --git a/src/main/java/com/speedment/codegen/model/Class_.java b/src/main/java/com/speedment/codegen/model/Class_.java
@@ -12,6 +12,7 @@ public class Class_ {
     private final List<Field_> fields;
     private final List<Constructor_> constructors;
     private final List<Method_> methods;
+    private Package_ pagage;
 
     public Class_() {
         this.fields = new ArrayList<>();
@@ -46,4 +47,12 @@ public List<Method_> getMethods() {
         return methods;
     }
 
+    public Package_ getPagage() {
+        return pagage;
+    }
+
+    public void setPagage(Package_ pagage) {
+        this.pagage = pagage;
+    }
+
 }
diff --git a/src/main/java/com/speedment/codegen/model/Field_.java b/src/main/java/com/speedment/codegen/model/Field_.java
@@ -9,7 +9,7 @@ public class Field_ {
     protected boolean private_;
     protected boolean static_;
     protected boolean final_;
-    protected Type_ type_;
+    private Type_ type_;
     private CharSequence name_;
     private Expression_ expression_;
 
@@ -95,4 +95,12 @@ public Expression_ getExpression_() {
     public void setExpression_(Expression_ expression_) {
         this.expression_ = expression_;
     }
+
+    public Type_ getType_() {
+        return type_;
+    }
+
+    public void setType_(Type_ type_) {
+        this.type_ = type_;
+    }
 }
diff --git a/src/main/java/com/speedment/codegen/model/Method_.java b/src/main/java/com/speedment/codegen/model/Method_.java
@@ -15,44 +15,23 @@ public class Method_ {
     private Type_ type_;
     private CharSequence name_;
     private Expression_ expression_;
-    private Block_ block;
     private List<Field_> parameters; // Todo: Introduce parameter
+    private List<Statement_> statements;
 
     public Method_(Type_ type_, CharSequence name_) {
         this.parameters = new ArrayList<>();
+        this.statements = new ArrayList<>();
         this.type_ = type_;
         this.name_ = name_;
     }
 
-    public Method_ private_() {
-        return private_(true);
-    }
-
-    public Method_ private_(boolean private_) {
-        this.setPrivate_(private_);
-        return this;
-    }
-
-    public Method_ static_() {
-        return static_(true);
-    }
-
-    public Method_ static_(boolean static_) {
-        this.setFinal_(static_);
-        return this;
-    }
-
-    public Method_ final_(boolean final_) {
-        this.setFinal_(final_);
+    public Method_ add(Field_ field_) {
+        getParameters().add(field_);
         return this;
     }
 
-    public Method_ final_() {
-        return final_(true);
-    }
-
-    Method_ add(Field_ field_) {
-        getParameters().add(field_);
+    public Method_ add(Statement_ statement) {
+        statements.add(statement);
         return this;
     }
 
@@ -104,14 +83,6 @@ public void setExpression_(Expression_ expression_) {
         this.expression_ = expression_;
     }
 
-    public Block_ getBlock() {
-        return block;
-    }
-
-    public void setBlock(Block_ block) {
-        this.block = block;
-    }
-
     public List<Field_> getParameters() {
         return parameters;
     }

diff --git a/src/main/java/com/speedment/codegen/model/Package_.java b/src/main/java/com/speedment/codegen/model/Package_.java
@@ -11,9 +11,11 @@ public class Package_ {
 
     private String name_;
     private final List<Class_> classes;
+    private final List<Package_> packages;
 
     public Package_() {
         this.classes = new ArrayList<>();
+        this.packages = new ArrayList<>();
     }
 
     public String getName_() {
@@ -24,4 +26,18 @@ public void setName_(String name_) {
         this.name_ = name_;
     }
 
+    Package_ add(Class_ class_) {
+        classes.add(class_);
+        return this;
+    }
+
+    Package_ add(Package_ package_) {
+        package_.add(package_);
+        return this;
+    }
+
+    public List<Package_> getPackages() {
+        return packages;
+    }
+
 }
diff --git a/src/main/java/com/speedment/codegen/model/Statement_.java b/src/main/java/com/speedment/codegen/model/Statement_.java
@@ -6,18 +6,31 @@
  */
 public class Statement_ {
 
-    private CharSequence statementText;
+    private CharSequence statementText_;
+    private Block_ block_;
 
     public Statement_(CharSequence statementText) {
-        this.statementText = statementText;
+        this.statementText_ = statementText;
+    }
+
+    public Statement_(Block_ block_) {
+        this.block_ = block_;
     }
 
     public CharSequence getStatementText() {
-        return statementText;
+        return statementText_;
     }
 
     public void setStatementText(CharSequence statementText) {
-        this.statementText = statementText;
+        this.statementText_ = statementText;
+    }
+
+    public Block_ getBlock() {
+        return block_;
+    }
+
+    public void setBlock(Block_ block_) {
+        this.block_ = block_;
     }
 
 }