Remove vote feature

[airbundle] / Entity / User.php

diff --git a/Entity/User.php b/Entity/User.php
index 473fd07e894cf0efd83d0deb5c0a256a98933fe7..fd35891ea950ab872890a53094f9b8b269d2af23 100644 (file)
--- 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
         */
-       private $votes;
+       protected $phone;
+
+       /**
+        * @var string
+        */
+       protected $donation;
+
+       /**
+        * @var string
+        */
+       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 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 getVotes() {
-               return $this->votes;
+       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 Location
+        */
+       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;
+       }
 }