vendor/friendsofsymfony/rest-bundle/EventListener/ResponseStatusCodeListener.php line 43

  1. <?php
  3. /*
  4.  * This file is part of the FOSRestBundle package.
  5.  *
  6.  * (c) FriendsOfSymfony <http://friendsofsymfony.github.com/>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace FOS\RestBundle\EventListener;
  14. use FOS\RestBundle\FOSRestBundle;
  15. use FOS\RestBundle\Util\ExceptionValueMap;
  16. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  17. use Symfony\Component\HttpKernel\Event\ExceptionEvent;
  18. use Symfony\Component\HttpKernel\Event\KernelEvent;
  19. use Symfony\Component\HttpKernel\Event\ResponseEvent;
  20. use Symfony\Component\HttpKernel\KernelEvents;
  22. /**
  23.  * @author Christian Flothmann <christian.flothmann@sensiolabs.de>
  24.  */
  25. class ResponseStatusCodeListener implements EventSubscriberInterface
  26. {
  27.     private $exceptionValueMap;
  28.     private $responseStatusCode;
  30.     public function __construct(ExceptionValueMap $exceptionValueMap)
  31.     {
  32.         $this->exceptionValueMap = $exceptionValueMap;
  33.     }
  35.     public static function getSubscribedEvents(): array
  36.     {
  37.         return [
  38.             KernelEvents::EXCEPTION => 'getResponseStatusCodeFromThrowable',
  39.             KernelEvents::RESPONSE => 'setResponseStatusCode',
  40.         ];
  41.     }
  43.     public function getResponseStatusCodeFromThrowable(ExceptionEvent $event): void
  44.     {
  45.         if (!$this->isMainRequest($event)) {
  46.             return;
  47.         }
  49.         $request = $event->getRequest();
  51.         if (!$request->attributes->get(FOSRestBundle::ZONE_ATTRIBUTE, true)) {
  52.             return;
  53.         }
  55.         $statusCode = $this->exceptionValueMap->resolveFromClassName(get_class($event->getThrowable()));
  57.         if (is_int($statusCode)) {
  58.             $this->responseStatusCode = $statusCode;
  59.         }
  60.     }
  62.     public function setResponseStatusCode(ResponseEvent $event): void
  63.     {
  64.         if (!$this->isMainRequest($event)) {
  65.             return;
  66.         }
  68.         if (null !== $this->responseStatusCode) {
  69.             $event->getResponse()->setStatusCode($this->responseStatusCode);
  71.             $this->responseStatusCode = null;
  72.         }
  73.     }
  75.     private function isMainRequest(KernelEvent $event): bool
  76.     {
  77.         if (method_exists($event, 'isMainRequest')) {
  78.             return $event->isMainRequest();
  79.         }
  81.         return $event->isMasterRequest();
  82.     }
  83. }