AddAssertArrayFromClassMethodDocblockRector.php

<?php

declare(strict_types=1);

namespace Rector\Assert\Rector\ClassMethod;

use PhpParser\Node;
use PhpParser\Node\Arg;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\FuncCall;
use PhpParser\Node\Expr\StaticCall;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Identifier;
use PhpParser\Node\Name;
use PhpParser\Node\Name\FullyQualified;
use PhpParser\Node\Stmt\ClassMethod;
use PhpParser\Node\Stmt\Expression;
use PhpParser\PrettyPrinter\Standard;
use PHPStan\Type\ArrayType;
use PHPStan\Type\BooleanType;
use PHPStan\Type\FloatType;
use PHPStan\Type\IntegerType;
use PHPStan\Type\StringType;
use PHPStan\Type\Type;
use Rector\Assert\Enum\AssertClassName;
use Rector\Assert\NodeAnalyzer\ExistingAssertStaticCallResolver;
use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfo;
use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfoFactory;
use Rector\Contract\Rector\ConfigurableRectorInterface;
use Rector\PHPStan\ScopeFetcher;
use Rector\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
use Webmozart\Assert\Assert;

/**
 * @extends AbstractRector<ClassMethod>
 * @implements ConfigurableRectorInterface<ClassMethod>
 * @experimental Check generic array key/value types in runtime with assert. Generics for impatient people.
 *
 * @see \Rector\Tests\Assert\Rector\ClassMethod\AddAssertArrayFromClassMethodDocblockRector\AddAssertArrayFromClassMethodDocblockRectorTest
 */
final class AddAssertArrayFromClassMethodDocblockRector extends AbstractRector implements ConfigurableRectorInterface
{
    private string $assertClass = AssertClassName::WEBMOZART;

    public function __construct(
        private readonly PhpDocInfoFactory $phpDocInfoFactory,
        private readonly ExistingAssertStaticCallResolver $existingAssertStaticCallResolver
    ) {
    }

    public function getRuleDefinition(): RuleDefinition
    {
        return new RuleDefinition(
            'Add key and value assert based on docblock @param type declarations (pick from "webmozart" or "beberlei" asserts)',
            [
                new ConfiguredCodeSample(
                    <<<'CODE_SAMPLE'
<?php

class SomeClass
{
    /**
     * @param int[] $items
     */
    public function run(array $items)
    {
    }
}

CODE_SAMPLE
                    ,
                    <<<'CODE_SAMPLE'
<?php

use Webmozart\Assert\Assert;

class SomeClass
{
    /**
     * @param int[] $items
     */
    public function run(array $items)
    {
        Assert::allInteger($items);
    }
}
CODE_SAMPLE
                    ,
                    [AssertClassName::WEBMOZART]
                ),

            ]
        );
    }

    public function getNodeTypes(): array
    {
        return [ClassMethod::class];
    }

    public function refactor(Node $node): ?ClassMethod
    {
        $scope = ScopeFetcher::fetch($node);
        if (! $scope->isInClass()) {
            return null;
        }

        if ($node->stmts === null || $node->isAbstract()) {
            return null;
        }

        $methodPhpDocInfo = $this->phpDocInfoFactory->createFromNode($node);
        if (! $methodPhpDocInfo instanceof PhpDocInfo) {
            return null;
        }

        $paramTagValueNodes = $methodPhpDocInfo->getParamTagValueNodes();
        if ($paramTagValueNodes === []) {
            return null;
        }

        $assertStaticCallStmts = [];

        foreach ($node->getParams() as $param) {
            if (! $param->type instanceof Identifier) {
                continue;
            }

            // handle arrays only
            if (! $this->isName($param->type, 'array')) {
                continue;
            }

            if (! $param->var instanceof Variable) {
                continue;
            }

            $paramName = $param->var->name;
            if (! is_string($paramName)) {
                continue;
            }

            $paramDocType = $methodPhpDocInfo->getParamType($paramName);
            if (! $paramDocType instanceof ArrayType) {
                continue;
            }

            $valueAssertMethod = $this->matchTypeToAssertMethod($paramDocType->getItemType());

            if (is_string($valueAssertMethod)) {
                $assertStaticCallStmts[] = $this->createAssertExpression($param->var, $valueAssertMethod);
            }

            $keyAssertMethod = $this->matchTypeToAssertMethod($paramDocType->getKeyType());

            if (is_string($keyAssertMethod)) {
                $arrayKeys = new FuncCall(new Name('array_keys'), [new Arg($param->var)]);
                $assertStaticCallStmts[] = $this->createAssertExpression($arrayKeys, $keyAssertMethod);
            }
        }

        // filter existing assert to avoid duplication
        if ($assertStaticCallStmts === []) {
            return null;
        }

        $existingAssertCallHashes = $this->existingAssertStaticCallResolver->resolve($node);

        $assertStaticCallStmts = $this->filterOutExistingStaticCall($assertStaticCallStmts, $existingAssertCallHashes);

        if ($assertStaticCallStmts === []) {
            return null;
        }

        $node->stmts = array_merge($assertStaticCallStmts, $node->stmts);

        return $node;
    }

    /**
     * @param array<string> $configuration
     */
    public function configure(array $configuration): void
    {
        if ($configuration === []) {
            // default
            return;
        }

        Assert::count($configuration, 1);
        Assert::inArray($configuration[0], [AssertClassName::BEBERLEI, AssertClassName::WEBMOZART]);

        $this->assertClass = $configuration[0];
    }

    private function createAssertExpression(Expr $expr, string $methodName): Expression
    {
        $assertFullyQualified = new FullyQualified($this->assertClass);

        $staticCall = new StaticCall($assertFullyQualified, $methodName, [new Arg($expr)]);

        return new Expression($staticCall);
    }

    /**
     * @param Expression[] $assertStaticCallStmts
     * @param string[] $existingAssertCallHashes
     * @return Expression[]
     */
    private function filterOutExistingStaticCall(array $assertStaticCallStmts, array $existingAssertCallHashes): array
    {
        $standard = new Standard();

        return array_filter($assertStaticCallStmts, function (Expression $assertStaticCallExpression) use (
            $standard,
            $existingAssertCallHashes
        ): bool {
            $currentStaticCallHash = $standard->prettyPrintExpr($assertStaticCallExpression->expr);

            return ! in_array($currentStaticCallHash, $existingAssertCallHashes, true);
        });
    }

    private function matchTypeToAssertMethod(Type $type): ?string
    {
        if ($type instanceof IntegerType) {
            return 'allInteger';
        }

        if ($type instanceof StringType) {
            return 'allString';
        }

        if ($type instanceof FloatType) {
            return 'allFloat';
        }

        if ($type instanceof BooleanType) {
            return 'allBoolean';
        }

        return null;
    }
}