date-poll-api/src/Entity/Choice.php

<?php

namespace App\Entity;

use DateTimeInterface;
use Doctrine\Common\Collections\ArrayCollection;
use Doctrine\Common\Collections\Collection;
use Doctrine\ORM\Mapping as ORM;
use JMS\Serializer\Annotation as Serializer;

/**
 * one poll choice, could be a text or a date
 * @ORM\Entity(repositoryClass="App\Repository\ChoiceRepository")
 */
class Choice {
	/**
	 * @ORM\Id()
	 * @ORM\GeneratedValue()
	 * @ORM\Column(type="integer")
	 */
	private $id;

	/**
	 * @ORM\Column(type="string", length=255, nullable=true)
	 * @Serializer\Type("string")
	 */
	public $name;

	/**
	 * @ORM\Column(type="datetime", nullable=true)
	 * @Serializer\Type("datetime")
	 */
	public $dateTime;

	/**
	 * @ORM\ManyToOne(targetEntity="App\Entity\Poll", inversedBy="choices", cascade={"persist"})
	 * @Serializer\Type("App\Entity\Poll")
	 */
	private $poll;

	/**
	 * @ORM\OneToMany(targetEntity="App\Entity\Vote", mappedBy="choice", cascade={"persist"})
	 * @Serializer\Type("App\Entity\Vote")
	 */
	public $votes;

	public function __construct() {
      		$this->poll  = new ArrayCollection();
      		$this->votes = new ArrayCollection();
      		$this->setDateTime( new \DateTime() );
      	}


	public function getId(): ?int {
      		return $this->id;
      	}

	public function getName(): ?string {
      		return $this->name;
      	}

	public function setName( $name ): self {
      		if ( is_a( $name, 'DateTime' ) ) {
      			$this->setDateTime( $name );
      			$name = $name->format( 'D Y-m-d' );
      		}
      		$this->name = $name;
      
      		return $this;
      	}

	public function getDateTime(): ?DateTimeInterface {
      		return $this->dateTime;
      	}

	public function setDateTime( ?DateTimeInterface $dateTime ): self {
      		$this->dateTime = $dateTime;
      
      		return $this;
      	}


	/**
	 * @return Collection|Vote[]
	 */
	public function getVotes(): Collection {
      		return $this->votes;
      	}

	public function addVote( Vote $vote ): self {
      		if ( ! $this->votes->contains( $vote ) ) {
      			$this->votes[] = $vote;
      			$vote->setChoice( $this );
      		}
      
      		return $this;
      	}

	public function removeVote( Vote $vote ): self {
      		if ( $this->votes->contains( $vote ) ) {
      			$this->votes->removeElement( $vote );
      			// set the owning side to null (unless already changed)
      			if ( $vote->getChoice() === $this ) {
      				$vote->setChoice( null );
      			}
      		}
      
      		return $this;
      	}

    public function getPoll(): ?Poll
    {
        return $this->poll;
    }

    public function setPoll(?Poll $poll): self
    {
        $this->poll = $poll;

        return $this;
    }
}