#3 Automatické přiřazení consumerů projektu do skupiny

Author: Marek Hučík

src/DI/SupervisorExtension.php

@@ -20,6 +20,7 @@ final class SupervisorExtension extends CompilerExtension
 		'prefix' => NULL,
 		'configuration' => [],
 		'defaults' => [],
+		'group' => NULL,
 	];
 
 
@@ -37,7 +38,8 @@ public function loadConfiguration()
 		$this->loadSupervisorConfiguration(
 			(array) $config['configuration'],
 			(array) $config['defaults'],
-			isset($config['prefix']) ? (string) $config['prefix'] : NULL
+			isset($config['prefix']) ? (string) $config['prefix'] : NULL,
+			isset($config['group']) ? (string) $config['group'] : NULL
 		);
 
 		$builder->addDefinition($this->prefix('renderCommand'))
@@ -51,23 +53,21 @@ public function loadConfiguration()
 	}
 
 
-	private function loadSupervisorConfiguration(array $config, array $defaults = [], string $prefix = NULL)
+	private function loadSupervisorConfiguration(array $config, array $defaults = [], string $prefix = NULL, string $group = NULL)
 	{
 		$builder = $this->getContainerBuilder();
 
 		$configuration = $builder->addDefinition($this->prefix('configuration'))
 			->setClass(Configuration::class)
 		;
+
 		foreach ($config as $sectionName => $sectionConfig) {
 			if ( ! $sectionClass = (new Configuration)->findSection($sectionName)) {
 				$sectionClass = GenericSection::class;
 			}
 			if (is_subclass_of($sectionClass, Named::class)) {
 				foreach ((array) $sectionConfig as $name => $properties) {
-					$name = Helpers::expand($name, $builder->parameters);
-					if ($prefix !== NULL) {
-						$name = sprintf('%s-%s', $prefix, $name);
-					}
+					$name = $this->prepareName($name, $prefix);
 					$configuration->addSetup('addSection', [
 						new Statement($sectionClass, [
 							$name,
@@ -84,6 +84,8 @@ private function loadSupervisorConfiguration(array $config, array $defaults = []
 				]);
 			}
 		}
+
+		$this->prepareGroup($config, $configuration, $prefix, $group);
 	}
 
 
@@ -99,4 +101,43 @@ private function mergeProperties(array $properties, array $defaults = []): array
 
 		return $properties;
 	}
+
+
+	private function prepareName(string $name, string $prefix = NULL): string
+	{
+		$builder = $this->getContainerBuilder();
+		$parameters = [
+			'webId' => $builder->parameters['webId'],
+		];
+		$name = Helpers::expand($name, $builder->parameters);
+
+		if ($prefix !== NULL) {
+			$name = sprintf('%s-%s', $prefix, $name);
+		}
+
+		return $name;
+	}
+
+
+	private function prepareGroup(array $config, \Nette\DI\ServiceDefinition $configuration, string $prefix = NULL, string $group = NULL): void
+	{
+		if ( ! $group) {
+			return;
+		}
+
+		$webPrograms = implode(',', array_map(function ($name) use ($prefix): string {
+			return $this->prepareName($name, $prefix);
+		}, array_keys($config['program'])));
+
+		$sectionClass = (new Configuration)->findSection('group');
+		$configuration->addSetup('addSection', [
+			new Statement(
+				$sectionClass,
+				[
+					$group,
+					['programs' => $webPrograms],
+				]
+			)
+		]);
+	}
 }