Handle union types

Author: n.gnato

CHANGELOG.md

@@ -6,6 +6,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Fixed
+- Handle Union Types for `data` and `attributes` fields.  
+
 ## [0.0.5] - 2025-03-24
 
 ### Fixed

src/FreeElephants/JsonApi/DTO/AbstractDocument.php

@@ -2,8 +2,6 @@
 
 namespace FreeElephants\JsonApi\DTO;
 
-use Psr\Http\Message\MessageInterface;
-
 /**
  * @property AbstractResourceObject|mixed $data
  */
@@ -13,9 +11,14 @@ final public function __construct(array $payload)
     {
         $concreteClass = new \ReflectionClass($this);
         $dataProperty = $concreteClass->getProperty('data');
-        /** @var \ReflectionNamedType $reflectionType */
+
         $reflectionType = $dataProperty->getType();
-        $dataClassName = $reflectionType->getName();
+        if ($reflectionType instanceof \ReflectionNamedType) {
+            $dataClassName = $reflectionType->getName();
+        } else {
+            /** @var \ReflectionUnionType $reflectionType */
+            $dataClassName = $reflectionType->getTypes()[0]->getName();
+        }
         if ($dataClassName !== 'array') {
             $data = new $dataClassName($payload['data']);
         } else {

src/FreeElephants/JsonApi/DTO/AbstractResourceObject.php

@@ -15,14 +15,19 @@ class AbstractResourceObject
 
     public function __construct(array $data)
     {
-        $this->id = $data['id'] ?? null;
+        $this->id = $data['id'];
         $this->type = $data['type'];
 
         $concreteClass = new \ReflectionClass($this);
 
         if (property_exists($this, 'attributes')) {
-            $attributesProperty = $concreteClass->getProperty('attributes');
-            $attributesClass = $attributesProperty->getType()->getName();
+            $attributesPropertyType = $concreteClass->getProperty('attributes')->getType();
+
+            if($attributesPropertyType instanceof \ReflectionUnionType) {
+                $attributesClass = $attributesPropertyType->getTypes()[0]->getName();
+            } else {
+                $attributesClass = $attributesPropertyType->getName();
+            }
             $this->attributes = new $attributesClass($data['attributes']);
         }
 

tests/FreeElephants/JsonApi/DTO/Example/AttributesExt.php

@@ -0,0 +1,9 @@
+<?php
+declare(strict_types=1);
+
+namespace FreeElephants\JsonApi\DTO\Example;
+
+class AttributesExt extends Attributes
+{
+    public int $baz;
+}

tests/FreeElephants/JsonApi/DTO/Example/ResourceObject.php

@@ -0,0 +1,11 @@
+<?php
+declare(strict_types=1);
+
+namespace FreeElephants\JsonApi\DTO\Example;
+
+use FreeElephants\JsonApi\DTO\AbstractResourceObject;
+
+class ResourceObject extends AbstractResourceObject
+{
+
+}

tests/FreeElephants/JsonApi/DTO/Example/ResourceObjectExt.php

@@ -0,0 +1,9 @@
+<?php
+declare(strict_types=1);
+
+namespace FreeElephants\JsonApi\DTO\Example;
+
+class ResourceObjectExt extends ResourceObject
+{
+
+}

tests/FreeElephants/JsonApi/DTO/Reflection/DocumentTestPHP8.php

@@ -0,0 +1,25 @@
+<?php
+declare(strict_types=1);
+
+namespace FreeElephants\JsonApi\DTO\Reflection;
+
+use FreeElephants\JsonApi\AbstractTestCase;
+use FreeElephants\JsonApi\DTO\AbstractDocument;
+use FreeElephants\JsonApi\DTO\Example;
+
+class DocumentTestPHP8 extends AbstractTestCase
+{
+    public function testUnionTypes(): void
+    {
+        $document = new class([
+            'data' => [
+                'id' => '1',
+                'type' => 'foos',
+            ],
+        ]) extends AbstractDocument {
+            public Example\ResourceObjectExt|Example\ResourceObject $data;
+        };
+
+        $this->assertInstanceOf(Example\ResourceObjectExt::class, $document->data);
+    }
+}

tests/FreeElephants/JsonApi/DTO/ResourceObjectTestPHP8.php renamed to tests/FreeElephants/JsonApi/DTO/Reflection/ResourceObjectTestPHP8.php

@@ -1,8 +1,11 @@
 <?php
 
-namespace FreeElephants\JsonApi\DTO;
+namespace FreeElephants\JsonApi\DTO\Reflection;
 
 use FreeElephants\JsonApi\AbstractTestCase;
+use FreeElephants\JsonApi\DTO\AbstractResourceObject;
+use FreeElephants\JsonApi\DTO\Example;
+use FreeElephants\JsonApi\DTO\Example\AttributesExt;
 
 class ResourceObjectTestPHP8 extends AbstractTestCase
 {
@@ -13,6 +16,7 @@ public function testUnionTypes()
             'type'          => 'type',
             'attributes'    => [
                 'foo' => 'bar',
+                'baz' => 1,
             ],
             'relationships' => [
                 'one' => [
@@ -23,11 +27,14 @@ public function testUnionTypes()
                 ],
             ],
         ]) extends AbstractResourceObject{
-            public Example\Attributes $attributes;
+            public Example\AttributesExt|Example\Attributes $attributes;
             public Example\OneRelationships|Example\TwoRelationships $relationships;
         };
 
         $this->assertSame('one', $resourceObject->relationships->one->data->type);
+        $this->assertSame('bar', $resourceObject->attributes->foo);
+        $this->assertSame(1, $resourceObject->attributes->baz);
+        $this->assertInstanceOf(AttributesExt::class, $resourceObject->attributes);
     }
 }