Avoid calling getContext() on presenters not supporting this getter

Kdyby · Oct 28, 2024 · 5df0639 · 5df0639
1 parent 2ea8630
commit 5df0639
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/Kdyby/Autowired/AutowireComponentFactories.php b/src/Kdyby/Autowired/AutowireComponentFactories.php
@@ -24,7 +24,11 @@ trait AutowireComponentFactories
 	protected function getComponentFactoriesLocator(): Nette\DI\Container
 	{
 		if (! isset($this->autowireComponentFactoriesLocator)) {
-			$this->injectComponentFactories($this->getPresenter()->getContext());
+			$presenter = $this->getPresenter();
+			if (! method_exists($presenter, 'getContext')) {
+				throw new InvalidStateException('Cannot get DI Container service - are inject methods allowed?');
+			}
+			$this->injectComponentFactories($presenter->getContext());
 		}
 
 		return $this->autowireComponentFactoriesLocator;