stk-table-vue 0.8.14 → 0.9.0-beta.1

package/src/StkTable/useTree.ts

@@ -1,161 +1,161 @@
-import { ShallowRef } from 'vue';
-import { PrivateRowDT, RowKeyGen, TreeConfig, UniqKey } from './types';
-
-type DT = PrivateRowDT & { children?: DT[] };
-type Option<DT extends Record<string, any>> = {
-    props: any;
-    rowKeyGen: RowKeyGen;
-    dataSourceCopy: ShallowRef<DT[]>;
-    emits: any;
-};
-
-export function useTree({ props, dataSourceCopy, rowKeyGen, emits }: Option<DT>) {
-    const { defaultExpandAll, defaultExpandKeys, defaultExpandLevel }: TreeConfig = props.treeConfig;
-    /** It used to check if it is first load. To execute defaultExpandXXX */
-    let isFirstLoad = true;
-    /** click expended icon to toggle expand row */
-    function toggleTreeNode(row: DT, col: any) {
-        const expand = row ? !row.__T_EXP__ : false;
-        privateSetTreeExpand(row, { expand, col, isClick: true });
-    }
-
-    /**
-     *
-     * @param row rowKey or row
-     * @param option
-     * @param option.expand expand or collapse
-     * @param option.silent if set true, not emit `toggle-tree-expand`, default:false
-     */
-    function privateSetTreeExpand(row: (UniqKey | DT) | (UniqKey | DT)[], option: { expand?: boolean; col?: any; isClick: boolean }) {
-        const rowKeyOrRowArr: (UniqKey | DT)[] = Array.isArray(row) ? row : [row];
-
-        const tempData = dataSourceCopy.value.slice();
-        for (let i = 0; i < rowKeyOrRowArr.length; i++) {
-            const rowKeyOrRow = rowKeyOrRowArr[i];
-            let rowKey: UniqKey;
-            if (typeof rowKeyOrRow === 'string') {
-                rowKey = rowKeyOrRow;
-            } else {
-                rowKey = rowKeyGen(rowKeyOrRow);
-            }
-            const index = tempData.findIndex(it => rowKeyGen(it) === rowKey);
-            if (index === -1) {
-                console.warn('treeExpandRow failed.rowKey:', rowKey);
-                return;
-            }
-
-            const row = tempData[index];
-            const level = row.__T_LV__ || 0;
-            let expanded = option?.expand;
-            if (expanded === void 0) {
-                expanded = !row.__T_EXP__;
-            }
-            if (expanded) {
-                const children = expandNode(row, level);
-                tempData.splice(index + 1, 0, ...children);
-            } else {
-                // delete all child nodes from i
-                const deleteCount = foldNode(index, tempData, level);
-                tempData.splice(index + 1, deleteCount);
-            }
-
-            setNodeExpanded(row, expanded, level);
-
-            if (option.isClick) {
-                emits('toggle-tree-expand', { expanded: Boolean(expanded), row, col: option.col });
-            }
-        }
-
-        dataSourceCopy.value = tempData;
-    }
-
-    function setTreeExpand(row: (UniqKey | DT) | (UniqKey | DT)[], option?: { expand?: boolean }) {
-        privateSetTreeExpand(row, { ...option, isClick: false });
-    }
-
-    function setNodeExpanded(row: DT, expanded: boolean, level?: number, parent?: DT) {
-        row.__T_EXP__ = expanded;
-        if (level !== void 0) {
-            row.__T_LV__ = level;
-        }
-        // if (parent) {
-        //     row.__T_P_K__ = rowKeyGen(parent);
-        // }
-    }
-
-    function recursionFlat(data: DT[] | undefined, level: number, parent?: DT): DT[] {
-        if (!data) return [];
-        let result: DT[] = []
-        for (let i = 0; i < data.length; i++) {
-            const item = data[i];
-            result.push(item);
-            const isExpanded = Boolean(item.__T_EXP__);
-            setNodeExpanded(item, isExpanded, level, parent);
-            if (isFirstLoad && !isExpanded) {
-                // first load will expand all node if defaultExpandAll is true
-                if (defaultExpandAll) {
-                    setNodeExpanded(item, true);
-                } else {
-                    if (defaultExpandLevel && level < defaultExpandLevel) {
-                        setNodeExpanded(item, true);
-                    }
-                    if (defaultExpandKeys?.includes(rowKeyGen(item))) {
-                        setNodeExpanded(item, true);
-                    }
-                }
-            }
-            if (item.__T_EXP__) {
-                const res = recursionFlat(item.children, level + 1, item);
-                result = result.concat(res);
-            }
-        }
-        return result;
-    };
-
-    /**
-     * 根据保存的展开状态，深度遍历，展平树形数据。
-     * en: flatten tree data by saved expand state.
-     * @param data
-     * @returns
-     */
-    function flatTreeData(data: DT[]) {
-        const result = recursionFlat(data, 0);
-        isFirstLoad = false;
-        return result;
-    }
-
-    function expandNode(row: DT, level: number) {
-        let result: DT[] = [];
-        row.children &&
-            row.children.forEach((child: DT) => {
-                result.push(child);
-                const childLv = level + 1;
-                if (child.__T_EXP__ && child.children) {
-                    const res = expandNode(child, childLv);
-                    result = result.concat(res);
-                } else {
-                    setNodeExpanded(child, false, childLv, row);
-                }
-            });
-        return result;
-    }
-
-    function foldNode(index: number, tempData: DT[], level: number) {
-        let deleteCount = 0;
-        for (let i = index + 1; i < tempData.length; i++) {
-            const child = tempData[i];
-            if (child.__T_LV__ && child.__T_LV__ > level) {
-                deleteCount++;
-            } else {
-                break;
-            }
-        }
-        return deleteCount;
-    }
-
-    return {
-        toggleTreeNode,
-        setTreeExpand,
-        flatTreeData,
-    };
-}
+import { ShallowRef } from 'vue';
+import { PrivateRowDT, RowKeyGen, TreeConfig, UniqKey } from './types';
+
+type DT = PrivateRowDT & { children?: DT[] };
+type Option<DT extends Record<string, any>> = {
+    props: any;
+    rowKeyGen: RowKeyGen;
+    dataSourceCopy: ShallowRef<DT[]>;
+    emits: any;
+};
+
+export function useTree({ props, dataSourceCopy, rowKeyGen, emits }: Option<DT>) {
+    const { defaultExpandAll, defaultExpandKeys, defaultExpandLevel }: TreeConfig = props.treeConfig;
+    /** It used to check if it is first load. To execute defaultExpandXXX */
+    let isFirstLoad = true;
+    /** click expended icon to toggle expand row */
+    function toggleTreeNode(row: DT, col: any) {
+        const expand = row ? !row.__T_EXP__ : false;
+        privateSetTreeExpand(row, { expand, col, isClick: true });
+    }
+
+    /**
+     *
+     * @param row rowKey or row
+     * @param option
+     * @param option.expand expand or collapse
+     * @param option.silent if set true, not emit `toggle-tree-expand`, default:false
+     */
+    function privateSetTreeExpand(row: (UniqKey | DT) | (UniqKey | DT)[], option: { expand?: boolean; col?: any; isClick: boolean }) {
+        const rowKeyOrRowArr: (UniqKey | DT)[] = Array.isArray(row) ? row : [row];
+
+        const tempData = dataSourceCopy.value.slice();
+        for (let i = 0; i < rowKeyOrRowArr.length; i++) {
+            const rowKeyOrRow = rowKeyOrRowArr[i];
+            let rowKey: UniqKey;
+            if (typeof rowKeyOrRow === 'string') {
+                rowKey = rowKeyOrRow;
+            } else {
+                rowKey = rowKeyGen(rowKeyOrRow);
+            }
+            const index = tempData.findIndex(it => rowKeyGen(it) === rowKey);
+            if (index === -1) {
+                console.warn('treeExpandRow failed.rowKey:', rowKey);
+                return;
+            }
+
+            const row = tempData[index];
+            const level = row.__T_LV__ || 0;
+            let expanded = option?.expand;
+            if (expanded === void 0) {
+                expanded = !row.__T_EXP__;
+            }
+            if (expanded) {
+                const children = expandNode(row, level);
+                tempData.splice(index + 1, 0, ...children);
+            } else {
+                // delete all child nodes from i
+                const deleteCount = foldNode(index, tempData, level);
+                tempData.splice(index + 1, deleteCount);
+            }
+
+            setNodeExpanded(row, expanded, level);
+
+            if (option.isClick) {
+                emits('toggle-tree-expand', { expanded: Boolean(expanded), row, col: option.col });
+            }
+        }
+
+        dataSourceCopy.value = tempData;
+    }
+
+    function setTreeExpand(row: (UniqKey | DT) | (UniqKey | DT)[], option?: { expand?: boolean }) {
+        privateSetTreeExpand(row, { ...option, isClick: false });
+    }
+
+    function setNodeExpanded(row: DT, expanded: boolean, level?: number, parent?: DT) {
+        row.__T_EXP__ = expanded;
+        if (level !== void 0) {
+            row.__T_LV__ = level;
+        }
+        // if (parent) {
+        //     row.__T_P_K__ = rowKeyGen(parent);
+        // }
+    }
+
+    function recursionFlat(data: DT[] | undefined, level: number, parent?: DT): DT[] {
+        if (!data) return [];
+        let result: DT[] = []
+        for (let i = 0; i < data.length; i++) {
+            const item = data[i];
+            result.push(item);
+            const isExpanded = Boolean(item.__T_EXP__);
+            setNodeExpanded(item, isExpanded, level, parent);
+            if (isFirstLoad && !isExpanded) {
+                // first load will expand all node if defaultExpandAll is true
+                if (defaultExpandAll) {
+                    setNodeExpanded(item, true);
+                } else {
+                    if (defaultExpandLevel && level < defaultExpandLevel) {
+                        setNodeExpanded(item, true);
+                    }
+                    if (defaultExpandKeys?.includes(rowKeyGen(item))) {
+                        setNodeExpanded(item, true);
+                    }
+                }
+            }
+            if (item.__T_EXP__) {
+                const res = recursionFlat(item.children, level + 1, item);
+                result = result.concat(res);
+            }
+        }
+        return result;
+    };
+
+    /**
+     * 根据保存的展开状态，深度遍历，展平树形数据。
+     * en: flatten tree data by saved expand state.
+     * @param data
+     * @returns
+     */
+    function flatTreeData(data: DT[]) {
+        const result = recursionFlat(data, 0);
+        isFirstLoad = false;
+        return result;
+    }
+
+    function expandNode(row: DT, level: number) {
+        let result: DT[] = [];
+        row.children &&
+            row.children.forEach((child: DT) => {
+                result.push(child);
+                const childLv = level + 1;
+                if (child.__T_EXP__ && child.children) {
+                    const res = expandNode(child, childLv);
+                    result = result.concat(res);
+                } else {
+                    setNodeExpanded(child, false, childLv, row);
+                }
+            });
+        return result;
+    }
+
+    function foldNode(index: number, tempData: DT[], level: number) {
+        let deleteCount = 0;
+        for (let i = index + 1; i < tempData.length; i++) {
+            const child = tempData[i];
+            if (child.__T_LV__ && child.__T_LV__ > level) {
+                deleteCount++;
+            } else {
+                break;
+            }
+        }
+        return deleteCount;
+    }
+
+    return {
+        toggleTreeNode,
+        setTreeExpand,
+        flatTreeData,
+    };
+}