TreeGrid: new data api methods #1714

enonic · Jan 14, 2021 · 57b8750 · 57b8750
1 parent b5b4e0f
commit 57b8750
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/main/resources/assets/admin/common/js/ui/treegrid/TreeGrid.ts b/src/main/resources/assets/admin/common/js/ui/treegrid/TreeGrid.ts
@@ -1725,10 +1725,32 @@ export class TreeGrid<DATA extends IDentifiable>
         this.grid.selectRow(row, debounce);
     }
 
-    hasNodeWithDataId(dataId: string): boolean {
+    hasItemWithDataId(dataId: string): boolean {
         return !!this.root.getNodeByDataId(dataId);
     }
 
+    hasItemWithDataIdInDefault(dataId: string): boolean {
+        return !!this.root.getNodeByDataIdFromDefault(dataId);
+    }
+
+    getItemWithDataId(dataId: string): DATA {
+        const node: TreeNode<DATA> = this.root.getNodeByDataId(dataId);
+
+        return !!node ? node.getData() : null;
+    }
+
+    getItemWithDataIdFromDefault(dataId: string): DATA {
+        const node: TreeNode<DATA> = this.root.getNodeByDataIdFromDefault(dataId);
+
+        return !!node ? node.getData() : null;
+    }
+
+    getItemWithDataIdFromFiltered(dataId: string): DATA {
+        const node: TreeNode<DATA> = this.root.getNodeByDataIdFromFiltered(dataId);
+
+        return !!node ? node.getData() : null;
+    }
+
     protected insertDataToParentNode(data: DATA, parent: TreeNode<DATA>, index: number) {
         const nodeToInsert: TreeNode<DATA> = this.dataToTreeNode(data, parent);
         this.insertNodeToParentNode(nodeToInsert, parent, index);

diff --git a/src/main/resources/assets/admin/common/js/ui/treegrid/TreeRoot.ts b/src/main/resources/assets/admin/common/js/ui/treegrid/TreeRoot.ts
@@ -76,6 +76,10 @@ export class TreeRoot<DATA extends IDentifiable> {
         return this.getDefaultRoot().findNode(dataId);
     }
 
+    getNodeByDataIdFromFiltered(dataId: string): TreeNode<DATA> {
+        return this.getFilteredRoot().findNode(dataId);
+    }
+
     getNodesByDataId(dataId: string): TreeNode<DATA>[] {
         const nodesToUpdate: TreeNode<DATA>[] = [];