Core Editor: Show "+" icon next to new elements

ShadelessFox · Mar 14, 2024 · f1d098e · f1d098e
1 parent a936346
commit f1d098e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/modules/decima-ui/src/main/java/com/shade/decima/ui/editor/core/command/BaseNodeCommand.java b/modules/decima-ui/src/main/java/com/shade/decima/ui/editor/core/command/BaseNodeCommand.java
@@ -20,7 +20,9 @@ public BaseNodeCommand(@NotNull Tree tree, @NotNull CoreNodeObject node) {
     public void redo() {
         super.redo();
 
-        node.setState(CoreNodeObject.State.CHANGED);
+        if (node.getState() == CoreNodeObject.State.UNCHANGED) {
+            node.setState(CoreNodeObject.State.CHANGED);
+        }
     }
 
     @Override

diff --git a/...es/decima-ui/src/main/java/com/shade/decima/ui/editor/core/command/ElementAddCommand.java b/...es/decima-ui/src/main/java/com/shade/decima/ui/editor/core/command/ElementAddCommand.java
@@ -3,6 +3,7 @@
 import com.shade.decima.model.rtti.types.RTTITypeArray;
 import com.shade.decima.ui.editor.core.CoreNodeObject;
 import com.shade.platform.ui.controls.tree.Tree;
+import com.shade.platform.ui.controls.tree.TreeNode;
 import com.shade.util.NotNull;
 
 import javax.swing.tree.TreePath;
@@ -60,7 +61,13 @@ private void perform(boolean redo) {
         node.unloadChildren();
         tree.getModel().fireStructureChanged(node);
 
-        final TreePath path = tree.getModel().getTreePathToRoot(tree.getModel().getChild(node, leaf));
+        final TreeNode child = tree.getModel().getChild(node, leaf);
+
+        if (redo && child instanceof CoreNodeObject object) {
+            object.setState(CoreNodeObject.State.NEW);
+        }
+
+        final TreePath path = tree.getModel().getTreePathToRoot(child);
         tree.setSelectionPath(path);
         tree.scrollPathToVisible(path);
     }