phpstan-latte/src/Compiler/NodeVisitor/TransformDynamicFormControlsToDynamicStringNodeVisitor.php at main · OpenVK/phpstan-latte · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
<?php

declare(strict_types=1);

namespace Efabrica\PHPStanLatte\Compiler\NodeVisitor;

use Efabrica\PHPStanLatte\Compiler\NodeVisitor\Behavior\ExprTypeNodeVisitorBehavior;
use Efabrica\PHPStanLatte\Compiler\NodeVisitor\Behavior\ExprTypeNodeVisitorInterface;
use Efabrica\PHPStanLatte\Resolver\NameResolver\NameResolver;
use PhpParser\Node;
use PhpParser\Node\Expr\ArrayDimFetch;
use PhpParser\Node\Expr\PropertyFetch;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Scalar\String_;
use PhpParser\NodeVisitorAbstract;

/**
 * changed output from:
 * <code>
 * $form[$dynamicInput];
 * </code>
 *
 * to:
 * <code>
 * $form["$dynamicInput"];
 * </code>
 *
 * if feature flag is turned on and type of $dynamicInput cannot be resolved as constant string / int
 */
final class TransformDynamicFormControlsToDynamicStringNodeVisitor extends NodeVisitorAbstract implements ExprTypeNodeVisitorInterface
{
    use ExprTypeNodeVisitorBehavior;

    private bool $featureTransformDynamicFormControlNamesToString;

    private NameResolver $nameResolver;

    public function __construct(
        bool $featureTransformDynamicFormControlNamesToString,
        NameResolver $nameResolver
    ) {
        $this->featureTransformDynamicFormControlNamesToString = $featureTransformDynamicFormControlNamesToString;
        $this->nameResolver = $nameResolver;
    }

    public function enterNode(Node $node): ?Node
    {
        if ($this->featureTransformDynamicFormControlNamesToString === false) {
            return null;
        }

        if (!$node instanceof ArrayDimFetch) {
            return null;
        }

        if ($this->nameResolver->resolve($node->var) !== 'form') {
            return null;
        }

        if ($node->dim === null) {
            return null;
        }

        $type = $this->getType($node->dim);
        if ($type !== null && $type->getConstantScalarValues() !== []) {
            return null;
        }

        if ($node->dim instanceof Variable) {
            $node->dim = new String_('$' . $this->nameResolver->resolve($node->dim));
        }

        if ($node->dim instanceof PropertyFetch) {
            $varName = $this->nameResolver->resolve($node->dim->var);
            $propertyName = $this->nameResolver->resolve($node->dim->name);
            if ($varName === null || $propertyName === null) {
                return null;
            }
            $node->dim = new String_('$' . $varName . '->' . $propertyName);
        }

        return $node;
    }
}