Updated templates to generate properties

Author: drpayyne

resources/fileTemplates/internal/Magento Data Model Interface.php.ft

@@ -2,9 +2,41 @@
 #parse("PHP File Header.php")
 
 #if (${NAMESPACE})
-namespace ${NAMESPACE};
+    namespace ${NAMESPACE};
 #end
 
-interface ${NAME} {
+interface ${NAME}
+{
+    #if (${PROPERTIES})
+        /**
+         * String constants for property names
+         */
+        #set ($properties = ${PROPERTIES})
+        #foreach ($property in $properties.split(","))
+            #set ($propertyData = $property.split(";"))
+            #set ($propertyUpperSnake = $propertyData.get(0))
+            #set ($propertyLowerSnake = $propertyData.get(1))
+            const $propertyUpperSnake = "$propertyLowerSnake";
+        #end
+    #end
+    #if (${PROPERTIES})
+        #set ($properties = ${PROPERTIES})
+        #foreach ($property in $properties.split(","))
+            #set ($propertyData = $property.split(";"))
+            #set ($propertyType = $propertyData.get(2))
+            #set ($propertyUpperCamel = $propertyData.get(3))
+            #set ($propertyLowerCamel = $propertyData.get(4))
 
+            /**
+             * @return $propertyType
+             */
+            public function get$propertyUpperCamel();
+
+            /**
+             * @param $propertyType $$propertyLowerCamel
+             * @return $this
+             */
+            public function set$propertyUpperCamel($$propertyLowerCamel);
+        #end
+    #end
 }

resources/fileTemplates/internal/Magento Data Model.php.ft

@@ -2,14 +2,41 @@
 #parse("PHP File Header.php")
 
 #if (${NAMESPACE})
-namespace ${NAMESPACE};
+    namespace ${NAMESPACE};
 #end
 
 #set ($uses = ${USES})
 #foreach ($use in $uses.split(","))
-use $use;
+    use $use;
 #end
 
-class ${NAME} #if (${EXTENDS})extends ${EXTENDS} #end #if (${IMPLEMENTS})implements ${IMPLEMENTS}#end {
+class ${NAME} #if (${EXTENDS})extends ${EXTENDS} #end #if (${IMPLEMENTS})implements ${IMPLEMENTS}#end
+{
+    #if (${PROPERTIES})
+        #set ($properties = ${PROPERTIES})
+        #foreach ($property in $properties.split(","))
+            #set ($propertyData = $property.split(";"))
+            #set ($propertyUpperSnake = $propertyData.get(0))
+            #set ($propertyUpperCamel = $propertyData.get(3))
+            #set ($propertyLowerCamel = $propertyData.get(4))
+            #if(!($foreach.first))
 
+            #end
+            /**
+             * @inheritDoc
+             */
+            public function get$propertyUpperCamel()
+            {
+                return $this->getData(self::$propertyUpperSnake);
+            }
+
+            /**
+             * @inheritDoc
+             */
+            public function set$propertyUpperCamel($$propertyLowerCamel)
+            {
+                return $this->setData(self::$propertyUpperSnake, $$propertyLowerCamel);
+            }
+        #end
+    #end
 }

src/com/magento/idea/magento2plugin/actions/generation/data/DataModelData.java

@@ -11,19 +11,22 @@ public class DataModelData {
     private final String moduleName;
     private final String fqn;
     private final String interfaceFQN;
+    private final String properties;
 
     public DataModelData(
             final String namespace,
             final String name,
             final String moduleName,
             final String fqn,
-            final String interfaceFQN
+            final String interfaceFQN,
+            final String properties
     ) {
         this.namespace = namespace;
         this.name = name;
         this.moduleName = moduleName;
         this.fqn = fqn;
         this.interfaceFQN = interfaceFQN;
+        this.properties = properties;
     }
 
     public String getNamespace() {
@@ -45,4 +48,8 @@ public String getFQN() {
     public String getInterfaceFQN() {
         return interfaceFQN;
     }
+
+    public String getProperties() {
+        return properties;
+    }
 }

src/com/magento/idea/magento2plugin/actions/generation/data/DataModelInterfaceData.java

@@ -10,17 +10,20 @@ public class DataModelInterfaceData {
     private final String name;
     private final String moduleName;
     private final String fqn;
+    private final String properties;
 
     public DataModelInterfaceData(
             final String namespace,
             final String name,
             final String moduleName,
-            final String fqn
+            final String fqn,
+            final String properties
     ) {
         this.namespace = namespace;
         this.name = name;
         this.moduleName = moduleName;
         this.fqn = fqn;
+        this.properties = properties;
     }
 
     public String getNamespace() {
@@ -38,4 +41,8 @@ public String getModuleName() {
     public String getFQN() {
         return fqn;
     }
+
+    public String getProperties() {
+        return properties;
+    }
 }

src/com/magento/idea/magento2plugin/actions/generation/generator/DataModelGenerator.java

@@ -105,6 +105,7 @@ protected void fillAttributes(Properties attributes) {
                 "IMPLEMENTS",
                 PhpClassGeneratorUtil.getNameFromFqn(modelData.getInterfaceFQN())
         );
+        attributes.setProperty("PROPERTIES", modelData.getProperties());
     }
 
     private List<String> getUses() {

src/com/magento/idea/magento2plugin/actions/generation/generator/DataModelInterfaceGenerator.java

@@ -95,6 +95,7 @@ public PsiFile generate(String actionName) {
     protected void fillAttributes(Properties attributes) {
         attributes.setProperty("NAME", interfaceData.getName());
         attributes.setProperty("NAMESPACE", interfaceData.getNamespace());
+        attributes.setProperty("PROPERTIES", interfaceData.getProperties());
     }
 
     private PhpClass createInterface(String actionName) {