azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor if-to-when refactoring

Browse Source
This commit is contained in:
Mikhail Glukhikh
2016-03-18 18:54:01 +03:00
parent fa069214fd
commit 90b5d3b9f5
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfToWhenIntention.kt
+3 -5
View File
@@ -24,6 +24,7 @@ import com.intellij.psi.PsiWhiteSpace
import org.jetbrains.kotlin.idea.intentions.SelfTargetingRangeIntention
import org.jetbrains.kotlin.idea.intentions.branchedTransformations.getSubjectToIntroduce
import org.jetbrains.kotlin.idea.intentions.branchedTransformations.introduceSubject
import org.jetbrains.kotlin.idea.intentions.branchedTransformations.unwrapBlockOrParenthesis
import org.jetbrains.kotlin.idea.util.CommentSaver
import org.jetbrains.kotlin.lexer.KtTokens
import org.jetbrains.kotlin.psi.*
@@ -66,9 +67,6 @@ class IfToWhenIntention : SelfTargetingRangeIntention<KtIfExpression>(KtIfExpres
}
}
private fun KtExpression?.unwrapBlockIfNeeded(): KtExpression? =
(this as? KtBlockExpression)?.statements?.singleOrNull() ?: this
private fun KtIfExpression.siblingsUpTo(other: KtExpression): List<PsiElement> {
val result = ArrayList<PsiElement>()
var nextSibling = nextSibling
@@ -85,7 +83,7 @@ class IfToWhenIntention : SelfTargetingRangeIntention<KtIfExpression>(KtIfExpres
private fun BuilderByPattern<*>.appendElseBlock(block: KtExpression?) {
appendFixedText("else->")
appendExpression(block.unwrapBlockIfNeeded())
appendExpression(block?.unwrapBlockOrParenthesis())
appendFixedText("\n")
}
@@ -112,7 +110,7 @@ class IfToWhenIntention : SelfTargetingRangeIntention<KtIfExpression>(KtIfExpres
appendFixedText("->")
val currentThenBranch = currentIfExpression.then
appendExpression(currentThenBranch.unwrapBlockIfNeeded())
appendExpression(currentThenBranch?.unwrapBlockOrParenthesis())
appendFixedText("\n")
canPassThrough = canPassThrough || canPassThrough(currentThenBranch)