From 013b25734761a8c64fdcc91f361156ad982c7fca Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Tue, 6 Oct 2015 16:27:23 +0300
Subject: [PATCH] Make it possible to get diagnostics factory was registered
 for

---
 .../src/org/jetbrains/kotlin/idea/quickfix/QuickFixes.kt      | 4 ++++
 idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt  | 3 +--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/idea/idea-analysis/src/org/jetbrains/kotlin/idea/quickfix/QuickFixes.kt b/idea/idea-analysis/src/org/jetbrains/kotlin/idea/quickfix/QuickFixes.kt
index ec90b9f1762..ab3f6de7d9e 100644
--- a/idea/idea-analysis/src/org/jetbrains/kotlin/idea/quickfix/QuickFixes.kt
+++ b/idea/idea-analysis/src/org/jetbrains/kotlin/idea/quickfix/QuickFixes.kt
@@ -48,6 +48,10 @@ public class QuickFixes {
         return actions.get(diagnosticFactory)
     }
 
+    public fun getDiagnostics(factory: JetIntentionActionsFactory): Collection<DiagnosticFactory<*>> {
+        return factories.keySet().filter { factory in factories.get(it) }
+    }
+
     companion object {
         public fun getInstance(): QuickFixes = ServiceManager.getService(javaClass<QuickFixes>())
     }
diff --git a/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt b/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
index bce0957368b..56fcee0f0d4 100644
--- a/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/quickfix/AutoImportFix.kt
@@ -29,7 +29,6 @@ import com.intellij.psi.util.PsiModificationTracker
 import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
 import org.jetbrains.kotlin.descriptors.DeclarationDescriptorWithVisibility
 import org.jetbrains.kotlin.diagnostics.Diagnostic
-import org.jetbrains.kotlin.diagnostics.Errors
 import org.jetbrains.kotlin.idea.JetBundle
 import org.jetbrains.kotlin.idea.actions.KotlinAddImportAction
 import org.jetbrains.kotlin.idea.caches.resolve.analyze
@@ -113,7 +112,7 @@ public class AutoImportFix(element: JetSimpleNameExpression) : JetHintAction<Jet
 
         override fun isApplicableForCodeFragment() = true
 
-        private val ERRORS = setOf(Errors.UNRESOLVED_REFERENCE, Errors.UNRESOLVED_REFERENCE_WRONG_RECEIVER)
+        private val ERRORS by lazy(LazyThreadSafetyMode.PUBLICATION ) { QuickFixes.getInstance().getDiagnostics(this) }
 
         public fun computeSuggestions(element: JetSimpleNameExpression): Collection<DeclarationDescriptor> {
             if (!element.isValid()) return emptyList()