@@ -29,42 +29,42 @@ |
||
| 29 | 29 | |
| 30 | 30 | class BackupCodeMapper extends QBMapper { |
| 31 | 31 | |
| 32 | - public function __construct(IDBConnection $db) { |
|
| 33 | - parent::__construct($db, 'twofactor_backupcodes'); |
|
| 34 | - } |
|
| 32 | + public function __construct(IDBConnection $db) { |
|
| 33 | + parent::__construct($db, 'twofactor_backupcodes'); |
|
| 34 | + } |
|
| 35 | 35 | |
| 36 | - /** |
|
| 37 | - * @param IUser $user |
|
| 38 | - * @return BackupCode[] |
|
| 39 | - */ |
|
| 40 | - public function getBackupCodes(IUser $user) { |
|
| 41 | - /* @var IQueryBuilder $qb */ |
|
| 42 | - $qb = $this->db->getQueryBuilder(); |
|
| 36 | + /** |
|
| 37 | + * @param IUser $user |
|
| 38 | + * @return BackupCode[] |
|
| 39 | + */ |
|
| 40 | + public function getBackupCodes(IUser $user) { |
|
| 41 | + /* @var IQueryBuilder $qb */ |
|
| 42 | + $qb = $this->db->getQueryBuilder(); |
|
| 43 | 43 | |
| 44 | - $qb->select('id', 'user_id', 'code', 'used') |
|
| 45 | - ->from('twofactor_backupcodes') |
|
| 46 | - ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($user->getUID()))); |
|
| 44 | + $qb->select('id', 'user_id', 'code', 'used') |
|
| 45 | + ->from('twofactor_backupcodes') |
|
| 46 | + ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($user->getUID()))); |
|
| 47 | 47 | |
| 48 | - return self::findEntities($qb); |
|
| 49 | - } |
|
| 48 | + return self::findEntities($qb); |
|
| 49 | + } |
|
| 50 | 50 | |
| 51 | - /** |
|
| 52 | - * @param IUser $user |
|
| 53 | - */ |
|
| 54 | - public function deleteCodes(IUser $user) { |
|
| 55 | - $this->deleteCodesByUserId($user->getUID()); |
|
| 56 | - } |
|
| 51 | + /** |
|
| 52 | + * @param IUser $user |
|
| 53 | + */ |
|
| 54 | + public function deleteCodes(IUser $user) { |
|
| 55 | + $this->deleteCodesByUserId($user->getUID()); |
|
| 56 | + } |
|
| 57 | 57 | |
| 58 | - /** |
|
| 59 | - * @param string $uid |
|
| 60 | - */ |
|
| 61 | - public function deleteCodesByUserId($uid) { |
|
| 62 | - /* @var IQueryBuilder $qb */ |
|
| 63 | - $qb = $this->db->getQueryBuilder(); |
|
| 58 | + /** |
|
| 59 | + * @param string $uid |
|
| 60 | + */ |
|
| 61 | + public function deleteCodesByUserId($uid) { |
|
| 62 | + /* @var IQueryBuilder $qb */ |
|
| 63 | + $qb = $this->db->getQueryBuilder(); |
|
| 64 | 64 | |
| 65 | - $qb->delete('twofactor_backupcodes') |
|
| 66 | - ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($uid))); |
|
| 67 | - $qb->execute(); |
|
| 68 | - } |
|
| 65 | + $qb->delete('twofactor_backupcodes') |
|
| 66 | + ->where($qb->expr()->eq('user_id', $qb->createNamedParameter($uid))); |
|
| 67 | + $qb->execute(); |
|
| 68 | + } |
|
| 69 | 69 | |
| 70 | 70 | } |
@@ -33,58 +33,58 @@ |
||
| 33 | 33 | |
| 34 | 34 | class RememberBackupCodesJob extends TimedJob { |
| 35 | 35 | |
| 36 | - /** @var IRegistry */ |
|
| 37 | - private $registry; |
|
| 36 | + /** @var IRegistry */ |
|
| 37 | + private $registry; |
|
| 38 | 38 | |
| 39 | - /** @var IUserManager */ |
|
| 40 | - private $userManager; |
|
| 39 | + /** @var IUserManager */ |
|
| 40 | + private $userManager; |
|
| 41 | 41 | |
| 42 | - /** @var IManager */ |
|
| 43 | - private $notificationManager; |
|
| 42 | + /** @var IManager */ |
|
| 43 | + private $notificationManager; |
|
| 44 | 44 | |
| 45 | - /** @var IJobList */ |
|
| 46 | - private $jobList; |
|
| 45 | + /** @var IJobList */ |
|
| 46 | + private $jobList; |
|
| 47 | 47 | |
| 48 | - public function __construct(IRegistry $registry, |
|
| 49 | - IUserManager $userManager, |
|
| 50 | - ITimeFactory $timeFactory, |
|
| 51 | - IManager $notificationManager, |
|
| 52 | - IJobList $jobList) { |
|
| 53 | - parent::__construct($timeFactory); |
|
| 54 | - $this->registry = $registry; |
|
| 55 | - $this->userManager = $userManager; |
|
| 56 | - $this->time = $timeFactory; |
|
| 57 | - $this->notificationManager = $notificationManager; |
|
| 58 | - $this->jobList = $jobList; |
|
| 48 | + public function __construct(IRegistry $registry, |
|
| 49 | + IUserManager $userManager, |
|
| 50 | + ITimeFactory $timeFactory, |
|
| 51 | + IManager $notificationManager, |
|
| 52 | + IJobList $jobList) { |
|
| 53 | + parent::__construct($timeFactory); |
|
| 54 | + $this->registry = $registry; |
|
| 55 | + $this->userManager = $userManager; |
|
| 56 | + $this->time = $timeFactory; |
|
| 57 | + $this->notificationManager = $notificationManager; |
|
| 58 | + $this->jobList = $jobList; |
|
| 59 | 59 | |
| 60 | - $this->setInterval(60*60*24*14); |
|
| 61 | - } |
|
| 60 | + $this->setInterval(60*60*24*14); |
|
| 61 | + } |
|
| 62 | 62 | |
| 63 | - protected function run($argument) { |
|
| 64 | - $uid = $argument['uid']; |
|
| 65 | - $user = $this->userManager->get($uid); |
|
| 63 | + protected function run($argument) { |
|
| 64 | + $uid = $argument['uid']; |
|
| 65 | + $user = $this->userManager->get($uid); |
|
| 66 | 66 | |
| 67 | - if ($user === null) { |
|
| 68 | - // We can't run with an invalid user |
|
| 69 | - return; |
|
| 70 | - } |
|
| 67 | + if ($user === null) { |
|
| 68 | + // We can't run with an invalid user |
|
| 69 | + return; |
|
| 70 | + } |
|
| 71 | 71 | |
| 72 | - $providers = $this->registry->getProviderStates($user); |
|
| 73 | - if (isset($providers['backup_codes']) && $providers['backup_codes'] === true) { |
|
| 74 | - // Backup codes already generated lets remove this job |
|
| 75 | - $this->jobList->remove(self::class, $argument); |
|
| 76 | - return; |
|
| 77 | - } |
|
| 72 | + $providers = $this->registry->getProviderStates($user); |
|
| 73 | + if (isset($providers['backup_codes']) && $providers['backup_codes'] === true) { |
|
| 74 | + // Backup codes already generated lets remove this job |
|
| 75 | + $this->jobList->remove(self::class, $argument); |
|
| 76 | + return; |
|
| 77 | + } |
|
| 78 | 78 | |
| 79 | - $date = new \DateTime(); |
|
| 80 | - $date->setTimestamp($this->time->getTime()); |
|
| 79 | + $date = new \DateTime(); |
|
| 80 | + $date->setTimestamp($this->time->getTime()); |
|
| 81 | 81 | |
| 82 | - $notification = $this->notificationManager->createNotification(); |
|
| 83 | - $notification->setApp('twofactor_backupcodes') |
|
| 84 | - ->setUser($user->getUID()) |
|
| 85 | - ->setDateTime($date) |
|
| 86 | - ->setObject('create', 'codes') |
|
| 87 | - ->setSubject('create_backupcodes'); |
|
| 88 | - $this->notificationManager->notify($notification); |
|
| 89 | - } |
|
| 82 | + $notification = $this->notificationManager->createNotification(); |
|
| 83 | + $notification->setApp('twofactor_backupcodes') |
|
| 84 | + ->setUser($user->getUID()) |
|
| 85 | + ->setDateTime($date) |
|
| 86 | + ->setObject('create', 'codes') |
|
| 87 | + ->setSubject('create_backupcodes'); |
|
| 88 | + $this->notificationManager->notify($notification); |
|
| 89 | + } |
|
| 90 | 90 | } |
@@ -34,35 +34,35 @@ |
||
| 34 | 34 | |
| 35 | 35 | class CheckBackupCodes extends QueuedJob { |
| 36 | 36 | |
| 37 | - /** @var IUserManager */ |
|
| 38 | - private $userManager; |
|
| 37 | + /** @var IUserManager */ |
|
| 38 | + private $userManager; |
|
| 39 | 39 | |
| 40 | - /** @var IJobList */ |
|
| 41 | - private $jobList; |
|
| 40 | + /** @var IJobList */ |
|
| 41 | + private $jobList; |
|
| 42 | 42 | |
| 43 | - /** @var Manager */ |
|
| 44 | - private $registry; |
|
| 43 | + /** @var Manager */ |
|
| 44 | + private $registry; |
|
| 45 | 45 | |
| 46 | - /** @var Manager */ |
|
| 47 | - private $twofactorManager; |
|
| 46 | + /** @var Manager */ |
|
| 47 | + private $twofactorManager; |
|
| 48 | 48 | |
| 49 | - public function __construct(ITimeFactory $timeFactory, IUserManager $userManager, IJobList $jobList, Manager $twofactorManager, IRegistry $registry) { |
|
| 50 | - parent::__construct($timeFactory); |
|
| 51 | - $this->userManager = $userManager; |
|
| 52 | - $this->jobList = $jobList; |
|
| 53 | - $this->twofactorManager = $twofactorManager; |
|
| 54 | - $this->registry = $registry; |
|
| 55 | - } |
|
| 49 | + public function __construct(ITimeFactory $timeFactory, IUserManager $userManager, IJobList $jobList, Manager $twofactorManager, IRegistry $registry) { |
|
| 50 | + parent::__construct($timeFactory); |
|
| 51 | + $this->userManager = $userManager; |
|
| 52 | + $this->jobList = $jobList; |
|
| 53 | + $this->twofactorManager = $twofactorManager; |
|
| 54 | + $this->registry = $registry; |
|
| 55 | + } |
|
| 56 | 56 | |
| 57 | - protected function run($argument) { |
|
| 58 | - $this->userManager->callForSeenUsers(function(IUser $user) { |
|
| 59 | - $providers = $this->registry->getProviderStates($user); |
|
| 60 | - $isTwoFactorAuthenticated = $this->twofactorManager->isTwoFactorAuthenticated($user); |
|
| 57 | + protected function run($argument) { |
|
| 58 | + $this->userManager->callForSeenUsers(function(IUser $user) { |
|
| 59 | + $providers = $this->registry->getProviderStates($user); |
|
| 60 | + $isTwoFactorAuthenticated = $this->twofactorManager->isTwoFactorAuthenticated($user); |
|
| 61 | 61 | |
| 62 | - if ($isTwoFactorAuthenticated && isset($providers['backup_codes']) && $providers['backup_codes'] === false) { |
|
| 63 | - $this->jobList->add(RememberBackupCodesJob::class, ['uid' => $user->getUID()]); |
|
| 64 | - } |
|
| 65 | - }); |
|
| 66 | - } |
|
| 62 | + if ($isTwoFactorAuthenticated && isset($providers['backup_codes']) && $providers['backup_codes'] === false) { |
|
| 63 | + $this->jobList->add(RememberBackupCodesJob::class, ['uid' => $user->getUID()]); |
|
| 64 | + } |
|
| 65 | + }); |
|
| 66 | + } |
|
| 67 | 67 | |
| 68 | 68 | } |