@lingxiteam/ebe-utils 0.2.22 → 0.2.23

package/lib/pcpublic/src/components/pcfactory/src/Table/hooks/useCheckStrict.ts

@@ -1,17 +1,17 @@
 import { useMemo } from 'react';
 import { useCreation } from '../../utils/ahooks';
+import { recursionDataSource, getFullCheckedKeySet } from '../utils';
 import { LevelItem } from '../types/prop';
-import { getFullCheckedKeySet, recursionDataSource } from '../utils';
 
 interface CheckStrictlyParams<T> {
-  rowSelectType?: 'checkbox' | 'radio';
-  checkLoosed?: boolean;
+  rowSelectType?: 'checkbox' | 'radio',
+  checkLoosed?: boolean,
   dataSource: T[];
   currentRowKey: string;
-  childrenKey?: string;
+  childrenKey?: string
 }
 
-function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(params: CheckStrictlyParams<T>) {
+function useCheckStrict<T extends {children?: T[], [key: string]: unknown}> (params: CheckStrictlyParams<T>) {
   const { rowSelectType, checkLoosed = false, dataSource, currentRowKey, childrenKey = 'children' } = params;
   // 父子数据是否关联选择，默认不关联
   const checkStrictly = useMemo(() => !(rowSelectType === 'checkbox' && checkLoosed), [checkLoosed, rowSelectType]);
@@ -23,13 +23,9 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(pa
       if (!checkStrictly) {
         relatedDisabledMap[node[currentRowKey] as string] = flag;
         if (flag && node[childrenKey]) {
-          recursionDataSource<T>(
-            node[childrenKey] as T[],
-            n => {
-              relatedDisabledMap[n[currentRowKey] as string] = flag;
-            },
-            childrenKey
-          );
+          recursionDataSource<T>(node[childrenKey] as T[], (n) => {
+            relatedDisabledMap[n[currentRowKey] as string] = flag;
+          }, childrenKey);
         }
       }
     };
@@ -43,26 +39,22 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(pa
           maxLevel,
         };
       }
-      recursionDataSource<T, LevelItem<T>>(
-        dataSource,
-        (node, parentPath) => {
-          const level = parentPath?.length || 0;
-          maxLevel = Math.max(maxLevel, level);
-          const n = {
-            ...node,
-            __key__: node[currentRowKey] as string,
-            __parent__: level && parentPath ? parentPath[level - 1] : undefined,
-            __children__: node[childrenKey] as LevelItem<T>[],
-          };
-          if (levelDataMap.get(level)) {
-            levelDataMap.get(level)?.push(n);
-          } else {
-            levelDataMap.set(level, [n]);
-          }
-          return n;
-        },
-        '__children__'
-      );
+      recursionDataSource<T, LevelItem<T>>(dataSource, (node, parentPath) => {
+        const level = parentPath?.length || 0;
+        maxLevel = Math.max(maxLevel, level);
+        const n = {
+          ...node,
+          __key__: node[currentRowKey] as string,
+          __parent__: level && parentPath ? parentPath[level - 1] : undefined,
+          __children__: node[childrenKey] as LevelItem<T>[],
+        };
+        if (levelDataMap.get(level)) {
+          levelDataMap.get(level)?.push(n);
+        } else {
+          levelDataMap.set(level, [n]);
+        }
+        return n;
+      }, '__children__');
       return {
         levelDataMap,
         maxLevel,
@@ -76,16 +68,12 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(pa
       if (dataCacheMap[key]) {
         return dataCacheMap[key];
       }
-      recursionDataSource(
-        [...dataSource, ...(extraData || [])],
-        n => {
-          dataCacheMap[n[currentRowKey] as string] = n;
-        },
-        childrenKey
-      );
+      recursionDataSource([...dataSource, ...(extraData || [])], (n) => {
+        dataCacheMap[n[currentRowKey] as string] = n;
+      }, childrenKey);
       return dataCacheMap[key];
     };
-
+    
     return {
       markDisable,
       relatedDisabledMap,
@@ -94,16 +82,19 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(pa
     };
   }, [dataSource, checkStrictly, currentRowKey, childrenKey]);
 
+
   // 获取单个勾选/反选时的选中项
   const triggerSingleSelection = (originSelectRow: T[], selectRow: T) => {
     const key = selectRow[currentRowKey];
-    const checked = originSelectRow.findIndex(c => c[currentRowKey] === key) === -1;
-    const selectionList = checked ? [...originSelectRow, selectRow] : originSelectRow.filter(c => c[currentRowKey] !== key);
+    const checked = originSelectRow.findIndex((c) => c[currentRowKey] === key) === -1;
+    const selectionList = checked
+      ? [...originSelectRow, selectRow]
+      : originSelectRow.filter((c) => c[currentRowKey] !== key);
     if (checkStrictly) {
       return selectionList;
     }
     const { levelDataMap, maxLevel } = getLevelDataMap();
-    const keys = selectionList.map(c => c[currentRowKey] as string);
+    const keys = selectionList.map((c) => c[currentRowKey] as string);
     // 基于当前选中数据获取上层所有选中key
     const checkedSet = getFullCheckedKeySet({
       levelEntries: levelDataMap,
@@ -113,25 +104,21 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(pa
       checked,
     });
     const childrenRow: T[] = [];
-    recursionDataSource(
-      (selectRow?.[childrenKey] as T[]) || [],
-      n => {
-        if (!relatedDisabledMap[n[currentRowKey] as string]) {
-          if (checked) {
-            childrenRow.push(n);
-          }
-          // 过滤选中集合中的部分
-          checkedSet.delete(n[currentRowKey] as string);
+    recursionDataSource(selectRow?.[childrenKey] as T[] || [], (n) => {
+      if (!relatedDisabledMap[n[currentRowKey] as string]) {
+        if (checked) {
+          childrenRow.push(n);
         }
-      },
-      childrenKey
-    );
-    const checkedRows = [...checkedSet].map(k => getRecordByKey(k, selectionList));
+        // 过滤选中集合中的部分
+        checkedSet.delete(n[currentRowKey] as string);
+      }
+    }, childrenKey);
+    const checkedRows = [...checkedSet].map((k) => getRecordByKey(k, selectionList));
     return [...checkedRows, ...childrenRow];
   };
 
   // 根据当前选中项收集最终选中key
-  const triggerFullSelection = (checkedKeys: (string | number)[]) => {
+  const triggerFullSelection = (checkedKeys: (string|number)[]) => {
     if (checkStrictly) {
       return checkedKeys;
     }