Support validArrayKey

Author: herndlm

README.md

@@ -55,6 +55,7 @@ This extension specifies types of values passed to:
 * `Assert::implementsInterface`
 * `Assert::classExists`
 * `Assert::interfaceExists`
+* `Assert::validArrayKey`
 * `Assert::minCount`
 * `Assert::inArray`
 * `Assert::oneOf`

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -320,6 +320,18 @@ private static function getExpressionResolvers(): array
 						[$key, $array]
 					);
 				},
+				'validArrayKey' => function (Scope $scope, Arg $value): \PhpParser\Node\Expr {
+					return new \PhpParser\Node\Expr\BinaryOp\BooleanOr(
+						new \PhpParser\Node\Expr\FuncCall(
+							new \PhpParser\Node\Name('is_int'),
+							[$value]
+						),
+						new \PhpParser\Node\Expr\FuncCall(
+							new \PhpParser\Node\Name('is_string'),
+							[$value]
+						)
+					);
+				},
 				'true' => function (Scope $scope, Arg $expr): \PhpParser\Node\Expr {
 					return new \PhpParser\Node\Expr\BinaryOp\Identical(
 						$expr->value,

tests/Type/WebMozartAssert/AssertTypeSpecifyingExtensionTest.php

@@ -12,6 +12,7 @@ class AssertTypeSpecifyingExtensionTest extends TypeInferenceTestCase
 	 */
 	public function dataFileAsserts(): iterable
 	{
+		yield from $this->gatherAssertTypes(__DIR__ . '/data/array.php');
 		yield from $this->gatherAssertTypes(__DIR__ . '/data/data.php');
 	}
 

tests/Type/WebMozartAssert/data/array.php

@@ -0,0 +1,20 @@
+<?php
+
+declare(strict_types=1);
+
+use Webmozart\Assert\Assert;
+
+class Foo
+{
+	/**
+	 * @param mixed $a
+	 */
+	public function validArrayKey($a, bool $b): void
+	{
+		Assert::validArrayKey($a);
+		\PHPStan\Testing\assertType('int|string', $a);
+
+		Assert::validArrayKey($b);
+		\PHPStan\Testing\assertType('*NEVER*', $b);
+	}
+}