java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/IfStatementSelectioner.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.codeInsight.editorActions.wordSelection;
  17
  18 import com.intellij.psi.PsiElement;
  19 import com.intellij.psi.PsiIfStatement;
  20 import com.intellij.psi.PsiKeyword;
  21 import com.intellij.psi.PsiStatement;
  22 import com.intellij.openapi.util.TextRange;
  23 import com.intellij.openapi.editor.Editor;
  24
  25 import java.util.List;
  26 import java.util.ArrayList;
  27
  28 public class IfStatementSelectioner extends BasicSelectioner {
  29   public boolean canSelect(PsiElement e) {
  30     return e instanceof PsiIfStatement;
  31   }
  32
  33   public List<TextRange> select(PsiElement e, CharSequence editorText, int cursorOffset, Editor editor) {
  34     List<TextRange> result = new ArrayList<TextRange>();
  35     result.addAll(expandToWholeLine(editorText, e.getTextRange(), false));
  36
  37     PsiIfStatement statement = (PsiIfStatement)e;
  38
  39     final PsiKeyword elseKeyword = statement.getElseElement();
  40     if (elseKeyword != null) {
  41       result.addAll(expandToWholeLine(editorText,
  42                                       new TextRange(elseKeyword.getTextRange().getStartOffset(),
  43                                                     statement.getTextRange().getEndOffset()),
  44                                       false));
  45
  46       final PsiStatement branch = statement.getElseBranch();
  47       if (branch instanceof PsiIfStatement) {
  48         PsiIfStatement elseIf = (PsiIfStatement)branch;
  49         final PsiKeyword element = elseIf.getElseElement();
  50         if (element != null) {
  51           result.addAll(expandToWholeLine(editorText,
  52                                           new TextRange(elseKeyword.getTextRange().getStartOffset(),
  53                                                         elseIf.getThenBranch().getTextRange().getEndOffset()),
  54                                           false));
  55         }
  56       }
  57     }
  58
  59     return result;
  60   }
  61 }