Correção do ViewModel.

phphacks · Jul 20, 2018 · be13ded · be13ded
1 parent 6ef6664
commit be13ded
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/src/Event/ErrorEventHandler.php b/src/Event/ErrorEventHandler.php
@@ -4,6 +4,7 @@
 
 use Zend\HttpErrors\Exceptions\Base\HttpErrorException;
 use Zend\HttpErrors\Exceptions\HttpInternalServerErrorException;
+use Zend\HttpErrors\Factory\JsonResponseFactory;
 use Zend\HttpErrors\Factory\ResponseFactory;
 use Zend\Mvc\MvcEvent;
 
@@ -60,6 +61,9 @@ public function handle(MvcEvent $event)
             $exception = new HttpInternalServerErrorException();
         }
 
+        $viewModel = JsonResponseFactory::createFrom($exception);
+        $event->setViewModel($viewModel);
+
         $this->responseFactory->createFor($exception, $response);
         $event->stopPropagation(true);
 

diff --git a/src/Factory/JsonResponseFactory.php b/src/Factory/JsonResponseFactory.php
@@ -3,14 +3,19 @@
 namespace Zend\HttpErrors\Factory;
 
 use Zend\HttpErrors\Exceptions\Base\HttpErrorException;
+use Zend\View\Model\JsonModel;
 
 class JsonResponseFactory
 {
     public static function createFrom(HttpErrorException $exception)
     {
-        return json_encode([
+        $jsonViewModel = new JsonModel([
             'errorCode' => $exception->getCode(),
             'message'   => $exception->getMessage(),
         ]);
+
+        $jsonViewModel->setTerminal(true);
+
+        return $jsonViewModel;
     }
 }
diff --git a/src/Factory/ResponseFactory.php b/src/Factory/ResponseFactory.php
@@ -42,7 +42,6 @@ public function createFor(HttpErrorException $exception, $response = null)
 
         $response->setStatusCode($exception::HTTP_ERROR_CODE);
         $response->setReasonPhrase($exception::HTTP_REASON_PHRASE );
-        $response->setContent(JsonResponseFactory::createFrom($exception));
 
         $httpErrorCode = $exception::HTTP_ERROR_CODE;