ConsecutiveIfsFactory.php

<?php

declare(strict_types=1);

namespace Rector\PHPUnit\NodeFactory;

use PhpParser\Node\Arg;
use PhpParser\Node\Expr\Array_;
use PhpParser\Node\Expr\ArrayDimFetch;
use PhpParser\Node\Expr\ArrowFunction;
use PhpParser\Node\Expr\BinaryOp\Identical;
use PhpParser\Node\Expr\MethodCall;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Identifier;
use PhpParser\Node\Scalar\Int_;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Expression;
use PhpParser\Node\Stmt\If_;
use Rector\Exception\NotImplementedYetException;
use Rector\NodeNameResolver\NodeNameResolver;
use Rector\PHPUnit\CodeQuality\NodeFactory\NestedClosureAssertFactory;
use Rector\PHPUnit\Enum\ConsecutiveVariable;

final readonly class ConsecutiveIfsFactory
{
    public function __construct(
        private NodeNameResolver $nodeNameResolver,
        private NestedClosureAssertFactory $nestedClosureAssertFactory
    ) {
    }

    /**
     * @return Stmt[]
     */
    public function createIfs(MethodCall $withConsecutiveMethodCall, MethodCall $numberOfInvocationsMethodCall): array
    {
        $ifs = [];
        $parametersVariable = new Variable(ConsecutiveVariable::PARAMETERS);

        foreach ($withConsecutiveMethodCall->getArgs() as $key => $withConsecutiveArg) {
            $ifStmts = [];

            if ($withConsecutiveArg->value instanceof Array_) {
                $array = $withConsecutiveArg->value;
                foreach ($array->items as $assertKey => $assertArrayItem) {
                    if (! $assertArrayItem->value instanceof MethodCall) {
                        $parametersDimFetch = new ArrayDimFetch(new Variable('parameters'), new Int_($assertKey));
                        $args = [new Arg($assertArrayItem->value), new Arg($parametersDimFetch)];
                        $ifStmts[] = new Expression(new MethodCall(new Variable('this'), 'assertSame', $args));
                        continue;
                    }

                    $assertMethodCall = $assertArrayItem->value;

                    if ($this->nodeNameResolver->isName($assertMethodCall->name, 'equalTo')) {
                        $ifStmts[] = $this->createAssertMethodCall($assertMethodCall, $parametersVariable, $assertKey);
                    } elseif ($this->nodeNameResolver->isName($assertMethodCall->name, 'callback')) {
                        $ifStmts = array_merge(
                            $ifStmts,
                            $this->nestedClosureAssertFactory->create($assertMethodCall, $assertKey)
                        );
                    } else {
                        $args = [
                            new Arg($assertMethodCall),
                            new Arg(new ArrayDimFetch(new Variable('parameters'), new Int_($assertKey))),
                        ];

                        $assertSameMethodCall = new MethodCall(new Variable('this'), new Identifier(
                            'assertSame'
                        ), $args);
                        $ifStmts[] = new Expression($assertSameMethodCall);
                    }
                }
            } elseif ($withConsecutiveArg->value instanceof MethodCall) {
                $methodCall = $withConsecutiveArg->value;
                if ($this->nodeNameResolver->isName($methodCall->name, 'callback')) {
                    // special callable case
                    $firstArg = $methodCall->getArgs()[0];
                    if ($firstArg->value instanceof ArrowFunction) {
                        $arrowFunction = $firstArg->value;
                        if ($arrowFunction->expr instanceof Identical) {
                            $identicalCompare = $arrowFunction->expr;

                            // @todo improve in time
                            if ($identicalCompare->left instanceof Variable) {
                                $parametersArrayDimFetch = new ArrayDimFetch(new Variable('parameters'), new Int_(
                                    0
                                ));

                                $assertSameMethodCall = new MethodCall(new Variable('this'), new Identifier(
                                    'assertSame'
                                ));
                                $assertSameMethodCall->args[] = new Arg($identicalCompare->right);
                                $assertSameMethodCall->args[] = new Arg($parametersArrayDimFetch);

                                return [new Expression($assertSameMethodCall)];
                            }
                        }
                    }
                }

                throw new NotImplementedYetException();
            }

            $ifs[] = new If_(new Identical($numberOfInvocationsMethodCall, new Int_($key + 1)), [
                'stmts' => $ifStmts,
            ]);
        }

        return $ifs;
    }

    private function createAssertMethodCall(
        MethodCall $assertMethodCall,
        Variable $parametersVariable,
        int $parameterPositionKey
    ): Expression {
        $assertMethodCall->name = new Identifier('assertEquals');
        $parametersArrayDimFetch = new ArrayDimFetch($parametersVariable, new Int_($parameterPositionKey));

        $assertMethodCall->args[] = new Arg($parametersArrayDimFetch);

        return new Expression($assertMethodCall);
    }
}