Slick\Orm\Entity\EntityCollection

Complexity

Total Complexity

11

Size/Duplication

Total Lines	138
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	8

Test Coverage

Coverage

45.71%

Importance

Changes	2
Bugs	0	Features	2

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	offsetSet()	0	4	1
A	add()	0	7	2
A	addEntity()	0	12	2
A	addEntityWithId()	0	11	2
A	getRepository()	0	7	2
A	setRepository()	0	5	1

<?php

/**
 * This file is part of slick/orm package
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Slick\Orm\Entity;

use League\Event\EmitterAwareTrait;
use Slick\Common\Utils\Collection\AbstractCollection;
use Slick\Orm\EntityInterface;
use Slick\Orm\Event\EntityAdded;
use Slick\Orm\Exception\EntityNotFoundException;
use Slick\Orm\Exception\InvalidArgumentException;
use Slick\Orm\Orm;
use Slick\Orm\RepositoryInterface;

/**
 * Entity Collection
 *
 * @package Slick\Orm\Entity
 * @author  Filipe Silva <silvam.filipe@gmail.com>
 */
class EntityCollection extends AbstractCollection implements
    EntityCollectionInterface
{
    /**
     * @var string
     */
    protected $type;

    /**
     * @var RepositoryInterface
     */
    protected $repository;

    /**
     * Emitter methods
     */
    use EmitterAwareTrait;

    /**
     * Creates the collection for entity type with provided data
     *
     * @param string $entityType
     * @param array|\Traversable $data
     */
    public function __construct($entityType, $data = [])
    {
        parent::__construct($data);
        $this->type = $entityType;
    }

    /**
     * Offset to set
     *
     * @link http://php.net/manual/en/arrayaccess.offsetset.php
     *
     * @param mixed $offset The offset to assign the value to.
     * @param mixed $value  The value to set.
     */
    public function offsetSet($offset, $value)
    {
        $this->add($value);
    }

    /**
     * Adds an entity to the collection
     *
     * This method adds an entity to the collection by accepting the entity
     * itself or the entity id value
     *
     * @param EntityInterface|integer $entity
     *
     * @throws EntityNotFoundException If the provided id is does not
     *  references an existing entity.
     * @throws InvalidArgumentException If the entity is not from
     *  the provided entity type
     *
     * @return self
     */
    public function add($entity)
    {
        if ($entity instanceof EntityInterface) {
            return $this->addEntity($entity);
        }
        return $this->addEntityWithId($entity);
    }

    /**
     * Adds an entity to the collection
     *
     * @param EntityInterface $entity
     * @throws InvalidArgumentException If the entity is not from
     *  the provided entity type
     *
     * @return self
     */
    public function addEntity(EntityInterface $entity)
    {
        if (!is_a($entity, $this->type)) {
            throw new InvalidArgumentException(
                "Trying to add an entity that is not a {$this->type}."
            );
        }
        $this->data[] = $entity;
        $event = new EntityAdded($entity);
        $this->getEmitter()->emit($event->getName());
        return $this;
    }

    /**
     * Adds an entity by entity id
     *
     * @param mixed $entityId
     *
     * @throws EntityNotFoundException If the provided id is does not
     *  references an existing entity.
     * @throws InvalidArgumentException If the entity is not from
     *  the provided entity type
     *
     * @return self
     */
    public function addEntityWithId($entityId)
    {
        $entity = $this->getRepository()->get($entityId);
        if (!$entity) {
            throw new EntityNotFoundException(
                "The entity with is '{$entityId}' was not found."
            );
        }
        return $this->addEntity($entity);
        
    }

    /**
     * Gets entity repository for this collection
     * 
     * @return RepositoryInterface
     */
    public function getRepository()
    {
        if (null == $this->repository) {
            $this->setRepository(Orm::getRepository($this->type));
        }
        return $this->repository;
    }

    /**
     * Set repository for this entity collection
     * 
     * @param RepositoryInterface $repository
     * 
     * @return self|$this|EntityCollection
     */
    public function setRepository(RepositoryInterface $repository)
    {
        $this->repository = $repository;
        return $this;
    }
}