add utilities for nette di

Author: MartkCz

src/DI/CompilerExtensionUtility.php

@@ -2,11 +2,13 @@
 
 namespace Utilitte\Nette\DI;
 
+use JetBrains\PhpStorm\Deprecated;
 use Nette\DI\ContainerBuilder;
 use Nette\DI\Definitions\Definition;
 use Nette\DI\Definitions\FactoryDefinition;
 use Nette\DI\Definitions\ServiceDefinition;
 
+#[Deprecated('Use NetteDIStrict instead of')]
 final class CompilerExtensionUtility
 {
 

src/DI/UtilityExtension.php

@@ -16,6 +16,7 @@
 use Utilitte\Nette\Latte\Macros;
 use Utilitte\Nette\UI\FlexibleMultiplierByIdentifierFactory;
 use Utilitte\Nette\UI\FlexibleTransferObjectMultiplier;
+use Utilitte\Nette\Utility\NetteDIUtility;
 
 final class UtilityExtension extends CompilerExtension
 {
@@ -69,9 +70,7 @@ public function beforeCompile(): void
 		$config = $this->getConfig();
 
 		if ($config->latte->enable) {
-			$service = CompilerExtensionUtility::getFactoryDefinitionByType($builder, ILatteFactory::class);
-
-			$service->getResultDefinition()
+			NetteDIUtility::getLatteFactory($builder)->getResultDefinition()
 				->addSetup('?->onCompile[] = function ($engine) { ?::install($engine->getCompiler(), ?); }', [
 					'@self',
 					Macros::class,

src/Strict/NetteDIStrict.php

@@ -0,0 +1,42 @@
+<?php declare(strict_types = 1);
+
+namespace Utilitte\Nette\Strict;
+
+use Nette\DI\CompilerExtension;
+use Nette\DI\ContainerBuilder;
+use Nette\DI\Definitions\AccessorDefinition;
+use Nette\DI\Definitions\FactoryDefinition;
+use Nette\DI\Definitions\ImportedDefinition;
+use Nette\DI\Definitions\LocatorDefinition;
+use Nette\DI\Definitions\ServiceDefinition;
+use Utilitte\Php\Strict\Strict;
+
+final class NetteDIStrict
+{
+
+	public static function getServiceDefinitionByType(ContainerBuilder $builder, string $type): ServiceDefinition
+	{
+		return Strict::instanceOf($builder->getDefinitionByType($type), ServiceDefinition::class);
+	}
+
+	public static function getFactoryDefinitionByType(ContainerBuilder $builder, string $type): FactoryDefinition
+	{
+		return Strict::instanceOf($builder->getDefinitionByType($type), FactoryDefinition::class);
+	}
+
+	public static function getAccessorDefinitionByType(ContainerBuilder $builder, string $type): AccessorDefinition
+	{
+		return Strict::instanceOf($builder->getDefinitionByType($type), AccessorDefinition::class);
+	}
+
+	public static function getImportedDefinitionByType(ContainerBuilder $builder, string $type): ImportedDefinition
+	{
+		return Strict::instanceOf($builder->getDefinitionByType($type), ImportedDefinition::class);
+	}
+
+	public static function getLocatorDefinitionByType(ContainerBuilder $builder, string $type): LocatorDefinition
+	{
+		return Strict::instanceOf($builder->getDefinitionByType($type), LocatorDefinition::class);
+	}
+
+}

src/Utility/NetteDIUtility.php

@@ -0,0 +1,28 @@
+<?php declare(strict_types = 1);
+
+namespace Utilitte\Nette\Utility;
+
+use Nette\Bridges\ApplicationLatte\ILatteFactory;
+use Nette\DI\ContainerBuilder;
+use Nette\DI\Definitions\FactoryDefinition;
+use Utilitte\Nette\Strict\NetteDIStrict;
+
+final class NetteDIUtility
+{
+
+	public static function getLatteFactory(ContainerBuilder $builder): FactoryDefinition
+	{
+		return NetteDIStrict::getFactoryDefinitionByType($builder, ILatteFactory::class);
+	}
+
+	public static function registerMacro(ContainerBuilder $builder, string $class): void
+	{
+		self::getLatteFactory($builder)
+			->getResultDefinition()
+				->addSetup('?->onCompile[] = function (Latte\Engine $engine): void { ?::install($engine->getCompiler()); }', [
+					'@self',
+					$class,
+				]);
+	}
+
+}