X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/f110b8ba20232e0ceeb67390f8e672431868d32d..baf72cd841a1277e45d37c1fd3a60bd40feb9d5e:/Entity/User.php

diff --git a/Entity/User.php b/Entity/User.php
index 473fd07..bd9f802 100644
--- a/Entity/User.php
+++ b/Entity/User.php
@@ -3,7 +3,17 @@
 // src/Rapsys/AirBundle/Entity/User.php
 namespace Rapsys\AirBundle\Entity;
 
-class User extends \Rapsys\UserBundle\Entity\User {
+use Rapsys\AirBundle\Entity\Application;
+use Rapsys\AirBundle\Entity\Group;
+use Rapsys\AirBundle\Entity\Vote;
+use Rapsys\UserBundle\Entity\User as BaseUser;
+
+class User extends BaseUser {
+	/**
+	 * @var string
+	 */
+	protected $phone;
+
 	/**
 	 * @var \Doctrine\Common\Collections\Collection
 	 */
@@ -21,6 +31,28 @@ class User extends \Rapsys\UserBundle\Entity\User {
 		parent::__construct();
 	}
 
+	/**
+	 * Set phone
+	 *
+	 * @param string $phone
+	 *
+	 * @return User
+	 */
+	public function setPhone($phone) {
+		$this->phone = $phone;
+
+		return $this;
+	}
+
+	/**
+	 * Get phone
+	 *
+	 * @return string
+	 */
+	public function getPhone() {
+		return $this->phone;
+	}
+
 	/**
 	 * Add vote
 	 *
@@ -28,7 +60,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @return User
 	 */
-	public function addVote(\Rapsys\AirBundle\Entity\Vote $vote) {
+	public function addVote(Vote $vote) {
 		$this->votes[] = $vote;
 
 		return $this;
@@ -39,7 +71,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @param \Rapsys\AirBundle\Entity\Vote $vote
 	 */
-	public function removeVote(\Rapsys\AirBundle\Entity\Vote $vote) {
+	public function removeVote(Vote $vote) {
 		$this->votes->removeElement($vote);
 	}
 
@@ -59,7 +91,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @return User
 	 */
-	public function addApplication(\Rapsys\AirBundle\Entity\Application $application) {
+	public function addApplication(Application $application) {
 		$this->applications[] = $application;
 
 		return $this;
@@ -70,7 +102,7 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	 *
 	 * @param \Rapsys\AirBundle\Entity\Application $application
 	 */
-	public function removeApplication(\Rapsys\AirBundle\Entity\Application $application) {
+	public function removeApplication(Application $application) {
 		$this->applications->removeElement($application);
 	}
 
@@ -82,4 +114,15 @@ class User extends \Rapsys\UserBundle\Entity\User {
 	public function getApplications() {
 		return $this->applications;
 	}
+
+	/**
+	 * Get roles
+	 *
+	 * @return array
+	 */
+	public function getRoles() {
+		//Return roles array
+		//XXX: [ ROLE_USER, ROLE_XXX, ... ]
+		return parent::getRoles();
+	}
 }