Adding block to control structure when needed

2016-10-22 13:29:23 +03:00
parent a4aa9bab8d
commit 8f9f2027f2
5 changed files with 104 additions and 8 deletions
@@ -18,4 +18,7 @@ package org.jetbrains.kotlin.psi

 import com.intellij.lang.ASTNode

-class KtContainerNodeForControlStructureBody(node: ASTNode) : KtContainerNode(node)
+class KtContainerNodeForControlStructureBody(node: ASTNode) : KtContainerNode(node) {
+    val expression: KtExpression?
+        get() = findChildByClass<KtExpression>(KtExpression::class.java)
+}