Rework

Author: VincentLanglet

src/Reflection/Php/SealedAllowedSubTypesClassReflectionExtension.php

@@ -0,0 +1,38 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Reflection\Php;
+
+use PHPStan\DependencyInjection\AutowiredService;
+use PHPStan\Reflection\AllowedSubTypesClassReflectionExtension;
+use PHPStan\Reflection\ClassReflection;
+use PHPStan\Type\Enum\EnumCaseObjectType;
+use PHPStan\Type\UnionType;
+
+use function array_keys;
+
+#[AutowiredService]
+final class SealedAllowedSubTypesClassReflectionExtension implements AllowedSubTypesClassReflectionExtension
+{
+
+	public function supports(ClassReflection $classReflection): bool
+	{
+		return count($classReflection->getSealedTags()) > 0;
+	}
+
+	public function getAllowedSubTypes(ClassReflection $classReflection): array
+	{
+		$types = [];
+
+		foreach ($classReflection->getSealedTags() as $sealedTag) {
+			$type = $sealedTag->getType();
+			if ($type instanceof UnionType) {
+				$types = $type->getTypes();
+			} else {
+				$types = [$type];
+			}
+		}
+
+		return $types;
+	}
+
+}

tests/PHPStan/Rules/Classes/AllowedSubTypesRuleTest.php

@@ -26,6 +26,24 @@ public function testRule(): void
 		]);
 	}
 
+	public function testSealed(): void
+	{
+		$this->analyse([__DIR__ . '/data/sealed.php'], [
+			[
+				'Class Sealed\BaseClass is sealed and only permits Sealed\BarClass|Sealed\FooClass as subtypes, Sealed\BazClass given.',
+				11,
+			],
+			[
+				'Interface Sealed\BaseInterface is sealed and only permits Sealed\BarClass2|Sealed\FooClass2 as subtypes, Sealed\BazClass2 given.',
+				19,
+			],
+			[
+				'Interface Sealed\BaseInterface2 is sealed and only permits Sealed\BarInterface|Sealed\FooInterface as subtypes, Sealed\BazInterface given.',
+				27,
+			],
+		]);
+	}
+
 	public static function getAdditionalConfigFiles(): array
 	{
 		return [