Add positioning function

Author: shuzijun

src/main/java/com/shuzijun/leetcode/plugin/actions/editor/AbstractEditAction.java

@@ -1,8 +1,9 @@
 package com.shuzijun.leetcode.plugin.actions.editor;
 
 import com.intellij.openapi.actionSystem.AnActionEvent;
-import com.intellij.openapi.actionSystem.PlatformDataKeys;
+import com.intellij.openapi.fileEditor.FileEditorManager;
 import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.util.ArrayUtil;
 import com.shuzijun.leetcode.plugin.actions.AbstractAction;
 import com.shuzijun.leetcode.plugin.manager.ViewManager;
 import com.shuzijun.leetcode.plugin.model.Config;
@@ -19,7 +20,10 @@ abstract class AbstractEditAction extends AbstractAction {
 
     @Override
     public void actionPerformed(AnActionEvent anActionEvent, Config config) {
-        VirtualFile vf = anActionEvent.getData(PlatformDataKeys.VIRTUAL_FILE);
+        VirtualFile vf = ArrayUtil.getFirstElement(FileEditorManager.getInstance(anActionEvent.getProject()).getSelectedFiles());
+        if(vf == null){
+            return;
+        }
         LeetcodeEditor leetcodeEditor = ProjectConfig.getInstance(anActionEvent.getProject()).getEditor(vf.getPath());
         if (leetcodeEditor == null) {
             return;

src/main/java/com/shuzijun/leetcode/plugin/actions/editor/PositionAction.java

@@ -0,0 +1,52 @@
+package com.shuzijun.leetcode.plugin.actions.editor;
+
+
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.fileEditor.FileEditorManager;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.ui.components.JBScrollPane;
+import com.intellij.util.ArrayUtil;
+import com.shuzijun.leetcode.plugin.manager.ViewManager;
+import com.shuzijun.leetcode.plugin.model.Config;
+import com.shuzijun.leetcode.plugin.model.LeetcodeEditor;
+import com.shuzijun.leetcode.plugin.model.Question;
+import com.shuzijun.leetcode.plugin.setting.ProjectConfig;
+import com.shuzijun.leetcode.plugin.utils.DataKeys;
+import com.shuzijun.leetcode.plugin.window.WindowFactory;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+
+/**
+ * @author shuzijun
+ */
+public class PositionAction extends AbstractEditAction {
+
+    @Override
+    public void update(@NotNull AnActionEvent e) {
+        VirtualFile vf = ArrayUtil.getFirstElement(FileEditorManager.getInstance(e.getProject()).getSelectedFiles());
+        if (vf == null) {
+            e.getPresentation().setEnabled(false);
+            return;
+        }
+        LeetcodeEditor leetcodeEditor = ProjectConfig.getInstance(e.getProject()).getEditor(vf.getPath());
+        if (leetcodeEditor == null) {
+            e.getPresentation().setEnabled(false);
+            return;
+        }
+        e.getPresentation().setEnabled(true);
+    }
+
+    @Override
+    public void actionPerformed(AnActionEvent anActionEvent, Config config, Question question) {
+        JTree tree = WindowFactory.getDataContext(anActionEvent.getProject()).getData(DataKeys.LEETCODE_PROJECTS_TREE);
+        if (tree == null) {
+            return;
+        }
+        JBScrollPane scrollPane = WindowFactory.getDataContext(anActionEvent.getProject()).getData(DataKeys.LEETCODE_PROJECTS_SCROLL);
+        ApplicationManager.getApplication().invokeAndWait(() -> {
+            ViewManager.position(tree, scrollPane, question);
+        });
+    }
+}

src/main/java/com/shuzijun/leetcode/plugin/manager/ViewManager.java

@@ -194,7 +194,7 @@ public static void pick(JTree tree, JBScrollPane scrollPane) {
         return;
     }
 
-    public static Question getTreeQuestion(JTree tree,Project project) {
+    public static Question getTreeQuestion(JTree tree, Project project) {
         Question question = null;
         if (tree != null) {
             DefaultMutableTreeNode note = (DefaultMutableTreeNode) tree.getLastSelectedPathComponent();
@@ -216,8 +216,8 @@ public static Question getTreeQuestion(JTree tree,Project project) {
         return question;
     }
 
-    public static Question getQuestionById(String id,Project project) {
-        if(question.isEmpty()){
+    public static Question getQuestionById(String id, Project project) {
+        if (question.isEmpty()) {
             MessageUtils.getInstance(project).showInfoMsg("info", PropertiesUtils.getInfo("tree.load"));
             ApplicationManager.getApplication().invokeAndWait(new Runnable() {
                 @Override
@@ -248,4 +248,32 @@ private static void addChild(DefaultMutableTreeNode rootNode, List<Tag> Lists, M
         }
     }
 
+    public static void position(JTree tree, JBScrollPane scrollPane, Question question) {
+
+        DefaultTreeModel treeMode = (DefaultTreeModel) tree.getModel();
+        DefaultMutableTreeNode root = (DefaultMutableTreeNode) treeMode.getRoot();
+        if (root.isLeaf()) {
+            return;
+        }
+        DefaultMutableTreeNode node = (DefaultMutableTreeNode) root.getChildAt(0);
+        if (node.isLeaf()) {
+            return;
+        }
+
+        for (int i = 0, j = node.getChildCount(); i < j; i++) {
+            DefaultMutableTreeNode childNode = (DefaultMutableTreeNode) node.getChildAt(i);
+            Question nodeData = (Question) childNode.getUserObject();
+            if(nodeData.getQuestionId().equals(question.getQuestionId())){
+                TreePath toShowPath = new TreePath(childNode.getPath());
+                tree.setSelectionPath(toShowPath);
+                Rectangle bounds = tree.getPathBounds(toShowPath);
+                Point point = new Point(0, (int) bounds.getY());
+                JViewport viewport = scrollPane.getViewport();
+                viewport.setViewPosition(point);
+                return;
+            }
+
+        }
+    }
+
 }

src/main/java/icons/LeetCodeEditorIcons.java

@@ -25,6 +25,7 @@ public interface LeetCodeEditorIcons {
     Icon LOGIN = IconLoader.getIcon("/icons/login.png");
     Icon LOGOUT = IconLoader.getIcon("/icons/logout.png");
     Icon POPUP = IconLoader.getIcon("/icons/popup.png");
+    Icon POSITION = IconLoader.getIcon("/icons/position.png");
     Icon PROGRESS = IconLoader.getIcon("/icons/progress.png");
     Icon QUESTION = IconLoader.getIcon("/icons/question.png");
     Icon RANDOM = IconLoader.getIcon("/icons/random.png");

src/main/resources/META-INF/plugin.xml

@@ -347,7 +347,7 @@
         <!--    <postStartupActivity  implementation = "com.shuzijun.leetcode.plugin.listener.RegisterPluginInstallerStateListener"></postStartupActivity>-->
         <errorHandler implementation="com.shuzijun.leetcode.plugin.listener.ErrorReportHandler"/>
         <toolWindow id="leetcode" secondary="true" icon="LeetCodeEditorIcons.LEETCODE_TOOL_WINDOW" anchor="right"
-                    factoryClass="com.shuzijun.leetcode.plugin.window.WindowFactory"/>
+                    factoryClass="com.shuzijun.leetcode.plugin.window.WindowFactory" />
         <applicationService serviceInterface="com.shuzijun.leetcode.plugin.setting.PersistentConfig"
                             serviceImplementation="com.shuzijun.leetcode.plugin.setting.PersistentConfig"/>
         <projectService serviceInterface="com.shuzijun.leetcode.plugin.setting.ProjectConfig"
@@ -424,6 +424,8 @@
         <action id="leetcode.PickAction" class="com.shuzijun.leetcode.plugin.actions.toolbar.PickAction"
                 text="Pick one" description="Pick one" icon="LeetCodeEditorIcons.RANDOM">
         </action>
+        <action id="leetcode.positionAction" class="com.shuzijun.leetcode.plugin.actions.editor.PositionAction"
+                text="position" description="position" icon="LeetCodeEditorIcons.POSITION"/>
 
         <group id="leetcode.NavigatorActionsToolbar">
             <reference id="leetcode.LoginAction"/>
@@ -432,6 +434,7 @@
             <reference id="leetcode.RefreshAction"/>
             <reference id="leetcode.PickAction"/>
             <reference id="leetcode.FindAction"/>
+            <reference id="leetcode.positionAction"/>
             <reference id="leetcode.CollapseAction"/>
             <reference id="leetcode.ProgressAction"/>
             <separator/>
@@ -526,6 +529,7 @@
                     text="open in web" description="open in web(editor)" icon="LeetCodeEditorIcons.POPUP">
             </action>
             <separator/>
+            <reference id="leetcode.positionAction"/>
             <group id="leetcode.editor.timer" popup="true" text="Timer" description="timer(editor)" icon="LeetCodeEditorIcons.TIME">
                 <action id="leetcode.editor.StartTimeAction"
                         class="com.shuzijun.leetcode.plugin.actions.editor.StartTimeAction"