support for php 8 enums in hydration

FluffyDiscord · May 2, 2022 · ac69d17 · ac69d17
1 parent 2d88616
commit ac69d17
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Hydrator/SimpleObjectHydrator.php b/Hydrator/SimpleObjectHydrator.php
@@ -108,6 +108,25 @@ protected function doHydrateRowData($className, array $data)
             if ($property->isPublic()) {
                 $entity->$name = $value;
             } else {
+                /** @var \UnitEnum|\BackedEnum $enumClass */
+                $enumClass = null;
+                foreach ($property->getAttributes() as $attribute) {
+                    $enumClass = $attribute->getArguments()['enumType'] ?? null;
+                    if($enumClass !== null) {
+                        break;
+                    }
+                }
+
+                if($enumClass !== null) {
+                    $enumInterfaces = array_keys(class_implements($enumClass));
+
+                    if(in_array(\BackedEnum::class, $enumInterfaces)) {
+                        $value = $enumClass::tryFrom($value);
+                    } elseif (in_array(\UnitEnum::class, $enumInterfaces)) {
+                        $value = $enumClass::$value;
+                    }
+                }
+
                 $property->setAccessible(true);
                 $property->setValue($entity, $value);
                 $property->setAccessible(false);