X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/1b730fd128c2767014b75babd4cb80381bbdd4c5..b8b55ddc3d67e65f83b7e38d88affc654a264e0e:/Entity/User.php

diff --git a/Entity/User.php b/Entity/User.php
index 151f1a1..1fb9f1a 100644
--- a/Entity/User.php
+++ b/Entity/User.php
@@ -1,8 +1,18 @@
-<?php
+<?php declare(strict_types=1);
+
+/*
+ * this file is part of the rapsys packbundle package.
+ *
+ * (c) raphaël gertz <symfony@rapsys.eu>
+ *
+ * for the full copyright and license information, please view the license
+ * file that was distributed with this source code.
+ */
 
-// src/Rapsys/AirBundle/Entity/User.php
 namespace Rapsys\AirBundle\Entity;
 
+use Doctrine\Common\Collections\ArrayCollection;
+
 use Rapsys\AirBundle\Entity\Application;
 use Rapsys\AirBundle\Entity\Group;
 use Rapsys\AirBundle\Entity\Link;
@@ -10,125 +20,148 @@ use Rapsys\AirBundle\Entity\Snippet;
 use Rapsys\UserBundle\Entity\User as BaseUser;
 
 class User extends BaseUser {
+	/**
+	 * @var string
+	 */
+	protected $pseudonym;
+
 	/**
 	 * @var string
 	 */
 	protected $phone;
 
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var string
+	 */
+	protected $slug;
+
+	/**
+	 * @var ArrayCollection
 	 */
 	private $applications;
 
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var ArrayCollection
+	 */
+	private $dances;
+
+	/**
+	 * @var ArrayCollection
 	 */
 	private $locations;
 
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var ArrayCollection
 	 */
 	private $snippets;
 
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var ArrayCollection
 	 */
 	private $subscribers;
 
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var ArrayCollection
 	 */
 	private $subscriptions;
 
 	/**
 	 * Constructor
+	 *
+	 * @param string $mail The user mail
 	 */
-	public function __construct() {
+	public function __construct(string $mail) {
 		//Call parent constructor
-		parent::__construct();
+		parent::__construct($mail);
+
+		//Set defaults
+		$this->pseudonym = null;
+		$this->phone = null;
+		$this->slug = null;
 
 		//Set collections
-		$this->applications = new \Doctrine\Common\Collections\ArrayCollection();
-		$this->locations = new \Doctrine\Common\Collections\ArrayCollection();
-		$this->snippets = new \Doctrine\Common\Collections\ArrayCollection();
-		$this->subscribers = new \Doctrine\Common\Collections\ArrayCollection();
-		$this->subscriptions = new \Doctrine\Common\Collections\ArrayCollection();
+		$this->applications = new ArrayCollection();
+		$this->dances = new ArrayCollection();
+		$this->locations = new ArrayCollection();
+		$this->snippets = new ArrayCollection();
+		$this->subscribers = new ArrayCollection();
+		$this->subscriptions = new ArrayCollection();
 	}
 
 	/**
-	 * Set phone
+	 * Set pseudonym
 	 *
-	 * @param string $phone
+	 * @param string $pseudonym
 	 *
 	 * @return User
 	 */
-	public function setPhone($phone) {
-		$this->phone = $phone;
+	public function setPseudonym(?string $pseudonym): User {
+		$this->pseudonym = $pseudonym;
 
 		return $this;
 	}
 
 	/**
-	 * Get phone
+	 * Get pseudonym
 	 *
 	 * @return string
 	 */
-	public function getPhone() {
-		return $this->phone;
+	public function getPseudonym(): ?string {
+		return $this->pseudonym;
 	}
 
 	/**
-	 * Set donation
+	 * Set phone
 	 *
-	 * @param string $donation
+	 * @param string $phone
 	 *
 	 * @return User
 	 */
-	public function setDonation($donation) {
-		$this->donation = $donation;
+	public function setPhone(?string $phone): User {
+		$this->phone = $phone;
 
 		return $this;
 	}
 
 	/**
-	 * Get donation
+	 * Get phone
 	 *
 	 * @return string
 	 */
-	public function getDonation() {
-		return $this->donation;
+	public function getPhone(): ?string {
+		return $this->phone;
 	}
 
 	/**
-	 * Set site
+	 * Set slug
 	 *
-	 * @param string $site
+	 * @param string $slug
 	 *
 	 * @return User
 	 */
-	public function setSite($site) {
-		$this->site = $site;
+	public function setSlug(?string $slug): User {
+		$this->slug = $slug;
 
 		return $this;
 	}
 
 	/**
-	 * Get site
+	 * Get slug
 	 *
 	 * @return string
 	 */
-	public function getSite() {
-		return $this->site;
+	public function getSlug(): ?string {
+		return $this->slug;
 	}
 
 	/**
 	 * Add application
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Application $application
+	 * @param Application $application
 	 *
 	 * @return User
 	 */
-	public function addApplication(Application $application) {
+	public function addApplication(Application $application): User {
 		$this->applications[] = $application;
 
 		return $this;
@@ -137,29 +170,29 @@ class User extends BaseUser {
 	/**
 	 * Remove application
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Application $application
+	 * @param Application $application
 	 */
-	public function removeApplication(Application $application) {
-		$this->applications->removeElement($application);
+	public function removeApplication(Application $application): bool {
+		return $this->applications->removeElement($application);
 	}
 
 	/**
 	 * Get applications
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @return ArrayCollection
 	 */
-	public function getApplications() {
+	public function getApplications(): ArrayCollection {
 		return $this->applications;
 	}
 
 	/**
 	 * Add snippet
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+	 * @param Snippet $snippet
 	 *
 	 * @return User
 	 */
-	public function addSnippet(Snippet $snippet) {
+	public function addSnippet(Snippet $snippet): User {
 		$this->snippets[] = $snippet;
 
 		return $this;
@@ -168,29 +201,62 @@ class User extends BaseUser {
 	/**
 	 * Remove snippet
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+	 * @param Snippet $snippet
 	 */
-	public function removeSnippet(Snippet $snippet) {
-		$this->snippets->removeElement($snippet);
+	public function removeSnippet(Snippet $snippet): bool {
+		return $this->snippets->removeElement($snippet);
 	}
 
 	/**
 	 * Get snippets
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @return ArrayCollection
 	 */
-	public function getSnippets() {
+	public function getSnippets(): ArrayCollection {
 		return $this->snippets;
 	}
 
+	/**
+	 * Add dance
+	 *
+	 * @param Dance $dance
+	 *
+	 * @return User
+	 */
+	public function addDance(Dance $dance): User {
+		$this->dances[] = $dance;
+
+		return $this;
+	}
+
+	/**
+	 * Remove dance
+	 *
+	 * @param Dance $dance
+	 *
+	 * @return bool
+	 */
+	public function removeDance(Dance $dance): bool {
+		return $this->dances->removeElement($dance);
+	}
+
+	/**
+	 * Get dances
+	 *
+	 * @return ArrayCollection
+	 */
+	public function getDances(): ArrayCollection {
+		return $this->dances;
+	}
+
 	/**
 	 * Add location
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Location $location
+	 * @param Location $location
 	 *
 	 * @return User
 	 */
-	public function addLocation(Location $location) {
+	public function addLocation(Location $location): User {
 		$this->locations[] = $location;
 
 		return $this;
@@ -199,29 +265,29 @@ class User extends BaseUser {
 	/**
 	 * Remove location
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Location $location
+	 * @param Location $location
 	 */
-	public function removeLocation(Location $location) {
-		$this->locations->removeElement($location);
+	public function removeLocation(Location $location): bool {
+		return $this->locations->removeElement($location);
 	}
 
 	/**
 	 * Get locations
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @return ArrayCollection
 	 */
-	public function getLocations() {
+	public function getLocations(): ArrayCollection {
 		return $this->locations;
 	}
 
 	/**
 	 * Add subscriber
 	 *
-	 * @param \Rapsys\AirBundle\Entity\User $subscriber
+	 * @param User $subscriber
 	 *
 	 * @return User
 	 */
-	public function addSubscriber(User $subscriber) {
+	public function addSubscriber(User $subscriber): User {
 		$this->subscribers[] = $subscriber;
 
 		return $this;
@@ -230,29 +296,29 @@ class User extends BaseUser {
 	/**
 	 * Remove subscriber
 	 *
-	 * @param \Rapsys\AirBundle\Entity\User $subscriber
+	 * @param User $subscriber
 	 */
-	public function removeSubscriber(User $subscriber) {
-		$this->subscribers->removeElement($subscriber);
+	public function removeSubscriber(User $subscriber): bool {
+		return $this->subscribers->removeElement($subscriber);
 	}
 
 	/**
 	 * Get subscribers
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @return ArrayCollection
 	 */
-	public function getSubscribers() {
+	public function getSubscribers(): ArrayCollection {
 		return $this->subscribers;
 	}
 
 	/**
 	 * Add subscription
 	 *
-	 * @param \Rapsys\AirBundle\Entity\User $subscription
+	 * @param User $subscription
 	 *
 	 * @return User
 	 */
-	public function addSubscription(User $subscription) {
+	public function addSubscription(User $subscription): User {
 		$this->subscriptions[] = $subscription;
 
 		return $this;
@@ -261,18 +327,18 @@ class User extends BaseUser {
 	/**
 	 * Remove subscription
 	 *
-	 * @param \Rapsys\AirBundle\Entity\User $subscription
+	 * @param User $subscription
 	 */
-	public function removeSubscription(User $subscription) {
-		$this->subscriptions->removeElement($subscription);
+	public function removeSubscription(User $subscription): bool {
+		return $this->subscriptions->removeElement($subscription);
 	}
 
 	/**
 	 * Get subscriptions
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @return ArrayCollection
 	 */
-	public function getSubscriptions() {
+	public function getSubscriptions(): ArrayCollection {
 		return $this->subscriptions;
 	}
 }