1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* 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.openapi.actionSystem.DataContext;
import com.intellij.openapi.actionSystem.IdeActions;
import com.intellij.openapi.editor.Caret;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.EditorLastActionTracker;
import com.intellij.openapi.editor.ScrollType;
import com.intellij.openapi.editor.actionSystem.EditorActionHandler;
import com.intellij.openapi.util.Key;
import com.intellij.util.containers.HashSet;
import org.jetbrains.annotations.Nullable;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
public class CloneCaretActionHandler extends EditorActionHandler {
private static final Key<Integer> LEVEL = Key.create("CloneCaretActionHandler.level");
private static final Set<String> OUR_ACTIONS = new HashSet<String>(Arrays.asList(
IdeActions.ACTION_EDITOR_CLONE_CARET_ABOVE,
IdeActions.ACTION_EDITOR_CLONE_CARET_BELOW,
IdeActions.ACTION_EDITOR_MOVE_CARET_LEFT_WITH_SELECTION,
IdeActions.ACTION_EDITOR_MOVE_CARET_RIGHT_WITH_SELECTION
));
private final boolean myCloneAbove;
public CloneCaretActionHandler(boolean above) {
myCloneAbove = above;
}
@Override
public boolean isEnabled(Editor editor, DataContext dataContext) {
return editor.getCaretModel().supportsMultipleCarets();
}
@Override
protected void doExecute(Editor editor, @Nullable Caret targetCaret, DataContext dataContext) {
if (targetCaret != null) {
targetCaret.clone(myCloneAbove);
return;
}
int currentLevel = 0;
List<Caret> currentCarets = new ArrayList<Caret>();
for (Caret caret : editor.getCaretModel().getAllCarets()) {
int level = getLevel(caret);
if (Math.abs(level) > Math.abs(currentLevel)) {
currentLevel = level;
currentCarets.clear();
}
if (Math.abs(level) == Math.abs(currentLevel)) {
currentCarets.add(caret);
}
}
boolean removeCarets = currentLevel > 0 && myCloneAbove || currentLevel < 0 && !myCloneAbove;
Integer newLevel = myCloneAbove ? currentLevel - 1 : currentLevel + 1;
for (Caret caret : currentCarets) {
if (removeCarets) {
editor.getCaretModel().removeCaret(caret);
}
else {
Caret clone = caret;
do {
Caret original = clone;
clone = clone.clone(myCloneAbove);
if (original != caret) {
editor.getCaretModel().removeCaret(original);
}
} while (clone != null && caret.hasSelection() && !clone.hasSelection());
if (clone != null) {
clone.putUserData(LEVEL, newLevel);
}
}
}
if (removeCarets) {
editor.getScrollingModel().scrollToCaret(ScrollType.RELATIVE);
}
}
private static int getLevel(Caret caret) {
if (isRepeatedActionInvocation()) {
Integer value = caret.getUserData(LEVEL);
return value == null ? 0 : value;
}
else {
caret.putUserData(LEVEL, null);
return 0;
}
}
private static boolean isRepeatedActionInvocation() {
String lastActionId = EditorLastActionTracker.getInstance().getLastActionId();
return OUR_ACTIONS.contains(lastActionId);
}
}
|
