From c62b1fac696a129a7173498cae1e81d2d5f5e1d4 Mon Sep 17 00:00:00 2001
From: Michael Nedzelsky <Michael.Nedzelsky@jetbrains.com>
Date: Mon, 16 Mar 2015 18:01:07 +0300
Subject: [PATCH] JS: fix support auto import for class names

---
 .../kotlin/idea/quickfix/AutoImportFix.kt      | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt b/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
index 4198955e368..dc824fbc903 100644
--- a/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
@@ -33,9 +33,8 @@ import org.jetbrains.kotlin.diagnostics.Diagnostic
 import org.jetbrains.kotlin.diagnostics.Errors
 import org.jetbrains.kotlin.idea.JetBundle
 import org.jetbrains.kotlin.idea.actions.JetAddImportAction
-import org.jetbrains.kotlin.idea.caches.JetShortNamesCache
-import org.jetbrains.kotlin.idea.core.KotlinIndicesHelper
 import org.jetbrains.kotlin.idea.caches.resolve.getResolutionFacade
+import org.jetbrains.kotlin.idea.core.KotlinIndicesHelper
 import org.jetbrains.kotlin.idea.core.isVisible
 import org.jetbrains.kotlin.idea.core.psiClassToDescriptor
 import org.jetbrains.kotlin.idea.project.ProjectStructureUtil
@@ -141,7 +140,12 @@ public class AutoImportFix(element: JetSimpleNameExpression) : JetHintAction<Jet
         val indicesHelper = KotlinIndicesHelper(file.getProject(), resolutionFacade, bindingContext, searchScope, moduleDescriptor, ::isVisible)
 
         if (!element.isImportDirectiveExpression() && !JetPsiUtil.isSelectorInQualified(element)) {
-            getClasses(referenceName, file, searchScope).filterTo(result, ::isVisible)
+            if (ProjectStructureUtil.isJsKotlinModule(file)) {
+                result.addAll(indicesHelper.getClassDescriptors({ it == referenceName }, { true }))
+            }
+            else {
+                getClasses(referenceName, file, searchScope).filterTo(result, ::isVisible)
+            }
             result.addAll(indicesHelper.getTopLevelCallablesByName(referenceName))
         }
 
@@ -156,13 +160,7 @@ public class AutoImportFix(element: JetSimpleNameExpression) : JetHintAction<Jet
             .filterNotNull()
             .toSet()
 
-    private fun getShortNamesCache(jetFile: JetFile): PsiShortNamesCache {
-        // if we are in JS module, do not include non-kotlin classes
-        return if (ProjectStructureUtil.isJsKotlinModule(jetFile))
-            JetShortNamesCache.getKotlinInstance(jetFile.getProject())
-        else
-            PsiShortNamesCache.getInstance(jetFile.getProject())
-    }
+    private fun getShortNamesCache(jetFile: JetFile): PsiShortNamesCache = PsiShortNamesCache.getInstance(jetFile.getProject())
 
     companion object {
         private val ERRORS = setOf(Errors.UNRESOLVED_REFERENCE, Errors.UNRESOLVED_REFERENCE_WRONG_RECEIVER)