@syncfusion/ej2-treegrid 34.1.29 → 34.1.30

package/dist/es6/ej2-treegrid.es2015.js

@@ -970,6 +970,20 @@ class Selection {
         this.resetSelectionCaches();
         this.removeEventListener();
     }
+    /**
+     * Gets the actual record from the data model's uniqueIDCollection.
+     *
+     * @param {ITreeData} record - The record (may be stale reference).
+     * @returns {ITreeData} The actual record from the data model.
+     */
+    getRecordFromUidCollection(record) {
+        if (!record) {
+            return record;
+        }
+        const uidMap = this.parent.uniqueIDCollection;
+        const uid = record.uniqueID;
+        return (uidMap && uid != null) ? (uidMap[String(uid)] ? uidMap[String(uid)] : record) : record;
+    }
     /**
      * Handles checkbox click events from the DOM and dispatches selection logic.
      *
@@ -995,8 +1009,7 @@ class Selection {
         }
         else if (checkWrap && checkWrap.querySelectorAll('.e-treeselectall').length > 0 && this.parent.autoCheckHierarchy) {
             const frame = checkWrap.querySelector('.e-frame');
-            const currentStateIsUncheck = !frame.classList.contains('e-check') && !frame.classList.contains('e-stop');
-            const targetState = currentStateIsUncheck; // If currently uncheck, target state is to check all.
+            const targetState = frame.classList.contains('e-uncheck') || frame.classList.contains('e-stop');
             this.headerSelection(targetState);
             checkBox = checkWrap.querySelector('input[type="checkbox"]');
             this.triggerChkChangeEvent(checkBox, targetState, target.closest('tr'));
@@ -1163,6 +1176,7 @@ class Selection {
             const flatRec = getParentData(this.parent, viewRec.uniqueID);
             const nextState = (flatRec.checkboxState === 'check') ? 'uncheck' : 'check';
             flatRec.checkboxState = nextState;
+            viewRec.checkboxState = nextState;
             this.traverSelection(flatRec, nextState, false);
         }
     }
@@ -1214,10 +1228,11 @@ class Selection {
             if (child.hasChildRecords) {
                 this.traverSelection(child, checkboxState, true);
             }
-            if (child.checkboxState === 'check') {
+            const updatedChild = this.getRecordFromUidCollection(child);
+            if (updatedChild.checkboxState === 'check') {
                 checkedCount++;
             }
-            else if (child.checkboxState === 'indeterminate') {
+            else if (updatedChild.checkboxState === 'indeterminate') {
                 indeterminateCount++;
             }
         }
@@ -1340,11 +1355,12 @@ class Selection {
             if (!child || child.isSummaryRow) {
                 continue;
             }
+            const actualChild = this.getRecordFromUidCollection(child);
             summary.total++;
-            if (child.checkboxState === 'check') {
+            if (actualChild.checkboxState === 'check') {
                 summary.checked++;
             }
-            else if (child.checkboxState === 'indeterminate') {
+            else if (actualChild.checkboxState === 'indeterminate') {
                 summary.indeterminate++;
             }
         }
@@ -1472,12 +1488,13 @@ class Selection {
             if (!record) {
                 continue;
             }
-            const previousState = record.checkboxState;
+            const actualRecord = this.getRecordFromUidCollection(record);
+            const previousState = actualRecord.checkboxState;
             if (previousState === targetState) {
                 continue;
             }
-            record.checkboxState = targetState;
-            this.updateSelectedItems(record, targetState, true);
+            actualRecord.checkboxState = targetState;
+            this.updateSelectedItems(actualRecord, targetState, true);
         }
     }
     /**
@@ -1504,9 +1521,10 @@ class Selection {
             if (item.hasChildRecords && isFilterOrSearch && item.level === 0 && this.parent.autoCheckHierarchy) {
                 this.updateParentSelection(item);
             }
-            if (item.uniqueID && item.checkboxState === 'check') {
-                newSelectedItems.push(item);
-                newSelectedUidMap.set(item.uniqueID, true);
+            const actualItem = this.getRecordFromUidCollection(item);
+            if (actualItem.uniqueID && actualItem.checkboxState === 'check') {
+                newSelectedItems.push(actualItem);
+                newSelectedUidMap.set(actualItem.uniqueID, true);
             }
         }
         if (!isFilterOrSearch) {
@@ -1678,7 +1696,8 @@ class Selection {
         this.totalSelectableCount = this.countSelectableRecords(recordsForHeaderLogic);
         let checkedCountForHeaderLogic = 0;
         for (const record of recordsForHeaderLogic) {
-            if (record && !record.isSummaryRow && record.checkboxState === 'check') {
+            const actualRecord = this.getRecordFromUidCollection(record);
+            if (actualRecord && !actualRecord.isSummaryRow && actualRecord.checkboxState === 'check') {
                 checkedCountForHeaderLogic++;
             }
         }