| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.jetbrains.plugins.groovy.editor.selection; |
| |
| import com.intellij.codeInsight.editorActions.ExtendWordSelectionHandlerBase; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.util.TextRange; |
| import com.intellij.psi.PsiElement; |
| import org.jetbrains.plugins.groovy.lang.psi.api.statements.arguments.GrArgumentList; |
| import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrCall; |
| import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrReferenceExpression; |
| |
| import java.util.List; |
| |
| /** |
| * @author ilyas |
| */ |
| public class GroovyArgListSelectioner extends ExtendWordSelectionHandlerBase { |
| @Override |
| public boolean canSelect(PsiElement e) { |
| return e instanceof GrArgumentList || e.getParent() instanceof GrReferenceExpression && e.getParent().getParent() instanceof GrCall; |
| } |
| |
| @Override |
| public List<TextRange> select(PsiElement element, CharSequence editorText, int cursorOffset, Editor editor) { |
| List<TextRange> result = super.select(element, editorText, cursorOffset, editor); |
| |
| if (element instanceof GrArgumentList) { |
| GrArgumentList args = ((GrArgumentList) element); |
| TextRange range = args.getTextRange(); |
| if (range.contains(cursorOffset)) { |
| PsiElement leftParen = args.getLeftParen(); |
| PsiElement rightParen = args.getRightParen(); |
| |
| if (leftParen != null) { |
| int leftOffset = leftParen.getTextOffset(); |
| if (rightParen != null) { |
| if (leftOffset + 1 < rightParen.getTextOffset()) { |
| int rightOffset = rightParen.getTextRange().getEndOffset(); |
| range = new TextRange(leftParen.getTextRange().getStartOffset() + 1, rightOffset - 1); |
| result.add(range); |
| } |
| } else { |
| range = new TextRange(leftParen.getTextRange().getStartOffset() + 1, element.getTextRange().getEndOffset()); |
| result.add(range); |
| } |
| } |
| } |
| } |
| final PsiElement parent = element.getParent(); |
| if (parent instanceof GrReferenceExpression) { |
| final GrArgumentList argumentList = ((GrCall)parent.getParent()).getArgumentList(); |
| final PsiElement refName = ((GrReferenceExpression)parent).getReferenceNameElement(); |
| if (argumentList != null && refName == element) { |
| result.add(new TextRange(refName.getTextRange().getStartOffset(), argumentList.getTextRange().getEndOffset())); |
| } |
| } |
| return result; |
| } |
| } |