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

Open in your IDE?
  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. }