@lingxiteam/ebe-utils 0.2.3 → 0.2.5

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

@@ -11,115 +11,99 @@ interface CheckStrictlyParams<T> {
   childrenKey?: string;
 }
 
-function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(
-  params: CheckStrictlyParams<T>,
-) {
-  const {
-    rowSelectType,
-    checkLoosed = false,
-    dataSource,
-    currentRowKey,
-    childrenKey = 'children',
-  } = params;
+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],
-  );
+  const checkStrictly = useMemo(() => !(rowSelectType === 'checkbox' && checkLoosed), [checkLoosed, rowSelectType]);
 
   // 标记节点是否禁用
-  const { relatedDisabledMap, markDisable, getLevelDataMap, getRecordByKey } =
-    useCreation(() => {
-      const relatedDisabledMap: Record<string, boolean> = {};
-      const markDisable = (node: T, flag: boolean = false) => {
-        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,
-            );
-          }
-        }
-      };
-      const levelDataMap: Map<number, LevelItem<T>[]> = new Map();
-      let maxLevel = 0;
-      // 按数据层级收集映射数据
-      const getLevelDataMap = () => {
-        if (levelDataMap.size > 0) {
-          return {
-            levelDataMap,
-            maxLevel,
-          };
+  const { relatedDisabledMap, markDisable, getLevelDataMap, getRecordByKey } = useCreation(() => {
+    const relatedDisabledMap: Record<string, boolean> = {};
+    const markDisable = (node: T, flag: boolean = false) => {
+      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, 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__',
-        );
+      }
+    };
+    const levelDataMap: Map<number, LevelItem<T>[]> = new Map();
+    let maxLevel = 0;
+    // 按数据层级收集映射数据
+    const getLevelDataMap = () => {
+      if (levelDataMap.size > 0) {
         return {
           levelDataMap,
           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__'
+      );
+      return {
+        levelDataMap,
+        maxLevel,
       };
+    };
 
-      // 记录数据映射
-      const dataCacheMap: Record<string, T> = {};
-      // extraData 之前选中过的跨页数据不存在当前dataSource中
-      const getRecordByKey = (key: string | number, extraData?: T[]) => {
-        if (dataCacheMap[key]) {
-          return dataCacheMap[key];
-        }
-        recursionDataSource(
-          [...dataSource, ...(extraData || [])],
-          (n) => {
-            dataCacheMap[n[currentRowKey] as string] = n;
-          },
-          childrenKey,
-        );
+    // 记录数据映射
+    const dataCacheMap: Record<string, T> = {};
+    // extraData 之前选中过的跨页数据不存在当前dataSource中
+    const getRecordByKey = (key: string | number, extraData?: T[]) => {
+      if (dataCacheMap[key]) {
         return dataCacheMap[key];
-      };
+      }
+      recursionDataSource(
+        [...dataSource, ...(extraData || [])],
+        n => {
+          dataCacheMap[n[currentRowKey] as string] = n;
+        },
+        childrenKey
+      );
+      return dataCacheMap[key];
+    };
 
-      return {
-        markDisable,
-        relatedDisabledMap,
-        getLevelDataMap,
-        getRecordByKey,
-      };
-    }, [dataSource, checkStrictly, currentRowKey, childrenKey]);
+    return {
+      markDisable,
+      relatedDisabledMap,
+      getLevelDataMap,
+      getRecordByKey,
+    };
+  }, [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,
@@ -131,7 +115,7 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(
     const childrenRow: T[] = [];
     recursionDataSource(
       (selectRow?.[childrenKey] as T[]) || [],
-      (n) => {
+      n => {
         if (!relatedDisabledMap[n[currentRowKey] as string]) {
           if (checked) {
             childrenRow.push(n);
@@ -140,11 +124,9 @@ function useCheckStrict<T extends { children?: T[]; [key: string]: unknown }>(
           checkedSet.delete(n[currentRowKey] as string);
         }
       },
-      childrenKey,
-    );
-    const checkedRows = [...checkedSet].map((k) =>
-      getRecordByKey(k, selectionList),
+      childrenKey
     );
+    const checkedRows = [...checkedSet].map(k => getRecordByKey(k, selectionList));
     return [...checkedRows, ...childrenRow];
   };