Add getRecipientName function

[userbundle] / Entity / Group.php

diff --git a/Entity/Group.php b/Entity/Group.php
index c7000ea8627aa3d0fa01468e04938e1579ca8728..c97515c1f36f1e3ad260136508df93dc87f42cb0 100644 (file)
--- a/Entity/Group.php
+++ b/Entity/Group.php
@@ -1,9 +1,25 @@
-<?php
+<?php declare(strict_types=1);
+
+/*
+ * This file is part of the Rapsys UserBundle package.
+ *
+ * (c) Raphaël Gertz <symfony@rapsys.eu>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
 
-// src/Rapsys/UserBundle/Entity/Group.php
 namespace Rapsys\UserBundle\Entity;
 
-class Group extends \Symfony\Component\Security\Core\Role\Role {
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\ORM\Event\PreUpdateEventArgs;
+
+use Rapsys\UserBundle\Entity\User;
+
+/**
+ * Group
+ */
+class Group {
        /**
         * @var integer
         */
@@ -12,7 +28,7 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
        /**
         * @var string
         */
-       protected $role;
+       protected $title;
 
        /**
         * @var \DateTime
@@ -25,39 +41,52 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
        protected $updated;
 
        /**
-        * @var \Doctrine\Common\Collections\Collection
+        * @var ArrayCollection
         */
        protected $users;
 
        /**
         * Constructor
-        * @param string $role The role name
+        *
+        * @param string $title The group name
         */
-       public function __construct($role) {
-               $this->role = (string) $role;
-               $this->users = new \Doctrine\Common\Collections\ArrayCollection();
+       public function __construct(string $title) {
+               //Set defaults
+               $this->title = $title;
+               $this->created = new \DateTime('now');
+               $this->updated = new \DateTime('now');
+               $this->users = new ArrayCollection();
        }
 
        /**
-        * Set role
+        * Get id
         *
-        * @param string $role
+        * @return integer
+        */
+       public function getId(): int {
+               return $this->id;
+       }
+
+       /**
+        * Set title
         *
-        * @return User
+        * @param string $title The group name
+        *
+        * @return Group
         */
-       public function setRole($role) {
-               $this->role = $role;
+       public function setTitle(string $title): Group {
+               $this->title = $title;
 
                return $this;
        }
 
        /**
-        * Get role
+        * Get title
         *
         * @return string
         */
-       public function getRole() {
-               return $this->role;
+       public function getTitle(): ?string {
+               return $this->title;
        }
 
        /**
@@ -65,9 +94,9 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
         *
         * @param \DateTime $created
         *
-        * @return User
+        * @return Group
         */
-       public function setCreated($created) {
+       public function setCreated(\DateTime $created): Group {
                $this->created = $created;
 
                return $this;
@@ -78,7 +107,7 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
         *
         * @return \DateTime
         */
-       public function getCreated() {
+       public function getCreated(): \DateTime {
                return $this->created;
        }
 
@@ -87,9 +116,9 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
         *
         * @param \DateTime $updated
         *
-        * @return User
+        * @return Group
         */
-       public function setUpdated($updated) {
+       public function setUpdated(\DateTime $updated): Group {
                $this->updated = $updated;
 
                return $this;
@@ -100,18 +129,18 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
         *
         * @return \DateTime
         */
-       public function getUpdated() {
+       public function getUpdated(): \DateTime {
                return $this->updated;
        }
 
        /**
         * Add user
         *
-        * @param \Rapsys\UserBundle\Entity\User $user
+        * @param User $user
         *
         * @return Group
         */
-       public function addUser(\Rapsys\UserBundle\Entity\User $user) {
+       public function addUser(User $user) {
                $this->users[] = $user;
 
                return $this;
@@ -120,18 +149,47 @@ class Group extends \Symfony\Component\Security\Core\Role\Role {
        /**
         * Remove user
         *
-        * @param \Rapsys\UserBundle\Entity\User $user
+        * @param User $user
         */
-       public function removeUser(\Rapsys\UserBundle\Entity\User $user) {
+       public function removeUser(User $user) {
                $this->users->removeElement($user);
        }
 
        /**
         * Get users
         *
-        * @return \Doctrine\Common\Collections\Collection
+        * @return ArrayCollection
         */
-       public function getUsers() {
+       public function getUsers(): ArrayCollection {
                return $this->users;
        }
+
+       /**
+        * {@inheritdoc}
+        */
+       public function preUpdate(PreUpdateEventArgs $eventArgs) {
+               //Check that we have a group instance
+               if (($user = $eventArgs->getEntity()) instanceof Group) {
+                       //Set updated value
+                       $user->setUpdated(new \DateTime('now'));
+               }
+       }
+
+       /**
+        * Returns a string representation of the group
+        *
+        * @return string
+        */
+       public function __toString(): string {
+               return $this->title;
+       }
+
+       /**
+        * Get role
+        *
+        * @return string
+        */
+       public function getRole(): string {
+               return 'ROLE_'.strtoupper($this->title);
+       }
 }