Add new action to expand/collapse tree

Author: oshando

swan_assist/src/main/java/de/fraunhofer/iem/swan/assist/actions/ExpandAllAction.java

@@ -0,0 +1,51 @@
+package de.fraunhofer.iem.swan.assist.actions;
+
+import com.intellij.openapi.actionSystem.AnAction;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.util.messages.MessageBus;
+import de.fraunhofer.iem.swan.assist.comm.ExpandNotifier;
+import de.fraunhofer.iem.swan.assist.data.JSONFileLoader;
+import de.fraunhofer.iem.swan.assist.ui.MethodListTree;
+import icons.PluginIcons;
+import org.jetbrains.annotations.NotNull;
+
+/**
+ * @author Oshando Johnson on 2020-01-07
+ */
+public class ExpandAllAction extends AnAction {
+
+    @Override
+    public void actionPerformed(@NotNull AnActionEvent e) {
+
+        MessageBus messageBus = ApplicationManager.getApplication().getMessageBus();
+        ExpandNotifier publisher = messageBus.syncPublisher(ExpandNotifier.EXPAND_COLLAPSE_LIST);
+        publisher.expandTree(true);
+
+    }
+
+    /**
+     * Controls whether the action is enabled or disabled
+     * @param event source  event
+     */
+    @Override
+    public void update(@NotNull AnActionEvent event) {
+
+        //Disable/Enable action button
+        if (JSONFileLoader.isFileSelected())
+            event.getPresentation().setEnabled(true);
+        else
+            event.getPresentation().setEnabled(false);
+
+
+        //Disable/Enable action button
+        if (MethodListTree.TREE_EXPANDED){
+            event.getPresentation().setIcon(PluginIcons.COLLAPSE);
+            event.getPresentation().setText("Collapse Tree");
+        } else{
+            event.getPresentation().setIcon(PluginIcons.EXPAND);
+            event.getPresentation().setText("Expand Tree");
+        }
+    }
+
+}

swan_assist/src/main/java/de/fraunhofer/iem/swan/assist/comm/ExpandNotifier.java

@@ -0,0 +1,13 @@
+package de.fraunhofer.iem.swan.assist.comm;
+
+import com.intellij.util.messages.Topic;
+
+/**
+ * @author Oshando Johnson on 2020-01-07
+ */
+public interface ExpandNotifier {
+
+    Topic<ExpandNotifier> EXPAND_COLLAPSE_LIST = Topic.create("Expand list", ExpandNotifier.class);
+
+    void expandTree(boolean expand);
+}

swan_assist/src/main/java/icons/PluginIcons.java

@@ -24,6 +24,8 @@ public interface PluginIcons {
     Icon ADD_METHOD_ACTION = IconLoader.getIcon("/icons/add.png");
     Icon EXPORT_ACTION = IconLoader.getIcon("/icons/export.png");
     Icon SUGGEST_ACTION = IconLoader.getIcon("/icons/suggest.png");
+    Icon EXPAND = IconLoader.getIcon("/icons/expand.png");
+    Icon COLLAPSE = IconLoader.getIcon("/icons/collapse.png");
 
     Icon SOURCE = IconLoader.getIcon("/icons/sou.png");
     Icon SANITIZER = IconLoader.getIcon("/icons/san.png");
@@ -34,6 +36,5 @@ public interface PluginIcons {
     Icon CWE = IconLoader.getIcon("/icons/cwe.png");
     Icon DEFAULT = IconLoader.getIcon("/icons/default.png");
 
-
     Icon SELECTED = IconLoader.getIcon("/icons/selected.png");
 }

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

@@ -48,6 +48,11 @@
                    text="Filter Methods"
                    popup="true" icon="PluginIcons.FILTER_ACTION">
             </group>
+            <action id="SWAN_Assist.ExpandAll" class="de.fraunhofer.iem.swan.assist.actions.ExpandAllAction"
+                    text="Expand All"
+                    icon="PluginIcons.EXPAND"
+                    description="Expand or collapse the method tree">
+            </action>
             <separator/>
 
             <action id="SWAN_Assist.LaunchSWANAction"