X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/bd16cca32dea4c86fcec8df5292f53c4e3f1c45d..d2cc96771dfe99250fc381de504c374c0a4e4e49:/Entity/User.php

diff --git a/Entity/User.php b/Entity/User.php
index 3385a20..22bf46d 100644
--- a/Entity/User.php
+++ b/Entity/User.php
@@ -5,7 +5,8 @@ namespace Rapsys\AirBundle\Entity;
 
 use Rapsys\AirBundle\Entity\Application;
 use Rapsys\AirBundle\Entity\Group;
-use Rapsys\AirBundle\Entity\Vote;
+use Rapsys\AirBundle\Entity\Link;
+use Rapsys\AirBundle\Entity\Snippet;
 use Rapsys\UserBundle\Entity\User as BaseUser;
 
 class User extends BaseUser {
@@ -15,19 +16,19 @@ class User extends BaseUser {
 	protected $phone;
 
 	/**
-	 * @var string
+	 * @var \Doctrine\Common\Collections\Collection
 	 */
-	protected $donation;
+	private $applications;
 
 	/**
-	 * @var string
+	 * @var \Doctrine\Common\Collections\Collection
 	 */
-	protected $site;
+	private $locations;
 
 	/**
 	 * @var \Doctrine\Common\Collections\Collection
 	 */
-	private $applications;
+	private $snippets;
 
 	/**
 	 * @var \Doctrine\Common\Collections\Collection
@@ -39,23 +40,21 @@ class User extends BaseUser {
 	 */
 	private $subscriptions;
 
-	/**
-	 * @var \Doctrine\Common\Collections\Collection
-	 */
-	private $snippets;
-
 	/**
 	 * Constructor
+	 *
+	 * @param string $mail The user mail
 	 */
-	public function __construct() {
+	public function __construct(string $mail) {
 		//Call parent constructor
-		parent::__construct();
+		parent::__construct($mail);
 
 		//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->snippets = new \Doctrine\Common\Collections\ArrayCollection();
 	}
 
 	/**
@@ -155,6 +154,68 @@ class User extends BaseUser {
 		return $this->applications;
 	}
 
+	/**
+	 * Add snippet
+	 *
+	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+	 *
+	 * @return User
+	 */
+	public function addSnippet(Snippet $snippet) {
+		$this->snippets[] = $snippet;
+
+		return $this;
+	}
+
+	/**
+	 * Remove snippet
+	 *
+	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
+	 */
+	public function removeSnippet(Snippet $snippet) {
+		$this->snippets->removeElement($snippet);
+	}
+
+	/**
+	 * Get snippets
+	 *
+	 * @return \Doctrine\Common\Collections\Collection
+	 */
+	public function getSnippets() {
+		return $this->snippets;
+	}
+
+	/**
+	 * Add location
+	 *
+	 * @param \Rapsys\AirBundle\Entity\Location $location
+	 *
+	 * @return User
+	 */
+	public function addLocation(Location $location) {
+		$this->locations[] = $location;
+
+		return $this;
+	}
+
+	/**
+	 * Remove location
+	 *
+	 * @param \Rapsys\AirBundle\Entity\Location $location
+	 */
+	public function removeLocation(Location $location) {
+		$this->locations->removeElement($location);
+	}
+
+	/**
+	 * Get locations
+	 *
+	 * @return \Doctrine\Common\Collections\Collection
+	 */
+	public function getLocations() {
+		return $this->locations;
+	}
+
 	/**
 	 * Add subscriber
 	 *
@@ -216,35 +277,4 @@ class User extends BaseUser {
 	public function getSubscriptions() {
 		return $this->subscriptions;
 	}
-
-	/**
-	 * Add snippet
-	 *
-	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
-	 *
-	 * @return User
-	 */
-	public function addSnippet(\Rapsys\AirBundle\Entity\Snippet $snippet) {
-		$this->snippets[] = $snippet;
-
-		return $this;
-	}
-
-	/**
-	 * Remove snippet
-	 *
-	 * @param \Rapsys\AirBundle\Entity\Snippet $snippet
-	 */
-	public function removeSnippet(\Rapsys\AirBundle\Entity\Snippet $snippet) {
-		$this->snippets->removeElement($snippet);
-	}
-
-	/**
-	 * Get snippets
-	 *
-	 * @return \Doctrine\Common\Collections\Collection
-	 */
-	public function getSnippets() {
-		return $this->snippets;
-	}
 }