Added argument $context for custom DBAL types

68publishers · Dec 10, 2020 · b00848e · b00848e
1 parent 11cfcf8
commit b00848e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/src/Bridge/Nettrine/DI/DoctrineBridgeExtension.php b/src/Bridge/Nettrine/DI/DoctrineBridgeExtension.php
@@ -61,6 +61,7 @@ private function processDatabaseTypeProviders(): void
 		$connectionFactory = $builder->getDefinition($dbalExtensionName . '.connectionFactory');
 		$factory = $connectionFactory->getFactory();
 		[$types, $typesMapping] = $factory->arguments;
+		$contexts = [];
 
 		/** @var \SixtyEightPublishers\DoctrineBridge\DI\DatabaseTypeProviderInterface $extension */
 		foreach ($this->compiler->getExtensions(DatabaseTypeProviderInterface::class) as $extension) {
@@ -73,6 +74,8 @@ private function processDatabaseTypeProviders(): void
 				if (NULL !== $databaseType->mappingType) {
 					$typesMapping[$databaseType->name] = $databaseType->mappingType;
 				}
+
+				$contexts[$databaseType->name] = $databaseType->context;
 			}
 		}
 
@@ -89,10 +92,11 @@ private function processDatabaseTypeProviders(): void
 				continue;
 			}
 
-			$connection->addSetup('?::getType(?)->setContainer(?)', [
+			$connection->addSetup('?::getType(?)->setContainer(?, ?)', [
 				new PhpLiteral(Type::class),
 				$typeName,
 				new Reference($builder::THIS_CONTAINER),
+				$contexts[$typeName] ?? [],
 			]);
 		}
 	}

diff --git a/src/DI/DatabaseType.php b/src/DI/DatabaseType.php
@@ -15,15 +15,20 @@ final class DatabaseType
 	/** @var string|NULL  */
 	public $mappingType;
 
+	/** @var array  */
+	public $context;
+
 	/**
 	 * @param string      $name
 	 * @param string      $class
 	 * @param string|NULL $mappingType
+	 * @param array       $context
 	 */
-	public function __construct(string $name, string $class, ?string $mappingType = NULL)
+	public function __construct(string $name, string $class, ?string $mappingType = NULL, array $context = [])
 	{
 		$this->name = $name;
 		$this->class = $class;
 		$this->mappingType = $mappingType;
+		$this->context = $context;
 	}
 }
diff --git a/src/Type/ContainerAwareTypeInterface.php b/src/Type/ContainerAwareTypeInterface.php
@@ -10,8 +10,9 @@ interface ContainerAwareTypeInterface
 {
 	/**
 	 * @param \Nette\DI\Container $container
+	 * @param array               $context
 	 *
 	 * @return void
 	 */
-	public function setContainer(Container $container): void;
+	public function setContainer(Container $container, array $context = []): void;
 }