Fix location user add/remove from inverse side

[airbundle] / Entity / User.php

diff --git a/Entity/User.php b/Entity/User.php
index 1fb9f1ae33588692a5345a441d3b1b22d5bf69d3..387b1964e33963ef4356d1d838add7238ce3890e 100644 (file)
--- a/Entity/User.php
+++ b/Entity/User.php
@@ -11,59 +11,71 @@
 
 namespace Rapsys\AirBundle\Entity;
 
 
 namespace Rapsys\AirBundle\Entity;
 

+use Doctrine\Common\Collections\Collection;

 use Doctrine\Common\Collections\ArrayCollection;
 
 use Doctrine\Common\Collections\ArrayCollection;
 

-use Rapsys\AirBundle\Entity\Application;
-use Rapsys\AirBundle\Entity\Group;
-use Rapsys\AirBundle\Entity\Link;
-use Rapsys\AirBundle\Entity\Snippet;

 use Rapsys\UserBundle\Entity\User as BaseUser;
 
 class User extends BaseUser {
        /**
 use Rapsys\UserBundle\Entity\User as BaseUser;
 
 class User extends BaseUser {
        /**

-        * @var string
+        * @var ?string

         */
         */

-       protected $pseudonym;
+       private ?string $city;

 
        /**
 
        /**

-        * @var string
+        * @var ?string

         */
         */

-       protected $phone;
+       private ?string $phone;

 
        /**
 
        /**

-        * @var string
+        * @var Country

         */
         */

-       protected $slug;
+       private ?Country $country;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var ?string

         */
         */

-       private $applications;
+       private ?string $pseudonym;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var ?string

         */
         */

-       private $dances;
+       private ?string $zipcode;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var \Doctrine\Common\Collections\Collection

         */
         */

-       private $locations;
+       private Collection $applications;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var \Doctrine\Common\Collections\Collection

         */
         */

-       private $snippets;
+       private Collection $dances;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var \Doctrine\Common\Collections\Collection

         */
         */

-       private $subscribers;
+       private Collection $locations;

 
        /**
 
        /**

-        * @var ArrayCollection
+        * @var \Doctrine\Common\Collections\Collection

         */
         */

-       private $subscriptions;
+       private Collection $snippets;
+
+       /**
+        * @var \Doctrine\Common\Collections\Collection
+        */
+       private Collection $subscribers;
+
+       /**
+        * @var \Doctrine\Common\Collections\Collection
+        */
+       private Collection $subscriptions;
+
+       /**
+        * @var \Doctrine\Common\Collections\Collection
+        */
+       private Collection $googleTokens;

 
        /**
         * Constructor
 
        /**
         * Constructor


@@ -75,9 +87,11 @@ class User extends BaseUser {
                parent::__construct($mail);
 
                //Set defaults
                parent::__construct($mail);
 
                //Set defaults

-               $this->pseudonym = null;
+               $this->city = null;
+               $this->country = null;

                $this->phone = null;
                $this->phone = null;

-               $this->slug = null;
+               $this->pseudonym = null;
+               $this->zipcode = null;

 
                //Set collections
                $this->applications = new ArrayCollection();
 
                //Set collections
                $this->applications = new ArrayCollection();


@@ -86,28 +100,51 @@ class User extends BaseUser {
                $this->snippets = new ArrayCollection();
                $this->subscribers = new ArrayCollection();
                $this->subscriptions = new ArrayCollection();
                $this->snippets = new ArrayCollection();
                $this->subscribers = new ArrayCollection();
                $this->subscriptions = new ArrayCollection();

+               $this->googleTokens = new ArrayCollection();

        }
 
        /**
        }
 
        /**

-        * Set pseudonym
+        * Set city

         *
         *

-        * @param string $pseudonym
+        * @param string $city

         *
         * @return User
         */
         *
         * @return User
         */

-       public function setPseudonym(?string $pseudonym): User {
-               $this->pseudonym = $pseudonym;
+       public function setCity(?string $city): User {
+               $this->city = $city;

 
                return $this;
        }
 
        /**
 
                return $this;
        }
 
        /**

-        * Get pseudonym
+        * Get city

         *
         * @return string
         */
         *
         * @return string
         */

-       public function getPseudonym(): ?string {
-               return $this->pseudonym;
+       public function getCity(): ?string {
+               return $this->city;
+       }
+
+       /**
+        * Set country
+        *
+        * @param Country $country
+        *
+        * @return User
+        */
+       public function setCountry(?Country $country): User {
+               $this->country = $country;
+
+               return $this;
+       }
+
+       /**
+        * Get country
+        *
+        * @return Country
+        */
+       public function getCountry(): ?Country {
+               return $this->country;

        }
 
        /**
        }
 
        /**


@@ -133,87 +170,78 @@ class User extends BaseUser {
        }
 
        /**
        }
 
        /**

-        * Set slug
+        * Set pseudonym

         *
         *

-        * @param string $slug
+        * @param string $pseudonym

         *
         * @return User
         */
         *
         * @return User
         */

-       public function setSlug(?string $slug): User {
-               $this->slug = $slug;
+       public function setPseudonym(?string $pseudonym): User {
+               $this->pseudonym = $pseudonym;

 
                return $this;
        }
 
        /**
 
                return $this;
        }
 
        /**

-        * Get slug
+        * Get pseudonym

         *
         * @return string
         */
         *
         * @return string
         */

-       public function getSlug(): ?string {
-               return $this->slug;
+       public function getPseudonym(): ?string {
+               return $this->pseudonym;

        }
 
        /**
        }
 
        /**

-        * Add application
+        * Set zipcode

         *
         *

-        * @param Application $application
+        * @param string $zipcode

         *
         * @return User
         */
         *
         * @return User
         */

-       public function addApplication(Application $application): User {
-               $this->applications[] = $application;
+       public function setZipcode(?string $zipcode): User {
+               $this->zipcode = $zipcode;

 
                return $this;
        }
 
        /**
 
                return $this;
        }
 
        /**

-        * Remove application
+        * Get zipcode

         *
         *

-        * @param Application $application
-        */
-       public function removeApplication(Application $application): bool {
-               return $this->applications->removeElement($application);
-       }
-
-       /**
-        * Get applications
-        *
-        * @return ArrayCollection
+        * @return string

         */
         */

-       public function getApplications(): ArrayCollection {
-               return $this->applications;
+       public function getZipcode(): ?string {
+               return $this->zipcode;

        }
 
        /**
        }
 
        /**

-        * Add snippet
+        * Add application

         *
         *

-        * @param Snippet $snippet
+        * @param Application $application

         *
         * @return User
         */
         *
         * @return User
         */

-       public function addSnippet(Snippet $snippet): User {
-               $this->snippets[] = $snippet;
+       public function addApplication(Application $application): User {
+               $this->applications[] = $application;

 
                return $this;
        }
 
        /**
 
                return $this;
        }
 
        /**

-        * Remove snippet
+        * Remove application

         *
         *

-        * @param Snippet $snippet
+        * @param Application $application

         */
         */

-       public function removeSnippet(Snippet $snippet): bool {
-               return $this->snippets->removeElement($snippet);
+       public function removeApplication(Application $application): bool {
+               return $this->applications->removeElement($application);

        }
 
        /**
        }
 
        /**

-        * Get snippets
+        * Get applications

         *
         *

-        * @return ArrayCollection
+        * @return \Doctrine\Common\Collections\Collection

         */
         */

-       public function getSnippets(): ArrayCollection {
-               return $this->snippets;
+       public function getApplications(): Collection {
+               return $this->applications;

        }
 
        /**
        }
 
        /**


@@ -243,9 +271,9 @@ class User extends BaseUser {
        /**
         * Get dances
         *
        /**
         * Get dances
         *

-        * @return ArrayCollection
+        * @return \Doctrine\Common\Collections\Collection

         */
         */

-       public function getDances(): ArrayCollection {
+       public function getDances(): Collection {

                return $this->dances;
        }
 
                return $this->dances;
        }
 


@@ -274,12 +302,43 @@ class User extends BaseUser {
        /**
         * Get locations
         *
        /**
         * Get locations
         *

-        * @return ArrayCollection
+        * @return \Doctrine\Common\Collections\Collection

         */
         */

-       public function getLocations(): ArrayCollection {
+       public function getLocations(): Collection {

                return $this->locations;
        }
 
                return $this->locations;
        }
 

+       /**
+        * Add snippet
+        *
+        * @param Snippet $snippet
+        *
+        * @return User
+        */
+       public function addSnippet(Snippet $snippet): User {
+               $this->snippets[] = $snippet;
+
+               return $this;
+       }
+
+       /**
+        * Remove snippet
+        *
+        * @param Snippet $snippet
+        */
+       public function removeSnippet(Snippet $snippet): bool {
+               return $this->snippets->removeElement($snippet);
+       }
+
+       /**
+        * Get snippets
+        *
+        * @return \Doctrine\Common\Collections\Collection
+        */
+       public function getSnippets(): Collection {
+               return $this->snippets;
+       }
+

        /**
         * Add subscriber
         *
        /**
         * Add subscriber
         *


@@ -305,9 +364,9 @@ class User extends BaseUser {
        /**
         * Get subscribers
         *
        /**
         * Get subscribers
         *

-        * @return ArrayCollection
+        * @return \Doctrine\Common\Collections\Collection

         */
         */

-       public function getSubscribers(): ArrayCollection {
+       public function getSubscribers(): Collection {

                return $this->subscribers;
        }
 
                return $this->subscribers;
        }
 


@@ -319,6 +378,9 @@ class User extends BaseUser {
         * @return User
         */
        public function addSubscription(User $subscription): User {
         * @return User
         */
        public function addSubscription(User $subscription): User {

+               //Add from owning side
+               $subscription->addSubscriber($this);
+

                $this->subscriptions[] = $subscription;
 
                return $this;
                $this->subscriptions[] = $subscription;
 
                return $this;


@@ -330,15 +392,53 @@ class User extends BaseUser {
         * @param User $subscription
         */
        public function removeSubscription(User $subscription): bool {
         * @param User $subscription
         */
        public function removeSubscription(User $subscription): bool {

+               if (!$this->users->contains($user)) {
+                       return true;
+               }
+
+               //Remove from owning side
+               $subscription->removeSubscriber($this);
+

                return $this->subscriptions->removeElement($subscription);
        }
 
        /**
         * Get subscriptions
         *
                return $this->subscriptions->removeElement($subscription);
        }
 
        /**
         * Get subscriptions
         *

-        * @return ArrayCollection
+        * @return \Doctrine\Common\Collections\Collection

         */
         */

-       public function getSubscriptions(): ArrayCollection {
+       public function getSubscriptions(): Collection {

                return $this->subscriptions;
        }
                return $this->subscriptions;
        }

+
+       /**
+        * Add google token
+        *
+        * @param GoogleToken $googleToken
+        *
+        * @return User
+        */
+       public function addGoogleToken(GoogleToken $googleToken): User {
+               $this->googleTokens[] = $googleToken;
+
+               return $this;
+       }
+
+       /**
+        * Remove google token
+        *
+        * @param GoogleToken $googleToken
+        */
+       public function removeGoogleToken(GoogleToken $googleToken): bool {
+               return $this->googleTokens->removeElement($googleToken);
+       }
+
+       /**
+        * Get googleTokens
+        *
+        * @return \Doctrine\Common\Collections\Collection
+        */
+       public function getGoogleTokens(): Collection {
+               return $this->googleTokens;
+       }

 }
 }