Make entity models configurable in xml

Author: Bernard Delhez

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

@@ -22,20 +22,30 @@ 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) {
+            if (Mage::getModel($entityModelsConfig['class'])) {
+                $entityModels[] = $entityModelsConfig['class'];
+            }
         }
-        return $models;
+
+        return $entityModels;
+    }
+    protected function _sortEntityModelsConfig($entityModelsConfig)
+    {
+        function sortEntityModels($entityModel1, $entityModel2)
+        {
+            return strcmp($entityModel1['sort'], $entityModel2['sort']);
+        }
+
+        usort($entityModelsConfig, 'sortEntityModels');
+        return $entityModelsConfig;
     }
     /**
      * @param resource $stream stream resource used for output (for example opened file pointer or STDOUT)

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>