Extension with optional latteFactory (#119)

contributte · Jul 18, 2023 · cf6fc53 · cf6fc53
1 parent a819298
commit cf6fc53
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/DI/TranslationExtension.php b/src/DI/TranslationExtension.php
@@ -82,6 +82,7 @@ public function getConfigSchema(): Schema
 			'translatorFactory' => Expect::string()->default(null),
 			'returnOriginalMessage' => Expect::bool()->default(true),
 			'autowired' => Expect::type('bool|array')->default(true),
+			'latteFactory' => Expect::string(ILatteFactory::class)->nullable(),
 		]);
 	}
 
@@ -229,7 +230,7 @@ public function beforeCompile(): void
 			$tracyPanel = $builder->getDefinition($this->prefix('tracyPanel'));
 		}
 
-		$latteFactoryName = $builder->getByType(ILatteFactory::class);
+		$latteFactoryName = $this->config->latteFactory ? $builder->getByType($this->config->latteFactory) : null;
 
 		if ($latteFactoryName !== null) {
 			$iTranslator = $builder->getDefinitionByType(ITranslator::class);