Malef\Associate\DoctrineOrm\Collector\AssociationCollector

Complexity

Total Complexity

8

Size/Duplication

Total Lines	50
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	collect()	0	7	2
A	collectToManyAssociation()	0	13	2
A	__construct()	0	3	1
A	collectToOneAssociation()	0	14	3

<?php

namespace Malef\Associate\DoctrineOrm\Collector;

use Doctrine\Common\Collections\ArrayCollection;
use Malef\Associate\DoctrineOrm\Metadata\AssociationMetadataAdapter;
use Malef\Associate\DoctrineOrm\Source\UniqueEntitySet;
use Symfony\Component\PropertyAccess\PropertyAccessor;

class AssociationCollector
{
    /**
     * @var PropertyAccessor
     */
    protected $propertyAccessor;

    public function __construct(PropertyAccessor $propertyAccessor)
    {
        $this->propertyAccessor = $propertyAccessor;
    }

    public function collect(
        ArrayCollection $sourceEntities,
        AssociationMetadataAdapter $associationMetadataAdapter
    ): ArrayCollection {
        return $associationMetadataAdapter->isToMany()
            ? $this->collectToManyAssociation($sourceEntities, $associationMetadataAdapter)
            : $this->collectToOneAssociation($sourceEntities, $associationMetadataAdapter);
    }

    protected function collectToManyAssociation(
        ArrayCollection $sourceEntities,
        AssociationMetadataAdapter $associationMetadataAdapter
    ): ArrayCollection {
        $targetEntities = new UniqueEntitySet();

        foreach ($sourceEntities as $sourceEntity) {
            $targetEntities->addMany(
                $this->propertyAccessor->getValue($sourceEntity, $associationMetadataAdapter->getName())->getValues()
            );
        }

        return $targetEntities->getAll();
    }

    protected function collectToOneAssociation(
        ArrayCollection $sourceEntities,
        AssociationMetadataAdapter $associationMetadataAdapter
    ): ArrayCollection {
        $targetEntities = new UniqueEntitySet();

        foreach ($sourceEntities as $sourceEntity) {
            $targetEntity = $this->propertyAccessor->getValue($sourceEntity, $associationMetadataAdapter->getName());
            if ($targetEntity) {
                $targetEntities->addOne($targetEntity);
            }
        }

        return $targetEntities->getAll();
    }
}