optimization of sorting

Author: shuzijun

src/main/java/com/shuzijun/leetcode/plugin/actions/toolbar/FindActionGroup.java

@@ -2,11 +2,13 @@
 
 import com.google.common.collect.Lists;
 import com.intellij.openapi.actionSystem.ActionGroup;
+import com.intellij.openapi.actionSystem.ActionToolbar;
 import com.intellij.openapi.actionSystem.AnAction;
 import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.shuzijun.leetcode.plugin.manager.ViewManager;
 import com.shuzijun.leetcode.plugin.model.Constant;
 import com.shuzijun.leetcode.plugin.model.Tag;
+import com.shuzijun.leetcode.plugin.utils.DataKeys;
 import icons.LeetCodeEditorIcons;
 
 import java.util.List;
@@ -23,18 +25,21 @@ public boolean displayTextInToolbar() {
 
     @Override
     public void update(AnActionEvent e) {
+        ActionToolbar findToolbar = e.getDataContext().getData(DataKeys.LEETCODE_TOOLBAR_FIND);
         String id = e.getActionManager().getId(this);
         List<Tag> tags = getTags(id);
 
         if (tags != null && !tags.isEmpty()) {
             for (Tag tag : tags) {
                 if(tag.isSelect()){
                     e.getPresentation().setIcon(LeetCodeEditorIcons.FILTER);
+                    findToolbar.getComponent().updateUI();
                     return;
                 }
             }
         }
         e.getPresentation().setIcon(null);
+        findToolbar.getComponent().updateUI();
     }
 
 

src/main/java/com/shuzijun/leetcode/plugin/actions/toolbar/SortAction.java

@@ -0,0 +1,67 @@
+package com.shuzijun.leetcode.plugin.actions.toolbar;
+
+import com.intellij.openapi.actionSystem.ActionToolbar;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.shuzijun.leetcode.plugin.actions.AbstractAction;
+import com.shuzijun.leetcode.plugin.manager.ViewManager;
+import com.shuzijun.leetcode.plugin.model.Config;
+import com.shuzijun.leetcode.plugin.model.Constant;
+import com.shuzijun.leetcode.plugin.model.Sort;
+import com.shuzijun.leetcode.plugin.utils.DataKeys;
+import com.shuzijun.leetcode.plugin.window.WindowFactory;
+import icons.LeetCodeEditorIcons;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.*;
+
+/**
+ * @author shuzijun
+ */
+public class SortAction extends AbstractAction {
+
+    @Override
+    public boolean displayTextInToolbar() {
+        return true;
+    }
+
+    @Override
+    public void update(@NotNull AnActionEvent e) {
+        Sort sort = getSort(e);
+        if (sort == null) {
+            return;
+        }
+        ActionToolbar sortToolbar = e.getDataContext().getData(DataKeys.LEETCODE_TOOLBAR_SORT);
+        if (sort.getType() == Constant.SORT_ASC) {
+            e.getPresentation().setIcon(LeetCodeEditorIcons.SORT_ASC);
+        } else if (sort.getType() == Constant.SORT_DESC) {
+            e.getPresentation().setIcon(LeetCodeEditorIcons.SORT_DESC);
+        } else {
+            e.getPresentation().setIcon(null);
+        }
+        sortToolbar.getComponent().updateUI();
+        super.update(e);
+
+    }
+
+    @Override
+    public void actionPerformed(AnActionEvent anActionEvent, Config config) {
+        JTree tree = WindowFactory.getDataContext(anActionEvent.getProject()).getData(DataKeys.LEETCODE_PROJECTS_TREE);
+        if (tree == null || ViewManager.getFilter(Constant.FIND_TYPE_DIFFICULTY) == null) {
+            return;
+        }
+        Sort sort = getSort(anActionEvent);
+        if (sort == null) {
+            return;
+        }
+        ViewManager.operationType(getKey(anActionEvent));
+        ViewManager.update(tree);
+    }
+
+    private Sort getSort(AnActionEvent anActionEvent) {
+        return ViewManager.getSort(getKey(anActionEvent));
+    }
+
+    private String getKey(AnActionEvent anActionEvent){
+        return anActionEvent.getActionManager().getId(this).replace("leetcode.sort.","");
+    }
+}

src/main/java/com/shuzijun/leetcode/plugin/actions/tree/AbstractSortAction.java

@@ -8,7 +8,6 @@
 import com.intellij.openapi.project.Project;
 import com.intellij.ui.treeStructure.SimpleTree;
 import com.shuzijun.leetcode.plugin.manager.CodeManager;
-import com.shuzijun.leetcode.plugin.model.Constant;
 import com.shuzijun.leetcode.plugin.model.Question;
 import org.jetbrains.annotations.NotNull;
 
@@ -45,27 +44,20 @@ public void mouseClicked(MouseEvent e) {
                 if (e.getButton() == 3) { //鼠标右键
                     final ActionManager actionManager = ActionManager.getInstance();
                     final ActionGroup actionGroup = (ActionGroup) actionManager.getAction("leetcode.NavigatorActionsMenu");
-
                     if (actionGroup != null) {
                         actionManager.createActionPopupMenu("", actionGroup).getComponent().show(e.getComponent(), e.getX(), e.getY());
                     }
                 } else if (e.getButton() == MouseEvent.BUTTON1 && e.getClickCount() == 2) {
-                    ProgressManager.getInstance().run(new Task.Backgroundable(project, "leetcode.editor.openCode", false) {
+                    ProgressManager.getInstance().run(new Task.Backgroundable(project,"leetcode.editor.openCode",false) {
                         @Override
                         public void run(@NotNull ProgressIndicator progressIndicator) {
                             CodeManager.openCode(question, project);
                         }
                     });
                 }
-            } else if (Constant.NODETYPE_TAG.equals(question.getNodeType()) || Constant.NODETYPE_PROBLEMS.equals(question.getNodeType())) {
-                if (e.getButton() == 3) { //鼠标右键
-                    final ActionManager actionManager = ActionManager.getInstance();
-                    final ActionGroup actionGroup = (ActionGroup) actionManager.getAction("leetcode.editor.tree.menu");
-                    if (actionGroup != null) {
-                        actionManager.createActionPopupMenu("", actionGroup).getComponent().show(e.getComponent(), e.getX(), e.getY());
-                    }
-                }
             }
         }
+
+
     }
 }