Update PjaxServiceProvider.php

vtalbot · Dec 10, 2013 · 06ce370 · 06ce370
1 parent 41c0387
commit 06ce370
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/VTalbot/Pjax/PjaxServiceProvider.php b/src/VTalbot/Pjax/PjaxServiceProvider.php
@@ -26,9 +26,20 @@ public function register()
 		$app = $this->app;
 
 		$this->app->after(function(Request $request, Response $response) use ($app) {
+			if ($response->isRedirection())
+			{
+				return $response;
+			}
+
 			if ($request->server->get('HTTP_X_PJAX')) {
 				$crawler = new Crawler($response->getContent());
-				$response->setContent($crawler->filter($request->server->get('HTTP_X_PJAX_CONTAINER'))->html());
+
+				$html = $crawler->filter($request->server->get('HTTP_X_PJAX_CONTAINER'))->html();
+				$title = $crawler->filter('head title')->html();
+
+				$response->setContent($title . $html);
+
+				$response->header('X-PJAX-URL', $request->getRequestUri());
 			}
 
 			return $response;