@zat-design/sisyphus-react 3.14.2 → 3.14.3-beta.1

package/es/ProTree/components/ProTreeSelect/index.js

@@ -6,19 +6,19 @@ import "antd/es/input/style";
 import _Input from "antd/es/input";
 import "antd/es/message/style";
 import _message from "antd/es/message";
-import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
 import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
+import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
 import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
 import "antd/es/tree-select/style";
 import _TreeSelect from "antd/es/tree-select";
-var _excluded = ["disabled", "code", "dataSource", "defaultDisableValue", "onChange", "useRequest", "transformResponse", "fieldNames", "placeholder", "tooltip", "showSearch", "filterTreeNode", "treeNodeFilterProp", "defaultExpandAll", "expandedKeys", "treeCheckable", "onSearch", "otherProps", "width", "dropdownStyle", "popupClassName", "style", "allowClear", "listHeight", "showCodeName", "className", "checkStrictly", "checkable"],
-  _excluded2 = ["children"];
+var _excluded = ["disabled", "code", "dataSource", "defaultDisableValue", "onChange", "useRequest", "transformResponse", "fieldNames", "placeholder", "tooltip", "showSearch", "filterTreeNode", "treeNodeFilterProp", "defaultExpandAll", "expandedKeys", "treeCheckable", "onSearch", "otherProps", "width", "dropdownStyle", "popupClassName", "style", "allowClear", "listHeight", "showCodeName", "className", "checkStrictly", "checkable"];
 /* eslint-disable react/no-danger */
 /* eslint-disable react/jsx-closing-tag-location */
 import { useDeepCompareEffect, useRequest as useRequestFunc, useSetState } from 'ahooks';
 import classnames from 'classnames';
 import React, { useImperativeHandle, forwardRef, useCallback } from 'react';
 import { cloneDeep, debounce } from 'lodash';
+import { tools } from '@zat-design/utils';
 import locale from "../../../locale";
 import { useProConfig } from "../../../ProConfigProvider";
 import Container from "../../../ProForm/components/Container";
@@ -35,14 +35,7 @@ export var ProTreeSelect = (props, ref) => {
   var _useProConfig = useProConfig('ProEnum'),
     _useProConfig$dics = _useProConfig.dics,
     dics = _useProConfig$dics === void 0 ? {} : _useProConfig$dics;
-  var label = 'label';
-  var code = 'value';
-  var uuidKey = 'value';
-  if (proSelectFieldNames && Object.keys(proSelectFieldNames).length) {
-    code = proSelectFieldNames.value;
-    label = proSelectFieldNames.label;
-    uuidKey = proSelectFieldNames.key || proSelectFieldNames.value;
-  }
+
   // 配置属性
   var disabled = props.disabled,
     enumCode = props.code,
@@ -88,11 +81,32 @@ export var ProTreeSelect = (props, ref) => {
   var _ref3 = otherProps || {},
     isView = _ref3.isView,
     viewEmpty = _ref3.viewEmpty;
-  if (fieldNames && Object.keys(fieldNames).length) {
-    code = fieldNames.value || 'value';
-    label = fieldNames.label || 'label';
-    uuidKey = fieldNames.key || code;
-  }
+
+  // 合并全局和组件级别的 fieldNames，优先级：组件级 > 全局
+  var finalFieldNames = _objectSpread(_objectSpread({
+    label: 'label',
+    value: 'value',
+    children: 'children'
+  }, proSelectFieldNames), fieldNames);
+
+  /**
+   * 使用 tools.transformDataName 统一转换字段名
+   * 将原始数据转换为标准格式（label, value, children）
+   * tools.transformDataName 会自动检查数据是否已经是标准格式，如果是则直接返回
+   */
+  var transformFieldNames = useCallback(data => {
+    if (!data || !Array.isArray(data) || !data.length) {
+      return data || [];
+    }
+
+    // 使用 tools.transformDataName 转换字段名
+    // 它会自动检查数据是否已经是标准格式（label, value, children），如果是则直接返回
+    return tools.transformDataName(data, {
+      label: finalFieldNames.label,
+      value: finalFieldNames.value,
+      children: finalFieldNames.children
+    });
+  }, [finalFieldNames]);
   // 经过处理后的枚举数据，可能来自用户配置的处理函数后返回的结果
   var _useSetState = useSetState({}),
     _useSetState2 = _slicedToArray(_useSetState, 2),
@@ -105,6 +119,7 @@ export var ProTreeSelect = (props, ref) => {
 
   /**
    * 如果数据少于10条那么将数据平铺返回
+   * 注意：当 treeCheckable 为 false 时（单选模式），应保持树形结构，不进行平铺
    * @param dictsEnum 原始数据
    * @returns 平铺或者原始数据
    */
@@ -112,50 +127,33 @@ export var ProTreeSelect = (props, ref) => {
     if (treeCheckable) {
       return dictsEnum;
     }
-    var count = 0;
-    var result = [];
-    var loopCount = data => {
-      data.map(item => {
-        count += 1;
-        var children = item.children,
-          other = _objectWithoutProperties(item, _excluded2);
-        result.push(_objectSpread({}, other));
-        if (item.children) {
-          loopCount(item.children);
-        }
-        return item;
-      });
-    };
-    if ((dictsEnum === null || dictsEnum === void 0 ? void 0 : dictsEnum.length) > 10) {
-      return dictsEnum;
-    }
-    loopCount(dictsEnum);
-    if (count <= 10) {
-      return result;
-    }
+    // 单选模式（treeCheckable=false）时，保持树形结构，不进行平铺
+    // 因为 TreeSelect 在单选模式下也需要树形结构来正确展示层级关系
     return dictsEnum;
   };
 
   /**
    * 如果要展示code-name，那么将源数据的label转为code-name
-   * @param dataArr 源数据
+   * 注意：此时数据已经是标准格式（label, value, children）
+   * @param dataArr 源数据（标准格式）
    * @param showCodeAndName 是否展示code-name
    * @returns 转换结果
    */
   var transferDataSource = (dataArr, showCodeAndName) => {
+    if (!showCodeAndName) {
+      return dataArr;
+    }
     var copyData = cloneDeep(dataArr);
     var loop = arr => {
       arr.map(item => {
-        item[label] = `${item[code]}-${item[label]}`;
+        item.label = `${item.value}-${item.label}`;
         if (item.children) {
           loop(item.children);
         }
         return item;
       });
     };
-    if (showCodeAndName) {
-      loop(copyData);
-    }
+    loop(copyData);
     return copyData;
   };
   var defaultOnSuccessFun = res => {
@@ -167,7 +165,9 @@ export var ProTreeSelect = (props, ref) => {
       _message.error(msg);
       return;
     }
-    var resultData = transferDataSource(data, showCodeName);
+    // 先转换字段名，再处理 code-name 展示
+    var transformedData = transformFieldNames(data || []);
+    var resultData = transferDataSource(transformedData, showCodeName);
     setState({
       selectList: resultData || [],
       origDataSource: resultData || []
@@ -196,7 +196,9 @@ export var ProTreeSelect = (props, ref) => {
     onSuccess: data => {
       if (transformResponse && typeof transformResponse === 'function') {
         var responseData = transformResponse(data);
-        var resultData = transferDataSource(responseData, showCodeName);
+        // 先转换字段名，再处理 code-name 展示
+        var transformedData = transformFieldNames(responseData);
+        var resultData = transferDataSource(transformedData, showCodeName);
         setState({
           selectList: countChild(resultData),
           origDataSource: resultData
@@ -217,7 +219,9 @@ export var ProTreeSelect = (props, ref) => {
     if (fetchFunction !== null && fetchFunction !== void 0 && fetchFunction.data, useRequest !== null && useRequest !== void 0 && (_useRequest$options2 = useRequest.options) !== null && _useRequest$options2 !== void 0 && _useRequest$options2.cacheKey) {
       if (transformResponse && typeof transformResponse === 'function') {
         var responseData = transformResponse(fetchFunction.data);
-        var resultData = transferDataSource(responseData, showCodeName);
+        // 先转换字段名，再处理 code-name 展示
+        var transformedData = transformFieldNames(responseData);
+        var resultData = transferDataSource(transformedData, showCodeName);
         setState({
           selectList: countChild(resultData),
           origDataSource: resultData
@@ -226,23 +230,29 @@ export var ProTreeSelect = (props, ref) => {
         defaultOnSuccessFun(fetchFunction.data);
       }
     }
-  }, [fetchFunction === null || fetchFunction === void 0 ? void 0 : fetchFunction.data]);
+  }, [fetchFunction === null || fetchFunction === void 0 ? void 0 : fetchFunction.data, transformFieldNames]);
   useDeepCompareEffect(() => {
     if (enumCode) {
       var dictEnum = dics[enumCode] || [];
-      var resultData = transferDataSource(dictEnum, showCodeName);
+      // 先转换字段名，再处理 code-name 展示
+      var transformedData = transformFieldNames(dictEnum);
+      var resultData = transferDataSource(transformedData, showCodeName);
+      var processedList = countChild(resultData);
       setState({
-        selectList: countChild(resultData),
+        selectList: processedList,
         origDataSource: resultData
       });
     } else if (dataSource) {
-      var _resultData = transferDataSource(dataSource, showCodeName);
+      // 先转换字段名，再处理 code-name 展示
+      var _transformedData = transformFieldNames(dataSource);
+      var _resultData = transferDataSource(_transformedData, showCodeName);
+      var _processedList = countChild(_resultData);
       setState({
-        selectList: countChild(_resultData),
+        selectList: _processedList,
         origDataSource: _resultData
       });
     }
-  }, [enumCode, dataSource]);
+  }, [enumCode, dataSource, transformFieldNames]);
   useDeepCompareEffect(() => {
     // code存在huo dataSource存在，不执行接口请求
     if (enumCode || dataSource) {
@@ -278,6 +288,7 @@ export var ProTreeSelect = (props, ref) => {
 
   /**
    * 根据值进行回显
+   * 注意：此时数据已经是标准格式（label, value, children）
    * @param value 输入值
    * @returns 对应枚举label
    */
@@ -287,8 +298,8 @@ export var ProTreeSelect = (props, ref) => {
     }
     var labelResult = [];
     var queryLabel = dicts => dicts.map(item => {
-      if (value.includes(item[code])) {
-        var labelText = item[label];
+      if (value.includes(item.value)) {
+        var labelText = item.label;
         labelResult.push(labelText);
       }
       if (item.children) {
@@ -333,15 +344,22 @@ export var ProTreeSelect = (props, ref) => {
 
   /**
    * 自定义渲染子树
-   * @param node 每个节点数据对象
+   * 注意：此时数据已经是标准格式（label, value, children）
+   * @param node 每个节点数据对象（标准格式）
    * @returns 整个react节点树
    */
   var renderTreeNode = (node, grade) => {
     var children = node.children,
-      icon = node.icon;
-    var labelText = node[label];
-    var value = node[code];
-    var key = node[uuidKey];
+      icon = node.icon,
+      label = node.label,
+      nodeValue = node.value,
+      nodeKey = node.key;
+    var labelText = label;
+    var value = nodeValue;
+    var key = nodeValue; // 使用 value 作为 key，如果有 key 字段则使用 key
+    if (nodeKey !== undefined) {
+      key = nodeKey;
+    }
     // 三无走默认
     if (['undefined-undefined', 'undefined'].includes(labelText) && !value && !key) {
       if (showCodeName) {
@@ -409,7 +427,8 @@ export var ProTreeSelect = (props, ref) => {
 
   /**
    * 从树形数据中找出用户搜索得数据，并且将搜索字符串高亮标识
-   * @param data 全量树
+   * 注意：此时数据已经是标准格式（label, value, children）
+   * @param data 全量树（标准格式）
    * @param searchStr 搜索字符串
    * @returns 处理后得树
    */
@@ -426,11 +445,11 @@ export var ProTreeSelect = (props, ref) => {
         }
         // 迭代完子节点向上回溯，如果有匹配到得子节点，那么加入当前节点得孩子对象上,否则加入得是空数组
         item.children = currentNodeChild;
-        // 判断节点title是否包含关键字 如果有，那么直接进行下一循环
+        // 判断节点label是否包含关键字 如果有，那么直接进行下一循环
         // @ts-ignore
-        if (item[label].indexOf(keyWord) > -1 && (selectProps === null || selectProps === void 0 ? void 0 : selectProps.mode) !== 'treeSelect') {
+        if (item.label && item.label.indexOf(keyWord) > -1 && (selectProps === null || selectProps === void 0 ? void 0 : selectProps.mode) !== 'treeSelect') {
           var regExp = new RegExp(keyWord);
-          item[label] = item[label].replace(regExp, `<span class="highlight-search-text">${keyWord}</span>`);
+          item.label = item.label.replace(regExp, `<span class="highlight-search-text">${keyWord}</span>`);
         }
         currentAllNodeChild.push(item);
       }
@@ -468,16 +487,23 @@ export var ProTreeSelect = (props, ref) => {
       return _ref6.apply(this, arguments);
     };
   }(), 500);
+
+  /**
+   * 从树形数据中查找节点
+   * 注意：此时数据已经是标准格式（label, value, children）
+   * @param treeList 树形数据（标准格式）
+   * @param nodeValue 节点值
+   * @returns 找到的节点
+   */
   function findTreeNode(treeList, nodeValue) {
     var result = {};
     var filterData = function filterData(data) {
       for (var i = 0; i < data.length; i++) {
         var item = data[i];
-        if (item[code] === nodeValue || Array.isArray(nodeValue) && nodeValue.includes(item[code])) {
+        if (item.value === nodeValue || Array.isArray(nodeValue) && nodeValue.includes(item.value)) {
           result = _objectSpread({}, item);
-          result.value = item[code];
-          result.label = item[label];
-          delete result[(fieldNames === null || fieldNames === void 0 ? void 0 : fieldNames.children) || 'children'];
+          // 删除 children 字段，避免循环引用
+          delete result.children;
           return null;
         }
         // 一直递归到最后一层子节点
@@ -489,6 +515,13 @@ export var ProTreeSelect = (props, ref) => {
     filterData(treeList);
     return result;
   }
+  /**
+   * 根据值获取标签
+   * 注意：此时数据已经是标准格式（label, value, children）
+   * @param treeList 树形数据（标准格式）
+   * @param nodeValue 节点值
+   * @returns 标签对象或数组
+   */
   function getLabelByValue(treeList, nodeValue) {
     var result = nodeValue;
     if (nodeValue) {
@@ -500,19 +533,19 @@ export var ProTreeSelect = (props, ref) => {
           return item;
         });
       } else {
-        result = findTreeNode(treeList, nodeValue[code]);
+        result = findTreeNode(treeList, nodeValue.value);
       }
     }
     return result;
   }
   var handleChange = (newVal, label, extra) => {
     // newVal回来得一定是label、value
+    // 注意：此时数据已经是标准格式（label, value, children）
     var _selectList = selectList.map(item => {
-      var label = item.label;
-      var _label = label;
-      if (label) {
+      var _label = item.label;
+      if (_label) {
         var regex = /<span class="highlight-search-text">(.*?)<\/span>/g;
-        _label = item[label] = label === null || label === void 0 ? void 0 : label.replace(regex, '$1');
+        _label = item.label = _label.replace(regex, '$1');
       }
       return _objectSpread(_objectSpread({}, item), {}, {
         label: _label

package/es/ProTreeModal/style/index.less

@@ -1,6 +1,12 @@
 @root-entry-name: 'default';
 @import (reference) '~antd/es/style/themes/index.less';
 
+.pro-tree-select-drop-down-container{
+  .@{ant-prefix}-select-tree-checkbox{
+    margin-top: 0;
+  }
+}
+
 .pro-tree-modal-container .@{ant-prefix}-modal-content .@{ant-prefix}-modal-body {
   max-height: 498px;
 }