| @@ 12-72 (lines=61) @@ | ||
| 9 | use League\OAuth2\Server\Entities\AuthCodeEntityInterface; |
|
| 10 | use League\OAuth2\Server\Repositories\AuthCodeRepositoryInterface; |
|
| 11 | ||
| 12 | class AuthCodeRepository implements AuthCodeRepositoryInterface |
|
| 13 | { |
|
| 14 | /** |
|
| 15 | * @var EntityManager |
|
| 16 | */ |
|
| 17 | private $entityManager; |
|
| 18 | ||
| 19 | /** |
|
| 20 | * @var EntityRepository |
|
| 21 | */ |
|
| 22 | private $repository; |
|
| 23 | ||
| 24 | public function __construct(EntityManager $entityManager) |
|
| 25 | { |
|
| 26 | $this->entityManager = $entityManager; |
|
| 27 | $this->repository = $entityManager->getRepository(AuthCodeEntity::class); |
|
| 28 | } |
|
| 29 | ||
| 30 | /** |
|
| 31 | * @return AuthCodeEntityInterface |
|
| 32 | */ |
|
| 33 | public function getNewAuthCode() |
|
| 34 | { |
|
| 35 | return new AuthCodeEntity(); |
|
| 36 | } |
|
| 37 | ||
| 38 | /** |
|
| 39 | * @param AuthCodeEntityInterface $authCodeEntity |
|
| 40 | */ |
|
| 41 | public function persistNewAuthCode(AuthCodeEntityInterface $authCodeEntity) |
|
| 42 | { |
|
| 43 | $this->entityManager->persist($authCodeEntity); |
|
| 44 | $this->entityManager->flush(); |
|
| 45 | } |
|
| 46 | ||
| 47 | /** |
|
| 48 | * @param string $codeId |
|
| 49 | */ |
|
| 50 | public function revokeAuthCode($codeId) |
|
| 51 | { |
|
| 52 | $this->repository->fetchOne($this->createQuery()->byIdentifier($codeId))->setRevoked(true); |
|
| 53 | $this->entityManager->flush(); |
|
| 54 | } |
|
| 55 | ||
| 56 | /** |
|
| 57 | * @param string $codeId |
|
| 58 | * @return bool |
|
| 59 | */ |
|
| 60 | public function isAuthCodeRevoked($codeId) |
|
| 61 | { |
|
| 62 | return $this->repository->fetchOne($this->createQuery()->byIdentifier($codeId))->isRevoked(); |
|
| 63 | } |
|
| 64 | ||
| 65 | /** |
|
| 66 | * @return QueryObject |
|
| 67 | */ |
|
| 68 | protected function createQuery(): QueryObject |
|
| 69 | { |
|
| 70 | return new AuthCodeQuery(); |
|
| 71 | } |
|
| 72 | } |
|
| 73 | ||
| @@ 12-72 (lines=61) @@ | ||
| 9 | use League\OAuth2\Server\Entities\RefreshTokenEntityInterface; |
|
| 10 | use League\OAuth2\Server\Repositories\RefreshTokenRepositoryInterface; |
|
| 11 | ||
| 12 | class RefreshTokenRepository implements RefreshTokenRepositoryInterface |
|
| 13 | { |
|
| 14 | /** |
|
| 15 | * @var EntityManager |
|
| 16 | */ |
|
| 17 | private $entityManager; |
|
| 18 | ||
| 19 | /** |
|
| 20 | * @var EntityRepository |
|
| 21 | */ |
|
| 22 | private $repository; |
|
| 23 | ||
| 24 | public function __construct(EntityManager $entityManager) |
|
| 25 | { |
|
| 26 | $this->entityManager = $entityManager; |
|
| 27 | $this->repository = $entityManager->getRepository(RefreshTokenEntity::class); |
|
| 28 | } |
|
| 29 | ||
| 30 | /** |
|
| 31 | * @return RefreshTokenEntity |
|
| 32 | */ |
|
| 33 | public function getNewRefreshToken() |
|
| 34 | { |
|
| 35 | return new RefreshTokenEntity(); |
|
| 36 | } |
|
| 37 | ||
| 38 | /** |
|
| 39 | * @param RefreshTokenEntityInterface $refreshTokenEntity |
|
| 40 | */ |
|
| 41 | public function persistNewRefreshToken(RefreshTokenEntityInterface $refreshTokenEntity) |
|
| 42 | { |
|
| 43 | $this->entityManager->persist($refreshTokenEntity); |
|
| 44 | $this->entityManager->flush(); |
|
| 45 | } |
|
| 46 | ||
| 47 | /** |
|
| 48 | * @param string $tokenId |
|
| 49 | */ |
|
| 50 | public function revokeRefreshToken($tokenId) |
|
| 51 | { |
|
| 52 | $this->repository->fetchOne($this->createQuery()->byIdentifier($tokenId))->setRevoked(true); |
|
| 53 | $this->entityManager->flush(); |
|
| 54 | } |
|
| 55 | ||
| 56 | /** |
|
| 57 | * @param string $tokenId |
|
| 58 | * @return bool |
|
| 59 | */ |
|
| 60 | public function isRefreshTokenRevoked($tokenId) |
|
| 61 | { |
|
| 62 | return $this->repository->fetchOne($this->createQuery()->byIdentifier($tokenId))->isRevoked(); |
|
| 63 | } |
|
| 64 | ||
| 65 | /** |
|
| 66 | * @return QueryObject |
|
| 67 | */ |
|
| 68 | protected function createQuery(): QueryObject |
|
| 69 | { |
|
| 70 | return new RefreshTokenQuery(); |
|
| 71 | } |
|
| 72 | } |
|
| 73 | ||