src/Voter/AccessTokenVoter.php line 30

Open in your IDE?
  1. <?php
  3. namespace App\Voter;
  5. use App\Controller\Front\AccessTokenController;
  6. use App\Entity\Trace;
  7. use DateInterval;
  8. use DateTime;
  9. use Doctrine\ORM\EntityManagerInterface;
  10. use Symfony\Component\HttpFoundation\RequestStack;
  11. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  12. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  14. class AccessTokenVoter extends Voter
  15. {
  16.     public const TRY_ACCESS_TOKEN = 'try_access_token';
  18.     public const MAX_TRY_PER_TIMESPAN = 5;
  20.     public const TIMESPAN_SECONDS = 3600;
  22.     private $traceRepo;
  24.     private $requestStack;
  26.     public function __construct(
  27.         EntityManagerInterface $em,
  28.         RequestStack $requestStack
  29.     ) {
  30.         $this->traceRepo    = $em->getRepository(Trace::class);
  31.         $this->requestStack = $requestStack;
  32.     }
  34.     protected function supports($attribute, $entity): bool
  35.     {
  36.         // if the attribute isn't one we support, return false
  38.         if ($attribute === self::TRY_ACCESS_TOKEN) {
  39.             return true;
  40.         }
  43.         return false;
  44.     }
  47.     protected function voteOnAttribute($attribute, $entity, TokenInterface $token): bool
  48.     {
  51.         if ($attribute === self::TRY_ACCESS_TOKEN) {
  52.             return $this->canTry();
  53.         }
  56.         return false;
  57.     }
  59.     private function canTry(): bool
  60.     {
  62.         /*
  63.          * Count all trace with same ip and same type in the timespan
  64.          */
  65.         $request = $this->requestStack->getCurrentRequest();
  67.         $ip = $request->getClientIp();
  69.         $after = $this->buildTryAfterDT();
  72.         $similarTriesCount = $this->traceRepo->searchCount([
  73.             'type'  => AccessTokenController::ACCESS_ROUTE_HIT,
  74.             'ip'    => $ip,
  75.             'after' => $after
  76.         ]);
  79.         return ($similarTriesCount <= self::MAX_TRY_PER_TIMESPAN);
  82.     }
  84.     private function buildTryInterval(): DateInterval
  85.     {
  87.         return new DateInterval('PT' . self::TIMESPAN_SECONDS . 'S');
  89.     }
  91.     private function buildTryAfterDT(): DateTime
  92.     {
  94.         $interval = $this->buildTryInterval();
  96.         $after = new DateTime();
  97.         $after->sub($interval);
  99.         return $after;
  101.     }
  102. }