Extension: 'connection' & 'entityFactory' can be disabled

src/LeanMapperExtension.php

@@ -53,12 +53,10 @@ public function loadConfiguration()
 			// Services
 			$connection = $this->configConnection($builder, $config);
 			$this->configMapper($builder, $config);
-
-			$builder->addDefinition($this->prefix('entityFactory'))
-				->setClass($config['entityFactory']);
+			$this->configEntityFactory($builder, $config);
 
 			// profiler
-			if ($useProfiler) {
+			if ($connection && $useProfiler) {
 				$panel = $builder->addDefinition($this->prefix('panel'))
 					->setClass(\Dibi\Bridges\Tracy\Panel::class);
 
@@ -73,8 +71,14 @@ public function loadConfiguration()
 		 */
 		protected function configConnection(ContainerBuilder $builder, array $config)
 		{
-			if (!isset($config['connection']) || !is_string($config['connection'])) {
-				throw new \RuntimeException('Connection class definition is missing, or not (string).');
+			$connectionClass = $config['connection'];
+
+			if ($connectionClass === FALSE || $connectionClass === NULL) {
+				return NULL;
+			}
+
+			if (!is_string($config['connection'])) {
+				throw new \RuntimeException('Connection class definition must be string.');
 			}
 
 			return $builder->addDefinition($this->prefix('connection'))
@@ -92,6 +96,27 @@ protected function configConnection(ContainerBuilder $builder, array $config)
 		}
 
 
+		/**
+		 * Adds connection service into container
+		 * @return ServiceDefinition
+		 */
+		protected function configEntityFactory(ContainerBuilder $builder, array $config)
+		{
+			$entityFactoryClass = $config['entityFactory'];
+
+			if ($entityFactoryClass === FALSE || $entityFactoryClass === NULL) {
+				return NULL;
+			}
+
+			if (!is_string($config['entityFactory'])) {
+				throw new \RuntimeException('EntityFactory class definition must be string.');
+			}
+
+			return $builder->addDefinition($this->prefix('entityFactory'))
+				->setClass($config['entityFactory']);
+		}
+
+
 		/**
 		 * Adds mapper service into container
 		 * @return ServiceDefinition|NULL

tests/LeanMapperExtension/Extension.phpt

@@ -57,6 +57,12 @@ class NewsRepository extends LeanMapper\Repository {}
 test(function () {
 	$container = createContainer('readme.entities');
 
+	$connection = $container->getService('leanmapper.connection');
+	Assert::true($connection instanceof \LeanMapper\Connection);
+
+	$entityFactory = $container->getService('leanmapper.entityFactory');
+	Assert::true($entityFactory instanceof \LeanMapper\DefaultEntityFactory);
+
 	$newsRepository = $container->getByType(NewsRepository::class);
 	Assert::true($newsRepository instanceof NewsRepository);
 
@@ -82,8 +88,17 @@ test(function () {
 
 // Disable
 test(function () {
-	$container = createContainer('disable-mapper');
+	$container = createContainer('disable-services');
+
+	Assert::exception(function () use ($container) {;
+		$container->getByType(LeanMapper\Connection::class);
+	}, Nette\DI\MissingServiceException::class);
+
 	Assert::exception(function () use ($container) {;
 		$container->getByType(LeanMapper\IMapper::class);
 	}, Nette\DI\MissingServiceException::class);
+
+	Assert::exception(function () use ($container) {;
+		$container->getByType(LeanMapper\IEntityFactory::class);
+	}, Nette\DI\MissingServiceException::class);
 });

tests/LeanMapperExtension/config/disable-services.neon

@@ -0,0 +1,4 @@
+leanmapper:
+	mapper: no
+	connection: no
+	entityFactory: no