From 68a1be9a4ff0311b56baa086badfc24f12d69d04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nil=20Portugu=C3=A9s=20Calder=C3=B3?= <nilopc@gmail.com>
Date: Tue, 20 Oct 2015 14:46:41 +0200
Subject: [PATCH] Adding custom headers using the JsonResponseTrait is now
 possible

---
 .../JsonSerializer/JsonResponseTrait.php      | 29 ++++++++++++-------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php b/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php
index 5f2a5bc..34b3976 100644
--- a/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php
+++ b/src/NilPortugues/Laravel5/JsonSerializer/JsonResponseTrait.php
@@ -14,6 +14,15 @@
 
 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
      *
@@ -21,7 +30,7 @@ trait JsonResponseTrait
      */
     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)));
     }
 }