plugins/groovy/src/org/jetbrains/plugins/groovy/editor/selection/GroovyArgListSelectioner.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
  17 package org.jetbrains.plugins.groovy.editor.selection;
  18
  19 import com.intellij.openapi.editor.Editor;
  20 import com.intellij.openapi.util.TextRange;
  21 import com.intellij.psi.PsiElement;
  22 import org.jetbrains.plugins.groovy.lang.psi.api.statements.arguments.GrArgumentList;
  23 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrCall;
  24 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrReferenceExpression;
  25
  26 import java.util.List;
  27
  28 /**
  29  * @author ilyas
  30  */
  31 public class GroovyArgListSelectioner extends GroovyBasicSelectioner {
  32   public boolean canSelect(PsiElement e) {
  33     return e instanceof GrArgumentList || e.getParent() instanceof GrReferenceExpression && e.getParent().getParent() instanceof GrCall;
  34   }
  35
  36   public List<TextRange> select(PsiElement element, CharSequence editorText, int cursorOffset, Editor editor) {
  37     List<TextRange> result = super.select(element, editorText, cursorOffset, editor);
  38
  39     if (element instanceof GrArgumentList) {
  40       GrArgumentList args = ((GrArgumentList) element);
  41       TextRange range = args.getTextRange();
  42       if (range.contains(cursorOffset)) {
  43         PsiElement leftParen = args.getLeftParen();
  44         PsiElement rightParen = args.getRightParen();
  45
  46         if (leftParen != null) {
  47           int leftOffset = leftParen.getTextOffset();
  48           if (rightParen != null) {
  49             if (leftOffset + 1 < rightParen.getTextOffset()) {
  50               int rightOffset = rightParen.getTextRange().getEndOffset();
  51               range = new TextRange(leftParen.getTextRange().getStartOffset() + 1, rightOffset - 1);
  52               result.add(range);
  53             }
  54           } else {
  55             range = new TextRange(leftParen.getTextRange().getStartOffset() + 1, element.getTextRange().getEndOffset());
  56             result.add(range);
  57           }
  58         }
  59       }
  60     }
  61     final PsiElement parent = element.getParent();
  62     if (parent instanceof GrReferenceExpression) {
  63       final GrArgumentList argumentList = ((GrCall)parent.getParent()).getArgumentList();
  64       final PsiElement refName = ((GrReferenceExpression)parent).getReferenceNameElement();
  65       if (argumentList != null && refName == element) {
  66         result.add(new TextRange(refName.getTextRange().getStartOffset(), argumentList.getTextRange().getEndOffset()));
  67       }
  68     }
  69     return result;
  70   }
  71 }