vendor/noxlogic/ratelimit-bundle/EventListener/HeaderModificationListener.php line 23

Open in your IDE?
  1. <?php
  3. namespace Noxlogic\RateLimitBundle\EventListener;
  5. use Noxlogic\RateLimitBundle\Service\RateLimitInfo;
  6. use Symfony\Component\HttpKernel\Event\FilterResponseEvent;
  7. use Symfony\Component\HttpKernel\Event\ResponseEvent;
  9. class HeaderModificationListener extends BaseListener
  10. {
  12.     /**
  13.      * @param array $defaultParameters
  14.      */
  15.     public function __construct($defaultParameters = array())
  16.     {
  17.         $this->parameters = $defaultParameters;
  18.     }
  20.     /**
  21.      * @param FilterResponseEvent|ResponseEvent $event
  22.      */
  23.     public function onKernelResponse($event)
  24.     {
  25.         $request = $event->getRequest();
  27.         // Check if we have a rate-limit-info object in our request attributes. If not, we didn't need to limit.
  28.         $rateLimitInfo = $request->attributes->get('rate_limit_info', null);
  29.         if (! $rateLimitInfo) {
  30.             return;
  31.         }
  33.         // Check if we need to add our x-rate-limits to the headers
  34.         if (! $this->getParameter('display_headers')) {
  35.             return;
  36.         }
  38.         /** @var RateLimitInfo $rateLimitInfo */
  40.         $remaining = $rateLimitInfo->getLimit() - $rateLimitInfo->getCalls();
  41.         if ($remaining < 0) {
  42.             $remaining = 0;
  43.         }
  45.         $response = $event->getResponse();
  46.         $response->headers->set($this->getParameter('header_limit_name'), $rateLimitInfo->getLimit());
  47.         $response->headers->set($this->getParameter('header_remaining_name'), $remaining);
  48.         $response->headers->set($this->getParameter('header_reset_name'), $rateLimitInfo->getResetTimestamp());
  49.     }
  50. }