Merge pull request #6 from nickygerritsen/set-response

Set response back on controller

Author: dorxy

src/Panel/ClientCallPanel.php

@@ -99,24 +99,24 @@ public function shutdown(Event $event)
          */
         $controller = $event->getSubject();
         if ($controller instanceof RequestsController) {
-            $this->_injectScriptsAndStyles($controller->getResponse());
+            $controller->setResponse($this->_injectScriptsAndStyles($controller->getResponse()));
         }
     }
 
     /**
-     * Injects the JS to build the toolbar.
+     * Injects the JS to build the toolbar and return the new response.
      *
      * The toolbar will only be injected if the response's content type
      * contains HTML and there is a </body> tag.
      *
      * @param \Cake\Network\Response $response The response to augment.
      *
-     * @return void
+     * @return \Cake\Network\Response
      */
     protected function _injectScriptsAndStyles($response)
     {
         if (strpos($response->getType(), 'html') === false) {
-            return;
+            return $response;
         }
         $body = $response->getBody();
 
@@ -136,6 +136,6 @@ protected function _injectScriptsAndStyles($response)
             $body  = substr($body, 0, $pos) . $style . substr($body, $pos);
         }
 
-        $response = $response->withStringBody($body);
+        return $response->withStringBody($body);
     }
 }