fix(doctrine): resource_class from context instead of entity class

features/graphql/query.feature

@@ -677,3 +677,50 @@ Feature: GraphQL query support
     Then the response status code should be 200
     And the header "Content-Type" should be equal to "application/json"
     And the JSON node "data.getSecurityAfterResolver.name" should be equal to "test"
+
+  @createSchema
+  @!mongodb
+  Scenario: Retrieve an item with a subResource collection within an DTO using stateOptions entityClass with an GraphQL query
+    Given there is an issue6590 foo object with 2 bar sub objects
+    When I send the following GraphQL request:
+    """
+    {
+      issue6590OrmFoo(id: "/issue6590_orm_foos/1") {
+        id
+        bars {
+          edges {
+            node {
+              name
+            }
+          }
+        }
+      }
+    }
+    """
+    Then the response status code should be 200
+    And the response should be in JSON
+    And the header "Content-Type" should be equal to "application/json"
+    And the JSON should be equal to:
+    """
+    {
+      "data": {
+        "issue6590OrmFoo": {
+          "id": "/issue6590_orm_foos/1",
+          "bars": {
+            "edges": [
+              {
+                "node": {
+                  "name": "bar1"
+                }
+              },
+              {
+                "node": {
+                  "name": "bar2"
+                }
+              }
+            ]
+          }
+        }
+      }
+    }
+    """

src/Doctrine/Common/State/PersistProcessor.php

@@ -52,7 +52,7 @@
             \assert(method_exists($manager, 'getReference'));
             $newData = $data;
             $identifiers = array_reverse($uriVariables);
-            $links = $this->getLinks($class, $operation, $context);
+            $links = $this->getLinks($context['resource_class'] ?? $class, $operation, $context);
             $reflectionProperties = $this->getReflectionProperties($data);
 
             // TODO: the call to getReference is most likely to fail with complex identifiers

src/Doctrine/Odm/State/LinksHandlerTrait.php

@@ -37,7 +37,7 @@
             return;
         }
 
-        $links = $this->getLinks($resourceClass, $operation, $context);
+        $links = $this->getLinks($context['resource_class'] ?? $resourceClass, $operation, $context);
 
         if (!$links) {
             return;

src/Doctrine/Orm/State/LinksHandlerTrait.php

@@ -40,7 +40,7 @@ private function handleLinks(QueryBuilder $queryBuilder, array $identifiers, Que
         $doctrineClassMetadata = $manager->getClassMetadata($entityClass);
         $alias = $queryBuilder->getRootAliases()[0];
 
-        $links = $this->getLinks($entityClass, $operation, $context);
+        $links = $this->getLinks($context['resource_class'] ?? $entityClass, $operation, $context);
 
         if (!$links) {
             return;

tests/Behat/DoctrineContext.php

@@ -160,6 +160,8 @@
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue5722\ItemLog;
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue5735\Group;
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6039\Issue6039EntityUser;
+use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590\Bar as Issue6590BarDummy;
+use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590\Foo as Issue6590FooDummy;
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\LinkHandledDummy;
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\MaxDepthDummy;
 use ApiPlatform\Tests\Fixtures\TestBundle\Entity\MultiRelationsDummy;
@@ -2325,6 +2327,27 @@ public function thereAreIssue6039Users(): void
         $this->manager->flush();
     }
 
+    /**
+     * @Given there is an issue6590 foo object with 2 bar sub objects
+     */
+    public function thereIsAIssue6590DtoFooObjectWith2BarSubObjects(): void
+    {
+        $foo = new Issue6590FooDummy();
+        $this->manager->persist($foo);
+
+        $bar1 = new Issue6590BarDummy();
+        $bar1->setName('bar1');
+        $bar1->setFoo($foo);
+        $this->manager->persist($bar1);
+
+        $bar2 = new Issue6590BarDummy();
+        $bar2->setName('bar2');
+        $bar2->setFoo($foo);
+        $this->manager->persist($bar2);
+
+        $this->manager->flush();
+    }
+
     private function isOrm(): bool
     {
         return null !== $this->schemaTool;

tests/Fixtures/TestBundle/ApiResource/Issue6590/OrmBarResource.php

@@ -0,0 +1,40 @@
+<?php
+
+/*
+ * This file is part of the API Platform project.
+ *
+ * (c) Kévin Dunglas <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace ApiPlatform\Tests\Fixtures\TestBundle\ApiResource\Issue6590;
+
+use ApiPlatform\Doctrine\Orm\State\Options;
+use ApiPlatform\Metadata\ApiProperty;
+use ApiPlatform\Metadata\ApiResource;
+use ApiPlatform\Metadata\GraphQl\Query;
+use ApiPlatform\Metadata\GraphQl\QueryCollection;
+use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590\Bar;
+use ApiPlatform\Tests\Fixtures\TestBundle\State\Issue6590\BarResourceProvider;
+
+#[ApiResource(
+    shortName: 'Issue6590OrmBar',
+    operations: [],
+    graphQlOperations: [
+        new Query(),
+        new QueryCollection(),
+    ],
+    provider: BarResourceProvider::class,
+    stateOptions: new Options(entityClass: Bar::class)
+)]
+class OrmBarResource
+{
+    #[ApiProperty(identifier: true)]
+    public int $id;
+
+    public string $name;
+}

tests/Fixtures/TestBundle/ApiResource/Issue6590/OrmFooResource.php

@@ -0,0 +1,53 @@
+<?php
+
+/*
+ * This file is part of the API Platform project.
+ *
+ * (c) Kévin Dunglas <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace ApiPlatform\Tests\Fixtures\TestBundle\ApiResource\Issue6590;
+
+use ApiPlatform\Doctrine\Orm\State\Options;
+use ApiPlatform\Metadata\ApiProperty;
+use ApiPlatform\Metadata\ApiResource;
+use ApiPlatform\Metadata\GraphQl\Query;
+use ApiPlatform\Metadata\GraphQl\QueryCollection;
+use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590\Foo;
+use ApiPlatform\Tests\Fixtures\TestBundle\State\Issue6590\FooResourceProvider;
+
+#[ApiResource(
+    shortName: 'Issue6590OrmFoo',
+    operations: [],
+    graphQlOperations: [
+        new Query(),
+        new QueryCollection(),
+    ],
+    provider: FooResourceProvider::class,
+    stateOptions: new Options(entityClass: Foo::class)
+)]
+class OrmFooResource
+{
+    #[ApiProperty(identifier: true)]
+    public int $id;
+
+    /**
+     * @var OrmBarResource[]
+     */
+    public array $bars;
+
+    public function addBar(OrmBarResource $bar): void
+    {
+        $this->bars[] = $bar;
+    }
+
+    public function removeBar(OrmBarResource $bar): void
+    {
+        unset($this->bars[array_search($bar, $this->bars, true)]);
+    }
+}

tests/Fixtures/TestBundle/Entity/Issue6590/Bar.php

@@ -0,0 +1,61 @@
+<?php
+
+/*
+ * This file is part of the API Platform project.
+ *
+ * (c) Kévin Dunglas <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590;
+
+use Doctrine\ORM\Mapping as ORM;
+
+#[ORM\Entity()]
+#[ORM\Table(name: 'bar6590')]
+class Bar
+{
+    #[ORM\Id]
+    #[ORM\Column(type: 'integer')]
+    #[ORM\GeneratedValue(strategy: 'AUTO')]
+    private int $id;
+
+    #[ORM\Column]
+    private string $name;
+
+    #[ORM\ManyToOne(targetEntity: Foo::class, inversedBy: 'bars')]
+    private ?Foo $foo = null;
+
+    public function getId(): int
+    {
+        return $this->id;
+    }
+
+    public function getName(): string
+    {
+        return $this->name;
+    }
+
+    public function setName(string $name): self
+    {
+        $this->name = $name;
+
+        return $this;
+    }
+
+    public function getFoo(): ?Foo
+    {
+        return $this->foo;
+    }
+
+    public function setFoo(?Foo $foo): self
+    {
+        $this->foo = $foo;
+
+        return $this;
+    }
+}

tests/Fixtures/TestBundle/Entity/Issue6590/Foo.php

@@ -0,0 +1,59 @@
+<?php
+
+/*
+ * This file is part of the API Platform project.
+ *
+ * (c) Kévin Dunglas <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590;
+
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\Common\Collections\Collection;
+use Doctrine\ORM\Mapping as ORM;
+
+#[ORM\Entity]
+#[ORM\Table(name: 'foo6590')]
+class Foo
+{
+    #[ORM\Id]
+    #[ORM\Column(type: 'integer')]
+    #[ORM\GeneratedValue(strategy: 'AUTO')]
+    private int $id;
+
+    #[ORM\OneToMany(targetEntity: Bar::class, mappedBy: 'foo')]
+    private Collection $bars;
+
+    public function __construct()
+    {
+        $this->bars = new ArrayCollection();
+    }
+
+    public function getId(): int
+    {
+        return $this->id;
+    }
+
+    /**
+     * @return Collection<Bar>
+     */
+    public function getBars(): Collection
+    {
+        return $this->bars;
+    }
+
+    /**
+     * @param Collection<Bar> $bars
+     */
+    public function setBars(Collection $bars): self
+    {
+        $this->bars = $bars;
+
+        return $this;
+    }
+}

tests/Fixtures/TestBundle/State/Issue6590/BarResourceProvider.php

@@ -0,0 +1,64 @@
+<?php
+
+/*
+ * This file is part of the API Platform project.
+ *
+ * (c) Kévin Dunglas <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace ApiPlatform\Tests\Fixtures\TestBundle\State\Issue6590;
+
+use ApiPlatform\Doctrine\Orm\Paginator;
+use ApiPlatform\Metadata\CollectionOperationInterface;
+use ApiPlatform\Metadata\Operation;
+use ApiPlatform\State\Pagination\TraversablePaginator;
+use ApiPlatform\State\ProviderInterface;
+use ApiPlatform\Tests\Fixtures\TestBundle\ApiResource\Issue6590\OrmBarResource;
+use ApiPlatform\Tests\Fixtures\TestBundle\Entity\Issue6590\Bar as BarEntity;
+
+class BarResourceProvider implements ProviderInterface
+{
+    public function __construct(
+        private readonly ProviderInterface $itemProvider,
+        private readonly ProviderInterface $collectionProvider,
+    ) {
+    }
+
+    public function provide(Operation $operation, array $uriVariables = [], array $context = []): object|array|null
+    {
+        if ($operation instanceof CollectionOperationInterface) {
+            $entities = $this->collectionProvider->provide($operation, $uriVariables, $context);
+            \assert($entities instanceof Paginator);
+
+            $resources = [];
+            foreach ($entities as $entity) {
+                $resources[] = $this->getResource($entity);
+            }
+
+            return new TraversablePaginator(
+                new \ArrayIterator($resources),
+                $entities->getCurrentPage(),
+                $entities->getItemsPerPage(),
+                $entities->getTotalItems()
+            );
+        }
+
+        $entity = $this->itemProvider->provide($operation, $uriVariables, $context);
+
+        return $this->getResource($entity);
+    }
+
+    protected function getResource(BarEntity $entity): OrmBarResource
+    {
+        $resource = new OrmBarResource();
+        $resource->id = $entity->getId();
+        $resource->name = $entity->getName();
+
+        return $resource;
+    }
+}