Add bodyBlockExpression() helper function to KtDeclarationWithBody

This commit is contained in:
Nikolay Krasko
2018-11-08 16:02:11 +03:00
parent 2b8a81c5a8
commit 2284028bbe
19 changed files with 45 additions and 36 deletions
@@ -437,7 +437,7 @@ object PositioningStrategies {
@JvmField
val DECLARATION_WITH_BODY: PositioningStrategy<KtDeclarationWithBody> = object : PositioningStrategy<KtDeclarationWithBody>() {
override fun mark(element: KtDeclarationWithBody): List<TextRange> {
val lastBracketRange = (element.bodyExpression as? KtBlockExpression)?.lastBracketRange
val lastBracketRange = element.bodyBlockExpression?.lastBracketRange
return if (lastBracketRange != null)
markRange(lastBracketRange)
else
@@ -445,7 +445,7 @@ object PositioningStrategies {
}
override fun isValid(element: KtDeclarationWithBody): Boolean {
return super.isValid(element) && (element.bodyExpression as? KtBlockExpression)?.lastBracketRange != null
return super.isValid(element) && element.bodyBlockExpression?.lastBracketRange != null
}
}