<?php declare(strict_types = 1);

namespace PHPStan\Parser;

use Override;

use PhpParser\Node;

use PhpParser\NodeVisitorAbstract;

use PHPStan\DependencyInjection\AutowiredService;

use function in_array;

#[AutowiredService]

final class ArrayFindArgVisitor extends NodeVisitorAbstract

{

public const ATTRIBUTE_NAME = 'isArrayFindArg';

#[Override]

public function enterNode(Node $node): ?Node

{

if ($node instanceof Node\Expr\FuncCall && $node->name instanceof Node\Name && !$node->isFirstClassCallable()) {

$functionName = $node->name->toLowerString();

if (in_array($functionName, ['array_all', 'array_any', 'array_find', 'array_find_key'], true)) {

$args = $node->getArgs();

if (isset($args[0])) {

$args[0]->setAttribute(self::ATTRIBUTE_NAME, true);

}

}

}

return null;

}

}