First version of KotlinAnnotationElementsSearcher #KT-2504 Fixed

2012-10-03 13:29:00 +04:00
parent ae6aa1b160
commit 8e89cf40b7
2 changed files with 135 additions and 0 deletions
@@ -217,6 +217,7 @@
      <referencesSearch implementation="org.jetbrains.jet.plugin.search.KotlinReferencesSearcher"/>
      <directClassInheritorsSearch implementation="org.jetbrains.jet.plugin.search.KotlinDirectInheritorsSearcher"/>
      <definitionsSearch implementation="org.jetbrains.jet.plugin.search.KotlinDefinitionsSearcher"/>
+      <annotatedElementsSearch implementation="org.jetbrains.jet.plugin.search.KotlinAnnotatedElementsSearcher"/>

      <toolWindow id="Kotlin"
                  factoryClass="org.jetbrains.jet.plugin.internal.KotlinInternalToolWindowFactory"
@@ -0,0 +1,134 @@
+/*
+ * Copyright 2010-2012 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.jet.plugin.search;
+
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.util.Computable;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiClass;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiMethod;
+import com.intellij.psi.PsiModifierListOwner;
+import com.intellij.psi.impl.search.AnnotatedElementsSearcher;
+import com.intellij.psi.search.GlobalSearchScope;
+import com.intellij.psi.search.SearchScope;
+import com.intellij.psi.search.searches.AnnotatedElementsSearch;
+import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.psi.util.PsiUtilCore;
+import com.intellij.util.Processor;
+import com.intellij.util.indexing.FileBasedIndex;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.analyzer.AnalyzeExhaust;
+import org.jetbrains.jet.asJava.JetLightClass;
+import org.jetbrains.jet.lang.descriptors.ClassifierDescriptor;
+import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
+import org.jetbrains.jet.lang.psi.*;
+import org.jetbrains.jet.lang.resolve.BindingContext;
+import org.jetbrains.jet.lang.resolve.DescriptorUtils;
+import org.jetbrains.jet.plugin.project.WholeProjectAnalyzerFacade;
+import org.jetbrains.jet.plugin.stubindex.JetAnnotationsIndex;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+/**
+ * User: Natalia.Ukhorskaya
+ */
+public class KotlinAnnotatedElementsSearcher extends AnnotatedElementsSearcher {
+    private static final Logger LOG = Logger.getInstance("#com.intellij.psi.impl.search.AnnotatedMembersSearcher");
+
+    @Override
+    public boolean execute(@NotNull final AnnotatedElementsSearch.Parameters p, @NotNull final Processor<PsiModifierListOwner> consumer) {
+        final PsiClass annClass = p.getAnnotationClass();
+        assert annClass.isAnnotationType() : "Annotation type should be passed to annotated members search";
+
+        final String annotationFQN = annClass.getQualifiedName();
+        assert annotationFQN != null;
+
+        final SearchScope useScope = p.getScope();
+
+        for (final PsiElement elt : getJetAnnotationCandidates(annClass, useScope)) {
+            if (notJetAnnotationEntry(elt)) continue;
+
+            ApplicationManager.getApplication().runReadAction(new Runnable() {
+                @Override
+                public void run() {
+                    //TODO LazyResolve
+                    AnalyzeExhaust analyzeExhaust = WholeProjectAnalyzerFacade.analyzeProjectWithCacheOnAFile((JetFile) elt.getContainingFile());
+                    JetDeclaration parentOfType = PsiTreeUtil.getParentOfType(elt, JetDeclaration.class);
+                    if (parentOfType == null) return;
+                    BindingContext context = analyzeExhaust.getBindingContext();
+                    AnnotationDescriptor annotationDescriptor = context.get(BindingContext.ANNOTATION, (JetAnnotationEntry) elt);
+                    if (annotationDescriptor == null) return;
+
+                    ClassifierDescriptor descriptor = annotationDescriptor.getType().getConstructor().getDeclarationDescriptor();
+                    if (descriptor == null) return;
+                    if (!(DescriptorUtils.getFQName(descriptor).getFqName().equals(annotationFQN))) return;
+
+                    if (parentOfType instanceof JetClass) {
+                        JetLightClass lightClass = JetLightClass.wrapDelegate((JetClass) parentOfType);
+                        consumer.process(lightClass);
+                    }
+                    else if (parentOfType instanceof JetNamedFunction) {
+                        PsiMethod wrappedMethod = JetLightClass.wrapMethod((JetNamedFunction) parentOfType);
+                        consumer.process(wrappedMethod);
+                    }
+                }
+            });
+        }
+
+        return true;
+    }
+
+    private static Collection<? extends PsiElement> getJetAnnotationCandidates(final PsiClass annClass, final SearchScope useScope) {
+        return ApplicationManager.getApplication().runReadAction(new Computable<Collection<? extends PsiElement>>() {
+            @Override
+            public Collection<? extends PsiElement> compute() {
+                if (useScope instanceof GlobalSearchScope) {
+                    return JetAnnotationsIndex.getInstance().get(annClass.getName(), annClass.getProject(), (GlobalSearchScope)useScope);
+                }
+                /*
+                TODO getJetAnnotationCandidates works only with global search scope
+                for (PsiElement element : ((LocalSearchScope)useScope).getScope()) {
+                    element.accept(new PsiRecursiveElementWalkingVisitor() {
+                        @Override
+                        public void visitElement(PsiElement element) {
+                            if (element instanceof JetAnnotationEntry) {
+                                result.add(element);
+                            }
+                        }
+                    });
+                }*/
+                return new ArrayList<PsiElement>();
+            }
+        });
+    }
+
+    private static boolean notJetAnnotationEntry(final PsiElement found) {
+        if (found instanceof JetAnnotationEntry) return false;
+
+        VirtualFile faultyContainer = PsiUtilCore.getVirtualFile(found);
+        LOG.error("Non annotation in annotations list: " + faultyContainer+"; element:"+found);
+        if (faultyContainer != null && faultyContainer.isValid()) {
+            FileBasedIndex.getInstance().requestReindex(faultyContainer);
+        }
+
+        return true;
+    }
+
+}