Merge pull request #5 from BboyKeen/master

Add support for embedded entities
ambta · Apr 4, 2016 · 448401e · 448401e
2 parents 675f0c0 + 17dc145
commit 448401e
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/Subscribers/DoctrineEncryptSubscriber.php b/Subscribers/DoctrineEncryptSubscriber.php
@@ -235,6 +235,10 @@ public function processFields($entity, $isEncryptOperation = true) {
             //Foreach property in the reflection class
             foreach ($properties as $refProperty) {
 
+                if ($this->annReader->getPropertyAnnotation($refProperty, 'Doctrine\ORM\Mapping\Embedded')) {
+                    $this->handleEmbeddedAnnotation($entity, $refProperty, $isEncryptOperation);
+                    continue;
+                }
                 /**
                  * If followed standards, method name is getPropertyName, the propertyName is lowerCamelCase
                  * So just uppercase first character of the property, later on get and set{$methodName} wil be used
@@ -297,6 +301,30 @@ public function processFields($entity, $isEncryptOperation = true) {
         return null;
     }
 
+    private function handleEmbeddedAnnotation($entity, $embeddedProperty, $isEncryptOperation = true)
+    {
+        $reflectionClass = new ReflectionClass($entity);
+        $propName = $embeddedProperty->getName();
+        $methodName = ucfirst($propName);
+
+        if ($embeddedProperty->isPublic()) {
+            $embeddedEntity = $embeddedProperty->getValue();
+        } else {
+            if ($reflectionClass->hasMethod($getter = 'get' . $methodName) && $reflectionClass->hasMethod($setter = 'set' . $methodName)) {
+
+                //Get the information (value) of the property
+                try {
+                    $embeddedEntity = $entity->$getter();
+                } catch(\Exception $e) {
+                    $embeddedEntity = null;
+                }
+            }
+        }
+        if ($embeddedEntity) {
+            $this->processFields($embeddedEntity, $isEncryptOperation);
+        }
+    }
+
     /**
      * Recursive function to get an associative array of class properties
      * including inherited ones from extended classes