Add support for statement in conditional tags

Author: VincentLanglet

src/DependencyInjection/ConditionalTagsExtension.php

@@ -4,6 +4,7 @@
 
 use Nette;
 use Nette\DI\CompilerExtension;
+use Nette\DI\Definitions\Statement;
 use Nette\Schema\Expect;
 use Override;
 use PHPStan\ShouldNotHappenException;
@@ -23,7 +24,11 @@ public function getConfigSchema(): Nette\Schema\Schema
 		$tags = array_values(ValidateServiceTagsExtension::INTERFACE_TAG_MAPPING);
 
 		return Expect::arrayOf(Expect::structure(
-			array_fill_keys($tags, Expect::anyOf(Expect::bool(), Expect::listOf(Expect::bool()))),
+			array_fill_keys($tags, Expect::anyOf(
+				Expect::bool(),
+				Expect::type(Statement::class),
+				Expect::listOf(Expect::anyOf(Expect::bool(), Expect::type(Statement::class))),
+			)),
 		)->min(1));
 	}
 
@@ -41,16 +46,29 @@ public function beforeCompile(): void
 			}
 			foreach ($services as $service) {
 				foreach ($tags as $tag => $parameter) {
-					if (is_array($parameter)) {
-						$parameter = array_reduce($parameter, static fn ($carry, $item) => $carry && (bool) $item, true);
-					}
-					if ((bool) $parameter) {
+					$parameter = is_array($parameter)
+						? array_reduce($parameter, fn ($carry, $item) => $carry && $this->resolveValue($item), true)
+						: $this->resolveValue($parameter);
+
+					if ($parameter) {
 						$service->addTag($tag);
-						continue;
 					}
 				}
 			}
 		}
 	}
 
+	public function resolveValue(mixed $parameter): bool
+	{
+		if (!$parameter instanceof Statement) {
+			return (bool) $parameter;
+		}
+
+		if ('not' === $parameter->getEntity()) {
+			return ! (bool) $parameter->arguments[0];
+		}
+
+		throw new \InvalidArgumentException('Unsupported Statement.');
+	}
+
 }

tests/PHPStan/DependencyInjection/ConditionalTagsExtensionTest.php

@@ -16,10 +16,14 @@ public function testConditionalTags(): void
 		$enabledServices = array_map(static fn ($service) => get_class($service), $enabledServices);
 		$this->assertNotContains(TestedConditionalServiceDisabled::class, $enabledServices);
 		$this->assertContains(TestedConditionalServiceEnabled::class, $enabledServices);
+		$this->assertContains(TestedConditionalServiceNotDisabled::class, $enabledServices);
+		$this->assertNotContains(TestedConditionalServiceNotEnabled::class, $enabledServices);
 		$this->assertNotContains(TestedConditionalServiceDisabledDisabled::class, $enabledServices);
 		$this->assertNotContains(TestedConditionalServiceDisabledEnabled::class, $enabledServices);
 		$this->assertNotContains(TestedConditionalServiceEnabledDisabled::class, $enabledServices);
 		$this->assertContains(TestedConditionalServiceEnabledEnabled::class, $enabledServices);
+		$this->assertContains(TestedConditionalServiceEnabledNotDisabled::class, $enabledServices);
+		$this->assertNotContains(TestedConditionalServiceEnabledNotEnabled::class, $enabledServices);
 	}
 
 	/**

tests/PHPStan/DependencyInjection/TestedConditionalServiceEnabledNotDisabled.php

@@ -0,0 +1,25 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DependencyInjection;
+
+use PhpParser\Node;
+use PHPStan\Analyser\Scope;
+use PHPStan\Rules\Rule;
+
+/**
+ * @implements Rule<Node>
+ */
+class TestedConditionalServiceEnabledNotDisabled implements Rule
+{
+
+	public function getNodeType(): string
+	{
+		return Node::class;
+	}
+
+	public function processNode(Node $node, Scope $scope): array
+	{
+		return [];
+	}
+
+}

tests/PHPStan/DependencyInjection/TestedConditionalServiceEnabledNotEnabled.php

@@ -0,0 +1,25 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DependencyInjection;
+
+use PhpParser\Node;
+use PHPStan\Analyser\Scope;
+use PHPStan\Rules\Rule;
+
+/**
+ * @implements Rule<Node>
+ */
+class TestedConditionalServiceEnabledNotEnabled implements Rule
+{
+
+	public function getNodeType(): string
+	{
+		return Node::class;
+	}
+
+	public function processNode(Node $node, Scope $scope): array
+	{
+		return [];
+	}
+
+}

tests/PHPStan/DependencyInjection/TestedConditionalServiceNotDisabled.php

@@ -0,0 +1,25 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DependencyInjection;
+
+use PhpParser\Node;
+use PHPStan\Analyser\Scope;
+use PHPStan\Rules\Rule;
+
+/**
+ * @implements Rule<Node>
+ */
+class TestedConditionalServiceNotDisabled implements Rule
+{
+
+	public function getNodeType(): string
+	{
+		return Node::class;
+	}
+
+	public function processNode(Node $node, Scope $scope): array
+	{
+		return [];
+	}
+
+}

tests/PHPStan/DependencyInjection/TestedConditionalServiceNotEnabled.php

@@ -0,0 +1,25 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\DependencyInjection;
+
+use PhpParser\Node;
+use PHPStan\Analyser\Scope;
+use PHPStan\Rules\Rule;
+
+/**
+ * @implements Rule<Node>
+ */
+class TestedConditionalServiceNotEnabled implements Rule
+{
+
+	public function getNodeType(): string
+	{
+		return Node::class;
+	}
+
+	public function processNode(Node $node, Scope $scope): array
+	{
+		return [];
+	}
+
+}

tests/PHPStan/DependencyInjection/conditionalTags.neon

@@ -11,6 +11,10 @@ conditionalTags:
 		phpstan.rules.rule: %disabled%
 	PHPStan\DependencyInjection\TestedConditionalServiceEnabled:
 		phpstan.rules.rule: %enabled%
+	PHPStan\DependencyInjection\TestedConditionalServiceNotDisabled:
+		phpstan.rules.rule: not(%disabled%)
+	PHPStan\DependencyInjection\TestedConditionalServiceNotEnabled:
+		phpstan.rules.rule: not(%enabled%)
 	PHPStan\DependencyInjection\TestedConditionalServiceDisabledDisabled:
 		phpstan.rules.rule: [%disabled%, %disabled%]
 	PHPStan\DependencyInjection\TestedConditionalServiceDisabledEnabled:
@@ -19,11 +23,19 @@ conditionalTags:
 		phpstan.rules.rule: [%enabled%, %disabled%]
 	PHPStan\DependencyInjection\TestedConditionalServiceEnabledEnabled:
 		phpstan.rules.rule: [%enabled%, %enabled%]
+	PHPStan\DependencyInjection\TestedConditionalServiceEnabledNotEnabled:
+		phpstan.rules.rule: [%enabled%, not(%enabled%)]
+	PHPStan\DependencyInjection\TestedConditionalServiceEnabledNotDisabled:
+		phpstan.rules.rule: [%enabled%, not(%disabled%)]
 
 services:
 	- PHPStan\DependencyInjection\TestedConditionalServiceDisabled
 	- PHPStan\DependencyInjection\TestedConditionalServiceEnabled
+	- PHPStan\DependencyInjection\TestedConditionalServiceNotDisabled
+	- PHPStan\DependencyInjection\TestedConditionalServiceNotEnabled
 	- PHPStan\DependencyInjection\TestedConditionalServiceDisabledDisabled
 	- PHPStan\DependencyInjection\TestedConditionalServiceDisabledEnabled
 	- PHPStan\DependencyInjection\TestedConditionalServiceEnabledDisabled
 	- PHPStan\DependencyInjection\TestedConditionalServiceEnabledEnabled
+	- PHPStan\DependencyInjection\TestedConditionalServiceEnabledNotEnabled
+	- PHPStan\DependencyInjection\TestedConditionalServiceEnabledNotDisabled