120 lines
3.3 KiB
JavaScript
120 lines
3.3 KiB
JavaScript
function _extends() { _extends = Object.assign ? Object.assign.bind() : function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }
|
|
import { useContext } from '@rc-component/context';
|
|
import { clsx } from 'clsx';
|
|
import * as React from 'react';
|
|
import { getCellProps } from "../Body/BodyRow";
|
|
import Cell from "../Cell";
|
|
import { GridContext } from "./context";
|
|
/**
|
|
* Return the width of the column by `colSpan`.
|
|
* When `colSpan` is `0` will be trade as `1`.
|
|
*/
|
|
export function getColumnWidth(colIndex, colSpan, columnsOffset) {
|
|
const mergedColSpan = colSpan || 1;
|
|
return columnsOffset[colIndex + mergedColSpan] - (columnsOffset[colIndex] || 0);
|
|
}
|
|
const VirtualCell = props => {
|
|
const {
|
|
rowInfo,
|
|
column,
|
|
colIndex,
|
|
indent,
|
|
index,
|
|
component,
|
|
renderIndex,
|
|
record,
|
|
style,
|
|
className,
|
|
inverse,
|
|
getHeight
|
|
} = props;
|
|
const {
|
|
render,
|
|
dataIndex,
|
|
className: columnClassName,
|
|
width: colWidth
|
|
} = column;
|
|
const {
|
|
columnsOffset
|
|
} = useContext(GridContext, ['columnsOffset']);
|
|
|
|
// TODO: support `expandableRowOffset`
|
|
const {
|
|
key,
|
|
fixedInfo,
|
|
appendCellNode,
|
|
additionalCellProps
|
|
} = getCellProps(rowInfo, column, colIndex, indent, index);
|
|
const {
|
|
style: cellStyle,
|
|
colSpan = 1,
|
|
rowSpan = 1
|
|
} = additionalCellProps;
|
|
|
|
// ========================= ColWidth =========================
|
|
// column width
|
|
const startColIndex = colIndex - 1;
|
|
const concatColWidth = getColumnWidth(startColIndex, colSpan, columnsOffset);
|
|
|
|
// margin offset
|
|
const marginOffset = colSpan > 1 ? colWidth - concatColWidth : 0;
|
|
|
|
// ========================== Style ===========================
|
|
const mergedStyle = {
|
|
...cellStyle,
|
|
...style,
|
|
flex: `0 0 ${concatColWidth}px`,
|
|
width: `${concatColWidth}px`,
|
|
marginRight: marginOffset,
|
|
pointerEvents: 'auto'
|
|
};
|
|
|
|
// When `colSpan` or `rowSpan` is `0`, should skip render.
|
|
const needHide = React.useMemo(() => {
|
|
if (inverse) {
|
|
return rowSpan <= 1;
|
|
} else {
|
|
return colSpan === 0 || rowSpan === 0 || rowSpan > 1;
|
|
}
|
|
}, [rowSpan, colSpan, inverse]);
|
|
|
|
// 0 rowSpan or colSpan should not render
|
|
if (needHide) {
|
|
mergedStyle.visibility = 'hidden';
|
|
} else if (inverse) {
|
|
mergedStyle.height = getHeight?.(rowSpan);
|
|
}
|
|
const mergedRender = needHide ? () => null : render;
|
|
|
|
// ========================== Render ==========================
|
|
const cellSpan = {};
|
|
|
|
// Virtual should reset `colSpan` & `rowSpan`
|
|
if (rowSpan === 0 || colSpan === 0) {
|
|
cellSpan.rowSpan = 1;
|
|
cellSpan.colSpan = 1;
|
|
}
|
|
return /*#__PURE__*/React.createElement(Cell, _extends({
|
|
className: clsx(columnClassName, className),
|
|
ellipsis: column.ellipsis,
|
|
align: column.align,
|
|
scope: column.rowScope,
|
|
component: component,
|
|
prefixCls: rowInfo.prefixCls,
|
|
key: key,
|
|
record: record,
|
|
index: index,
|
|
renderIndex: renderIndex,
|
|
dataIndex: dataIndex,
|
|
render: mergedRender,
|
|
shouldCellUpdate: column.shouldCellUpdate
|
|
}, fixedInfo, {
|
|
appendNode: appendCellNode,
|
|
additionalProps: {
|
|
...additionalCellProps,
|
|
style: mergedStyle,
|
|
...cellSpan
|
|
}
|
|
}));
|
|
};
|
|
export default VirtualCell; |