Support isAnyOf

Author: herndlm

README.md

@@ -48,6 +48,7 @@ This extension specifies types of values passed to:
 * `Assert::isInstanceOfAny`
 * `Assert::notInstanceOf`
 * `Assert::isAOf`
+* `Assert::isAnyOf`
 * `Assert::subclassOf`
 * `Assert::true`
 * `Assert::false`

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -392,19 +392,7 @@ private static function getExpressionResolvers(): array
 					);
 				},
 				'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);
+					return self::buildAnyOfExpr($scope, $expr, $classes, self::$resolvers['isInstanceOf']);
 				},
 				'notInstanceOf' => static function (Scope $scope, Arg $expr, Arg $class): ?Expr {
 					$expr = self::$resolvers['isInstanceOf']($scope, $expr, $class);
@@ -423,6 +411,9 @@ static function (?ArrayItem $class) use ($scope, $expr) {
 						[$expr, $class, new Arg(new ConstFetch(new Name($allowString ? 'true' : 'false')))]
 					);
 				},
+				'isAnyOf' => static function (Scope $scope, Arg $value, Arg $classes): ?Expr {
+					return self::buildAnyOfExpr($scope, $value, $classes, self::$resolvers['isAOf']);
+				},
 				'implementsInterface' => static function (Scope $scope, Arg $expr, Arg $class): ?Expr {
 					$classType = $scope->getType($class->value);
 					if (!$classType instanceof ConstantStringType) {
@@ -818,4 +809,21 @@ static function (Expr $carry, Expr $item) use ($binaryOp) {
 		);
 	}
 
+	private static function buildAnyOfExpr(Scope $scope, Arg $value, Arg $items, callable $resolver): ?Expr
+	{
+		if (!$items->value instanceof Array_ || $items->value->items === null) {
+			return null;
+		}
+
+		$resolvers = array_map(
+			static function (?ArrayItem $item) use ($scope, $value, $resolver) {
+				return $item !== null ? $resolver($scope, $value, new Arg($item->value)) : null;
+			},
+			$items->value->items
+		);
+		$resolvers = array_filter($resolvers);
+
+		return self::implodeExpr($resolvers, BooleanOr::class);
+	}
+
 }

tests/Type/WebMozartAssert/data/type.php

@@ -2,6 +2,7 @@
 
 namespace PHPStan\Type\WebMozartAssert;
 
+use DateTimeImmutable;
 use stdClass;
 use Webmozart\Assert\Assert;
 
@@ -233,6 +234,21 @@ public function isAOf($a, $b, string $c): void
 		\PHPStan\Testing\assertType('class-string<stdClass>', $c);
 	}
 
+	public function isAnyOf($a, $b): void
+	{
+		Assert::isAnyOf($a, [DateTimeImmutable::class, stdClass::class]);
+		\PHPStan\Testing\assertTyp0e('DateTimeImmutable|stdClass', $a);
+
+		Assert::isAnyOf($b, []);
+		\PHPStan\Testing\assertType('mixed', $b);
+
+		Assert::isAnyOf(Foo::class, [stdClass::class, Bar::class]);
+		\PHPStan\Testing\assertType('*NEVER*', Foo::class);
+
+		Assert::isAnyOf(Bar::class, [stdClass::class, Foo::class]);
+		\PHPStan\Testing\assertType('\'PHPStan\\\Type\\\WebMozartAssert\\\Bar\'', Bar::class);
+	}
+
 	public function isArrayAccessible($a, $b): void
 	{
 		Assert::isArrayAccessible($a);