X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/05b55f89bc508b0d6c43c4d854fb5e9ffdf6484c..7e8ef252c23703363c351b23ff1d1ce9fac80ce5:/Entity/Dance.php?ds=inline diff --git a/Entity/Dance.php b/Entity/Dance.php index c269f32..0b132c8 100644 --- a/Entity/Dance.php +++ b/Entity/Dance.php @@ -26,7 +26,12 @@ class Dance { /** * @var string */ - protected $title; + protected $name; + + /** + * @var string + */ + protected $type; /** * @var \DateTime @@ -50,9 +55,14 @@ class Dance { /** * Constructor + * + * @param string $name The dance name + * @param string $type The dance type */ - public function __construct() { + public function __construct(string $name, string $type) { //Set defaults + $this->name = $name; + $this->type = $type; $this->created = new \DateTime('now'); $this->updated = new \DateTime('now'); @@ -71,25 +81,47 @@ class Dance { } /** - * Set title + * Set name + * + * @param string $name + * + * @return Dance + */ + public function setName(string $name): Dance { + $this->name = $name; + + return $this; + } + + /** + * Get name + * + * @return string + */ + public function getName(): string { + return $this->name; + } + + /** + * Set type * - * @param string $title + * @param string $type * * @return Dance */ - public function setTitle(string $title): Dance { - $this->title = $title; + public function setType(string $type): Dance { + $this->type = $type; return $this; } /** - * Get title + * Get type * * @return string */ - public function getTitle(): string { - return $this->title; + public function getType(): string { + return $this->type; } /** @@ -177,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; @@ -190,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); } @@ -207,7 +249,7 @@ class Dance { */ public function preUpdate(PreUpdateEventArgs $eventArgs) { //Check that we have a dance instance - if (($dance = $eventArgs->getEntity()) instanceof Dance) { + if (($dance = $eventArgs->getObject()) instanceof Dance) { //Set updated value $dance->setUpdated(new \DateTime('now')); } @@ -219,6 +261,6 @@ class Dance { * @return string */ public function __toString(): string { - return $this->title; + return $this->name.' '.lcfirst($this->type); } }