linkmore-design/lib/table/rc-table/Body/index.js

🌈 🚀lm组件库。🚀

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;
var React = _interopRequireWildcard(require("react"));
var _BodyContext = _interopRequireDefault(require("../context/BodyContext"));
var _HoverContext = _interopRequireDefault(require("../context/HoverContext"));
var _PerfContext = _interopRequireDefault(require("../context/PerfContext"));
var _ResizeContext = _interopRequireDefault(require("../context/ResizeContext"));
var _TableContext = _interopRequireDefault(require("../context/TableContext"));
var _ContextSelector = require("../ContextSelector");
var _useFlattenRecords = _interopRequireDefault(require("../hooks/useFlattenRecords"));
var _valueUtil = require("../utils/valueUtil");
var _BodyRow = _interopRequireDefault(require("./BodyRow"));
var _ExpandedRow = _interopRequireDefault(require("./ExpandedRow"));
var _MeasureRow = _interopRequireDefault(require("./MeasureRow"));
function Body({
  data,
  getRowKey,
  measureColumnWidth,
  expandedKeys,
  onRow,
  rowExpandable,
  emptyNode,
  childrenColumnName
}) {
  const onColumnResize = (0, _ContextSelector.useContextSelector)(_ResizeContext.default, 'onColumnResize');
  const {
    prefixCls,
    getComponent
  } = (0, _ContextSelector.useContextSelector)(_TableContext.default, ['prefixCls', 'getComponent']);
  const flattenColumns = (0, _ContextSelector.useContextSelector)(_BodyContext.default, 'flattenColumns');
  const flattenData = (0, _useFlattenRecords.default)(data, childrenColumnName, expandedKeys, getRowKey);

  // =================== Performance ====================
  const perfRef = React.useRef({
    renderWithProps: false
  });

  // ====================== Hover =======================
  const [startRow, setStartRow] = React.useState(-1);
  const [endRow, setEndRow] = React.useState(-1);
  const onHover = React.useCallback((start, end) => {
    setStartRow(start);
    setEndRow(end);
  }, []);

  // ====================== Render ======================
  const bodyNode = React.useMemo(() => {
    const WrapperComponent = getComponent(['body', 'wrapper'], 'tbody');
    const trComponent = getComponent(['body', 'row'], 'tr');
    const tdComponent = getComponent(['body', 'cell'], 'td');
    let rows;
    if (data.length) {
      rows = flattenData.map((item, idx) => {
        const {
          record,
          indent,
          index: renderIndex
        } = item;
        const key = getRowKey(record, idx);
        return /*#__PURE__*/React.createElement(_BodyRow.default, {
          key: key,
          rowKey: key,
          record: record,
          recordKey: key,
          index: idx,
          renderIndex: renderIndex,
          rowComponent: trComponent,
          cellComponent: tdComponent,
          expandedKeys: expandedKeys,
          onRow: onRow,
          getRowKey: getRowKey,
          rowExpandable: rowExpandable,
          childrenColumnName: childrenColumnName,
          indent: indent
        });
      });
    } else {
      rows = /*#__PURE__*/React.createElement(_ExpandedRow.default, {
        expanded: true,
        className: `${prefixCls}-placeholder`,
        prefixCls: prefixCls,
        component: trComponent,
        cellComponent: tdComponent,
        colSpan: flattenColumns.length,
        isEmpty: true
      }, emptyNode);
    }
    const columnsKey = (0, _valueUtil.getColumnsKey)(flattenColumns);
    return /*#__PURE__*/React.createElement(WrapperComponent, {
      className: `${prefixCls}-tbody`
    }, measureColumnWidth && /*#__PURE__*/React.createElement(_MeasureRow.default, {
      prefixCls: prefixCls,
      columnsKey: columnsKey,
      onColumnResize: onColumnResize
    }), rows);
  }, [data, prefixCls, onRow, measureColumnWidth, expandedKeys, getRowKey, getComponent, emptyNode, flattenColumns, childrenColumnName, onColumnResize, rowExpandable, flattenData]);
  return /*#__PURE__*/React.createElement(_PerfContext.default.Provider, {
    value: perfRef.current
  }, /*#__PURE__*/React.createElement(_HoverContext.default.Provider, {
    value: {
      startRow,
      endRow,
      onHover
    }
  }, bodyNode));
}
const MemoBody = /*#__PURE__*/React.memo(Body);
MemoBody.displayName = 'Body';
var _default = MemoBody;
exports.default = _default;