fix: servers toolbar actions state

le-yams · Apr 30, 2024 · 8674349 · 8674349
1 parent 5dc89d9
commit 8674349
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/main/java/com/github/le_yams/openfga4intellij/servers/ui/ServersTreePanel.java b/src/main/java/com/github/le_yams/openfga4intellij/servers/ui/ServersTreePanel.java
@@ -46,15 +46,18 @@ public ServersTreePanel(ToolWindow toolWindow, OpenFGAServers servers) {
 
         var toolbarDecorator = ToolbarDecorator.createDecorator(tree);
 
+        toolbarDecorator.setAddActionUpdater(updater -> isSelectedServerNode(NodeType.ROOT_NODE));
         toolbarDecorator.setAddAction(this::addAction);
 
-        toolbarDecorator.setEditActionUpdater(updater -> getSelectedServerNode().isPresent());
+        toolbarDecorator.setEditActionUpdater(updater -> isSelectedServerNode(SERVER_NODE));
         toolbarDecorator.setEditAction(this::editAction);
 
-        toolbarDecorator.setRemoveActionUpdater(updater -> getSelectedServerNode().isPresent());
+        toolbarDecorator.setRemoveActionUpdater(updater -> isSelectedServerNode(SERVER_NODE));
         toolbarDecorator.setRemoveAction(this::removeAction);
 
+        toolbarDecorator.setMoveDownActionUpdater(updater -> isSelectedServerNode(SERVER_NODE));
         toolbarDecorator.setMoveDownAction(this::moveDownAction);
+        toolbarDecorator.setMoveUpActionUpdater(updater -> isSelectedServerNode(SERVER_NODE));
         toolbarDecorator.setMoveUpAction(this::moveUpAction);
 
         add(toolbarDecorator.createPanel(), BorderLayout.CENTER);
@@ -163,4 +166,10 @@ private Optional<ServerNode> getSelectedServerNode() {
                 .findFirst()
                 .map(ServerNode.class::cast);
     }
+
+    private boolean isSelectedServerNode(NodeType nodeType) {
+        return Arrays
+                .stream(tree.getSelectedNodes(TreeNode.class, node -> node.is(nodeType)))
+                .findAny().isPresent();
+    }
 }