Presenter: persistent parameters are transmitted between traits (in addition to the class hierarchy) [Closes #183]

Author: dg

src/Application/UI/ComponentReflection.php

@@ -52,7 +52,7 @@ public function getPersistentParams(string $class = null): array
 				if (!$rp->isStatic() && self::parseAnnotation($rp, 'persistent')) {
 					$params[$name] = [
 						'def' => $default,
-						'since' => $isPresenter ? $class : null,
+						'since' => $isPresenter ? Nette\Utils\Reflection::getPropertyDeclaringClass($rp)->getName() : null,
 					];
 				}
 			}
@@ -94,13 +94,14 @@ public function getPersistentComponents(string $class = null): array
 	 */
 	public function saveState(Component $component, array &$params): void
 	{
+		$traits = array_flip(Nette\Utils\Reflection::getAllTraits(new \ReflectionClass($component)));
 		foreach ($this->getPersistentParams() as $name => $meta) {
 			if (isset($params[$name])) {
 				// injected value
 
 			} elseif (
 				array_key_exists($name, $params) // nulls are skipped
-				|| (isset($meta['since']) && !$component instanceof $meta['since']) // not related
+				|| (isset($meta['since']) && !$component instanceof $meta['since'] && !isset($traits[$meta['since']])) // not related
 				|| !isset($component->$name)
 			) {
 				continue;

src/Application/UI/Presenter.php

@@ -1100,6 +1100,17 @@ protected function getGlobalState(string $forClass = null): array
 		}
 
 		if ($forClass !== null) {
+			$tree = [$forClass => $forClass] + class_parents($forClass);
+			$addTraits = function ($type) use (&$tree, &$addTraits) {
+				$tree += class_uses($type);
+				foreach (class_uses($type) as $trait) {
+					$addTraits($trait);
+				}
+			};
+			foreach ($tree as $class) {
+				$addTraits($class);
+			}
+
 			$since = null;
 			foreach ($state as $key => $foo) {
 				if (!isset($sinces[$key])) {
@@ -1109,7 +1120,7 @@ protected function getGlobalState(string $forClass = null): array
 				}
 				if ($since !== $sinces[$key]) {
 					$since = $sinces[$key];
-					$ok = $since && is_a($forClass, $since, true);
+					$ok = $since && isset($tree[$since]);
 				}
 				if (!$ok) {
 					unset($state[$key]);

tests/UI/Presenter.link().persistent.phpt

@@ -64,7 +64,7 @@ class TestPresenter extends BasePresenter
 		$this->t2 = 4;
 		Assert::same('/index.php?p2=2&p1=1&t1=3&t2=4&action=default&presenter=Test', $this->link('this'));
 		Assert::same('/index.php?p1=1&t1=3&action=default&presenter=Second', $this->link('Second:'));
-		Assert::same('/index.php?p1=1&t1=3&action=default&presenter=Third', $this->link('Third:'));
+		Assert::same('/index.php?p1=1&t1=3&t2=4&action=default&presenter=Third', $this->link('Third:'));
 
 		$this->p1 = 20;
 		Assert::same('/index.php?t1=3&action=default&presenter=Second', $this->link('Second:'));
@@ -97,27 +97,27 @@ class ThirdPresenter extends BasePresenter
 
 Assert::same([
 	'p1' => ['def' => null, 'since' => 'BasePresenter'],
-	't1' => ['def' => null, 'since' => 'BasePresenter'],
+	't1' => ['def' => null, 'since' => 'PersistentParam1'],
 ], BasePresenter::getReflection()->getPersistentParams());
 
 Assert::same([
 	'p2' => ['def' => null, 'since' => 'TestPresenter'],
 	'p1' => ['def' => null, 'since' => 'BasePresenter'],
-	't1' => ['def' => null, 'since' => 'BasePresenter'],
-	't2' => ['def' => null, 'since' => 'TestPresenter'],
+	't1' => ['def' => null, 'since' => 'PersistentParam1'],
+	't2' => ['def' => null, 'since' => 'PersistentParam2A'],
 ], TestPresenter::getReflection()->getPersistentParams());
 
 Assert::same([
 	'p1' => ['def' => 20, 'since' => 'BasePresenter'],
 	'p3' => ['def' => null, 'since' => 'SecondPresenter'],
-	't1' => ['def' => null, 'since' => 'BasePresenter'],
-	't3' => ['def' => null, 'since' => 'SecondPresenter'],
+	't1' => ['def' => null, 'since' => 'PersistentParam1'],
+	't3' => ['def' => null, 'since' => 'PersistentParam3'],
 ], SecondPresenter::getReflection()->getPersistentParams());
 
 Assert::same([
 	'p1' => ['def' => null, 'since' => 'BasePresenter'],
-	't1' => ['def' => null, 'since' => 'BasePresenter'],
-	't2' => ['def' => null, 'since' => 'ThirdPresenter'],
+	't1' => ['def' => null, 'since' => 'PersistentParam1'],
+	't2' => ['def' => null, 'since' => 'PersistentParam2A'],
 ], ThirdPresenter::getReflection()->getPersistentParams());