Fast implementation for KtLightClassForExplicitDeclaration#getContainingClass()

Prevent computing stubs for light classes for nested/inner classes when calling completion from java

compiler/light-classes/src/org/jetbrains/kotlin/asJava/KtLightClassForAnonymousDeclaration.java

@@ -47,6 +47,12 @@ class KtLightClassForAnonymousDeclaration extends KtLightClassForExplicitDeclara
         return JavaPsiFacade.getElementFactory(getClassOrObject().getProject()).createReferenceElementByType(getBaseClassType());
     }
 
+    @Nullable
+    @Override
+    public PsiClass getContainingClass() {
+        return getDelegate().getContainingClass();
+    }
+
     @NotNull
     private String getFirstSupertypeFQName() {
         ClassDescriptor descriptor = getDescriptor();

compiler/light-classes/src/org/jetbrains/kotlin/asJava/KtLightClassForExplicitDeclaration.kt

@@ -230,6 +230,13 @@ public open class KtLightClassForExplicitDeclaration(
 
     override fun getContainingClass(): PsiClass? {
         if (classOrObject.parent === classOrObject.containingFile) return null
+
+        val containingClassOrObject = (classOrObject.parent as? KtClassBody)?.parent as? KtClassOrObject
+        if (containingClassOrObject != null) {
+            return create(containingClassOrObject)
+        }
+
+        // TODO: should return null
         return super.getContainingClass()
     }
 

compiler/light-classes/src/org/jetbrains/kotlin/asJava/KtLightClassForInterfaceDefaultImpls.kt

@@ -46,4 +46,6 @@ public open class KtLightClassForInterfaceDefaultImpls(
     override fun setName(name: String): PsiElement {
         throw IncorrectOperationException("Impossible to rename DefaultImpls")
     }
+
+    override fun getContainingClass() = KtLightClassForExplicitDeclaration.create(classOrObject)
 }

idea/idea-completion/testData/injava/NestedNoImport.java

@@ -0,0 +1,12 @@
+package test;
+
+import mockLib.foo
+
+public class Testing {
+    public static void test() {
+        Nest<caret>
+    }
+}
+
+// EXIST: Nested
+// EXIST: NestedObject

idea/idea-completion/tests/org/jetbrains/kotlin/idea/completion/test/CompiledKotlinInJavaCompletionTestGenerated.java

@@ -137,6 +137,12 @@ public class CompiledKotlinInJavaCompletionTestGenerated extends AbstractCompile
         doTest(fileName);
     }
 
+    @TestMetadata("NestedNoImport.java")
+    public void testNestedNoImport() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("idea/idea-completion/testData/injava/NestedNoImport.java");
+        doTest(fileName);
+    }
+
     @TestMetadata("NestedObjectInstance.java")
     public void testNestedObjectInstance() throws Exception {
         String fileName = KotlinTestUtils.navigationMetadata("idea/idea-completion/testData/injava/NestedObjectInstance.java");

idea/idea-completion/tests/org/jetbrains/kotlin/idea/completion/test/KotlinSourceInJavaCompletionTestGenerated.java

@@ -137,6 +137,12 @@ public class KotlinSourceInJavaCompletionTestGenerated extends AbstractKotlinSou
         doTest(fileName);
     }
 
+    @TestMetadata("NestedNoImport.java")
+    public void testNestedNoImport() throws Exception {
+        String fileName = KotlinTestUtils.navigationMetadata("idea/idea-completion/testData/injava/NestedNoImport.java");
+        doTest(fileName);
+    }
+
     @TestMetadata("NestedObjectInstance.java")
     public void testNestedObjectInstance() throws Exception {
         String fileName = KotlinTestUtils.navigationMetadata("idea/idea-completion/testData/injava/NestedObjectInstance.java");