added option for registration of InvoiceFactory

contributte · Jun 13, 2017 · 2193ee0 · 2193ee0
1 parent d73285f
commit 2193ee0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/DI/InvoiceExtension.php b/src/DI/InvoiceExtension.php
@@ -11,6 +11,7 @@
 use WebChemistry\Invoice\Data\Template;
 use WebChemistry\Invoice\IFormatter;
 use WebChemistry\Invoice\Invoice;
+use WebChemistry\Invoice\InvoiceFactory;
 use WebChemistry\Invoice\ITranslator;
 use WebChemistry\Invoice\Translator;
 
@@ -27,7 +28,8 @@ class InvoiceExtension extends CompilerExtension {
 			'tin' => NULL,
 			'vaTin' => NULL,
 		],
-		'lang' => 'en'
+		'lang' => 'en',
+		'invoiceFactory' => FALSE,
 	];
 
 	public function loadConfiguration() {
@@ -54,6 +56,11 @@ public function loadConfiguration() {
 
 		$builder->addDefinition($this->prefix('invoice'))
 			->setClass(Invoice::class);
+
+		if ($config['invoiceFactory']) {
+			$builder->addDefinition($this->prefix('invoiceFactory'))
+				->setClass(InvoiceFactory::class);
+		}
 	}
 
 }