Introduce ForbiddenClassNameExtension for append additional forbidden class prefixes

Author: kamil-zacek

src/Classes/ForbiddenClassNameExtension.php

@@ -0,0 +1,32 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Classes;
+
+/**
+ * This is the extension interface to implement if you want to dynamically
+ * add forbidden class prefixes to the ClassForbiddenNameCheck rule.
+ *
+ * The idea is that you want to report usages of classes that you're not supposed to use in application.
+ * For example: Generated Doctrine proxies from their configured namespace.
+ *
+ * To register it in the configuration file use the `phpstan.forbiddenClassNamesExtension` service tag:
+ *
+ * ```
+ * services:
+ * 	-
+ *		class: App\PHPStan\MyExtension
+ *		tags:
+ *			- phpstan.forbiddenClassNamesExtension
+ * ```
+ *
+ * @api
+ */
+interface ForbiddenClassNameExtension
+{
+
+	public const EXTENSION_TAG = 'phpstan.forbiddenClassNamesExtension';
+
+	/** @return array<string, string> */
+	public function getClassPrefixes(): array;
+
+}

src/Rules/ClassForbiddenNameCheck.php

@@ -2,8 +2,13 @@
 
 namespace PHPStan\Rules;
 
+use PHPStan\Classes\ForbiddenClassNameExtension;
+use PHPStan\DependencyInjection\Container;
+use function array_map;
+use function array_merge;
 use function sprintf;
 use function str_starts_with;
+use function strlen;
 use function strpos;
 use function substr;
 
@@ -16,22 +21,45 @@ class ClassForbiddenNameCheck
 		'PHP-Scoper' => '_PhpScoper',
 	];
 
+	public function __construct(private Container $container)
+	{
+	}
+
 	/**
 	 * @param ClassNameNodePair[] $pairs
 	 * @return RuleError[]
 	 */
 	public function checkClassNames(array $pairs): array
 	{
+		$extensions = $this->container->getServicesByTag(ForbiddenClassNameExtension::EXTENSION_TAG);
+
+		$classPrefixes = array_merge(
+			self::INTERNAL_CLASS_PREFIXES,
+			...array_map(
+				static fn (ForbiddenClassNameExtension $extension): array => $extension->getClassPrefixes(),
+				$extensions,
+			),
+		);
+
 		$errors = [];
 		foreach ($pairs as $pair) {
 			$className = $pair->getClassName();
 
 			$projectName = null;
-			foreach (self::INTERNAL_CLASS_PREFIXES as $project => $prefix) {
-				if (str_starts_with($className, $prefix)) {
-					$projectName = $project;
-					break;
+			$withoutPrefixClassName = null;
+			foreach ($classPrefixes as $project => $prefix) {
+				if (!str_starts_with($className, $prefix)) {
+					continue;
 				}
+
+				$projectName = $project;
+				$withoutPrefixClassName = substr($className, strlen($prefix));
+
+				if (strpos($withoutPrefixClassName, '\\') === false) {
+					continue;
+				}
+
+				$withoutPrefixClassName = substr($withoutPrefixClassName, strpos($withoutPrefixClassName, '\\'));
 			}
 
 			if ($projectName === null) {
@@ -44,10 +72,10 @@ public function checkClassNames(array $pairs): array
 				$className,
 			))->line($pair->getNode()->getLine())->nonIgnorable();
 
-			if (strpos($className, '\\') !== false) {
+			if ($withoutPrefixClassName !== null) {
 				$error->tip(sprintf(
 					'This is most likely unintentional. Did you mean to type %s?',
-					substr($className, strpos($className, '\\')),
+					$withoutPrefixClassName,
 				));
 			}
 

tests/PHPStan/Rules/Classes/ClassAttributesRuleTest.php

@@ -42,7 +42,7 @@ protected function getRule(): Rule
 				),
 				new ClassNameCheck(
 					new ClassCaseSensitivityCheck($reflectionProvider, false),
-					new ClassForbiddenNameCheck(),
+					new ClassForbiddenNameCheck(self::getContainer()),
 				),
 				true,
 			),

tests/PHPStan/Rules/Classes/ClassConstantAttributesRuleTest.php

@@ -41,7 +41,7 @@ protected function getRule(): Rule
 				),
 				new ClassNameCheck(
 					new ClassCaseSensitivityCheck($reflectionProvider, false),
-					new ClassForbiddenNameCheck(),
+					new ClassForbiddenNameCheck(self::getContainer()),
 				),
 				true,
 			),

tests/PHPStan/Rules/Classes/ClassConstantRuleTest.php

@@ -27,7 +27,7 @@ protected function getRule(): Rule
 			new RuleLevelHelper($reflectionProvider, true, false, true, false, false, true, false),
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			new PhpVersion($this->phpVersion),
 		);

tests/PHPStan/Rules/Classes/ExistingClassInClassExtendsRuleTest.php

@@ -21,7 +21,7 @@ protected function getRule(): Rule
 		return new ExistingClassInClassExtendsRule(
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			$reflectionProvider,
 		);

tests/PHPStan/Rules/Classes/ExistingClassInInstanceOfRuleTest.php

@@ -21,7 +21,7 @@ protected function getRule(): Rule
 			$reflectionProvider,
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			true,
 		);

tests/PHPStan/Rules/Classes/ExistingClassInTraitUseRuleTest.php

@@ -21,7 +21,7 @@ protected function getRule(): Rule
 		return new ExistingClassInTraitUseRule(
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			$reflectionProvider,
 		);

tests/PHPStan/Rules/Classes/ExistingClassesInClassImplementsRuleTest.php

@@ -21,7 +21,7 @@ protected function getRule(): Rule
 		return new ExistingClassesInClassImplementsRule(
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			$reflectionProvider,
 		);

tests/PHPStan/Rules/Classes/ExistingClassesInEnumImplementsRuleTest.php

@@ -22,7 +22,7 @@ protected function getRule(): Rule
 		return new ExistingClassesInEnumImplementsRule(
 			new ClassNameCheck(
 				new ClassCaseSensitivityCheck($reflectionProvider, true),
-				new ClassForbiddenNameCheck(),
+				new ClassForbiddenNameCheck(self::getContainer()),
 			),
 			$reflectionProvider,
 		);