Adding custom headers using the JsonResponseTrait is now possible

nilportugues · Oct 20, 2015 · 68a1be9 · 68a1be9
1 parent 485e351
commit 68a1be9
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php b/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php
@@ -14,14 +14,23 @@
 
 trait JsonResponseTrait
 {
+    /**
+     * @param \Psr\Http\Message\ResponseInterface $response
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    protected function addHeaders(\Psr\Http\Message\ResponseInterface $response)
+    {
+        return $response;
+    }
+
     /**
      * @param string $json
      *
      * @return \Symfony\Component\HttpFoundation\Response
      */
     private function errorResponse($json) {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ErrorResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ErrorResponse($json)));
     }
 
     /**
@@ -32,7 +41,7 @@ private function errorResponse($json) {
     private function resourceCreatedResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourceCreatedResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourceCreatedResponse($json)));
     }
 
     /**
@@ -43,7 +52,7 @@ private function resourceCreatedResponse($json)
     private function resourceDeletedResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourceDeletedResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourceDeletedResponse($json)));
     }
 
     /**
@@ -54,7 +63,7 @@ private function resourceDeletedResponse($json)
     private function resourceNotFoundResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourceNotFoundResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourceNotFoundResponse($json)));
     }
 
     /**
@@ -65,7 +74,7 @@ private function resourceNotFoundResponse($json)
     private function resourcePatchErrorResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourcePatchErrorResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourcePatchErrorResponse($json)));
     }
 
     /**
@@ -76,7 +85,7 @@ private function resourcePatchErrorResponse($json)
     private function resourcePostErrorResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourcePostErrorResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourcePostErrorResponse($json)));
     }
 
     /**
@@ -87,7 +96,7 @@ private function resourcePostErrorResponse($json)
     private function resourceProcessingResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourceProcessingResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourceProcessingResponse($json)));
     }
 
     /**
@@ -98,7 +107,7 @@ private function resourceProcessingResponse($json)
     private function resourceUpdatedResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\ResourceUpdatedResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\ResourceUpdatedResponse($json)));
     }
 
     /**
@@ -109,7 +118,7 @@ private function resourceUpdatedResponse($json)
     private function response($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\Response($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\Response($json)));
     }
 
     /**
@@ -120,7 +129,7 @@ private function response($json)
     private function unsupportedActionResponse($json)
     {
         return (new HttpFoundationFactory())
-            ->createResponse(new \NilPortugues\Api\Json\Http\Message\UnsupportedActionResponse($json));
+            ->createResponse($this->addHeaders(new \NilPortugues\Api\Json\Http\Message\UnsupportedActionResponse($json)));
     }
 }