#305_add ui component reference

Author: iamdylanngo

src/com/magento/idea/magento2plugin/indexes/UIComponentIndex.java

@@ -92,4 +92,45 @@ public static Collection<String> getAllKeys(
     ) {
         return FileBasedIndex.getInstance().getAllKeys(identifier, project);
     }
+
+    /**
+     * Get ui component files.
+     *
+     * @param project Project
+     * @param fileName String
+     * @return List
+     */
+    public static List<XmlFile> getUIComponentFiles(Project project, @Nullable String fileName) {
+        List<XmlFile> results = new ArrayList<XmlFile>();
+        Collection<VirtualFile> xmlFiles = FilenameIndex.getAllFilesByExt(project, "xml");
+
+        PsiManager psiManager = PsiManager.getInstance(project);
+        for (VirtualFile xmlFile: xmlFiles) {
+            if (isUIComponentFile(xmlFile)) {
+                if (fileName != null && !xmlFile.getNameWithoutExtension().equals(fileName)) {
+                    continue;
+                }
+
+                PsiFile file = psiManager.findFile(xmlFile);
+                if (file != null) {
+                    results.add((XmlFile)file);
+                }
+            }
+        }
+
+        return results;
+    }
+
+    /**
+     * Available ui component file.
+     *
+     * @param virtualFile VirtualFile
+     * @return boolean
+     */
+    public static boolean isUIComponentFile(VirtualFile virtualFile) {
+        VirtualFile parent = virtualFile.getParent();
+        return virtualFile.getFileType() == XmlFileType.INSTANCE && parent.isDirectory()
+                && parent.getName().endsWith("ui_component");
+    }
+
 }

src/com/magento/idea/magento2plugin/reference/provider/UIComponentReferenceProvider.java

@@ -0,0 +1,38 @@
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+package com.magento.idea.magento2plugin.reference.provider;
+
+import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiReference;
+import com.intellij.psi.PsiReferenceProvider;
+import com.intellij.psi.xml.XmlFile;
+import com.intellij.util.ProcessingContext;
+import com.magento.idea.magento2plugin.indexes.UIComponentIndex;
+import com.magento.idea.magento2plugin.reference.xml.PolyVariantReferenceBase;
+import java.util.List;
+import org.jetbrains.annotations.NotNull;
+
+
+public class UIComponentReferenceProvider extends PsiReferenceProvider {
+
+    @NotNull
+    @Override
+    public PsiReference[] getReferencesByElement(
+            @NotNull final PsiElement element,
+            @NotNull final ProcessingContext context
+    ) {
+        final String value = StringUtil.unquoteString(element.getText());
+        final List<XmlFile> targets = UIComponentIndex.getUIComponentFiles(
+                element.getProject(),
+                value
+        );
+        if (!targets.isEmpty()) {
+            return new PsiReference[] {new PolyVariantReferenceBase(element, targets)};
+        }
+        return PsiReference.EMPTY_ARRAY;
+    }
+}

src/com/magento/idea/magento2plugin/reference/xml/XmlReferenceContributor.java

@@ -132,6 +132,16 @@ public void registerReferenceProviders(@NotNull PsiReferenceRegistrar registrar)
             new LayoutUpdateReferenceProvider()
         );
 
+        // <uiComponent name="completion"/>
+        registrar.registerReferenceProvider(
+                XmlPatterns.xmlAttributeValue().withParent(
+                        XmlPatterns.xmlAttribute().withName("name").withParent(
+                                XmlPatterns.xmlTag().withName("uiComponent")
+                        )
+                ),
+                new UIComponentReferenceProvider()
+        );
+
         // <event name="reference" />
         registrar.registerReferenceProvider(
             XmlPatterns.xmlAttributeValue().withParent(