X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/1c376852cd0277866cbb7345ea00993f0b37ad76..a19fbcfc503a73d83e1fb7213a5e566fedfb3454:/Entity/Dance.php?ds=sidebyside

diff --git a/Entity/Dance.php b/Entity/Dance.php
index dccee33..796b81f 100644
--- a/Entity/Dance.php
+++ b/Entity/Dance.php
@@ -209,6 +209,9 @@ class Dance {
 	 * @return Dance
 	 */
 	public function addUser(User $user): Dance {
+		//Add from owning side
+		$user->addDance($this);
+
 		$this->users[] = $user;
 
 		return $this;
@@ -222,6 +225,13 @@ class Dance {
 	 * @return bool
 	 */
 	public function removeUser(User $user): bool {
+		if (!$this->dances->contains($user)) {
+			return true;
+		}
+
+		//Remove from owning side
+		$user->removeDance($this);
+
 		return $this->users->removeElement($user);
 	}