Optimize Imports: do not keep imports of unimportable entities

idea/src/org/jetbrains/jet/plugin/codeInsight/KotlinCopyPasteReferenceProcessor.kt

@@ -164,21 +164,6 @@ public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<Refere
         return collectedData
     }
 
-    private fun DeclarationDescriptor.canBeReferencedViaImport(): Boolean {
-        if (this is PackageViewDescriptor || DescriptorUtils.isTopLevelDeclaration(this)) {
-            return true
-        }
-        val parent = getContainingDeclaration()!!
-        if (parent !is ClassDescriptor || !parent.canBeReferencedViaImport()) {
-            return false
-        }
-        // inner class constructors can't be referenced via import
-        if (this is ConstructorDescriptor && parent.isInner()) {
-            return false
-        }
-        return this is ClassDescriptor || this is ConstructorDescriptor
-    }
-
     private fun createReferenceData(element: PsiElement, startOffset: Int, fqName: FqName): ReferenceData {
         val range = element.range
         return ReferenceData(range.start - startOffset, range.end - startOffset, fqName.asString(), null)

idea/src/org/jetbrains/jet/plugin/imports/ImportsUtils.kt

@@ -20,6 +20,8 @@ import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor
 import org.jetbrains.jet.lang.resolve.name.FqName
 import org.jetbrains.jet.lang.descriptors.ConstructorDescriptor
 import org.jetbrains.jet.lang.resolve.DescriptorUtils
+import org.jetbrains.jet.lang.descriptors.PackageViewDescriptor
+import org.jetbrains.jet.lang.descriptors.ClassDescriptor
 
 public val DeclarationDescriptor.importableFqName: FqName?
     get() {
@@ -31,4 +33,19 @@ public val DeclarationDescriptor.importableFqName: FqName?
         else {
             null
         }
-    }
+    }
+
+public fun DeclarationDescriptor.canBeReferencedViaImport(): Boolean {
+    if (this is PackageViewDescriptor || DescriptorUtils.isTopLevelDeclaration(this)) {
+        return true
+    }
+    val parent = getContainingDeclaration()!!
+    if (parent !is ClassDescriptor || !parent.canBeReferencedViaImport()) {
+        return false
+    }
+    // inner class constructors can't be referenced via import
+    if (this is ConstructorDescriptor && parent.isInner()) {
+        return false
+    }
+    return this is ClassDescriptor || this is ConstructorDescriptor
+}

idea/src/org/jetbrains/jet/plugin/imports/KotlinImportOptimizer.kt

@@ -98,7 +98,8 @@ public class KotlinImportOptimizer() : ImportOptimizer {
                 val reference = element.getReference()
                 if (reference is JetReference) {
                     val referencedDescriptors = reference.resolveToDescriptors()
-                    usedQualifiedNames.addAll(referencedDescriptors.map { it.importableFqName }.filterNotNull())
+                    val importableDescriptors = referencedDescriptors.filter { it.canBeReferencedViaImport() }
+                    usedQualifiedNames.addAll(importableDescriptors.map { it.importableFqName }.filterNotNull())
                 }
                 super.visitJetElement(element)
             }

idea/testData/editor/optimizeImports/ClassMemberImported/ClassMemberImported.after.kt

@@ -0,0 +1,10 @@
+package test
+
+import dependency.*
+
+fun f(a: A, t: T) {
+    a.f()
+    a.p
+    t.f
+    t.p
+}

idea/testData/editor/optimizeImports/ClassMemberImported/ClassMemberImported.dependency.kt

@@ -0,0 +1,15 @@
+package dependency
+
+class A {
+    fun f() {
+    }
+
+    val p: Int = 1
+}
+
+trait T {
+    fun f() {
+    }
+
+    val p: Int = 1
+}

idea/testData/editor/optimizeImports/ClassMemberImported/ClassMemberImported.kt

@@ -0,0 +1,14 @@
+package test
+
+import dependency.*
+import dependency.A.f
+import dependency.A.p
+import dependency.T.f
+import dependency.T.p
+
+fun f(a: A, t: T) {
+    a.f()
+    a.p
+    t.f
+    t.p
+}

idea/tests/org/jetbrains/jet/plugin/imports/OptimizeImportsTestGenerated.java

@@ -46,6 +46,11 @@ public class OptimizeImportsTestGenerated extends AbstractOptimizeImportsTest {
         doTest("idea/testData/editor/optimizeImports/ArrayAccessExpression");
     }
     
+    @TestMetadata("ClassMemberImported")
+    public void testClassMemberImported() throws Exception {
+        doTest("idea/testData/editor/optimizeImports/ClassMemberImported");
+    }
+    
     @TestMetadata("ComponentFunction")
     public void testComponentFunction() throws Exception {
         doTest("idea/testData/editor/optimizeImports/ComponentFunction");