use controlflow to collect variables

Haehnchen · Apr 27, 2024 · 6fbe368 · 6fbe368
1 parent 8372a39
commit 6fbe368
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 8 deletions.
diff --git a/src/main/java/fr/adrienbrault/idea/symfony2plugin/security/utils/VoterUtil.java b/src/main/java/fr/adrienbrault/idea/symfony2plugin/security/utils/VoterUtil.java
@@ -173,7 +173,7 @@ private static void visitAttributeForeach(@NotNull Method method, @NotNull Consu
             return;
         }
 
-        for (Variable variable : PhpElementsUtil.getVariablesInScope(method, parameters[2])) {
+        for (Variable variable : PhpElementsUtil.getVariablesInScopeByName(method, parameters[2].getName())) {
             // foreach ($attributes as $attribute)
             PsiElement psiElement = PsiTreeUtil.nextVisibleLeaf(variable);
             if (psiElement != null && psiElement.getNode().getElementType() == PhpTokenTypes.kwAS) {
@@ -182,9 +182,8 @@ private static void visitAttributeForeach(@NotNull Method method, @NotNull Consu
                     continue;
                 }
 
-                PhpPsiElement variableDecl = variable.getNextPsiSibling();
-                if (variableDecl instanceof Variable) {
-                    for (Variable variable1 : PhpElementsUtil.getVariablesInScope(parent, (Variable) variableDecl)) {
+                if (variable.getNextPsiSibling() instanceof Variable variableDecl) {
+                    for (Variable variable1 : PhpElementsUtil.getVariablesInScope(parent, variableDecl)) {
                         visitVariable(variable1, consumer);
                     }
                 }
@@ -213,7 +212,7 @@ private static void visitAttribute(@NotNull Method method, @NotNull Consumer<Pai
             return;
         }
 
-        for (Variable variable : PhpElementsUtil.getVariablesInScope(method, parameters[0])) {
+        for (Variable variable : PhpElementsUtil.getVariablesInScopeByName(method, parameters[0].getName())) {
             visitVariable(variable, consumer);
         }
     }

diff --git a/...ava/fr/adrienbrault/idea/symfony2plugin/templating/util/PhpMethodVariableResolveUtil.java b/...ava/fr/adrienbrault/idea/symfony2plugin/templating/util/PhpMethodVariableResolveUtil.java
@@ -168,7 +168,7 @@ private static List<PsiElement> collectPossibleTemplateArrays(@NotNull Function
     private static Map<String, PsiVariable> collectOnVariableReferences(@NotNull Function function, @NotNull Variable variable) {
         Map<String, PsiVariable> collectedTypes = new HashMap<>();
 
-        for (Variable scopeVar : PhpElementsUtil.getVariablesInScope(function, variable)) {
+        for (Variable scopeVar : PhpElementsUtil.getVariablesInScopeByName(function, variable.getName())) {
             PsiElement parent = scopeVar.getParent();
             if (parent instanceof ArrayAccessExpression) {
                 // $template['variable'] = $foo

diff --git a/src/main/java/fr/adrienbrault/idea/symfony2plugin/util/PhpElementsUtil.java b/src/main/java/fr/adrienbrault/idea/symfony2plugin/util/PhpElementsUtil.java
@@ -1261,7 +1261,7 @@ public static Collection<Variable> getVariableReferencesInScope(@NotNull Variabl
 
         final List<Variable> variables = new ArrayList<>();
 
-        for (Variable variableRef : PhpElementsUtil.getVariablesInScope(function, variableDecl)) {
+        for (Variable variableRef : PhpElementsUtil.getVariablesInScopeByName(function, variableDecl.getName())) {
             if (!variableRef.equals(variable)) {
                 variables.add(variableRef);
             }
@@ -1506,7 +1506,7 @@ public static String getFirstVariableTypeInScope(@NotNull Variable variable) {
             @Override
             public boolean processAccessVariableInstruction(PhpAccessVariableInstruction instruction) {
                 PhpPsiElement element = instruction.getAnchor();
-                if(element instanceof Variable variableVisit && name.equals(variableVisit.getName())) {
+                if (name.contentEquals(instruction.getVariableName()) && element instanceof Variable) {
                     PsiElement assignmentExpression = element.getParent();
                     if(assignmentExpression instanceof AssignmentExpression) {
                         PhpPsiElement value = ((AssignmentExpression) assignmentExpression).getValue();
@@ -1786,6 +1786,23 @@ public static Set<Variable> getVariablesInScope(@NotNull PsiElement psiElement,
         return MyVariableRecursiveElementVisitor.visit(psiElement, variable.getName());
     }
 
+    public static Set<Variable> getVariablesInScopeByName(@NotNull PhpScopeHolder phpScopeHolder, @NotNull String name) {
+        Set<Variable> variables = new HashSet<>();
+
+        PhpControlFlowUtil.processFlow(phpScopeHolder.getControlFlow(), new PhpInstructionProcessor() {
+            @Override
+            public boolean processAccessVariableInstruction(PhpAccessVariableInstruction instruction) {
+                if (name.contentEquals(instruction.getVariableName()) && instruction.getAnchor() instanceof Variable variable) {
+                    variables.add(variable);
+                }
+
+                return super.processAccessVariableInstruction(instruction);
+            }
+        });
+
+        return variables;
+    }
+
     /**
      * Provide array key pattern. we need incomplete array key support, too.
      *