]>
Commit | Line | Data |
---|---|---|
1 | <?php | |
2 | ||
3 | namespace Wallabag\UserBundle\Entity; | |
4 | ||
5 | use Doctrine\Common\Collections\ArrayCollection; | |
6 | use Doctrine\ORM\Mapping as ORM; | |
7 | use Scheb\TwoFactorBundle\Model\Email\TwoFactorInterface; | |
8 | use Scheb\TwoFactorBundle\Model\TrustedComputerInterface; | |
9 | use Symfony\Bridge\Doctrine\Validator\Constraints\UniqueEntity; | |
10 | use Symfony\Component\Security\Core\User\UserInterface; | |
11 | use JMS\Serializer\Annotation\ExclusionPolicy; | |
12 | use JMS\Serializer\Annotation\Expose; | |
13 | use FOS\UserBundle\Model\User as BaseUser; | |
14 | use Wallabag\CoreBundle\Entity\Config; | |
15 | use Wallabag\CoreBundle\Entity\Entry; | |
16 | use Wallabag\CoreBundle\Entity\Tag; | |
17 | ||
18 | /** | |
19 | * User. | |
20 | * | |
21 | * @ORM\Entity(repositoryClass="Wallabag\UserBundle\Repository\UserRepository") | |
22 | * @ORM\Table | |
23 | * @ORM\HasLifecycleCallbacks() | |
24 | * @ExclusionPolicy("all") | |
25 | * | |
26 | * @UniqueEntity("email") | |
27 | * @UniqueEntity("username") | |
28 | */ | |
29 | class User extends BaseUser implements TwoFactorInterface, TrustedComputerInterface | |
30 | { | |
31 | /** | |
32 | * @var int | |
33 | * | |
34 | * @Expose | |
35 | * @ORM\Column(name="id", type="integer") | |
36 | * @ORM\Id | |
37 | * @ORM\GeneratedValue(strategy="AUTO") | |
38 | */ | |
39 | protected $id; | |
40 | ||
41 | /** | |
42 | * @var string | |
43 | * | |
44 | * @ORM\Column(name="name", type="text", nullable=true) | |
45 | */ | |
46 | protected $name; | |
47 | ||
48 | /** | |
49 | * @var date | |
50 | * | |
51 | * @ORM\Column(name="created_at", type="datetime") | |
52 | */ | |
53 | protected $createdAt; | |
54 | ||
55 | /** | |
56 | * @var date | |
57 | * | |
58 | * @ORM\Column(name="updated_at", type="datetime") | |
59 | */ | |
60 | protected $updatedAt; | |
61 | ||
62 | /** | |
63 | * @ORM\OneToMany(targetEntity="Wallabag\CoreBundle\Entity\Entry", mappedBy="user", cascade={"remove"}) | |
64 | */ | |
65 | protected $entries; | |
66 | ||
67 | /** | |
68 | * @ORM\OneToOne(targetEntity="Wallabag\CoreBundle\Entity\Config", mappedBy="user") | |
69 | */ | |
70 | protected $config; | |
71 | ||
72 | /** | |
73 | * @ORM\OneToMany(targetEntity="Wallabag\CoreBundle\Entity\Tag", mappedBy="user", cascade={"remove"}) | |
74 | */ | |
75 | protected $tags; | |
76 | ||
77 | /** | |
78 | * @ORM\Column(type="integer", nullable=true) | |
79 | */ | |
80 | private $authCode; | |
81 | ||
82 | /** | |
83 | * @var bool Enabled yes/no | |
84 | * @ORM\Column(type="boolean") | |
85 | */ | |
86 | private $twoFactorAuthentication = false; | |
87 | ||
88 | /** | |
89 | * @ORM\Column(type="json_array", nullable=true) | |
90 | */ | |
91 | private $trusted; | |
92 | ||
93 | public function __construct() | |
94 | { | |
95 | parent::__construct(); | |
96 | $this->entries = new ArrayCollection(); | |
97 | $this->tags = new ArrayCollection(); | |
98 | $this->roles = array('ROLE_USER'); | |
99 | } | |
100 | ||
101 | /** | |
102 | * @ORM\PrePersist | |
103 | * @ORM\PreUpdate | |
104 | */ | |
105 | public function timestamps() | |
106 | { | |
107 | if (is_null($this->createdAt)) { | |
108 | $this->createdAt = new \DateTime(); | |
109 | } | |
110 | ||
111 | $this->updatedAt = new \DateTime(); | |
112 | } | |
113 | ||
114 | /** | |
115 | * Set name. | |
116 | * | |
117 | * @param string $name | |
118 | * | |
119 | * @return User | |
120 | */ | |
121 | public function setName($name) | |
122 | { | |
123 | $this->name = $name; | |
124 | ||
125 | return $this; | |
126 | } | |
127 | ||
128 | /** | |
129 | * Get name. | |
130 | * | |
131 | * @return string | |
132 | */ | |
133 | public function getName() | |
134 | { | |
135 | return $this->name; | |
136 | } | |
137 | ||
138 | /** | |
139 | * @return string | |
140 | */ | |
141 | public function getCreatedAt() | |
142 | { | |
143 | return $this->createdAt; | |
144 | } | |
145 | ||
146 | /** | |
147 | * @return string | |
148 | */ | |
149 | public function getUpdatedAt() | |
150 | { | |
151 | return $this->updatedAt; | |
152 | } | |
153 | ||
154 | /** | |
155 | * @param Entry $entry | |
156 | * | |
157 | * @return User | |
158 | */ | |
159 | public function addEntry(Entry $entry) | |
160 | { | |
161 | $this->entries[] = $entry; | |
162 | ||
163 | return $this; | |
164 | } | |
165 | ||
166 | /** | |
167 | * @return ArrayCollection<Entry> | |
168 | */ | |
169 | public function getEntries() | |
170 | { | |
171 | return $this->entries; | |
172 | } | |
173 | ||
174 | /** | |
175 | * @param Entry $entry | |
176 | * | |
177 | * @return User | |
178 | */ | |
179 | public function addTag(Tag $tag) | |
180 | { | |
181 | $this->tags[] = $tag; | |
182 | ||
183 | return $this; | |
184 | } | |
185 | ||
186 | /** | |
187 | * @return ArrayCollection<Tag> | |
188 | */ | |
189 | public function getTags() | |
190 | { | |
191 | return $this->tags; | |
192 | } | |
193 | ||
194 | public function isEqualTo(UserInterface $user) | |
195 | { | |
196 | return $this->username === $user->getUsername(); | |
197 | } | |
198 | ||
199 | /** | |
200 | * Set config. | |
201 | * | |
202 | * @param Config $config | |
203 | * | |
204 | * @return User | |
205 | */ | |
206 | public function setConfig(Config $config = null) | |
207 | { | |
208 | $this->config = $config; | |
209 | ||
210 | return $this; | |
211 | } | |
212 | ||
213 | /** | |
214 | * Get config. | |
215 | * | |
216 | * @return Config | |
217 | */ | |
218 | public function getConfig() | |
219 | { | |
220 | return $this->config; | |
221 | } | |
222 | ||
223 | /** | |
224 | * @return bool | |
225 | */ | |
226 | public function isTwoFactorAuthentication() | |
227 | { | |
228 | return $this->twoFactorAuthentication; | |
229 | } | |
230 | ||
231 | /** | |
232 | * @param bool $twoFactorAuthentication | |
233 | */ | |
234 | public function setTwoFactorAuthentication($twoFactorAuthentication) | |
235 | { | |
236 | $this->twoFactorAuthentication = $twoFactorAuthentication; | |
237 | } | |
238 | ||
239 | public function isEmailAuthEnabled() | |
240 | { | |
241 | return $this->twoFactorAuthentication; | |
242 | } | |
243 | ||
244 | public function getEmailAuthCode() | |
245 | { | |
246 | return $this->authCode; | |
247 | } | |
248 | ||
249 | public function setEmailAuthCode($authCode) | |
250 | { | |
251 | $this->authCode = $authCode; | |
252 | } | |
253 | ||
254 | public function addTrustedComputer($token, \DateTime $validUntil) | |
255 | { | |
256 | $this->trusted[$token] = $validUntil->format('r'); | |
257 | } | |
258 | ||
259 | public function isTrustedComputer($token) | |
260 | { | |
261 | if (isset($this->trusted[$token])) { | |
262 | $now = new \DateTime(); | |
263 | $validUntil = new \DateTime($this->trusted[$token]); | |
264 | ||
265 | return $now < $validUntil; | |
266 | } | |
267 | ||
268 | return false; | |
269 | } | |
270 | } |