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")
	 */
	private $name;

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

	/**
	 * @ORM\OneToMany(targetEntity="App\Entity\Poll", mappedBy="choices")
	 * @Serializer\Type("App\Entity\Poll")
	 */
	private $poll;
	/**
	 * @ORM\OneToMany(targetEntity="App\Entity\StackOfVotes", mappedBy="choice")
	 * @Serializer\Type("App\Entity\StackOfVotes")
	 */
	private $stackOfVotes;

	/**
	 * @ORM\OneToMany(targetEntity="App\Entity\Vote", mappedBy="choice")
	 * @Serializer\Type("App\Entity\Vote")
	 */
	private $votes;

	public function __construct() {
		$this->poll         = new ArrayCollection();
		$this->stackOfVotes = 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|Poll[]
	 */
	public function getPoll(): Collection {
		return $this->poll;
	}

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

		return $this;
	}

	public function addPoll( Poll $poll ): self {
		if ( ! $this->poll->contains( $poll ) ) {
			$this->poll[] = $poll;
			$poll->setChoices( $this );
		}

		return $this;
	}

	public function removePoll( Poll $poll ): self {
		if ( $this->poll->contains( $poll ) ) {
			$this->poll->removeElement( $poll );
			// set the owning side to null (unless already changed)
			if ( $poll->getChoices() === $this ) {
				$poll->setChoices( null );
			}
		}

		return $this;
	}

	/**
	 * @return Collection|Choice[]
	 */
	public function getStackOfVotes(): Collection {
		return $this->stackOfVotes;
	}

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

		return $this;
	}

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

		return $this;
	}

	public function addStackOfVote( StackOfVotes $stackOfVote ): self {
		if ( ! $this->stackOfVotes->contains( $stackOfVote ) ) {
			$this->stackOfVotes[] = $stackOfVote;
			$stackOfVote->setChoice( $this );
		}

		return $this;
	}

	public function removeStackOfVote( StackOfVotes $stackOfVote ): self {
		if ( $this->stackOfVotes->contains( $stackOfVote ) ) {
			$this->stackOfVotes->removeElement( $stackOfVote );
			// set the owning side to null (unless already changed)
			if ( $stackOfVote->getChoice() === $this ) {
				$stackOfVote->setChoice( null );
			}
		}

		return $this;
	}

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