<?php declare(strict_types = 1);

namespace PHPStan\Parser;

use Override;
use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;
use PHPStan\DependencyInjection\AutowiredService;
use function array_slice;
use function count;

#[AutowiredService]
final class ArrayMapArgVisitor extends NodeVisitorAbstract
{

	public const ATTRIBUTE_NAME = 'arrayMapArgs';

	#[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 ($functionName === 'array_map') {
				$args = $node->getArgs();
				$arrayArgs = [];
				foreach ($args as $i => $arg) {
					if ($arg->name === null && $i === 0) {
						continue;
					}
					if ($arg->name !== null && $arg->name->toString() === 'callback') {
						continue;
					}

					$arrayArgs[] = $arg;
				}
				if (isset($args[0])) {
					$slicedArgs = array_slice($args, 1);
					if (count($slicedArgs) > 0) {
						$args[0]->value->setAttribute(self::ATTRIBUTE_NAME, $arrayArgs);
					}
				}
			}
		}
		return null;
	}

}