src/Platform/SecurityBundle/Doctrine/Login/AttemptRepository.php line 39

Open in your IDE?
  1. <?php
  3. namespace Platform\SecurityBundle\Doctrine\Login;
  5. use Cms\CoreBundle\Model\Search\SearchableTrait;
  6. use Cms\CoreBundle\Util\DateTimeUtils;
  7. use Cms\CoreBundle\Util\Doctrine\EntityRepository;
  8. use Cms\TenantBundle\Entity\Tenant;
  9. use DateTime;
  10. use Platform\SecurityBundle\Entity\Login\Attempt;
  12. /**
  13.  * Class AttemptRepository
  14.  *
  15.  * @package Platform\SecurityBundle\Doctrine\Login
  16.  *
  17.  * @method Attempt findExact($id);
  18.  */
  19. class AttemptRepository extends EntityRepository
  20. {
  21.     use SearchableTrait;
  23.     /**
  24.      * @param string $uid
  25.      * @return Attempt|null
  26.      */
  27.     public function findOneByUid(string $uid): ?Attempt
  28.     {
  29.         return $this->findOneBy([
  30.             'uid' => $uid,
  31.         ]);
  32.     }
  34.     /**
  35.      * @param Tenant $tenant
  36.      * @param string|null $ip
  37.      * @return int
  38.      */
  39.     public function countLastHourFailed(Tenant $tenant, ?string $ip = null)
  40.     {
  41.         // create the builder
  42.         $qb = $this->_em->createQueryBuilder()
  43.             ->select('COUNT(attempts)')
  44.             ->from($this->_entityName, 'attempts')
  45.             ->where('attempts.status = false')
  46.             ->andWhere('attempts.tenant = :tenant')
  47.             ->setParameter('tenant', $tenant)
  48.             ->andWhere('attempts.createdAt >= :date')
  49.             ->setParameter('date', DateTimeUtils::beforeCurrent('PT1H'));
  51.         if ( ! empty($ip)) {
  52.             $qb->andWhere('attempts.ip = :ip')
  53.                 ->setParameter('ip', $ip);
  54.         }
  56.         // run it
  57.         return intval($this->querySingleScalar($qb)[0]);
  58.     }
  60.     /**
  61.      * @param DateTime $startDate
  62.      * @param DateTime $endDate
  63.      * @return int
  64.      */
  65.     public function countLoginsByDateInterval(DateTime $startDate, DateTime $endDate)
  66.     {
  67.         // need to get login attempts for all tenants
  68.         $tenantFilter = $this->_em->getFilters()->getFilter('tenant_filter');
  69.         $this->_em->getFilters()->disable('tenant_filter');
  71.         // create the builder
  72.         $qb = $this->createQueryBuilder('attempts')
  73.             ->select('COUNT(attempts)')
  74.             ->where('attempts.status = :status')
  75.             ->andWhere('attempts.createdAt >= :startDate')
  76.             ->andWhere('attempts.createdAt < :endDate')
  77.             ->setParameters(array(
  78.                 'status' => true,
  79.                 'startDate' => $startDate,
  80.                 'endDate' => $endDate,
  81.             ))
  82.         ;
  84.         $result = $this->querySingleScalar($qb);
  86.         // I'm not sure that this is necessary, but theoretically this can affect further queries with tenant params
  87.         $this->_em->getFilters()
  88.             ->enable('tenant_filter')
  89.             ->setParameter('id', $tenantFilter->getParameter('id'))
  90.         ;
  92.         return intval($result[0]);
  93.     }
  94. }