X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/58242917a50ec39071529953e77343fd3eb6dda0..d017e31292688fd07b5bf7c4268f16bab17c0ec5:/Entity/User.php

diff --git a/Entity/User.php b/Entity/User.php
index 473fd07..3385a20 100644
--- a/Entity/User.php
+++ b/Entity/User.php
@@ -3,53 +3,125 @@
 // src/Rapsys/AirBundle/Entity/User.php
 namespace Rapsys\AirBundle\Entity;
 
-class User extends \Rapsys\UserBundle\Entity\User {
+use Rapsys\AirBundle\Entity\Application;
+use Rapsys\AirBundle\Entity\Group;
+use Rapsys\AirBundle\Entity\Vote;
+use Rapsys\UserBundle\Entity\User as BaseUser;
+
+class User extends BaseUser {
 	/**
-	 * @var \Doctrine\Common\Collections\Collection
+	 * @var string
+	 */
+	protected $phone;
+
+	/**
+	 * @var string
+	 */
+	protected $donation;
+
+	/**
+	 * @var string
 	 */
-	private $votes;
+	protected $site;
 
 	/**
 	 * @var \Doctrine\Common\Collections\Collection
 	 */
 	private $applications;
 
+	/**
+	 * @var \Doctrine\Common\Collections\Collection
+	 */
+	private $subscribers;
+
+	/**
+	 * @var \Doctrine\Common\Collections\Collection
+	 */
+	private $subscriptions;
+
+	/**
+	 * @var \Doctrine\Common\Collections\Collection
+	 */
+	private $snippets;
+
 	/**
 	 * Constructor
 	 */
 	public function __construct() {
+		//Call parent constructor
 		parent::__construct();
+
+		//Set collections
+		$this->applications = 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();
 	}
 
 	/**
-	 * Add vote
+	 * Set phone
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Vote $vote
+	 * @param string $phone
 	 *
 	 * @return User
 	 */
-	public function addVote(\Rapsys\AirBundle\Entity\Vote $vote) {
-		$this->votes[] = $vote;
+	public function setPhone($phone) {
+		$this->phone = $phone;
 
 		return $this;
 	}
 
 	/**
-	 * Remove vote
+	 * Get phone
 	 *
-	 * @param \Rapsys\AirBundle\Entity\Vote $vote
+	 * @return string
 	 */
-	public function removeVote(\Rapsys\AirBundle\Entity\Vote $vote) {
-		$this->votes->removeElement($vote);
+	public function getPhone() {
+		return $this->phone;
 	}
 
 	/**
-	 * Get votes
+	 * Set donation
 	 *
-	 * @return \Doctrine\Common\Collections\Collection
+	 * @param string $donation
+	 *
+	 * @return User
 	 */
-	public function getVotes() {
-		return $this->votes;
+	public function setDonation($donation) {
+		$this->donation = $donation;
+
+		return $this;
+	}
+
+	/**
+	 * Get donation
+	 *
+	 * @return string
+	 */
+	public function getDonation() {
+		return $this->donation;
+	}
+
+	/**
+	 * Set site
+	 *
+	 * @param string $site
+	 *
+	 * @return User
+	 */
+	public function setSite($site) {
+		$this->site = $site;
+
+		return $this;
+	}
+
+	/**
+	 * Get site
+	 *
+	 * @return string
+	 */
+	public function getSite() {
+		return $this->site;
 	}
 
 	/**
@@ -59,7 +131,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @return User
 	 */
-	public function addApplication(\Rapsys\AirBundle\Entity\Application $application) {
+	public function addApplication(Application $application) {
 		$this->applications[] = $application;
 
 		return $this;
@@ -70,7 +142,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @param \Rapsys\AirBundle\Entity\Application $application
 	 */
-	public function removeApplication(\Rapsys\AirBundle\Entity\Application $application) {
+	public function removeApplication(Application $application) {
 		$this->applications->removeElement($application);
 	}
 
@@ -82,4 +154,97 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	public function getApplications() {
 		return $this->applications;
 	}
+
+	/**
+	 * Add subscriber
+	 *
+	 * @param \Rapsys\AirBundle\Entity\User $subscriber
+	 *
+	 * @return User
+	 */
+	public function addSubscriber(User $subscriber) {
+		$this->subscribers[] = $subscriber;
+
+		return $this;
+	}
+
+	/**
+	 * Remove subscriber
+	 *
+	 * @param \Rapsys\AirBundle\Entity\User $subscriber
+	 */
+	public function removeSubscriber(User $subscriber) {
+		$this->subscribers->removeElement($subscriber);
+	}
+
+	/**
+	 * Get subscribers
+	 *
+	 * @return \Doctrine\Common\Collections\Collection
+	 */
+	public function getSubscribers() {
+		return $this->subscribers;
+	}
+
+	/**
+	 * Add subscription
+	 *
+	 * @param \Rapsys\AirBundle\Entity\User $subscription
+	 *
+	 * @return User
+	 */
+	public function addSubscription(User $subscription) {
+		$this->subscriptions[] = $subscription;
+
+		return $this;
+	}
+
+	/**
+	 * Remove subscription
+	 *
+	 * @param \Rapsys\AirBundle\Entity\User $subscription
+	 */
+	public function removeSubscription(User $subscription) {
+		$this->subscriptions->removeElement($subscription);
+	}
+
+	/**
+	 * Get subscriptions
+	 *
+	 * @return \Doctrine\Common\Collections\Collection
+	 */
+	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;
+	}
 }