Silverback\ApiComponentsBundle\RefreshToken\Storage\DoctrineRefreshTokenStorage

Complexity

Total Complexity

16

Size/Duplication

Total Lines	90
Duplicated Lines	0 %

Test Coverage

Coverage

12.24%

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	10	2
A	expireAll()	0	12	3
A	create()	0	14	1
A	getEntityManager()	0	9	2
A	expireToken()	0	8	3
A	findOneByUser()	0	20	5

<?php

/*
 * This file is part of the Silverback API Components Bundle Project
 *
 * (c) Daniel West <daniel@silverback.is>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

declare(strict_types=1);

namespace Silverback\ApiComponentsBundle\RefreshToken\Storage;

use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityNotFoundException;
use Doctrine\Persistence\ManagerRegistry;
use Silverback\ApiComponentsBundle\Entity\Core\AbstractRefreshToken;
use Silverback\ApiComponentsBundle\RefreshToken\RefreshToken;
use Silverback\ApiComponentsBundle\Repository\Core\RefreshTokenRepository;
use Symfony\Component\Security\Core\User\UserInterface;

/**
 * @author Vincent Chalamon <vincent@les-tilleuls.coop>
 */
final class DoctrineRefreshTokenStorage implements RefreshTokenStorageInterface
{
    private ManagerRegistry $registry;
    private int $ttl;
    private string $className;

    public function __construct(ManagerRegistry $registry, int $ttl, array $options)
    {
        $this->registry = $registry;
        $this->ttl = $ttl;

        if (!isset($options['class'])) {
            throw new \InvalidArgumentException('You must specify silverback_api_components.refresh.token.options.class option.');
        }

        $this->className = $options['class'];
    }

    public function findOneByUser(UserInterface $user): ?RefreshToken
    {
        $em = $this->getEntityManager();
        $repository = $em->getRepository($this->className);
        if (!$repository instanceof RefreshTokenRepository) {
            throw new \InvalidArgumentException('RefreshToken entity repository must be instance of ' . RefreshTokenRepository::class);
        }
        $refreshTokens = $repository->findByUser($user);

        // Concurrent requests will result in multiple refresh tokens having been expired and re-created
        if (\count($refreshTokens) > 1) {
            foreach ($refreshTokens as $i => $refreshToken) {
                if ($i > 0) {
                    $em->remove($refreshToken);
                }
            }
            $em->flush();
        }

        return $refreshTokens[0];
    }

    public function create(UserInterface $user): RefreshToken
    {
        $className = $this->className;
        /** @var RefreshToken $refreshToken */
        $refreshToken = new $className();
        $refreshToken->setCreatedAt(new \DateTimeImmutable());
        $refreshToken->setExpiresAt(new \DateTimeImmutable("$this->ttl seconds"));
        $refreshToken->setUser($user);

        $em = $this->getEntityManager();
        $em->persist($refreshToken);
        $em->flush($refreshToken);

        return $refreshToken;
    }

    public function expireAll(?UserInterface $user): void
    {
        $em = $this->getEntityManager();
        $repository = $em->getRepository($this->className);
        $refreshTokens = $user ? $repository->findBy(['user' => $user]) : $repository->findAll();

        foreach ($refreshTokens as $refreshToken) {
            /* @var AbstractRefreshToken $refreshToken */
            $this->expireToken($refreshToken, false);
        }

        $em->flush();
    }

    public function expireToken(AbstractRefreshToken $refreshToken, bool $flush = true): void
    {
        $em = $this->getEntityManager();
        if (!$refreshToken->isExpired()) {
            $refreshToken->setExpiresAt(new \DateTimeImmutable());
        }
        if ($flush) {
            $em->flush();
        }
    }

    private function getEntityManager(): EntityManager
    {
        /** @var EntityManager|null $em */
        $em = $this->registry->getManagerForClass($this->className);
        if (!$em) {

$em is of type Doctrine\ORM\EntityManager, thus it always evaluated to true.

            throw new EntityNotFoundException('No entity found for class RefreshToken::class.');
        }

        return $em;
    }
}