Child constructors should be disallowed when parent constructor or an…

… interface is disallowed Close #146
spaze · Dec 26, 2022 · d55780c · d55780c
1 parent 53c2903
commit d55780c
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 6 deletions.
diff --git a/src/Calls/NewCalls.php b/src/Calls/NewCalls.php
@@ -11,7 +11,7 @@
 use PHPStan\Rules\Rule;
 use PHPStan\Rules\RuleError;
 use PHPStan\ShouldNotHappenException;
-use PHPStan\Type\ConstantScalarType;
+use PHPStan\Type\Constant\ConstantStringType;
 use Spaze\PHPStan\Rules\Disallowed\DisallowedCall;
 use Spaze\PHPStan\Rules\Disallowed\DisallowedCallFactory;
 use Spaze\PHPStan\Rules\Disallowed\DisallowedHelper;
@@ -24,6 +24,7 @@
  */
 class NewCalls implements Rule
 {
+	private const CONSTRUCT = '::__construct';
 
 	/** @var DisallowedHelper */
 	private $disallowedHelper;
@@ -61,13 +62,41 @@ public function getNodeType(): string
 	public function processNode(Node $node, Scope $scope): array
 	{
 		if ($node->class instanceof Name) {
-			$name = "{$node->class}::__construct";
-		} elseif ($node->class instanceof Expr && $scope->getType($node->class) instanceof ConstantScalarType) {
-			$name = $scope->getType($node->class)->getValue() . '::__construct';
-		} else {
+			$className = $node->class;
+		} elseif ($node->class instanceof Expr) {
+			$type = $scope->getType($node->class);
+			if ($type instanceof ConstantStringType) {
+				$className = new Name($type->getValue());
+			}
+		}
+		if (!isset($className)) {
 			return [];
 		}
-		return $this->disallowedHelper->getDisallowedMessage($node, $scope, $name, $name, $this->disallowedCalls);
+
+		$type = $scope->resolveTypeByName($className);
+		$names = [
+			$type->getClassName(),
+		];
+		$reflection = $type->getClassReflection();
+		if ($reflection) {
+			foreach ($reflection->getParents() as $parent) {
+				$names[] = $parent->getName();
+			}
+			foreach ($reflection->getInterfaces() as $interface) {
+				$names[] = $interface->getName();
+			}
+		}
+
+		$errors = [];
+		foreach ($names as $name) {
+			$name .= self::CONSTRUCT;
+			$errors = array_merge(
+				$errors,
+				$this->disallowedHelper->getDisallowedMessage($node, $scope, $name, $type->getClassName() . self::CONSTRUCT, $this->disallowedCalls)
+			);
+		}
+
+		return $errors;
 	}
 
 }
diff --git a/tests/Calls/NewCallsTest.php b/tests/Calls/NewCallsTest.php
@@ -35,6 +35,14 @@ protected function getRule(): Rule
 						'../src/*-allow/*.*',
 					],
 				],
+				[
+					'method' => 'Inheritance\Base::__construct()',
+					'message' => 'all your base are belong to us',
+					'allowIn' => [
+						'../src/disallowed-allowed/*.php',
+						'../src/*-allow/*.*',
+					],
+				],
 				[
 					'function' => 'DateTime::__construct()',
 					'message' => 'no future',
@@ -55,6 +63,10 @@ public function testRule(): void
 	{
 		// Based on the configuration above, in this file:
 		$this->analyse([__DIR__ . '/../src/disallowed/methodCalls.php'], [
+			[
+				'Calling Inheritance\Base::__construct() (as Inheritance\Sub::__construct()) is forbidden, all your base are belong to us',
+				19,
+			],
 			[
 				'Calling Constructor\ClassWithConstructor::__construct() is forbidden, class ClassWithConstructor should not be created',
 				32,