acud/es/menu/rc-menu/utils/nodeUtil.js

acg react 组件库

import _extends from "@babel/runtime/helpers/esm/extends";
import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
import * as React from 'react';
import toArray from "rc-util/es/Children/toArray";
export function parseChildren(children, keyPath) {
  return toArray(children).map(function (child, index) {
    var _a, _b;
    if (/*#__PURE__*/React.isValidElement(child)) {
      var key = child.key;
      var eventKey = (_b = (_a = child.props) === null || _a === void 0 ? void 0 : _a.eventKey) !== null && _b !== void 0 ? _b : key;
      var emptyKey = eventKey === null || eventKey === undefined;
      if (emptyKey) {
        eventKey = "tmp_key-".concat([].concat(_toConsumableArray(keyPath), [index]).join('-'));
      }
      var cloneProps = {
        key: eventKey,
        eventKey: eventKey
      };
      if (process.env.NODE_ENV !== 'production' && emptyKey) {
        cloneProps.warnKey = true;
      }
      return /*#__PURE__*/React.cloneElement(child, cloneProps);
    }
    return child;
  });
}
/**
 * 如果是带icon的submenu，则将 children className 更新为 'item-with-icon'。
 * 如果 subStyle 中存在 paddingLeft，则将其传递给子元素的 sub 属性。
 *
 * @param children submenu 的子元素
 * @param className submenu 的 className
 * @param style submenu title 的 style
 * @returns React.ReactNode[] 返回一个包含修改后的子元素的数组
 */
export function parseChildrenWithIcon(children, className, subStyle) {
  return toArray(children).map(function (child) {
    if (/*#__PURE__*/React.isValidElement(child)) {
      var subWithIcon = className === null || className === void 0 ? void 0 : className.includes('submenu-with-icon');
      var cloneProps = _extends(_extends({}, subWithIcon ? {
        className: 'item-with-icon'
      } : {}), (subStyle === null || subStyle === void 0 ? void 0 : subStyle.paddingLeft) ? {
        sub: subStyle.paddingLeft
      } : {});
      return /*#__PURE__*/React.cloneElement(child, cloneProps);
    }
    return child;
  });
}