From 2a9b28ddc313419a78e0c1ff971800d91ba624aa Mon Sep 17 00:00:00 2001
From: Vladyslav_Prykhodko <vprykhodko@thingsboard.io>
Date: Wed, 3 Nov 2021 17:06:09 +0200
Subject: [PATCH] UI Improvement: added argument updatedData in function
 updateNode in nav-tree component

---
 ui-ngx/src/app/shared/components/nav-tree.component.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ui-ngx/src/app/shared/components/nav-tree.component.ts b/ui-ngx/src/app/shared/components/nav-tree.component.ts
index afac4e78f4..7de7a8902f 100644
--- a/ui-ngx/src/app/shared/components/nav-tree.component.ts
+++ b/ui-ngx/src/app/shared/components/nav-tree.component.ts
@@ -42,7 +42,7 @@ export interface NavTreeEditCallbacks {
   nodeIsOpen?: (id: string) => boolean;
   nodeIsLoaded?: (id: string) => boolean;
   refreshNode?: (id: string) => void;
-  updateNode?: (id: string, newName: string) => void;
+  updateNode?: (id: string, newName: string, updatedData?: any) => void;
   createNode?: (parentId: string, node: NavTreeNode, pos: number | string) => void;
   deleteNode?: (id: string) => void;
   disableNode?: (id: string) => void;
@@ -221,11 +221,14 @@ export class NavTreeComponent implements OnInit {
             }
           }
         };
-        this.editCallbacks.updateNode = (id, newName) => {
+        this.editCallbacks.updateNode = (id, newName, updatedData) => {
           const node: NavTreeNode = this.treeElement.jstree('get_node', id);
           if (node) {
             this.treeElement.jstree('rename_node', node, newName);
           }
+          if (updatedData && node.data) {
+            Object.assign(node.data, updatedData);
+          }
         };
         this.editCallbacks.createNode = (parentId, node, pos) => {
           const parentNode: NavTreeNode = this.treeElement.jstree('get_node', parentId);