pixel-react 1.9.3 → 1.9.5

package/src/utils/handleTreeNodeSelect/handleTreeNodeSelect.ts

@@ -0,0 +1,90 @@
+export type TreeNode = {
+  createdBy: string;
+  modifiedBy: string;
+  createdByUname: string;
+  modifiedByUname: string;
+  createdOn: string;
+  modifiedOn: string;
+  state: string;
+  path: string;
+  searchKey: string;
+  key: string;
+  name: string;
+  projectId: string;
+  hierarchy: number;
+  executionOrder: number;
+  subContainerCount: number;
+  resourceCount: number;
+  totalSubContainerCount: number;
+  totalResourceCount: number;
+  totalProjectElementCount: number;
+  totalSharedElementCount: number;
+  container: boolean;
+  localDelete: boolean;
+  defaultEntity: boolean;
+  lastResource: boolean;
+  platform?: string;
+  parentId?: string;
+  parentName?: string;
+  checked?: boolean | 'partial';
+};
+
+export const handleTreeNodeSect = (
+  data: TreeNode[],
+  key: string,
+  isChecked: boolean | 'partial'
+): TreeNode[] => {
+  const nodesMap = new Map<string, TreeNode>();
+
+  // Build a map for quick access to nodes by key
+  data.forEach((node) => nodesMap.set(node.key, node));
+
+  // Helper to update child nodes
+  function updateChildren(nodeKey: string, checkedStatus: boolean | 'partial') {
+    for (const node of data) {
+      if (node.parentId === nodeKey) {
+        node.checked = checkedStatus;
+        updateChildren(node.key, checkedStatus);
+      }
+    }
+  }
+
+  // Helper to update parent nodes
+  function updateParents(nodeKey: string) {
+    const node = nodesMap.get(nodeKey);
+    if (node && node.parentId) {
+      const parentNode = nodesMap.get(node.parentId);
+      if (parentNode) {
+        const siblings = data.filter(
+          (sibling) => sibling.parentId === parentNode.key
+        );
+        const allChecked = siblings.every(
+          (sibling) => sibling.checked === true
+        );
+        const someChecked = siblings.some(
+          (sibling) => sibling.checked === true || sibling.checked === 'partial'
+        );
+
+        parentNode.checked = allChecked
+          ? true
+          : someChecked
+          ? 'partial'
+          : false;
+
+        updateParents(parentNode.key);
+      }
+    }
+  }
+
+  // Find the target node and update its state
+  const targetNode = nodesMap.get(key);
+  if (targetNode) {
+    targetNode.checked = isChecked;
+    // Update children recursively
+    updateChildren(key, isChecked);
+    // Update parents recursively
+    updateParents(key);
+  }
+
+  return data;
+};