From a2cdb14610d8d4045f295d28fcc1cb28fa5729c4 Mon Sep 17 00:00:00 2001
From: Simon Ogorodnik <Simon.Ogorodnik@jetbrains.com>
Date: Thu, 13 Sep 2018 22:12:09 +0300
Subject: [PATCH] Add function to get typed capability from ModuleInfo

---
 .../src/org/jetbrains/kotlin/analyzer/AnalyzerFacade.kt      | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/compiler/frontend/src/org/jetbrains/kotlin/analyzer/AnalyzerFacade.kt b/compiler/frontend/src/org/jetbrains/kotlin/analyzer/AnalyzerFacade.kt
index 1523faa2669..ceb0ff9d082 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/analyzer/AnalyzerFacade.kt
+++ b/compiler/frontend/src/org/jetbrains/kotlin/analyzer/AnalyzerFacade.kt
@@ -502,4 +502,7 @@ private object DiagnoseUnknownModuleInfoReporter {
     private fun errorInSpecialModuleInfoResolver(message: String): Nothing = throw AssertionError(message)
 
     private fun otherError(message: String): Nothing = throw AssertionError(message)
-}
\ No newline at end of file
+}
+
+@Suppress("UNCHECKED_CAST")
+fun <T> ModuleInfo.getCapability(capability: ModuleDescriptor.Capability<T>) = capabilities[capability] as? T