add a flag indicates to skip null values, adding possibility to postprocess generated array

andreasschacht · andreasschacht · commit 439db1b68c40 · 2016-08-01T11:45:32.000+02:00
diff --git a/src/Sp/FixtureDumper/Generator/AbstractGenerator.php b/src/Sp/FixtureDumper/Generator/AbstractGenerator.php
@@ -50,8 +50,23 @@ abstract class AbstractGenerator
      */
     protected $models;
 
+    /**
+     * @var array
+     */
     protected $propertyReader = array();
 
+    /**
+     * @var array
+     */
+    protected $fieldNamesPostProcessors = array();
+
+    /**
+     * @var bool
+     */
+    protected $skipNullValues = false;
+
+
+
 
     /**
      * @param \Doctrine\Common\Persistence\ObjectManager|null   $manager
@@ -75,6 +90,36 @@ public function addPropertyReader(PropertyReaderInterface $propertyReader)
         $this->propertyReader[] = $propertyReader;
     }
 
+    /**
+     * adds a postprocessor
+     *
+     * @param FieldNamesPostProcessorInterface $postProcessor
+     */
+    public function addFieldNamesPostProcessor(FieldNamesPostProcessorInterface $postProcessor)
+    {
+        $this->fieldNamesPostProcessors[] = $postProcessor;
+    }
+
+    /**
+     * @return boolean
+     */
+    public function isSkipNullValues()
+    {
+        return $this->skipNullValues;
+    }
+
+    /**
+     * if set to true, null values won't be dumped
+     *
+     * @param boolean $skipNullValues
+     */
+    public function setSkipNullValues($skipNullValues)
+    {
+        $this->skipNullValues = $skipNullValues;
+    }
+
+
+
     /**
      * @param \Doctrine\Common\Persistence\Mapping\ClassMetadata $metadata
      * @param array|null                                         $models
@@ -235,8 +280,22 @@ protected function processFieldNames(ClassMetadata $metadata, $model)
                 continue;
             }
 
-            $data[$fieldName] = $this->navigator->accept($this->getVisitor(), $this->readProperty($model, $fieldName));
+            $value = $this->readProperty($model, $fieldName);
+            if ($this->skipNullValues && $value === null) {
+                continue;
+            }
+
+            $data[$fieldName] = $this->navigator->accept($this->getVisitor(), $value);
         }
+        
+        foreach ($this->fieldNamesPostProcessors as $postProcessor) {
+            /** @var FieldNamesPostProcessorInterface $postProcessor */
+            if ($postProcessor->isSupporting($model, $metadata, $data)) {
+                $data = $postProcessor->getData($model, $metadata, $data);
+            }
+        }
+
+
 
         return $data;
     }
diff --git a/src/Sp/FixtureDumper/Generator/FieldNamesPostProcessorInterface.php b/src/Sp/FixtureDumper/Generator/FieldNamesPostProcessorInterface.php
@@ -0,0 +1,38 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: andreasschacht
+ * Date: 01.08.16
+ * Time: 11:27
+ */
+
+namespace Sp\FixtureDumper\Generator;
+
+use Doctrine\Common\Persistence\Mapping\ClassMetadata;
+
+interface FieldNamesPostProcessorInterface
+{
+
+    /**
+     * Indicates if this class supports processing data
+     *
+     * @param               $object
+     * @param ClassMetaData $metadata
+     * @param array         $data
+     *
+     *
+     * @return boolean true, if this class can process the data
+     */
+    public function isSupporting($object, ClassMetaData $metadata, array $data);
+
+    /**
+     * returns the processed data
+     *
+     * @param               $object
+     * @param ClassMetaData $metadata
+     * @param array         $data
+     *
+     * @return array
+     */
+    public function getData($object, ClassMetaData $metadata, array $data);
+}
