cycle / orm

src/Relation/Morphed/BelongsToMorphed.php

<?php
/**
 * Cycle DataMapper ORM
 *
 * @license   MIT
 * @author    Anton Titov (Wolfy-J)
 */
declare(strict_types=1);

namespace Cycle\ORM\Relation\Morphed;

use Cycle\ORM\Command\CommandInterface;
use Cycle\ORM\Command\ContextCarrierInterface as CC;
use Cycle\ORM\Exception\RelationException;
use Cycle\ORM\Heap\Node;
use Cycle\ORM\ORMInterface;
use Cycle\ORM\Relation;
use Cycle\ORM\Relation\BelongsTo;

class BelongsToMorphed extends BelongsTo
{
    /** @var string */
    private $morphKey;

    /**
     * @param ORMInterface $orm
     * @param string       $target
     * @param string       $name
     * @param array        $schema
     */
    public function __construct(ORMInterface $orm, string $name, string $target, array $schema)
    {
        parent::__construct($orm, $name, $target, $schema);
        $this->morphKey = $schema[Relation::MORPH_KEY];
    }

    /**
     * @inheritdoc
     */
    public function initPromise(Node $node): array
    {
        if (is_null($innerKey = $this->fetchKey($node, $this->innerKey))) {
            return [null, null];
        }

        /** @var string $target */
        $target = $this->fetchKey($node, $this->morphKey);
        if (is_null($target)) {

The condition is_null($target) is always false.

            return [null, null];
        }

        $e = $this->orm->getHeap()->find($target, $this->outerKey, $innerKey);

The method getHeap() does not exist on Cycle\ORM\Select\SourceProviderInterface. Since it exists in all sub-types, consider adding an abstract or default implementation to Cycle\ORM\Select\SourceProviderInterface.

        if ($e !== null) {
            return [$e, $e];
        }

        $e = $this->orm->promise($target, [$this->outerKey => $innerKey]);

        return [$e, $e];
    }

    /**
     * @inheritdoc
     */
    public function queue(CC $store, $entity, Node $node, $related, $original): CommandInterface
    {
        $wrappedStore = parent::queue($store, $entity, $node, $related, $original);

        if (is_null($related)) {
            if ($this->fetchKey($node, $this->morphKey) !== null) {
                $store->register($this->morphKey, null, true);
                $node->register($this->morphKey, null, true);
            }
        } else {
            $rNode = $this->getNode($related);
            if ($this->fetchKey($node, $this->morphKey) != $rNode->getRole()) {
                $store->register($this->morphKey, $rNode->getRole(), true);
                $node->register($this->morphKey, $rNode->getRole(), true);
            }
        }

        return $wrappedStore;
    }

    /**
     * Assert that given entity is allowed for the relation.
     *
     * @param Node $related
     *
     * @throws RelationException
     */
    protected function assertValid(Node $related)
    {
        // no need to validate morphed relation yet
    }
}