| Total Complexity | 40 |
| Total Lines | 363 |
| Duplicated Lines | 0 % |
| Changes | 1 | ||
| Bugs | 0 | Features | 0 |
Complex classes like AuthTrait often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use AuthTrait, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 40 | trait AuthTrait |
||
| 41 | { |
||
| 42 | |||
| 43 | /** |
||
| 44 | * @var MailerInterface |
||
| 45 | */ |
||
| 46 | protected $mailer; |
||
| 47 | |||
| 48 | /** |
||
| 49 | * @var Hasher |
||
| 50 | */ |
||
| 51 | protected $hasher; |
||
| 52 | |||
| 53 | /** |
||
| 54 | * @var JwtToken |
||
| 55 | */ |
||
| 56 | protected $jwt; |
||
| 57 | |||
| 58 | /** |
||
| 59 | * @var AuthServiceInterface |
||
| 60 | */ |
||
| 61 | protected $authService; |
||
| 62 | |||
| 63 | /** |
||
| 64 | * @var int |
||
| 65 | */ |
||
| 66 | protected $otpLength = 6; |
||
| 67 | |||
| 68 | /** |
||
| 69 | * @var array |
||
| 70 | */ |
||
| 71 | protected $keyFields = []; |
||
| 72 | |||
| 73 | /** |
||
| 74 | * @var array |
||
| 75 | */ |
||
| 76 | protected $visibleFields = []; |
||
| 77 | |||
| 78 | /** |
||
| 79 | * @inheritDoc |
||
| 80 | * @return bool |
||
| 81 | * @throws ConfigException |
||
| 82 | * @throws CryptorException |
||
| 83 | * @throws DatabaseException |
||
| 84 | * @throws DiException |
||
| 85 | * @throws ReflectionException |
||
| 86 | * @throws SessionException |
||
| 87 | * @throws JwtException |
||
| 88 | */ |
||
| 89 | public function check(): bool |
||
| 90 | { |
||
| 91 | return !is_null($this->user()); |
||
|
|
|||
| 92 | } |
||
| 93 | |||
| 94 | /** |
||
| 95 | * Sign Up |
||
| 96 | * @param array $userData |
||
| 97 | * @param array|null $customData |
||
| 98 | * @return User |
||
| 99 | */ |
||
| 100 | public function signup(array $userData, array $customData = null): User |
||
| 101 | { |
||
| 102 | $activationToken = $this->generateToken(); |
||
| 103 | |||
| 104 | $userData[$this->keyFields[AuthKeys::PASSWORD]] = $this->hasher->hash($userData[$this->keyFields[AuthKeys::PASSWORD]]); |
||
| 105 | $userData[$this->keyFields[AuthKeys::ACTIVATION_TOKEN]] = $activationToken; |
||
| 106 | |||
| 107 | $user = $this->authService->add($userData); |
||
| 108 | |||
| 109 | $body = [ |
||
| 110 | 'user' => $user, |
||
| 111 | 'activationToken' => $activationToken |
||
| 112 | ]; |
||
| 113 | |||
| 114 | if ($customData) { |
||
| 115 | $body = array_merge($body, $customData); |
||
| 116 | } |
||
| 117 | |||
| 118 | $this->mailer->setSubject(t('common.activate_account')); |
||
| 119 | $this->mailer->setTemplate(base_dir() . DS . 'shared' . DS . 'views' . DS . 'email' . DS . 'activate'); |
||
| 120 | |||
| 121 | $this->sendMail($user, $body); |
||
| 122 | |||
| 123 | return $user; |
||
| 124 | } |
||
| 125 | |||
| 126 | /** |
||
| 127 | * Activate |
||
| 128 | * @param string $token |
||
| 129 | */ |
||
| 130 | public function activate(string $token) |
||
| 131 | { |
||
| 132 | $this->authService->update( |
||
| 133 | $this->keyFields[AuthKeys::ACTIVATION_TOKEN], |
||
| 134 | $token, |
||
| 135 | [$this->keyFields[AuthKeys::ACTIVATION_TOKEN] => ''] |
||
| 136 | ); |
||
| 137 | } |
||
| 138 | |||
| 139 | /** |
||
| 140 | * Forget |
||
| 141 | * @param string $username |
||
| 142 | * @return string|null |
||
| 143 | */ |
||
| 144 | public function forget(string $username): ?string |
||
| 145 | { |
||
| 146 | $user = $this->authService->get($this->keyFields[AuthKeys::USERNAME], $username); |
||
| 147 | |||
| 148 | if (!$user) { |
||
| 149 | return null; |
||
| 150 | } |
||
| 151 | |||
| 152 | $resetToken = $this->generateToken(); |
||
| 153 | |||
| 154 | $this->authService->update( |
||
| 155 | $this->keyFields[AuthKeys::USERNAME], |
||
| 156 | $username, |
||
| 157 | [$this->keyFields[AuthKeys::RESET_TOKEN] => $resetToken] |
||
| 158 | ); |
||
| 159 | |||
| 160 | $body = [ |
||
| 161 | 'user' => $user, |
||
| 162 | 'resetToken' => $resetToken |
||
| 163 | ]; |
||
| 164 | |||
| 165 | $this->mailer->setSubject(t('common.reset_password')); |
||
| 166 | $this->mailer->setTemplate(base_dir() . DS . 'shared' . DS . 'views' . DS . 'email' . DS . 'reset'); |
||
| 167 | |||
| 168 | $this->sendMail($user, $body); |
||
| 169 | |||
| 170 | return $resetToken; |
||
| 171 | } |
||
| 172 | |||
| 173 | /** |
||
| 174 | * Reset |
||
| 175 | * @param string $token |
||
| 176 | * @param string $password |
||
| 177 | */ |
||
| 178 | public function reset(string $token, string $password) |
||
| 179 | { |
||
| 180 | $user = $this->authService->get($this->keyFields[AuthKeys::RESET_TOKEN], $token); |
||
| 181 | |||
| 182 | if ($user) { |
||
| 183 | if (!$this->isActivated($user)) { |
||
| 184 | $this->activate($token); |
||
| 185 | } |
||
| 186 | |||
| 187 | $this->authService->update( |
||
| 188 | $this->keyFields[AuthKeys::RESET_TOKEN], |
||
| 189 | $token, |
||
| 190 | [ |
||
| 191 | $this->keyFields[AuthKeys::PASSWORD] => $this->hasher->hash($password), |
||
| 192 | $this->keyFields[AuthKeys::RESET_TOKEN] => '' |
||
| 193 | ] |
||
| 194 | ); |
||
| 195 | } |
||
| 196 | } |
||
| 197 | |||
| 198 | /** |
||
| 199 | * Resend OTP |
||
| 200 | * @param string $otpToken |
||
| 201 | * @return string |
||
| 202 | * @throws AuthException |
||
| 203 | * @throws Exception |
||
| 204 | */ |
||
| 205 | public function resendOtp(string $otpToken): string |
||
| 206 | { |
||
| 207 | $user = $this->authService->get($this->keyFields[AuthKeys::OTP_TOKEN], $otpToken); |
||
| 208 | |||
| 209 | if (!$user) { |
||
| 210 | throw AuthException::incorrectCredentials(); |
||
| 211 | } |
||
| 212 | |||
| 213 | return $this->twoStepVerification($user); |
||
| 214 | |||
| 215 | } |
||
| 216 | |||
| 217 | /** |
||
| 218 | * Gets the user by username and password |
||
| 219 | * @param string $username |
||
| 220 | * @param string $password |
||
| 221 | * @return User |
||
| 222 | * @throws AuthException |
||
| 223 | */ |
||
| 224 | protected function getUser(string $username, string $password): User |
||
| 225 | { |
||
| 226 | $user = $this->authService->get($this->keyFields[AuthKeys::USERNAME], $username); |
||
| 227 | |||
| 228 | if (!$user) { |
||
| 229 | throw AuthException::incorrectCredentials(); |
||
| 230 | } |
||
| 231 | |||
| 232 | if (!$this->hasher->check($password, $user->getFieldValue($this->keyFields[AuthKeys::PASSWORD]))) { |
||
| 233 | throw AuthException::incorrectCredentials(); |
||
| 234 | } |
||
| 235 | |||
| 236 | if (!$this->isActivated($user)) { |
||
| 237 | throw AuthException::inactiveAccount(); |
||
| 238 | } |
||
| 239 | |||
| 240 | return $user; |
||
| 241 | } |
||
| 242 | |||
| 243 | /** |
||
| 244 | * Two-Step Verification |
||
| 245 | * @param User $user |
||
| 246 | * @return string |
||
| 247 | * @throws Exception |
||
| 248 | */ |
||
| 249 | protected function twoStepVerification(User $user): string |
||
| 280 | } |
||
| 281 | |||
| 282 | /** |
||
| 283 | * Verify and update OTP |
||
| 284 | * @param int $otp |
||
| 285 | * @param string $otpToken |
||
| 286 | * @return User |
||
| 287 | * @throws AuthException |
||
| 288 | * @throws Exception |
||
| 289 | */ |
||
| 290 | protected function verifyAndUpdateOtp(int $otp, string $otpToken): User |
||
| 291 | { |
||
| 292 | $user = $this->authService->get($this->keyFields[AuthKeys::OTP_TOKEN], $otpToken); |
||
| 293 | |||
| 294 | if (!$user || $otp != $user->getFieldValue($this->keyFields[AuthKeys::OTP])) { |
||
| 295 | throw AuthException::incorrectVerificationCode(); |
||
| 296 | } |
||
| 297 | |||
| 298 | if (new DateTime() >= new DateTime($user->getFieldValue($this->keyFields[AuthKeys::OTP_EXPIRY]))) { |
||
| 299 | throw AuthException::verificationCodeExpired(); |
||
| 300 | } |
||
| 301 | |||
| 302 | $this->authService->update( |
||
| 303 | $this->keyFields[AuthKeys::USERNAME], |
||
| 304 | $user->getFieldValue($this->keyFields[AuthKeys::USERNAME]), |
||
| 305 | [ |
||
| 306 | $this->keyFields[AuthKeys::OTP] => null, |
||
| 307 | $this->keyFields[AuthKeys::OTP_EXPIRY] => null, |
||
| 308 | $this->keyFields[AuthKeys::OTP_TOKEN] => null, |
||
| 309 | ] |
||
| 310 | ); |
||
| 311 | |||
| 312 | return $user; |
||
| 313 | } |
||
| 314 | |||
| 315 | /** |
||
| 316 | * Filters and gets the visible fields |
||
| 317 | * @param User $user |
||
| 318 | * @return array |
||
| 319 | */ |
||
| 320 | protected function getVisibleFields(User $user): array |
||
| 321 | { |
||
| 322 | $userData = $user->getData(); |
||
| 323 | |||
| 324 | if (count($this->visibleFields)) { |
||
| 325 | foreach ($userData as $field => $value) { |
||
| 326 | if (!in_array($field, $this->visibleFields)) { |
||
| 327 | unset($userData[$field]); |
||
| 328 | } |
||
| 329 | } |
||
| 330 | } |
||
| 331 | |||
| 332 | return $userData; |
||
| 333 | } |
||
| 334 | |||
| 335 | /** |
||
| 336 | * Is user account activated |
||
| 337 | * @param User $user |
||
| 338 | * @return bool |
||
| 339 | */ |
||
| 340 | protected function isActivated(User $user): bool |
||
| 341 | { |
||
| 342 | return empty($user->getFieldValue($this->keyFields[AuthKeys::ACTIVATION_TOKEN])); |
||
| 343 | } |
||
| 344 | |||
| 345 | /** |
||
| 346 | * Generate Token |
||
| 347 | * @param string|null $val |
||
| 348 | * @return string |
||
| 349 | */ |
||
| 350 | protected function generateToken(string $val = null): string |
||
| 351 | { |
||
| 352 | return base64_encode($this->hasher->hash($val ?: config()->get('app_key'))); |
||
| 353 | } |
||
| 354 | |||
| 355 | /** |
||
| 356 | * Send email |
||
| 357 | * @param User $user |
||
| 358 | * @param array $body |
||
| 359 | */ |
||
| 360 | protected function sendMail(User $user, array $body) |
||
| 361 | { |
||
| 362 | $fullName = ($user->hasField('firstname') && $user->hasField('lastname')) ? $user->getFieldValue('firstname') . ' ' . $user->getFieldValue('lastname') : ''; |
||
| 363 | |||
| 364 | $appEmail = config()->get('app_email') ?: ''; |
||
| 365 | $appName = config()->get('app_name') ?: ''; |
||
| 366 | |||
| 367 | $this->mailer->setFrom($appEmail, $appName) |
||
| 368 | ->setAddress($user->getFieldValue($this->keyFields[AuthKeys::USERNAME]), $fullName) |
||
| 369 | ->setBody($body) |
||
| 370 | ->send(); |
||
| 371 | } |
||
| 372 | |||
| 373 | /** |
||
| 374 | * Verify user schema |
||
| 375 | * @param array $schema |
||
| 376 | * @throws AuthException |
||
| 377 | */ |
||
| 378 | protected function verifySchema(array $schema) |
||
| 379 | { |
||
| 380 | $constants = (new ReflectionClass(AuthKeys::class))->getConstants(); |
||
| 381 | |||
| 382 | foreach ($constants as $constant) { |
||
| 383 | if (!isset($schema[$constant]) || !isset($schema[$constant]['name'])) { |
||
| 384 | throw AuthException::incorrectUserSchema(); |
||
| 385 | } |
||
| 386 | |||
| 387 | $this->keyFields[$constant] = $schema[$constant]['name']; |
||
| 388 | } |
||
| 389 | |||
| 390 | foreach ($schema as $field) { |
||
| 391 | if (isset($field['visible']) && $field['visible']) { |
||
| 392 | $this->visibleFields[] = $field['name']; |
||
| 393 | } |
||
| 394 | } |
||
| 395 | } |
||
| 396 | |||
| 397 | /** |
||
| 398 | * @return bool |
||
| 399 | */ |
||
| 400 | protected function isTwoFactorEnabled(): bool |
||
| 403 | } |
||
| 404 | } |