-
Notifications
You must be signed in to change notification settings - Fork 44
/
CsvChangeSeparatorAction.java
55 lines (49 loc) · 2.33 KB
/
CsvChangeSeparatorAction.java
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
package net.seesharpsoft.intellij.plugins.csv.actions;
import com.intellij.lang.Language;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.actionSystem.PlatformDataKeys;
import com.intellij.openapi.actionSystem.ToggleAction;
import com.intellij.openapi.components.ServiceManager;
import com.intellij.openapi.fileEditor.FileEditor;
import com.intellij.psi.PsiFile;
import com.intellij.util.FileContentUtilCore;
import net.seesharpsoft.intellij.plugins.csv.CsvLanguage;
import net.seesharpsoft.intellij.plugins.csv.CsvSeparatorHolder;
import net.seesharpsoft.intellij.plugins.csv.CsvValueSeparator;
import net.seesharpsoft.intellij.plugins.csv.components.CsvFileAttributes;
import org.jetbrains.annotations.NotNull;
public class CsvChangeSeparatorAction extends ToggleAction {
private CsvValueSeparator mySeparator;
CsvChangeSeparatorAction(CsvValueSeparator separator) {
super(separator.getDisplay());
mySeparator = separator;
}
@Override
public boolean isSelected(@NotNull AnActionEvent anActionEvent) {
PsiFile psiFile = anActionEvent.getData(CommonDataKeys.PSI_FILE);
if (psiFile == null) {
return false;
}
CsvFileAttributes csvFileAttributes = ServiceManager.getService(psiFile.getProject(), CsvFileAttributes.class);
return csvFileAttributes.getFileSeparator(psiFile) != null && csvFileAttributes.getFileSeparator(psiFile).equals(mySeparator);
}
@Override
public void setSelected(@NotNull AnActionEvent anActionEvent, boolean selected) {
PsiFile psiFile = anActionEvent.getData(CommonDataKeys.PSI_FILE);
if (psiFile == null) {
return;
}
Language language = psiFile.getLanguage();
if (!language.isKindOf(CsvLanguage.INSTANCE) || language instanceof CsvSeparatorHolder) {
return;
}
CsvFileAttributes csvFileAttributes = ServiceManager.getService(psiFile.getProject(), CsvFileAttributes.class);
csvFileAttributes.setFileSeparator(psiFile, this.mySeparator);
FileContentUtilCore.reparseFiles(psiFile.getVirtualFile());
FileEditor fileEditor = anActionEvent.getData(PlatformDataKeys.FILE_EDITOR);
if (fileEditor != null) {
fileEditor.selectNotify();
}
}
}