From ab4fa36dd69f27c120d2d4c514e41430275ccfdf Mon Sep 17 00:00:00 2001
From: Daniel Espendiller <daniel@espendiller.net>
Date: Sun, 25 Feb 2024 11:39:14 +0100
Subject: [PATCH] support vue.js compoenents in any components key

---
 .../idea/vuejs/index/ComponentUsageIndex.java | 38 +++++++++++++++++--
 .../index/fixtures/component-annotation.vue   | 14 +++++++
 .../index/fixtures/component-composition.vue  |  8 ++++
 .../index/fixtures/component-options.vue      | 14 +++++++
 4 files changed, 70 insertions(+), 4 deletions(-)
 create mode 100644 src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-annotation.vue
 create mode 100644 src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-composition.vue
 create mode 100644 src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-options.vue

diff --git a/src/main/java/de/espend/idea/vuejs/index/ComponentUsageIndex.java b/src/main/java/de/espend/idea/vuejs/index/ComponentUsageIndex.java
index 98af1f1..cdf8cd7 100644
--- a/src/main/java/de/espend/idea/vuejs/index/ComponentUsageIndex.java
+++ b/src/main/java/de/espend/idea/vuejs/index/ComponentUsageIndex.java
@@ -1,10 +1,18 @@
 package de.espend.idea.vuejs.index;
 
 import com.intellij.lang.ecmascript6.psi.ES6ImportedBinding;
+import com.intellij.lang.ecmascript6.psi.ES6Property;
 import com.intellij.lang.ecmascript6.resolve.JSFileReferencesUtil;
+import com.intellij.lang.javascript.psi.JSExpression;
+import com.intellij.lang.javascript.psi.JSObjectLiteralExpression;
+import com.intellij.lang.javascript.psi.JSProperty;
+import com.intellij.lang.javascript.psi.JSReferenceExpression;
+import com.intellij.lang.javascript.psi.impl.JSReferenceExpressionImpl;
+import com.intellij.lang.javascript.psi.util.JSStubBasedPsiTreeUtil;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiRecursiveElementVisitor;
+import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.psi.xml.XmlTag;
 import com.intellij.util.PathUtil;
 import com.intellij.util.indexing.*;
@@ -47,14 +55,36 @@ public class ComponentUsageIndex extends FileBasedIndexExtension<String, List<St
                             @Override
                             public void visitElement(@NotNull PsiElement element) {
                                 if (element instanceof ES6ImportedBinding es6ImportedBinding) {
-                                    VueJsUtil.ES6Import singleImported = VueJsUtil.getSingleImported(es6ImportedBinding);
-                                    if (singleImported != null) {
-                                        String fileName = PathUtil.getFileName(singleImported.fileNameWithoutExtension());
-                                        imports.put(JSFileReferencesUtil.getFileNameWithoutExtension(fileName, new String[] {".vue"}), Arrays.asList(singleImported.referenceText(), singleImported.importName()));
+                                    visitImportedBinding(es6ImportedBinding);
+                                } else if (element instanceof JSProperty jsProperty && "components".equals(jsProperty.getName())) {
+                                    // components: {Foobar}
+                                    JSExpression value = jsProperty.getValue();
+                                    if (value instanceof JSObjectLiteralExpression jsObjectLiteralExpression) {
+                                        for (ES6Property es6Property : PsiTreeUtil.collectElementsOfType(jsObjectLiteralExpression, ES6Property.class)) {
+                                            JSExpression value1 = es6Property.getValue();
+                                            if (value1 instanceof JSReferenceExpression jsReferenceExpression) {
+                                                String referenceName = JSReferenceExpressionImpl.getReferenceName(jsReferenceExpression.getNode());
+                                                if (referenceName != null) {
+                                                    for (PsiElement resolveLocallyWithMergedResult : JSStubBasedPsiTreeUtil.resolveLocallyWithMergedResults(referenceName, jsReferenceExpression)) {
+                                                        if (resolveLocallyWithMergedResult instanceof ES6ImportedBinding es6ImportedBinding) {
+                                                            visitImportedBinding(es6ImportedBinding);
+                                                        }
+                                                    }
+                                                }
+                                            }
+                                        }
                                     }
                                 }
                                 super.visitElement(element);
                             }
+
+                            private void visitImportedBinding(@NotNull ES6ImportedBinding es6ImportedBinding) {
+                                VueJsUtil.ES6Import singleImported = VueJsUtil.getSingleImported(es6ImportedBinding);
+                                if (singleImported != null) {
+                                    String fileName = PathUtil.getFileName(singleImported.fileNameWithoutExtension());
+                                    imports.put(JSFileReferencesUtil.getFileNameWithoutExtension(fileName, new String[] {".vue"}), Arrays.asList(singleImported.referenceText(), singleImported.importName()));
+                                }
+                            }
                         });
                     }
 
diff --git a/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-annotation.vue b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-annotation.vue
new file mode 100644
index 0000000..cd46870
--- /dev/null
+++ b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-annotation.vue
@@ -0,0 +1,14 @@
+<script>
+import ButtonCounter from './ButtonCounter.vue'
+
+@Component({
+  components: {
+    ButtonCounter,
+  },
+})
+export default {}
+
+<template>
+  <h1>Here is a child component!</h1>
+  <ButtonCounter />
+</template>
\ No newline at end of file
diff --git a/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-composition.vue b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-composition.vue
new file mode 100644
index 0000000..e207e90
--- /dev/null
+++ b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-composition.vue
@@ -0,0 +1,8 @@
+<script setup>
+import ButtonCounter from './ButtonCounter.vue'
+</script>
+
+<template>
+  <h1>Here is a child component!</h1>
+  <ButtonCounter />
+</template>
\ No newline at end of file
diff --git a/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-options.vue b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-options.vue
new file mode 100644
index 0000000..2194753
--- /dev/null
+++ b/src/test/java/tests/de/espend/idea/vuejs/index/fixtures/component-options.vue
@@ -0,0 +1,14 @@
+<script>
+import ButtonCounter from './ButtonCounter.vue'
+
+export default {
+  components: {
+    ButtonCounter
+  }
+}
+</script>
+
+<template>
+  <h1>Here is a child component!</h1>
+  <ButtonCounter />
+</template>
\ No newline at end of file