Merge pull request #13 from integer-net/make-entity-models-configurable-in-xml

Make entity models configurable in xml

Author: bernarddelhez

src/app/code/community/IntegerNet/Anonymizer/Model/Anonymizer.php

@@ -22,20 +22,32 @@ public function __construct()
     }
     protected function _getEntityModels()
     {
-        $models = array(
-            'integernet_anonymizer/bridge_entity_customer',
-            'integernet_anonymizer/bridge_entity_address_customerAddress',
-            'integernet_anonymizer/bridge_entity_address_quoteAddress',
-            'integernet_anonymizer/bridge_entity_address_orderAddress',
-            'integernet_anonymizer/bridge_entity_newsletterSubscriber',
-            // Models with grid table must come last!
-            'integernet_anonymizer/bridge_entity_order',
-        );
-        if (Mage::getEdition() == MAGE::EDITION_ENTERPRISE) {
-            $models[] = 'integernet_anonymizer/bridge_entity_enterprise_giftregistry';
-            $models[] = 'integernet_anonymizer/bridge_entity_enterprise_giftregistryPerson';
+        $entityModelsConfigXml = Mage::getConfig()->getNode('global/integernet_anonymizer/entity_models');
+
+        $entityModelsConfigArray = $entityModelsConfigXml->asArray();
+        $entityModelsConfigArray = $this->_sortEntityModelsConfig($entityModelsConfigArray);
+
+        $entityModels = [];
+
+        foreach ($entityModelsConfigArray as $entityModelsConfig) {
+            $entityModel = Mage::getModel($entityModelsConfig['class']);
+            if ($entityModel instanceof IntegerNet_Anonymizer_Model_Bridge_Entity_Abstract
+                && $entityModel->entityExists()) {
+                $entityModels[] = $entityModelsConfig['class'];
+            }
         }
-        return $models;
+
+        return $entityModels;
+    }
+    protected function _sortEntityModelsConfig($entityModelsConfig)
+    {
+        usort(
+            $entityModelsConfig,
+            function ($entityModel1, $entityModel2) {
+                return strcmp($entityModel1['sort'], $entityModel2['sort']);
+            }
+        );
+        return $entityModelsConfig;
     }
     /**
      * @param resource $stream stream resource used for output (for example opened file pointer or STDOUT)

src/app/code/community/IntegerNet/Anonymizer/Model/Bridge/Entity/Abstract.php

@@ -174,4 +174,12 @@ public function getCollectionIterator()
         return $iterator;
     }
 
+    /**
+     * Check if the entity exists
+     * @return bool
+     */
+    public function entityExists()
+    {
+        return (bool)$this->_entity;
+    }
 }

src/app/code/community/IntegerNet/Anonymizer/etc/config.xml

@@ -15,6 +15,43 @@
             <IntegerNet />
             <Faker />
         </psr0_namespaces>
+        <integernet_anonymizer>
+            <entity_models>
+                <customer>
+                    <class>integernet_anonymizer/bridge_entity_customer</class>
+                    <sort>100</sort>
+                </customer>
+                <customer_address>
+                    <class>integernet_anonymizer/bridge_entity_address_customerAddress</class>
+                    <sort>200</sort>
+                </customer_address>
+                <quote_address>
+                    <class>integernet_anonymizer/bridge_entity_address_quoteAddress</class>
+                    <sort>300</sort>
+                </quote_address>
+                <order_address>
+                    <class>integernet_anonymizer/bridge_entity_address_orderAddress</class>
+                    <sort>400</sort>
+                </order_address>
+                <newsletter_subscriber>
+                    <class>integernet_anonymizer/bridge_entity_newsletterSubscriber</class>
+                    <sort>500</sort>
+                </newsletter_subscriber>
+                <enterprise_giftregistry>
+                    <class>integernet_anonymizer/bridge_entity_enterprise_giftregistry</class>
+                    <sort>600</sort>
+                </enterprise_giftregistry>
+                <enterprise_giftregistry_person>
+                    <class>integernet_anonymizer/bridge_entity_enterprise_giftregistryPerson</class>
+                    <sort>700</sort>
+                </enterprise_giftregistry_person>
+                <!-- Models with grid table must come last! -->
+                <order>
+                    <class>integernet_anonymizer/bridge_entity_order</class>
+                    <sort>800</sort>
+                </order>
+            </entity_models>
+        </integernet_anonymizer>
     </global>
     <phpunit>
         <suite>