Support isAOf

Author: herndlm

README.md

@@ -46,6 +46,7 @@ This extension specifies types of values passed to:
 * `Assert::isCountable`
 * `Assert::isInstanceOf`
 * `Assert::notInstanceOf`
+* `Assert::isAOf`
 * `Assert::subclassOf`
 * `Assert::true`
 * `Assert::false`

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -38,6 +38,7 @@
 use PHPStan\Type\MixedType;
 use PHPStan\Type\ObjectType;
 use PHPStan\Type\StaticMethodTypeSpecifyingExtension;
+use PHPStan\Type\StringType;
 use PHPStan\Type\Type;
 use PHPStan\Type\TypeCombinator;
 use PHPStan\Type\TypeUtils;
@@ -382,6 +383,15 @@ private static function getExpressionResolvers(): array
 						)
 					);
 				},
+				'isAOf' => static function (Scope $scope, Arg $expr, Arg $class): Expr {
+					$exprType = $scope->getType($expr->value);
+					$allowString = (new StringType())->isSuperTypeOf($exprType)->yes();
+
+					return new FuncCall(
+						new Name('is_a'),
+						[$expr, $class, new Arg(new ConstFetch(new Name($allowString ? 'true' : 'false')))]
+					);
+				},
 				'implementsInterface' => static function (Scope $scope, Arg $expr, Arg $class): ?Expr {
 					$classType = $scope->getType($class->value);
 					if (!$classType instanceof ConstantStringType) {

tests/Type/WebMozartAssert/data/type.php

@@ -191,6 +191,24 @@ public function notInstanceOf($a, $b, $c): void
 		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\Bar|PHPStan\Type\WebMozartAssert\Foo', $c);
 	}
 
+	public function isAOf($a, $b, string $c): void
+	{
+		Assert::isAOf($a, stdClass::class);
+		\PHPStan\Testing\assertType('stdClass', $a);
+
+		Assert::isAOf(Foo::class, stdClass::class);
+		\PHPStan\Testing\assertType('*NEVER*', Foo::class);
+
+		Assert::isAOf(Bar::class, stdClass::class);
+		\PHPStan\Testing\assertType('\'PHPStan\\\Type\\\WebMozartAssert\\\Bar\'', Bar::class);
+
+		Assert::nullOrIsAOf($b, stdClass::class);
+		\PHPStan\Testing\assertType('stdClass|null', $b);
+
+		Assert::isAOf($c, stdClass::class);
+		\PHPStan\Testing\assertType('class-string<stdClass>', $c);
+	}
+
 	public function isArrayAccessible($a, $b): void
 	{
 		Assert::isArrayAccessible($a);
@@ -200,3 +218,7 @@ public function isArrayAccessible($a, $b): void
 		\PHPStan\Testing\assertType('array|ArrayAccess|null', $b);
 	}
 }
+
+class Foo {}
+
+class Bar extends stdClass {}