1 /*
2  * Copyright 2014, Google Inc.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are
7  * met:
8  *
9  *     * Redistributions of source code must retain the above copyright
10  * notice, this list of conditions and the following disclaimer.
11  *     * Redistributions in binary form must reproduce the above
12  * copyright notice, this list of conditions and the following disclaimer
13  * in the documentation and/or other materials provided with the
14  * distribution.
15  *     * Neither the name of Google Inc. nor the names of its
16  * contributors may be used to endorse or promote products derived from
17  * this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 package org.jf.smalidea.psi.index;
33 
34 import com.intellij.openapi.project.Project;
35 import com.intellij.psi.PsiClass;
36 import com.intellij.psi.PsiElementFinder;
37 import com.intellij.psi.search.GlobalSearchScope;
38 import org.jetbrains.annotations.NotNull;
39 import org.jf.smalidea.psi.impl.SmaliClass;
40 
41 import java.util.Collection;
42 
43 public class SmaliClassFinder extends PsiElementFinder {
44 
45     private final Project project;
46 
SmaliClassFinder(Project project)47     public SmaliClassFinder(Project project) {
48         this.project = project;
49     }
50 
51     @Override
findClass(@otNull String qualifiedName, @NotNull GlobalSearchScope scope)52     public PsiClass findClass(@NotNull String qualifiedName, @NotNull GlobalSearchScope scope) {
53         Collection<SmaliClass> classes = SmaliClassNameIndex.INSTANCE.get(qualifiedName, project, scope);
54         if (classes != null && classes.size() == 1) {
55             return classes.iterator().next();
56         }
57         return null;
58     }
59 
60     @NotNull
61     @Override
findClasses(@otNull String qualifiedName, @NotNull GlobalSearchScope scope)62     public PsiClass[] findClasses(@NotNull String qualifiedName, @NotNull GlobalSearchScope scope) {
63         PsiClass cls = findClass(qualifiedName, scope);
64         if (cls != null) {
65             return new PsiClass[] {cls};
66         }
67         return PsiClass.EMPTY_ARRAY;
68     }
69 }
70