Support isInstanceOfAny

Author: herndlm

README.md

@@ -45,6 +45,7 @@ This extension specifies types of values passed to:
 * `Assert::isIterable`
 * `Assert::isCountable`
 * `Assert::isInstanceOf`
+* `Assert::isInstanceOfAny`
 * `Assert::notInstanceOf`
 * `Assert::isAOf`
 * `Assert::subclassOf`

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -7,6 +7,9 @@
 use Countable;
 use PhpParser\Node\Arg;
 use PhpParser\Node\Expr;
+use PhpParser\Node\Expr\Array_;
+use PhpParser\Node\Expr\ArrayItem;
+use PhpParser\Node\Expr\BinaryOp;
 use PhpParser\Node\Expr\BinaryOp\BooleanAnd;
 use PhpParser\Node\Expr\BinaryOp\BooleanOr;
 use PhpParser\Node\Expr\BinaryOp\Greater;
@@ -45,7 +48,11 @@
 use PHPStan\Type\TypeWithClassName;
 use ReflectionObject;
 use Traversable;
+use function array_filter;
 use function array_key_exists;
+use function array_map;
+use function array_reduce;
+use function array_shift;
 use function count;
 use function key;
 use function lcfirst;
@@ -366,6 +373,21 @@ private static function getExpressionResolvers(): array
 						$className
 					);
 				},
+				'isInstanceOfAny' => static function (Scope $scope, Arg $expr, Arg $classes): ?Expr {
+					if (!$classes->value instanceof Array_ || $classes->value->items === null) {
+						return null;
+					}
+
+					$resolvers = array_map(
+						static function (?ArrayItem $class) use ($scope, $expr) {
+							return $class !== null ? self::$resolvers['isInstanceOf']($scope, $expr, new Arg($class->value)) : null;
+						},
+						$classes->value->items
+					);
+					$resolvers = array_filter($resolvers);
+
+					return self::implodeExpr($resolvers, BooleanOr::class);
+				},
 				'notInstanceOf' => static function (Scope $scope, Arg $expr, Arg $class): ?Expr {
 					$classType = $scope->getType($class->value);
 					if ($classType instanceof ConstantStringType) {
@@ -779,4 +801,24 @@ private function arrayOrIterable(
 		);
 	}
 
+	/**
+	 * @param Expr[] $expressions
+	 * @param class-string<BinaryOp> $binaryOp
+	 */
+	private static function implodeExpr(array $expressions, string $binaryOp): ?Expr
+	{
+		$firstExpression = array_shift($expressions);
+		if ($firstExpression === null) {
+			return null;
+		}
+
+		return array_reduce(
+			$expressions,
+			static function (Expr $carry, Expr $item) use ($binaryOp) {
+				return new $binaryOp($carry, $item);
+			},
+			$firstExpression
+		);
+	}
+
 }

tests/Type/WebMozartAssert/data/type.php

@@ -174,6 +174,30 @@ public function isInstanceOf($a, $b, $c, $d): void
 		\PHPStan\Testing\assertType('mixed', $d);
 	}
 
+	public function isInstanceOfAny($a, $b, $c, $d, $e, $f, $g): void
+	{
+		Assert::isInstanceOfAny($a, [self::class, new stdClass()]);
+		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\TypeTest|stdClass', $a);
+
+		Assert::isInstanceOfAny($b, [new stdClass()]);
+		\PHPStan\Testing\assertType('stdClass', $b);
+
+		Assert::isInstanceOfAny($c, []);
+		\PHPStan\Testing\assertType('mixed', $c);
+
+		Assert::isInstanceOfAny($d, 'foo');
+		\PHPStan\Testing\assertType('mixed', $d);
+
+		Assert::isInstanceOfAny($e, [17]);
+		\PHPStan\Testing\assertType('mixed', $e);
+
+		Assert::isInstanceOfAny($f, [17, self::class]);
+		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\TypeTest', $f);
+
+		Assert::nullOrIsInstanceOfAny($g, [self::class, new stdClass()]);
+		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\TypeTest|stdClass|null', $g);
+	}
+
 	/**
 	 * @param Foo|Bar $a
 	 * @param Foo|stdClass $b