PiedWeb\CMSBundle\Entity\PageImageTrait

Complexity

Total Complexity

16

Size/Duplication

Total Lines	94
Duplicated Lines	0 %

Test Coverage

Coverage

8.82%

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	addPageHasMedia()	0	4	1
A	setPageHasMedias()	0	5	2
A	__constructImage()	0	3	1
A	removePageHasMedia()	0	3	1
A	setMainImage()	0	10	3
A	getMainImage()	0	3	1
A	getPageHasMedias()	0	3	1
A	issetImage()	0	7	2
A	getImages()	0	12	4

<?php

namespace PiedWeb\CMSBundle\Entity;

use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\Common\Collections\Collection;
use Doctrine\ORM\Mapping as ORM;

trait PageImageTrait
{
    /**
     * @ORM\ManyToOne(
     *     targetEntity="PiedWeb\CMSBundle\Entity\MediaInterface",
     *     cascade={"all"},
     *     inversedBy="mainImagePages"
     * )
     */
    protected $mainImage;

    /**
     * @var ArrayCollection
     */
    protected $images;

    /**
     * @ORM\OneToMany(
     *     targetEntity="PiedWeb\CMSBundle\Entity\PageHasMediaInterface",
     *     mappedBy="page",
     *     cascade={"all"},
     *     orphanRemoval=true
     * )
     * @ORM\OrderBy({"position": "ASC"})
     */
    protected $pageHasMedias;

    public function __constructImage()
    {
        $this->pageHasMedias = new ArrayCollection();
    }

    public function setPageHasMedias($pageHasMedias)
    {
        $this->pageHasMedias = new ArrayCollection();
        foreach ($pageHasMedias as $pageHasMedia) {
            $this->addPageHasMedia($pageHasMedia);
        }
    }

    public function getPageHasMedias()
    {
        return $this->pageHasMedias;
    }

    public function addPageHasMedia(PageHasMedia $pageHasMedia)
    {
        $pageHasMedia->setPage($this);
        $this->pageHasMedias[] = $pageHasMedia;
    }

    public function removePageHasMedia(PageHasMedia $pageHasMedia)
    {
        $this->pageHasMedias->removeElement($pageHasMedia);
    }

    public function getMainImage(): ?MediaInterface
    {
        return $this->mainImage;
    }

    public function setMainImage(?MediaInterface $mainImage): self
    {
        // TODO: Déplacer en Assert pour éviter une erreur dégueu ?!
        if (null !== $mainImage && null === $mainImage->getWidth()) {
            throw new \Exception('mainImage must be an Image. Media imported is not an image');
        }

        $this->mainImage = $mainImage;

        return $this;
    }

    public function getImages(): Collection
    {
        if (! $this->images) {
            $this->images = new ArrayCollection();
            foreach ($this->pageHasMedias as $p) {
                if (null !== $p->getMedia()) {
                    $this->images[] = $p->getMedia();
                }
            }
        }

        return $this->images;
    }

    public function issetImage()
    {
        if ($this->getImages()->count() > 0) {
            return true;
        }

        return false;
    }
}