CheckBackupCodes.php 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. <?php
  2. declare(strict_types=1);
  3. /**
  4. * @copyright Copyright (c) 2018, Roeland Jago Douma <roeland@famdouma.nl>
  5. *
  6. * @author Christoph Wurst <christoph@winzerhof-wurst.at>
  7. * @author Roeland Jago Douma <roeland@famdouma.nl>
  8. *
  9. * @license GNU AGPL version 3 or any later version
  10. *
  11. * This program is free software: you can redistribute it and/or modify
  12. * it under the terms of the GNU Affero General Public License as
  13. * published by the Free Software Foundation, either version 3 of the
  14. * License, or (at your option) any later version.
  15. *
  16. * This program is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  19. * GNU Affero General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU Affero General Public License
  22. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  23. *
  24. */
  25. namespace OCA\TwoFactorBackupCodes\BackgroundJob;
  26. use OC\Authentication\TwoFactorAuth\Manager;
  27. use OCP\AppFramework\Utility\ITimeFactory;
  28. use OCP\Authentication\TwoFactorAuth\IRegistry;
  29. use OCP\BackgroundJob\IJobList;
  30. use OCP\BackgroundJob\QueuedJob;
  31. use OCP\IUser;
  32. use OCP\IUserManager;
  33. class CheckBackupCodes extends QueuedJob {
  34. /** @var IUserManager */
  35. private $userManager;
  36. /** @var IJobList */
  37. private $jobList;
  38. /** @var IRegistry */
  39. private $registry;
  40. /** @var Manager */
  41. private $twofactorManager;
  42. public function __construct(ITimeFactory $timeFactory, IUserManager $userManager, IJobList $jobList, Manager $twofactorManager, IRegistry $registry) {
  43. parent::__construct($timeFactory);
  44. $this->userManager = $userManager;
  45. $this->jobList = $jobList;
  46. $this->twofactorManager = $twofactorManager;
  47. $this->registry = $registry;
  48. }
  49. protected function run($argument) {
  50. $this->userManager->callForSeenUsers(function (IUser $user) {
  51. if (!$user->isEnabled()) {
  52. return;
  53. }
  54. $providers = $this->registry->getProviderStates($user);
  55. $isTwoFactorAuthenticated = $this->twofactorManager->isTwoFactorAuthenticated($user);
  56. if ($isTwoFactorAuthenticated && isset($providers['backup_codes']) && $providers['backup_codes'] === false) {
  57. $this->jobList->add(RememberBackupCodesJob::class, ['uid' => $user->getUID()]);
  58. }
  59. });
  60. }
  61. }