Add snipper hat field

[airbundle] / Entity / User.php

diff --git a/Entity/User.php b/Entity/User.php
index 3385a20dbffbf31c3892fcab27b1451f054af0bd..22bf46da496d7298ff95163b1258a6e10567bdd4 100644 (file)
--- 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;
-       }
 }