74 lines
3.0 KiB
Java
74 lines
3.0 KiB
Java
package io.github.linwancen.plugin.show;
|
|
|
|
import com.intellij.openapi.actionSystem.AnAction;
|
|
import com.intellij.openapi.actionSystem.AnActionEvent;
|
|
import com.intellij.openapi.actionSystem.CommonDataKeys;
|
|
import com.intellij.openapi.application.ApplicationManager;
|
|
import com.intellij.openapi.command.WriteCommandAction;
|
|
import com.intellij.openapi.editor.Document;
|
|
import com.intellij.openapi.project.Project;
|
|
import com.intellij.openapi.ui.popup.JBPopupFactory;
|
|
import com.intellij.openapi.ui.popup.ListPopup;
|
|
import com.intellij.openapi.vfs.VfsUtilCore;
|
|
import com.intellij.openapi.vfs.VirtualFile;
|
|
import com.intellij.openapi.vfs.VirtualFileVisitor;
|
|
import com.intellij.psi.FileViewProvider;
|
|
import com.intellij.psi.PsiManager;
|
|
import io.github.linwancen.plugin.show.line.FileViewToDocStrUtils;
|
|
import io.github.linwancen.plugin.show.settings.AppSettingsState;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
/**
|
|
* on ProjectViewPopupMenu
|
|
*/
|
|
public class LineEndAdd extends AnAction {
|
|
|
|
@Override
|
|
public void actionPerformed(@NotNull AnActionEvent event) {
|
|
Project project = event.getProject();
|
|
if (project == null) {
|
|
return;
|
|
}
|
|
VirtualFile[] files = event.getData(CommonDataKeys.VIRTUAL_FILE_ARRAY);
|
|
if (files == null) {
|
|
return;
|
|
}
|
|
ListPopup confirmation = JBPopupFactory.getInstance().createConfirmation(
|
|
"Add Line Comment?", "Add and replace files!", "Don't add.",
|
|
() -> ApplicationManager.getApplication().runReadAction(() -> addDocAll(project, files)), 2);
|
|
confirmation.showInFocusCenter();
|
|
}
|
|
|
|
private void addDocAll(@NotNull Project project, @NotNull VirtualFile[] files) {
|
|
AppSettingsState settings = AppSettingsState.getInstance();
|
|
for (VirtualFile file : files) {
|
|
VfsUtilCore.visitChildrenRecursively(file, new VirtualFileVisitor<Void>() {
|
|
@Override
|
|
public boolean visitFile(@NotNull VirtualFile file) {
|
|
if (!file.isDirectory()) {
|
|
ApplicationManager.getApplication().runReadAction(() -> addDoc(project, file, settings));
|
|
}
|
|
return true;
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
private void addDoc(@NotNull Project project, @NotNull VirtualFile file, @NotNull AppSettingsState settings) {
|
|
FileViewProvider viewProvider = PsiManager.getInstance(project).findViewProvider(file);
|
|
if (viewProvider == null) {
|
|
return;
|
|
}
|
|
Document document = viewProvider.getDocument();
|
|
if (document == null) {
|
|
return;
|
|
}
|
|
int startLine = 0;
|
|
int endLine = document.getLineCount() - 1;
|
|
String textWithDoc = FileViewToDocStrUtils.textWithDoc(settings, document,
|
|
startLine, endLine, project, file, viewProvider);
|
|
WriteCommandAction.runWriteCommandAction(project, () ->
|
|
document.replaceString(0, document.getTextLength() - 1, textWithDoc)
|
|
);
|
|
}
|
|
} |