@@ -37,44 +37,44 @@ |
||
| 37 | 37 | |
| 38 | 38 | class Registry implements IRegistry { |
| 39 | 39 | |
| 40 | - /** @var ProviderUserAssignmentDao */ |
|
| 41 | - private $assignmentDao; |
|
| 40 | + /** @var ProviderUserAssignmentDao */ |
|
| 41 | + private $assignmentDao; |
|
| 42 | 42 | |
| 43 | - /** @var IEventDispatcher */ |
|
| 44 | - private $dispatcher; |
|
| 43 | + /** @var IEventDispatcher */ |
|
| 44 | + private $dispatcher; |
|
| 45 | 45 | |
| 46 | - public function __construct(ProviderUserAssignmentDao $assignmentDao, |
|
| 47 | - IEventDispatcher $dispatcher) { |
|
| 48 | - $this->assignmentDao = $assignmentDao; |
|
| 49 | - $this->dispatcher = $dispatcher; |
|
| 50 | - } |
|
| 46 | + public function __construct(ProviderUserAssignmentDao $assignmentDao, |
|
| 47 | + IEventDispatcher $dispatcher) { |
|
| 48 | + $this->assignmentDao = $assignmentDao; |
|
| 49 | + $this->dispatcher = $dispatcher; |
|
| 50 | + } |
|
| 51 | 51 | |
| 52 | - public function getProviderStates(IUser $user): array { |
|
| 53 | - return $this->assignmentDao->getState($user->getUID()); |
|
| 54 | - } |
|
| 52 | + public function getProviderStates(IUser $user): array { |
|
| 53 | + return $this->assignmentDao->getState($user->getUID()); |
|
| 54 | + } |
|
| 55 | 55 | |
| 56 | - public function enableProviderFor(IProvider $provider, IUser $user) { |
|
| 57 | - $this->assignmentDao->persist($provider->getId(), $user->getUID(), 1); |
|
| 56 | + public function enableProviderFor(IProvider $provider, IUser $user) { |
|
| 57 | + $this->assignmentDao->persist($provider->getId(), $user->getUID(), 1); |
|
| 58 | 58 | |
| 59 | - $event = new RegistryEvent($provider, $user); |
|
| 60 | - $this->dispatcher->dispatch(self::EVENT_PROVIDER_ENABLED, $event); |
|
| 61 | - } |
|
| 59 | + $event = new RegistryEvent($provider, $user); |
|
| 60 | + $this->dispatcher->dispatch(self::EVENT_PROVIDER_ENABLED, $event); |
|
| 61 | + } |
|
| 62 | 62 | |
| 63 | - public function disableProviderFor(IProvider $provider, IUser $user) { |
|
| 64 | - $this->assignmentDao->persist($provider->getId(), $user->getUID(), 0); |
|
| 63 | + public function disableProviderFor(IProvider $provider, IUser $user) { |
|
| 64 | + $this->assignmentDao->persist($provider->getId(), $user->getUID(), 0); |
|
| 65 | 65 | |
| 66 | - $event = new RegistryEvent($provider, $user); |
|
| 67 | - $this->dispatcher->dispatch(self::EVENT_PROVIDER_DISABLED, $event); |
|
| 68 | - } |
|
| 66 | + $event = new RegistryEvent($provider, $user); |
|
| 67 | + $this->dispatcher->dispatch(self::EVENT_PROVIDER_DISABLED, $event); |
|
| 68 | + } |
|
| 69 | 69 | |
| 70 | - public function deleteUserData(IUser $user): void { |
|
| 71 | - foreach ($this->assignmentDao->deleteByUser($user->getUID()) as $provider) { |
|
| 72 | - $event = new TwoFactorProviderDisabled($provider['provider_id']); |
|
| 73 | - $this->dispatcher->dispatchTyped($event); |
|
| 74 | - } |
|
| 75 | - } |
|
| 70 | + public function deleteUserData(IUser $user): void { |
|
| 71 | + foreach ($this->assignmentDao->deleteByUser($user->getUID()) as $provider) { |
|
| 72 | + $event = new TwoFactorProviderDisabled($provider['provider_id']); |
|
| 73 | + $this->dispatcher->dispatchTyped($event); |
|
| 74 | + } |
|
| 75 | + } |
|
| 76 | 76 | |
| 77 | - public function cleanUp(string $providerId) { |
|
| 78 | - $this->assignmentDao->deleteAll($providerId); |
|
| 79 | - } |
|
| 77 | + public function cleanUp(string $providerId) { |
|
| 78 | + $this->assignmentDao->deleteAll($providerId); |
|
| 79 | + } |
|
| 80 | 80 | } |
@@ -36,99 +36,99 @@ |
||
| 36 | 36 | * 2FA providers |
| 37 | 37 | */ |
| 38 | 38 | class ProviderUserAssignmentDao { |
| 39 | - public const TABLE_NAME = 'twofactor_providers'; |
|
| 40 | - |
|
| 41 | - /** @var IDBConnection */ |
|
| 42 | - private $conn; |
|
| 43 | - |
|
| 44 | - public function __construct(IDBConnection $dbConn) { |
|
| 45 | - $this->conn = $dbConn; |
|
| 46 | - } |
|
| 47 | - |
|
| 48 | - /** |
|
| 49 | - * Get all assigned provider IDs for the given user ID |
|
| 50 | - * |
|
| 51 | - * @return string[] where the array key is the provider ID (string) and the |
|
| 52 | - * value is the enabled state (bool) |
|
| 53 | - */ |
|
| 54 | - public function getState(string $uid): array { |
|
| 55 | - $qb = $this->conn->getQueryBuilder(); |
|
| 56 | - |
|
| 57 | - $query = $qb->select('provider_id', 'enabled') |
|
| 58 | - ->from(self::TABLE_NAME) |
|
| 59 | - ->where($qb->expr()->eq('uid', $qb->createNamedParameter($uid))); |
|
| 60 | - $result = $query->execute(); |
|
| 61 | - $providers = []; |
|
| 62 | - foreach ($result->fetchAll() as $row) { |
|
| 63 | - $providers[$row['provider_id']] = 1 === (int)$row['enabled']; |
|
| 64 | - } |
|
| 65 | - $result->closeCursor(); |
|
| 66 | - |
|
| 67 | - return $providers; |
|
| 68 | - } |
|
| 69 | - |
|
| 70 | - /** |
|
| 71 | - * Persist a new/updated (provider_id, uid, enabled) tuple |
|
| 72 | - */ |
|
| 73 | - public function persist(string $providerId, string $uid, int $enabled) { |
|
| 74 | - $qb = $this->conn->getQueryBuilder(); |
|
| 75 | - |
|
| 76 | - try { |
|
| 77 | - // Insert a new entry |
|
| 78 | - $insertQuery = $qb->insert(self::TABLE_NAME)->values([ |
|
| 79 | - 'provider_id' => $qb->createNamedParameter($providerId), |
|
| 80 | - 'uid' => $qb->createNamedParameter($uid), |
|
| 81 | - 'enabled' => $qb->createNamedParameter($enabled, IQueryBuilder::PARAM_INT), |
|
| 82 | - ]); |
|
| 83 | - |
|
| 84 | - $insertQuery->execute(); |
|
| 85 | - } catch (UniqueConstraintViolationException $ex) { |
|
| 86 | - // There is already an entry -> update it |
|
| 87 | - $updateQuery = $qb->update(self::TABLE_NAME) |
|
| 88 | - ->set('enabled', $qb->createNamedParameter($enabled)) |
|
| 89 | - ->where($qb->expr()->eq('provider_id', $qb->createNamedParameter($providerId))) |
|
| 90 | - ->andWhere($qb->expr()->eq('uid', $qb->createNamedParameter($uid))); |
|
| 91 | - $updateQuery->execute(); |
|
| 92 | - } |
|
| 93 | - } |
|
| 94 | - |
|
| 95 | - /** |
|
| 96 | - * Delete all provider states of a user and return the provider IDs |
|
| 97 | - * |
|
| 98 | - * @param string $uid |
|
| 99 | - * |
|
| 100 | - * @return int[] |
|
| 101 | - */ |
|
| 102 | - public function deleteByUser(string $uid): array { |
|
| 103 | - $qb1 = $this->conn->getQueryBuilder(); |
|
| 104 | - $selectQuery = $qb1->select('*') |
|
| 105 | - ->from(self::TABLE_NAME) |
|
| 106 | - ->where($qb1->expr()->eq('uid', $qb1->createNamedParameter($uid))); |
|
| 107 | - $selectResult = $selectQuery->execute(); |
|
| 108 | - $rows = $selectResult->fetchAll(); |
|
| 109 | - $selectResult->closeCursor(); |
|
| 110 | - |
|
| 111 | - $qb2 = $this->conn->getQueryBuilder(); |
|
| 112 | - $deleteQuery = $qb2 |
|
| 113 | - ->delete(self::TABLE_NAME) |
|
| 114 | - ->where($qb2->expr()->eq('uid', $qb2->createNamedParameter($uid))); |
|
| 115 | - $deleteQuery->execute(); |
|
| 116 | - |
|
| 117 | - return array_map(function (array $row) { |
|
| 118 | - return [ |
|
| 119 | - 'provider_id' => $row['provider_id'], |
|
| 120 | - 'uid' => $row['uid'], |
|
| 121 | - 'enabled' => 1 === (int) $row['enabled'], |
|
| 122 | - ]; |
|
| 123 | - }, $rows); |
|
| 124 | - } |
|
| 125 | - |
|
| 126 | - public function deleteAll(string $providerId) { |
|
| 127 | - $qb = $this->conn->getQueryBuilder(); |
|
| 128 | - |
|
| 129 | - $deleteQuery = $qb->delete(self::TABLE_NAME) |
|
| 130 | - ->where($qb->expr()->eq('provider_id', $qb->createNamedParameter($providerId))); |
|
| 131 | - |
|
| 132 | - $deleteQuery->execute(); |
|
| 133 | - } |
|
| 39 | + public const TABLE_NAME = 'twofactor_providers'; |
|
| 40 | + |
|
| 41 | + /** @var IDBConnection */ |
|
| 42 | + private $conn; |
|
| 43 | + |
|
| 44 | + public function __construct(IDBConnection $dbConn) { |
|
| 45 | + $this->conn = $dbConn; |
|
| 46 | + } |
|
| 47 | + |
|
| 48 | + /** |
|
| 49 | + * Get all assigned provider IDs for the given user ID |
|
| 50 | + * |
|
| 51 | + * @return string[] where the array key is the provider ID (string) and the |
|
| 52 | + * value is the enabled state (bool) |
|
| 53 | + */ |
|
| 54 | + public function getState(string $uid): array { |
|
| 55 | + $qb = $this->conn->getQueryBuilder(); |
|
| 56 | + |
|
| 57 | + $query = $qb->select('provider_id', 'enabled') |
|
| 58 | + ->from(self::TABLE_NAME) |
|
| 59 | + ->where($qb->expr()->eq('uid', $qb->createNamedParameter($uid))); |
|
| 60 | + $result = $query->execute(); |
|
| 61 | + $providers = []; |
|
| 62 | + foreach ($result->fetchAll() as $row) { |
|
| 63 | + $providers[$row['provider_id']] = 1 === (int)$row['enabled']; |
|
| 64 | + } |
|
| 65 | + $result->closeCursor(); |
|
| 66 | + |
|
| 67 | + return $providers; |
|
| 68 | + } |
|
| 69 | + |
|
| 70 | + /** |
|
| 71 | + * Persist a new/updated (provider_id, uid, enabled) tuple |
|
| 72 | + */ |
|
| 73 | + public function persist(string $providerId, string $uid, int $enabled) { |
|
| 74 | + $qb = $this->conn->getQueryBuilder(); |
|
| 75 | + |
|
| 76 | + try { |
|
| 77 | + // Insert a new entry |
|
| 78 | + $insertQuery = $qb->insert(self::TABLE_NAME)->values([ |
|
| 79 | + 'provider_id' => $qb->createNamedParameter($providerId), |
|
| 80 | + 'uid' => $qb->createNamedParameter($uid), |
|
| 81 | + 'enabled' => $qb->createNamedParameter($enabled, IQueryBuilder::PARAM_INT), |
|
| 82 | + ]); |
|
| 83 | + |
|
| 84 | + $insertQuery->execute(); |
|
| 85 | + } catch (UniqueConstraintViolationException $ex) { |
|
| 86 | + // There is already an entry -> update it |
|
| 87 | + $updateQuery = $qb->update(self::TABLE_NAME) |
|
| 88 | + ->set('enabled', $qb->createNamedParameter($enabled)) |
|
| 89 | + ->where($qb->expr()->eq('provider_id', $qb->createNamedParameter($providerId))) |
|
| 90 | + ->andWhere($qb->expr()->eq('uid', $qb->createNamedParameter($uid))); |
|
| 91 | + $updateQuery->execute(); |
|
| 92 | + } |
|
| 93 | + } |
|
| 94 | + |
|
| 95 | + /** |
|
| 96 | + * Delete all provider states of a user and return the provider IDs |
|
| 97 | + * |
|
| 98 | + * @param string $uid |
|
| 99 | + * |
|
| 100 | + * @return int[] |
|
| 101 | + */ |
|
| 102 | + public function deleteByUser(string $uid): array { |
|
| 103 | + $qb1 = $this->conn->getQueryBuilder(); |
|
| 104 | + $selectQuery = $qb1->select('*') |
|
| 105 | + ->from(self::TABLE_NAME) |
|
| 106 | + ->where($qb1->expr()->eq('uid', $qb1->createNamedParameter($uid))); |
|
| 107 | + $selectResult = $selectQuery->execute(); |
|
| 108 | + $rows = $selectResult->fetchAll(); |
|
| 109 | + $selectResult->closeCursor(); |
|
| 110 | + |
|
| 111 | + $qb2 = $this->conn->getQueryBuilder(); |
|
| 112 | + $deleteQuery = $qb2 |
|
| 113 | + ->delete(self::TABLE_NAME) |
|
| 114 | + ->where($qb2->expr()->eq('uid', $qb2->createNamedParameter($uid))); |
|
| 115 | + $deleteQuery->execute(); |
|
| 116 | + |
|
| 117 | + return array_map(function (array $row) { |
|
| 118 | + return [ |
|
| 119 | + 'provider_id' => $row['provider_id'], |
|
| 120 | + 'uid' => $row['uid'], |
|
| 121 | + 'enabled' => 1 === (int) $row['enabled'], |
|
| 122 | + ]; |
|
| 123 | + }, $rows); |
|
| 124 | + } |
|
| 125 | + |
|
| 126 | + public function deleteAll(string $providerId) { |
|
| 127 | + $qb = $this->conn->getQueryBuilder(); |
|
| 128 | + |
|
| 129 | + $deleteQuery = $qb->delete(self::TABLE_NAME) |
|
| 130 | + ->where($qb->expr()->eq('provider_id', $qb->createNamedParameter($providerId))); |
|
| 131 | + |
|
| 132 | + $deleteQuery->execute(); |
|
| 133 | + } |
|
| 134 | 134 | } |
@@ -60,7 +60,7 @@ discard block |
||
| 60 | 60 | $result = $query->execute(); |
| 61 | 61 | $providers = []; |
| 62 | 62 | foreach ($result->fetchAll() as $row) { |
| 63 | - $providers[$row['provider_id']] = 1 === (int)$row['enabled']; |
|
| 63 | + $providers[$row['provider_id']] = 1 === (int) $row['enabled']; |
|
| 64 | 64 | } |
| 65 | 65 | $result->closeCursor(); |
| 66 | 66 | |
@@ -114,7 +114,7 @@ discard block |
||
| 114 | 114 | ->where($qb2->expr()->eq('uid', $qb2->createNamedParameter($uid))); |
| 115 | 115 | $deleteQuery->execute(); |
| 116 | 116 | |
| 117 | - return array_map(function (array $row) { |
|
| 117 | + return array_map(function(array $row) { |
|
| 118 | 118 | return [ |
| 119 | 119 | 'provider_id' => $row['provider_id'], |
| 120 | 120 | 'uid' => $row['uid'], |
@@ -32,21 +32,21 @@ |
||
| 32 | 32 | */ |
| 33 | 33 | final class TwoFactorProviderDisabled extends Event { |
| 34 | 34 | |
| 35 | - /** @var string */ |
|
| 36 | - private $providerId; |
|
| 37 | - |
|
| 38 | - /** |
|
| 39 | - * @since 20.0.0 |
|
| 40 | - */ |
|
| 41 | - public function __construct(string $providerId) { |
|
| 42 | - parent::__construct(); |
|
| 43 | - $this->providerId = $providerId; |
|
| 44 | - } |
|
| 45 | - |
|
| 46 | - /** |
|
| 47 | - * @since 20.0.0 |
|
| 48 | - */ |
|
| 49 | - public function getProviderId(): string { |
|
| 50 | - return $this->providerId; |
|
| 51 | - } |
|
| 35 | + /** @var string */ |
|
| 36 | + private $providerId; |
|
| 37 | + |
|
| 38 | + /** |
|
| 39 | + * @since 20.0.0 |
|
| 40 | + */ |
|
| 41 | + public function __construct(string $providerId) { |
|
| 42 | + parent::__construct(); |
|
| 43 | + $this->providerId = $providerId; |
|
| 44 | + } |
|
| 45 | + |
|
| 46 | + /** |
|
| 47 | + * @since 20.0.0 |
|
| 48 | + */ |
|
| 49 | + public function getProviderId(): string { |
|
| 50 | + return $this->providerId; |
|
| 51 | + } |
|
| 52 | 52 | } |