fixed adding user data to a linked node.

Author: JavaSaBr

src/main/java/com/ss/editor/ui/component/bar/action/OpenAssetAction.java

@@ -82,7 +82,7 @@ private void openAssetByNative() {
     }
 
     /**
-     * Open asset folder use custom file chooser.
+     * Open an asset folder using custom file chooser.
      */
     private void openAsset() {
 

src/main/java/com/ss/editor/ui/control/model/node/spatial/SpatialTreeNode.java

@@ -65,7 +65,8 @@ public void fillContextMenu(@NotNull final NodeTree<?> nodeTree,
                                 @NotNull final ObservableList<MenuItem> items) {
         if (!(nodeTree instanceof ModelNodeTree)) return;
 
-        final AssetLinkNode linkNode = findParent(getElement(), AssetLinkNode.class::isInstance);
+        final T element = getElement();
+        final AssetLinkNode linkNode = findParent(element, AssetLinkNode.class::isInstance);
 
         if (linkNode == null) {
             final Menu createMenu = createCreationMenu(nodeTree);
@@ -74,12 +75,14 @@ public void fillContextMenu(@NotNull final NodeTree<?> nodeTree,
             if (toolMenu != null) items.add(toolMenu);
         }
 
-        if (canRemove()) items.add(new RemoveNodeAction(nodeTree, this));
-
-        if (linkNode == null) {
+        if (linkNode == null || element == linkNode) {
             items.add(new AddUserDataAction(nodeTree, this));
         }
 
+        if (canRemove()) {
+            items.add(new RemoveNodeAction(nodeTree, this));
+        }
+
         super.fillContextMenu(nodeTree, items);
     }
 

src/main/java/com/ss/editor/ui/control/model/tree/action/AddUserDataAction.java

@@ -27,44 +27,22 @@
 import org.jetbrains.annotations.Nullable;
 
 /**
- * The action to add a new user data.
+ * The action to add new user data.
  *
  * @author JavaSaBr
  */
 public class AddUserDataAction extends AbstractNodeAction<ModelChangeConsumer> {
 
     private enum DataType {
-        /**
-         * Float data type.
-         */
         FLOAT,
-        /**
-         * Integer data type.
-         */
         INTEGER,
-        /**
-         * Vector 3 f data type.
-         */
         VECTOR3F,
-        /**
-         * Vector 2 f data type.
-         */
         VECTOR2F,
-        /**
-         * Color data type.
-         */
         COLOR,
-        /**
-         * Boolean data type.
-         */
         BOOLEAN,
-        /**
-         * String data type.
-         */
-        STRING,
+        STRING
     }
 
-
     @NotNull
     private static final String PROPERTY_NAME = "name";
 
@@ -96,14 +74,37 @@ protected void process() {
         definitions.add(new PropertyDefinition(STRING, Messages.MODEL_PROPERTY_NAME, PROPERTY_NAME, ""));
         definitions.add(new PropertyDefinition(ENUM, Messages.MODEL_PROPERTY_DATA_TYPE, PROPERTY_DATA_TYPE, DataType.STRING));
 
-        final GenericFactoryDialog dialog = new GenericFactoryDialog(definitions, this::addUserData,
-                vars -> !StringUtils.isEmpty(vars.getString(PROPERTY_NAME)));
+        final GenericFactoryDialog dialog = new GenericFactoryDialog(definitions, this::addUserData, this::validate);
 
         dialog.setTitle(Messages.ADD_USER_DATA_DIALOG_TITLE);
         dialog.setButtonOkText(Messages.SIMPLE_DIALOG_BUTTON_ADD);
         dialog.show();
     }
 
+    /**
+     * Validate the input paramaters.
+     *
+     * @param vars the input paramaters.
+     * @return true if all is ok.
+     */
+    @FXThread
+    private boolean validate(@NotNull final VarTable vars) {
+        if(!vars.has(PROPERTY_NAME)) return false;
+
+        final String name = vars.getString(PROPERTY_NAME);
+
+        final TreeNode<?> node = getNode();
+        final Spatial element = (Spatial) node.getElement();
+
+        return !StringUtils.isEmpty(name) && element.getUserData(name) == null;
+    }
+
+    /**
+     * Add new user date.
+     *
+     * @param vars the input paramaters.
+     */
+    @FXThread
     private void addUserData(@NotNull final VarTable vars) {
 
         final String name = vars.get(PROPERTY_NAME);