@visactor/vtable 1.19.1 → 1.19.3-alpha.0

package/es/data/DataSource.js

@@ -87,13 +87,16 @@ export class DataSource extends EventTarget {
         this.rowHierarchyType = rowHierarchyType, this.updatePagerData();
     }
     initTreeHierarchyState() {
-        var _a;
+        var _a, _b;
+        this._sourceLength = (null === (_a = this._source) || void 0 === _a ? void 0 : _a.length) || 0, 
         this.currentIndexedData = Array.from({
             length: this._sourceLength
         }, ((_, i) => i));
         let nodeLength = this._sourceLength;
         for (let i = 0; i < nodeLength; i++) {
-            const indexKey = this.currentIndexedData[i], nodeData = this.getOriginalRecord(indexKey), children = null !== (_a = nodeData.filteredChildren) && void 0 !== _a ? _a : nodeData.children;
+            const indexKey = this.currentIndexedData[i], nodeData = this.getOriginalRecord(indexKey);
+            if (!nodeData) continue;
+            const children = null !== (_b = nodeData.filteredChildren) && void 0 !== _b ? _b : nodeData.children;
             if ((null == children ? void 0 : children.length) > 0) {
                 if (this.hierarchyExpandLevel > 1 ? !nodeData.hierarchyState && (nodeData.hierarchyState = HierarchyState.expand) : !nodeData.hierarchyState && (nodeData.hierarchyState = HierarchyState.collapse), 
                 this.hasHierarchyStateExpand = !0, nodeData.hierarchyState === HierarchyState.collapse) continue;
@@ -652,8 +655,11 @@ export class DataSource extends EventTarget {
                 }
                 if (sourceI = sourceIndexs.splice(sourceIndexs.length - 1, 1)[0], targetI = targetIndexs.splice(targetIndexs.length - 1, 1)[0], 
                 sourceIndexs.length >= 1) {
-                    const parent = this.getOriginalRecord(sourceIndexs), sourceIds = parent.filteredChildren ? parent.filteredChildren.splice(sourceI, 1) : parent.children.splice(sourceI, 1);
-                    sourceIds.unshift(targetI, 0), Array.prototype.splice.apply(null !== (_c = parent.filteredChildren) && void 0 !== _c ? _c : parent.children, sourceIds);
+                    const parent = this.getOriginalRecord(sourceIndexs);
+                    if (parent) {
+                        const sourceIds = parent.filteredChildren ? parent.filteredChildren.splice(sourceI, 1) : parent.children.splice(sourceI, 1);
+                        sourceIds.unshift(targetI, 0), Array.prototype.splice.apply(null !== (_c = parent.filteredChildren) && void 0 !== _c ? _c : parent.children, sourceIds);
+                    }
                 } else {
                     const sourceIds = this.records.splice(sourceI, 1);
                     sourceIds.unshift(targetI, 0), Array.prototype.splice.apply(this.records, sourceIds);
@@ -674,6 +680,7 @@ export class DataSource extends EventTarget {
         if (this.hierarchyExpandLevel) {
             for (let i = 0; i < this._sourceLength; i++) {
                 const nodeData = this.getOriginalRecord(i);
+                if (!nodeData) continue;
                 (null !== (_a = nodeData.filteredChildren) && void 0 !== _a ? _a : nodeData.children) && !nodeData.hierarchyState && (nodeData.hierarchyState = HierarchyState.collapse);
             }
             this.currentIndexedData = Array.from({
@@ -681,7 +688,9 @@ export class DataSource extends EventTarget {
             }, ((_, i) => i));
             let nodeLength = this._sourceLength;
             for (let i = 0; i < nodeLength; i++) {
-                const indexKey = this.currentIndexedData[i], nodeData = this.getOriginalRecord(indexKey), children = null !== (_b = nodeData.filteredChildren) && void 0 !== _b ? _b : nodeData.children;
+                const indexKey = this.currentIndexedData[i], nodeData = this.getOriginalRecord(indexKey);
+                if (!nodeData) continue;
+                const children = null !== (_b = nodeData.filteredChildren) && void 0 !== _b ? _b : nodeData.children;
                 if ((null == children ? void 0 : children.length) > 0 && nodeData.hierarchyState === HierarchyState.expand) {
                     this.hasHierarchyStateExpand = !0;
                     const childrenLength = this.restoreChildrenNodeHierarchy(indexKey, nodeData);
@@ -702,6 +711,26 @@ export class DataSource extends EventTarget {
         }
         return childTotalLength;
     }
+    _setNodeStateRecursive(node, targetState) {
+        var _a;
+        if (!node) return;
+        const children = null !== (_a = node.filteredChildren) && void 0 !== _a ? _a : node.children;
+        if (children && (Array.isArray(children) ? children.length > 0 : !0 === children) && (node.hierarchyState = targetState), 
+        children && Array.isArray(children)) for (const child of children) this._setNodeStateRecursive(child, targetState);
+    }
+    expandAllNodes() {
+        if (Array.isArray(this._source)) {
+            for (const rootNode of this._source) this._setNodeStateRecursive(rootNode, HierarchyState.expand);
+            this.hasHierarchyStateExpand = !0, this.clearSortedIndexMap(), this.restoreTreeHierarchyState(), 
+            this.lastSortStates && this.lastSortStates.length > 0 ? this.sort(this.lastSortStates) : this.updatePagerData();
+        }
+    }
+    collapseAllNodes() {
+        if (Array.isArray(this._source)) {
+            for (const rootNode of this._source) this._setNodeStateRecursive(rootNode, HierarchyState.collapse);
+            this.clearSortedIndexMap(), this.restoreTreeHierarchyState(), this.lastSortStates && this.lastSortStates.length > 0 ? this.sort(this.lastSortStates) : this.updatePagerData();
+        }
+    }
 }
 
 DataSource.EMPTY = new DataSource({