Fix for exception: JetCodeFragment with context null

#KT-6876 Fixed
2015-03-06 15:49:46 +03:00
parent f6162dc726
commit cdf081ddc3
1 changed files with 5 additions and 0 deletions
@@ -107,6 +107,11 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
                return getContextElement(elementAt.getOrigin())
            }

+            // If label for some variable is set
+            if (elementAt.getParent() is JetCodeFragment) {
+                return elementAt.getParent().getContext()
+            }
+
            val expressionAtOffset = PsiTreeUtil.findElementOfClassAtOffset(elementAt.getContainingFile()!!, elementAt.getTextOffset(), javaClass<JetExpression>(), false)
            if (expressionAtOffset != null) {
                return expressionAtOffset