Move "else" extractor to JetWhenExpression

This commit is contained in:
Alexey Sedunov
2013-04-30 16:08:30 +04:00
parent d30e8d88fb
commit 4ee59e351b
3 changed files with 12 additions and 11 deletions
@@ -69,7 +69,7 @@ public class WhenUtils {
if (!JetPsiUtil.checkWhenExpressionHasSingleElse(whenExpression)) return false;
JetExpression elseBranch = JetPsiUtil.getWhenElseBranch(whenExpression);
JetExpression elseBranch = whenExpression.getElseExpression();
if (!(elseBranch instanceof JetWhenExpression)) return false;
JetWhenExpression nestedWhenExpression = (JetWhenExpression) elseBranch;
@@ -89,7 +89,7 @@ public class WhenUtils {
public static void flattenWhen(@NotNull JetWhenExpression whenExpression) {
JetExpression subjectExpression = whenExpression.getSubjectExpression();
JetExpression elseBranch = JetPsiUtil.getWhenElseBranch(whenExpression);
JetExpression elseBranch = whenExpression.getElseExpression();
assert elseBranch instanceof JetWhenExpression : TRANSFORM_WITHOUT_CHECK;
JetWhenExpression nestedWhenExpression = (JetWhenExpression) elseBranch;