<?php

declare(strict_types=1);

namespace Doctrine\Tests\ORM\Functional\Ticket;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\Common\Collections\Collection;
use Doctrine\ORM\Mapping\Column;
use Doctrine\ORM\Mapping\DiscriminatorColumn;
use Doctrine\ORM\Mapping\DiscriminatorMap;
use Doctrine\ORM\Mapping\Entity;
use Doctrine\ORM\Mapping\GeneratedValue;
use Doctrine\ORM\Mapping\Id;
use Doctrine\ORM\Mapping\InheritanceType;
use Doctrine\ORM\Mapping\JoinColumn;
use Doctrine\ORM\Mapping\ManyToOne;
use Doctrine\ORM\Mapping\OneToMany;
use Doctrine\Tests\OrmFunctionalTestCase;

class DDC531Test extends OrmFunctionalTestCase
{
    protected function setUp(): void
    {
        parent::setUp();

        $this->createSchemaForModels(
            DDC531Item::class,
            DDC531SubItem::class,
        );
    }

    public function testIssue(): void
    {
        $item1         = new DDC531Item();
        $item2         = new DDC531Item();
        $item2->parent = $item1;
        $item1->getChildren()->add($item2);
        $this->_em->persist($item1);
        $this->_em->persist($item2);
        $this->_em->flush();
        $this->_em->clear();

        $item3 = $this->_em->find(DDC531Item::class, $item2->id); // Load child item first (id 2)
        // parent will already be loaded, cannot be lazy because it has mapped subclasses and we would not
        // know which proxy type to put in.
        self::assertInstanceOf(DDC531Item::class, $item3->parent);
        self::assertFalse($this->isUninitializedObject($item3->parent));
        $item4 = $this->_em->find(DDC531Item::class, $item1->id); // Load parent item (id 1)
        self::assertNull($item4->parent);
        self::assertNotNull($item4->getChildren());
        self::assertTrue($item4->getChildren()->contains($item3)); // lazy-loads children
    }
}

#[Entity]
#[InheritanceType('SINGLE_TABLE')]
#[DiscriminatorColumn(name: 'type', type: 'integer')]
#[DiscriminatorMap(['0' => 'DDC531Item', '1' => 'DDC531SubItem'])]
class DDC531Item
{
    /** @var int */
    #[Id]
    #[Column(type: 'integer')]
    #[GeneratedValue(strategy: 'AUTO')]
    public $id;

    /** @psalm-var Collection<int, DDC531Item> */
    #[OneToMany(targetEntity: 'DDC531Item', mappedBy: 'parent')]
    protected $children;

    /** @var DDC531Item */
    #[ManyToOne(targetEntity: 'DDC531Item', inversedBy: 'children')]
    #[JoinColumn(name: 'parentId', referencedColumnName: 'id')]
    public $parent;

    public function __construct()
    {
        $this->children = new ArrayCollection();
    }

    public function getParent(): DDC531Item
    {
        return $this->parent;
    }

    /** @psalm-return Collection<int, DDC531Item> */
    public function getChildren(): Collection
    {
        return $this->children;
    }
}

#[Entity]
class DDC531SubItem extends DDC531Item
{
}
