State.php 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. <?php
  2. declare(strict_types=1);
  3. /**
  4. * @copyright 2018 Christoph Wurst <christoph@winzerhof-wurst.at>
  5. *
  6. * @author Christoph Wurst <christoph@winzerhof-wurst.at>
  7. * @author Joas Schilling <coding@schilljs.com>
  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 OC\Core\Command\TwoFactorAuth;
  26. use OCP\Authentication\TwoFactorAuth\IRegistry;
  27. use OCP\IUserManager;
  28. use Symfony\Component\Console\Input\InputArgument;
  29. use Symfony\Component\Console\Input\InputInterface;
  30. use Symfony\Component\Console\Output\OutputInterface;
  31. class State extends Base {
  32. private IRegistry $registry;
  33. public function __construct(IRegistry $registry, IUserManager $userManager) {
  34. parent::__construct('twofactorauth:state');
  35. $this->registry = $registry;
  36. $this->userManager = $userManager;
  37. }
  38. protected function configure() {
  39. parent::configure();
  40. $this->setName('twofactorauth:state');
  41. $this->setDescription('Get the two-factor authentication (2FA) state of a user');
  42. $this->addArgument('uid', InputArgument::REQUIRED);
  43. }
  44. protected function execute(InputInterface $input, OutputInterface $output): int {
  45. $uid = $input->getArgument('uid');
  46. $user = $this->userManager->get($uid);
  47. if (is_null($user)) {
  48. $output->writeln("<error>Invalid UID</error>");
  49. return 1;
  50. }
  51. $providerStates = $this->registry->getProviderStates($user);
  52. $filtered = $this->filterEnabledDisabledUnknownProviders($providerStates);
  53. [$enabled, $disabled] = $filtered;
  54. if (!empty($enabled)) {
  55. $output->writeln("Two-factor authentication is enabled for user $uid");
  56. } else {
  57. $output->writeln("Two-factor authentication is not enabled for user $uid");
  58. }
  59. $output->writeln("");
  60. $this->printProviders("Enabled providers", $enabled, $output);
  61. $this->printProviders("Disabled providers", $disabled, $output);
  62. return 0;
  63. }
  64. private function filterEnabledDisabledUnknownProviders(array $providerStates): array {
  65. $enabled = [];
  66. $disabled = [];
  67. foreach ($providerStates as $providerId => $isEnabled) {
  68. if ($isEnabled) {
  69. $enabled[] = $providerId;
  70. } else {
  71. $disabled[] = $providerId;
  72. }
  73. }
  74. return [$enabled, $disabled];
  75. }
  76. private function printProviders(string $title, array $providers,
  77. OutputInterface $output) {
  78. if (empty($providers)) {
  79. // Ignore and don't print anything
  80. return;
  81. }
  82. $output->writeln($title . ":");
  83. foreach ($providers as $provider) {
  84. $output->writeln("- " . $provider);
  85. }
  86. }
  87. }