| /* |
| * 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 com.intellij.openapi.editor.actions; |
| |
| import com.intellij.find.*; |
| import com.intellij.find.editorHeaderActions.EditorHeaderAction; |
| import com.intellij.find.editorHeaderActions.SelectAllAction; |
| import com.intellij.openapi.actionSystem.DataContext; |
| import com.intellij.openapi.editor.Caret; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.editor.ScrollType; |
| import com.intellij.openapi.editor.actionSystem.EditorAction; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.util.TextRange; |
| import org.jetbrains.annotations.Nullable; |
| |
| import java.util.Iterator; |
| |
| public class SelectAllOccurrencesAction extends EditorAction { |
| protected SelectAllOccurrencesAction() { |
| super(new Handler()); |
| } |
| |
| private static class Handler extends SelectOccurrencesActionHandler { |
| @Override |
| public boolean isEnabled(Editor editor, DataContext dataContext) { |
| return super.isEnabled(editor, dataContext) && editor.getProject() != null && editor.getCaretModel().supportsMultipleCarets(); |
| } |
| |
| @Override |
| public void doExecute(final Editor editor, @Nullable Caret c, DataContext dataContext) { |
| Caret caret = c == null ? editor.getCaretModel().getPrimaryCaret() : c; |
| |
| boolean wholeWordsSearch = false; |
| if (!caret.hasSelection()) { |
| TextRange wordSelectionRange = getSelectionRange(editor, caret); |
| if (wordSelectionRange != null) { |
| setSelection(editor, caret, wordSelectionRange); |
| wholeWordsSearch = true; |
| } |
| } |
| |
| String selectedText = caret.getSelectedText(); |
| Project project = editor.getProject(); |
| if (project == null || selectedText == null) { |
| return; |
| } |
| |
| int caretShiftFromSelectionStart = caret.getOffset() - caret.getSelectionStart(); |
| final FindManager findManager = FindManager.getInstance(project); |
| |
| final FindModel model = getFindModel(selectedText, wholeWordsSearch); |
| |
| FindUtil.selectSearchResultsInEditor(editor, new Iterator<FindResult>() { |
| FindResult findResult = findManager.findString(editor.getDocument().getCharsSequence(), 0, model); |
| |
| @Override |
| public boolean hasNext() { |
| return findResult.isStringFound(); |
| } |
| |
| @Override |
| public FindResult next() { |
| FindResult result = findResult; |
| findResult = findManager.findString(editor.getDocument().getCharsSequence(), findResult.getEndOffset(), model); |
| return result; |
| } |
| |
| @Override |
| public void remove() { |
| throw new UnsupportedOperationException(); |
| } |
| }, caretShiftFromSelectionStart); |
| editor.getScrollingModel().scrollToCaret(ScrollType.RELATIVE); |
| } |
| } |
| } |