support iterator in twig loop completion #1097 #1035

Author: Haehnchen

src/fr/adrienbrault/idea/symfony2plugin/templating/util/TwigTypeResolveUtil.java

@@ -258,8 +258,33 @@ public static Map<String, PsiVariable> collectScopeVariables(@NotNull PsiElement
         globalVars.putAll(convertHashMapToTypeSet(findInlineStatementVariableDocBlock(psiElement, TwigElementTypes.MACRO_STATEMENT)));
         globalVars.putAll(convertHashMapToTypeSet(findInlineStatementVariableDocBlock(psiElement, TwigElementTypes.FOR_STATEMENT)));
 
+        // @TODO find core stuff for resolve possible class return values
+        // getIterator / __iterator
+        // Foobar[]
         for(Map.Entry<String, Set<String>> entry: globalVars.entrySet()) {
-            controllerVars.put(entry.getKey(), new PsiVariable(entry.getValue(), null));
+            Set<String> types = entry.getValue();
+
+            Set<String> arrayValues = new HashSet<>();
+
+            for (String s : types) {
+                PhpClass phpClass = PhpElementsUtil.getClassInterface(psiElement.getProject(), s);
+                if(phpClass != null)  {
+                    for (String methodName : new String[]{"getIterator", "__iterator", "__invoke"}) {
+                        Method method = phpClass.findMethodByName(methodName);
+                        if(method != null) {
+                            // Foobar[]
+                            for (String type : method.getType().getTypes()) {
+                                if(type.endsWith("[]")) {
+                                    arrayValues.add(type);
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+
+            types.addAll(arrayValues);
+            controllerVars.put(entry.getKey(), new PsiVariable(types, null));
         }
 
         // check if we are in "for" scope and resolve types ending with []