1.13 Add Copy With Line Comment & Add Line Comment | 增加带行末注释复制和添加行末注释

2022-04-10 19:42:26 +08:00
parent 595d3730d4
commit baa50ba361
29 changed files with 385 additions and 185 deletions
--- a/src/main/java/io/github/linwancen/plugin/show/LineEndCopy.java
+++ b/src/main/java/io/github/linwancen/plugin/show/LineEndCopy.java
@@ -0,0 +1,63 @@
+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.editor.Caret;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.ide.CopyPasteManager;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.FileViewProvider;
+import com.intellij.psi.PsiFile;
+import io.github.linwancen.plugin.show.line.FileViewToDocStrUtils;
+import io.github.linwancen.plugin.show.settings.AppSettingsState;
+import org.jetbrains.annotations.NotNull;
+
+import java.awt.datatransfer.StringSelection;
+
+/**
+ * on EditorPopupMenu
+ */
+public class LineEndCopy extends AnAction {
+
+    @Override
+    public void actionPerformed(@NotNull AnActionEvent event) {
+        Project project = event.getProject();
+        VirtualFile file = event.getData(CommonDataKeys.VIRTUAL_FILE);
+        if (file == null) {
+            return;
+        }
+        PsiFile psiFile = event.getData(CommonDataKeys.PSI_FILE);
+        if (psiFile == null) {
+            return;
+        }
+        FileViewProvider viewProvider = psiFile.getViewProvider();
+        Document document = viewProvider.getDocument();
+        if (document == null) {
+            return;
+        }
+        int startLine = 0;
+        int endLine = document.getLineCount() - 1;
+
+        // if select
+        Editor editor = event.getData(CommonDataKeys.EDITOR);
+        if (editor != null) {
+            Caret primaryCaret = editor.getCaretModel().getPrimaryCaret();
+            int start = primaryCaret.getSelectionStart();
+            int end = primaryCaret.getSelectionEnd();
+            if (start != end) {
+                startLine = document.getLineNumber(start);
+                endLine = document.getLineNumber(end);
+            }
+        }
+
+        AppSettingsState settings = AppSettingsState.getInstance();
+
+        String textWithDoc = FileViewToDocStrUtils.textWithDoc(settings, document,
+                startLine, endLine, project, file, viewProvider);
+        StringSelection content = new StringSelection(textWithDoc);
+        CopyPasteManager.getInstance().setContents(content);
+    }
+}