Adding ConstStrategy::beforeFind for append Enum Field on Entity

arusinowski · Mar 16, 2021 · d354c5d · d354c5d
1 parent 14311b4
commit d354c5d
Show file tree

Hide file tree

Showing 2 changed files with 66 additions and 0 deletions.
diff --git a/src/Model/Behavior/EnumBehavior.php b/src/Model/Behavior/EnumBehavior.php
@@ -291,4 +291,20 @@ public function __call(string $method, array $args): bool
 
         return array_key_exists($entity->{$config['field']}, $this->enum($alias));
     }
+
+    /**
+     * @param \Cake\Event\EventInterface $event The beforeFind event that was fired.
+     * @param \Cake\ORM\Query $query Query
+     * @param \ArrayObject $options The options for the query
+     * @return void
+     */
+    public function beforeFind(\Cake\Event\EventInterface $event, \Cake\ORM\Query $query, \ArrayObject $options)
+    {
+        foreach ($this->getConfig('lists') as $alias => $config) {
+            $strategy = $this->strategy($alias, $config['strategy']);
+            if (method_exists($strategy, 'beforeFind')) {
+                $strategy->beforeFind($event, $query, $options);
+            }
+        }
+    }
 }
diff --git a/src/Model/Behavior/Strategy/ConstStrategy.php b/src/Model/Behavior/Strategy/ConstStrategy.php
@@ -1,4 +1,5 @@
 <?php
+
 declare(strict_types=1);
 
 /**
@@ -14,6 +15,8 @@
 namespace CakeDC\Enum\Model\Behavior\Strategy;
 
 use Cake\ORM\Table;
+use Cake\Utility\Hash;
+use Cake\Utility\Inflector;
 use ReflectionClass;
 
 class ConstStrategy extends AbstractStrategy
@@ -92,4 +95,51 @@ protected function _getConstants(): array
 
         return $this->_constants = $constants;
     }
+
+    /**
+     * @param \Cake\Event\EventInterface $event The beforeFind event that was fired.
+     * @param \Cake\ORM\Query $query Query
+     * @param \ArrayObject $options The options for the query
+     * @return void
+     */
+    public function beforeFind(\Cake\Event\EventInterface $event, \Cake\ORM\Query $query, \ArrayObject $options)
+    {
+        $assocName = Inflector::pluralize(Inflector::classify($this->_alias));
+        if (!in_array($assocName, array_keys($query->getContain()))) {
+            return ;
+        } 
+
+        $contain = array_filter($query->getContain(), function($value) use ($assocName) {
+            return $value !== $assocName;
+        }, ARRAY_FILTER_USE_KEY);
+
+        $query->clearContain()->contain($contain);
+
+        $query->formatResults(function (\Cake\Collection\CollectionInterface $results) {
+            return $results->map(function ($row) {
+                if (is_string($row)) {
+                    return $row;
+                }
+
+                $constant = Hash::get($row, $this->getConfig('field'));
+
+                $field = Inflector::singularize(Inflector::underscore($this->_alias));
+                $value = new \Cake\ORM\Entity([
+                    'label' => Hash::get($this->_getConstants(), $constant, $constant),
+                    'prefix' => $this->getConfig('prefix'),
+                    'value' => $constant
+                ], ['markClean' => true, 'markNew' => false]);
+
+                if (is_array($row)) {
+                    $row[$field] = $value->toArray();
+                    return $row; 
+                }
+
+                $row->set($field, $value);                
+                $row->setDirty($field, false);
+
+                return $row;
+            });
+        });
+    }
 }