diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml index 0a59aeb5f4f..4d3ba4cbd6b 100644 --- a/idea/src/META-INF/plugin.xml +++ b/idea/src/META-INF/plugin.xml @@ -204,6 +204,7 @@ + diff --git a/idea/src/org/jetbrains/jet/plugin/stubindex/JetClassByPackageIndex.java b/idea/src/org/jetbrains/jet/plugin/stubindex/JetClassByPackageIndex.java new file mode 100644 index 00000000000..bec685258e5 --- /dev/null +++ b/idea/src/org/jetbrains/jet/plugin/stubindex/JetClassByPackageIndex.java @@ -0,0 +1,49 @@ +/* + * Copyright 2010-2013 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.stubindex; + +import com.intellij.openapi.project.Project; +import com.intellij.psi.search.GlobalSearchScope; +import com.intellij.psi.stubs.StringStubIndexExtension; +import com.intellij.psi.stubs.StubIndexKey; +import org.jetbrains.annotations.NotNull; +import org.jetbrains.jet.lang.psi.JetClassOrObject; + +import java.util.Collection; + +public class JetClassByPackageIndex extends StringStubIndexExtension { + private static final StubIndexKey KEY = KotlinIndexUtil.createIndexKey(JetClassByPackageIndex.class); + + private static final JetClassByPackageIndex ourInstance = new JetClassByPackageIndex(); + + public static JetClassByPackageIndex getInstance() { + return ourInstance; + } + + private JetClassByPackageIndex() {} + + @NotNull + @Override + public StubIndexKey getKey() { + return KEY; + } + + @Override + public Collection get(final String fqName, final Project project, @NotNull final GlobalSearchScope scope) { + return super.get(fqName, project, new JetSourceFilterScope(scope)); + } +} diff --git a/idea/src/org/jetbrains/jet/plugin/stubindex/StubIndexServiceImpl.java b/idea/src/org/jetbrains/jet/plugin/stubindex/StubIndexServiceImpl.java index bc6dfcf93e1..7daadf3613b 100644 --- a/idea/src/org/jetbrains/jet/plugin/stubindex/StubIndexServiceImpl.java +++ b/idea/src/org/jetbrains/jet/plugin/stubindex/StubIndexServiceImpl.java @@ -17,6 +17,7 @@ package org.jetbrains.jet.plugin.stubindex; import com.intellij.psi.stubs.IndexSink; +import com.intellij.psi.stubs.StubElement; import org.jetbrains.jet.lang.psi.stubs.*; import org.jetbrains.jet.lang.psi.stubs.elements.StubIndexService; import org.jetbrains.jet.lang.resolve.name.FqName; @@ -44,6 +45,15 @@ public class StubIndexServiceImpl implements StubIndexService { for (String superName : stub.getSuperNames()) { sink.occurrence(JetSuperClassIndex.getInstance().getKey(), superName); } + + StubElement parentStub = stub.getParentStub(); + if (parentStub instanceof PsiJetFileStub) { + PsiJetFileStub jetFileStub = (PsiJetFileStub) parentStub; + String packageName = jetFileStub.getPackageName(); + if (packageName != null) { + sink.occurrence(JetClassByPackageIndex.getInstance().getKey(), packageName); + } + } } @Override