diff options
Diffstat (limited to 'src/main/java/me/laria/code/idea_caseconv/SelectionReplacerAction.java')
-rw-r--r-- | src/main/java/me/laria/code/idea_caseconv/SelectionReplacerAction.java | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/src/main/java/me/laria/code/idea_caseconv/SelectionReplacerAction.java b/src/main/java/me/laria/code/idea_caseconv/SelectionReplacerAction.java new file mode 100644 index 0000000..692fd6c --- /dev/null +++ b/src/main/java/me/laria/code/idea_caseconv/SelectionReplacerAction.java @@ -0,0 +1,59 @@ +package me.laria.code.idea_caseconv; + +import com.intellij.openapi.actionSystem.AnAction; +import com.intellij.openapi.actionSystem.AnActionEvent; +import com.intellij.openapi.actionSystem.CommonDataKeys; +import com.intellij.openapi.command.WriteCommandAction; +import com.intellij.openapi.editor.Caret; +import com.intellij.openapi.editor.CaretModel; +import com.intellij.openapi.editor.Document; +import com.intellij.openapi.editor.Editor; +import com.intellij.openapi.project.Project; + +import java.util.ArrayList; + +abstract class SelectionReplacerAction extends AnAction { + @Override + public void update(final AnActionEvent e) { + //Get required data keys + final Project project = e.getData(CommonDataKeys.PROJECT); + final Editor editor = e.getData(CommonDataKeys.EDITOR); + //Set visibility only in case of existing project and editor and if some text in the editor is selected + e.getPresentation().setVisible( + project != null + && editor != null + && editor.getSelectionModel().hasSelection() + ); + } + + @Override + public void actionPerformed(AnActionEvent anActionEvent) { + Editor editor = anActionEvent.getData(CommonDataKeys.EDITOR); + Project project = anActionEvent.getData(CommonDataKeys.PROJECT); + assert editor != null; + assert project != null; + + CaretModel caretModel = editor.getCaretModel(); + Document document = editor.getDocument(); + + ArrayList<Replacement> replacements = new ArrayList<>(); + for (Caret caret : caretModel.getAllCarets()) { + if (!caret.hasSelection()) { + continue; + } + + replacements.add(new Replacement( + caret.getSelectionStart(), + caret.getSelectionEnd(), + this.replace(caret.getSelectedText()) + )); + } + + // Sort in reverse order so a replacement won't mess up the indices of the other replacements + replacements.sort((o1, o2) -> -o1.compareTo(o2)); + + WriteCommandAction.runWriteCommandAction(project, () -> replacements.forEach(r -> r.doReplace(document))); + } + + abstract protected String replace(String s); +} |