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.","");
+    }
+}