Added support for request restore with ajax

Author: sallyx

src/Application/UI/Presenter.php

@@ -1092,7 +1092,7 @@ public function storeRequest($expiration = '+ 10 minutes')
 			$key = Nette\Utils\Random::generate(5);
 		} while (isset($session[$key]));
 
-		$session[$key] = [$this->getUser()->getId(), $this->request];
+		$session[$key] = [$this->getUser()->getId(), $this->request, (string) $this->httpRequest->getUrl()];
 		$session->setExpiration($expiration, $key);
 		return $key;
 	}
@@ -1110,7 +1110,11 @@ public function restoreRequest($key)
 			return;
 		}
 		$request = clone $session[$key][1];
+		$url = $session[$key][2];
 		unset($session[$key]);
+		if($this->isAjax()) {
+			$this->redirectUrl($url);
+		}
 		$request->setFlag(Application\Request::RESTORED, TRUE);
 		$params = $request->getParameters();
 		$params[self::FLASH_KEY] = $this->getParameter(self::FLASH_KEY);

tests/UI/Presenter.storeRequest().phpt

@@ -83,11 +83,13 @@ class MockUser extends Security\User
 
 
 $presenter = new TestPresenter();
+$url = 'http://www.example.org/foo/bar';
+$urlScript = new Http\UrlScript($url);
 $presenter->injectPrimary(
 	NULL,
 	NULL,
 	new Application\Routers\SimpleRouter,
-	new Http\Request(new Http\UrlScript),
+	new Http\Request($urlScript),
 	new Http\Response,
 	$session = new MockSession,
 	$user = new MockUser
@@ -105,4 +107,4 @@ Assert::same($expiration, $section->testExpiration);
 Assert::same($key, $section->testExpirationVariables);
 Assert::same($key, $section->testedKeyExistence);
 Assert::same($key, $section->storedKey);
-Assert::same([$user->getId(), $applicationRequest], $section->storedValue);
+Assert::same([$user->getId(), $applicationRequest, $url], $section->storedValue);