ServiceDefinition: setup: prepends @self in depth

nette · Oct 16, 2023 · f3f0447 · f3f0447
1 parent 3b69f36
commit f3f0447
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 7 deletions.
diff --git a/src/DI/Definitions/ServiceDefinition.php b/src/DI/Definitions/ServiceDefinition.php
@@ -177,15 +177,21 @@ public function complete(Nette\DI\Resolver $resolver): void
 
 		$this->creator = $resolver->completeStatement($this->creator);
 
-		foreach ($this->setup as &$setup) {
-			if (
-				is_string($setup->getEntity())
-				&& strpbrk($setup->getEntity(), ':@?\\') === false
-			) { // auto-prepend @self
-				$setup = new Statement([new Reference(Reference::Self), $setup->getEntity()], $setup->arguments);
+		foreach ($this->setup as $i => $setup) {
+			// auto-prepend @self
+			$stmts = [];
+			while (is_array($setup->getEntity()) && $setup->getEntity()[0] instanceof Statement) {
+				$stmts[] = $setup;
+				$setup = $setup->getEntity()[0];
+			}
+			if (is_string($member = $setup->getEntity()) && preg_match('~^\$?[a-z][a-zA-Z0-9[\]]*$~', $member)) {
+				$setup = new Statement([new Reference(Reference::Self), $member], $setup->arguments);
+			}
+			while ($stmt = array_pop($stmts)) {
+				$setup = new Statement([$setup, $stmt->getEntity()[1]], $stmt->arguments);
 			}
 
-			$setup = $resolver->completeStatement($setup, true);
+			$this->setup[$i] = $resolver->completeStatement($setup, true);
 		}
 	}
 

diff --git a/tests/DI/Compiler.services.setup.phpt b/tests/DI/Compiler.services.setup.phpt
@@ -22,6 +22,7 @@ class Lorem
 	public function test(...$args)
 	{
 		Notes::add(__METHOD__ . ' ' . implode(' ', $args));
+		return new self;
 	}
 }
 
@@ -79,6 +80,8 @@ Assert::same([
 	'globtest 7',
 	'Ipsum::test  a',
 	'Ipsum::test 10 b',
+	'Lorem::test outer',
+	'Lorem::test inner',
 ], Notes::fetch());
 
 Assert::same(8, $container->getService('lorem')->test);

diff --git a/tests/DI/files/compiler.services.setup.neon b/tests/DI/files/compiler.services.setup.neon
@@ -36,6 +36,7 @@ services:
 			# rich
 			- Ipsum(a)::test
 			- Ipsum(b)::test(10)
+			- test(outer)::test(inner)
 
 			# object
 			- \Ipsum(@self)