Return moduleInfo UserDataProperty if it was explicitly set

2015-03-13 19:32:38 +03:00
parent b04d14f79d
commit 15f8ba811b
1 changed files with 3 additions and 0 deletions
@@ -48,6 +48,9 @@ fun PsiElement.getModuleInfo(): IdeaModuleInfo {
        )
    }

+    val explicitModuleInfo = containingJetFile?.moduleInfo
+    if (explicitModuleInfo is IdeaModuleInfo) return explicitModuleInfo
+
    if (containingJetFile is JetCodeFragment) {
        return containingJetFile.getContext()?.getModuleInfo()
               ?: logAndReturnDefault("Analyzing code fragment of type ${containingJetFile.javaClass} with no context element\nText:\n${containingJetFile.getText()}")