add references provider for console HelperSet #243

Author: Haehnchen

src/fr/adrienbrault/idea/symfony2plugin/config/SymfonyPhpReferenceContributor.java

@@ -6,13 +6,14 @@
 import com.jetbrains.php.lang.PhpLanguage;
 import com.jetbrains.php.lang.psi.elements.PhpClass;
 import com.jetbrains.php.lang.psi.elements.StringLiteralExpression;
-import fr.adrienbrault.idea.symfony2plugin.Symfony2ProjectComponent;
+import fr.adrienbrault.idea.symfony2plugin.dic.MapClassReference;
 import fr.adrienbrault.idea.symfony2plugin.dic.ServiceReference;
 import fr.adrienbrault.idea.symfony2plugin.doctrine.EntityHelper;
 import fr.adrienbrault.idea.symfony2plugin.doctrine.EntityReference;
 import fr.adrienbrault.idea.symfony2plugin.doctrine.ModelFieldReference;
 import fr.adrienbrault.idea.symfony2plugin.doctrine.dict.DoctrineTypes;
 import fr.adrienbrault.idea.symfony2plugin.templating.TemplateReference;
+import fr.adrienbrault.idea.symfony2plugin.util.CommandUtil;
 import fr.adrienbrault.idea.symfony2plugin.util.MethodMatcher;
 import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
 import org.jetbrains.annotations.NotNull;
@@ -45,6 +46,12 @@ public class SymfonyPhpReferenceContributor extends PsiReferenceContributor {
         new MethodMatcher.CallToSignature("\\Symfony\\Bundle\\FrameworkBundle\\Controller\\Controller", "stream"),
     };
 
+    public static MethodMatcher.CallToSignature[] CONSOLE_HELP_SET = new MethodMatcher.CallToSignature[] {
+        new MethodMatcher.CallToSignature("\\Symfony\\Component\\Console\\Helper\\HelperSet", "get"),
+        new MethodMatcher.CallToSignature("\\Symfony\\Component\\Console\\Helper\\HelperSet", "has"),
+        new MethodMatcher.CallToSignature("\\Symfony\\Component\\Console\\Command\\Command", "getHelper"),
+    };
+
     @Override
     public void registerReferenceProviders(PsiReferenceRegistrar psiReferenceRegistrar) {
 
@@ -149,6 +156,21 @@ public PsiReference[] getReferencesByElement(@NotNull PsiElement psiElement, @No
             }
         );
 
+        psiReferenceRegistrar.registerReferenceProvider(
+            PlatformPatterns.psiElement(StringLiteralExpression.class),
+            new PsiReferenceProvider() {
+                @NotNull
+                @Override
+                public PsiReference[] getReferencesByElement(@NotNull PsiElement psiElement, @NotNull ProcessingContext processingContext) {
+
+                    if (MethodMatcher.getMatchedSignatureWithDepth(psiElement, CONSOLE_HELP_SET) == null) {
+                        return new PsiReference[0];
+                    }
+
+                    return new PsiReference[]{ new MapClassReference((StringLiteralExpression) psiElement, CommandUtil.getCommandHelper(psiElement.getProject()))};
+                }
+            }
+        );
 
     }
 

src/fr/adrienbrault/idea/symfony2plugin/dic/MapClassReference.java

@@ -0,0 +1,60 @@
+package fr.adrienbrault.idea.symfony2plugin.dic;
+
+import com.intellij.codeInsight.lookup.LookupElement;
+import com.intellij.codeInsight.lookup.LookupElementBuilder;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiElementResolveResult;
+import com.intellij.psi.PsiPolyVariantReferenceBase;
+import com.intellij.psi.ResolveResult;
+import com.jetbrains.php.PhpIcons;
+import com.jetbrains.php.lang.psi.elements.PhpClass;
+import com.jetbrains.php.lang.psi.elements.StringLiteralExpression;
+import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+public class MapClassReference extends PsiPolyVariantReferenceBase<PsiElement> {
+
+    private Map<String, String> variants;
+
+    public MapClassReference(@NotNull StringLiteralExpression element, Map<String, String> variants) {
+        super(element);
+        this.variants = variants;
+    }
+
+    @NotNull
+    @Override
+    public ResolveResult[] multiResolve(boolean incompleteCode) {
+
+        List<ResolveResult> resolveResults = new ArrayList<ResolveResult>();
+
+        for (Map.Entry<String, String> entry: this.variants.entrySet()) {
+            PhpClass phpClass = PhpElementsUtil.getClassInterface(getElement().getProject(), entry.getValue());
+            if(phpClass != null) {
+                resolveResults.add(new PsiElementResolveResult(phpClass));
+            }
+        }
+
+
+        return resolveResults.toArray(new ResolveResult[resolveResults.size()]);
+    }
+
+    @NotNull
+    @Override
+    public Object[] getVariants() {
+
+        List<LookupElement> lookupElements = new ArrayList<LookupElement>();
+        for (Map.Entry<String, String> entry: this.variants.entrySet()) {
+            PhpClass phpClass = PhpElementsUtil.getClassInterface(getElement().getProject(), entry.getValue());
+            if(phpClass != null) {
+                lookupElements.add(LookupElementBuilder.create(entry.getKey()).withIcon(PhpIcons.CLASS).withTypeText(phpClass.getPresentableFQN()));
+            }
+        }
+
+        return lookupElements.toArray();
+
+    }
+}

src/fr/adrienbrault/idea/symfony2plugin/dic/StringArrayListReference.java

@@ -0,0 +1,37 @@
+package fr.adrienbrault.idea.symfony2plugin.util;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.util.PsiTreeUtil;
+import com.jetbrains.php.PhpIndex;
+import com.jetbrains.php.lang.psi.elements.Method;
+import com.jetbrains.php.lang.psi.elements.PhpClass;
+import com.jetbrains.php.lang.psi.elements.PhpReturn;
+
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.Map;
+
+public class CommandUtil {
+
+    public static Map<String, String> getCommandHelper(Project project) {
+
+        Map<String, String> map = new HashMap<String, String>();
+        Collection<PhpClass> phpClasses = PhpIndex.getInstance(project).getAllSubclasses("\\Symfony\\Component\\Console\\Helper\\HelperInterface");
+        for(PhpClass phpClass: phpClasses) {
+            Method method = PhpElementsUtil.getClassMethod(phpClass, "getName");
+            for(PhpReturn phpReturn: PsiTreeUtil.findChildrenOfType(method, PhpReturn.class)) {
+                String nameValue = PhpElementsUtil.getStringValue(phpReturn.getArgument());
+                if(nameValue != null) {
+                    String presentableFQN = phpClass.getPresentableFQN();
+                    if(presentableFQN != null) {
+                        map.put(nameValue, presentableFQN);
+                    }
+                }
+            }
+        }
+
+        return map;
+    }
+
+
+}