From b7bc251311c4a01c081f749a7689334730ffb9bb Mon Sep 17 00:00:00 2001
From: Alexey Sedunov <alexey.sedunov@jetbrains.com>
Date: Fri, 26 Apr 2013 13:36:14 +0400
Subject: [PATCH] Add extraction of 'else' branch of 'when' expression

---
 .../src/org/jetbrains/jet/lang/psi/JetPsiUtil.java       | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
index 73364aa0f56..e5c880dc45b 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPsiUtil.java
@@ -693,6 +693,15 @@ public class JetPsiUtil {
         return (elseCount == 1);
     }
 
+    public static JetExpression getWhenElseBranch(@NotNull JetWhenExpression whenExpression) {
+        for (JetWhenEntry entry : whenExpression.getEntries()) {
+            if (entry.isElse()) {
+                return entry.getExpression();
+            }
+        }
+        return null;
+    }
+
     public static PsiElement skipTrailingWhitespacesAndComments(PsiElement element)  {
         return PsiTreeUtil.skipSiblingsForward(element, PsiWhiteSpace.class, PsiComment.class);
     }