vendor/noxlogic/ratelimit-bundle/Noxlogic/RateLimitBundle/EventListener/HeaderModificationListener.php line 22

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