269 lines
10 KiB
JavaScript
269 lines
10 KiB
JavaScript
|
import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
|
||
|
|
import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
|
||
|
|
import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
|
||
|
|
import _typeof from "@babel/runtime/helpers/esm/typeof";
|
||
|
|
import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
|
||
|
|
import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
|
||
|
|
var _excluded = ["children"],
|
||
|
|
_excluded2 = ["fixed"];
|
||
|
|
import toArray from "rc-util/es/Children/toArray";
|
||
|
|
import warning from "rc-util/es/warning";
|
||
|
|
import * as React from 'react';
|
||
|
|
import { EXPAND_COLUMN } from "../../constant";
|
||
|
|
import { INTERNAL_COL_DEFINE } from "../../utils/legacyUtil";
|
||
|
|
import useWidthColumns from "./useWidthColumns";
|
||
|
|
export function convertChildrenToColumns(children) {
|
||
|
|
return toArray(children).filter(function (node) {
|
||
|
|
return /*#__PURE__*/React.isValidElement(node);
|
||
|
|
}).map(function (_ref) {
|
||
|
|
var key = _ref.key,
|
||
|
|
props = _ref.props;
|
||
|
|
var nodeChildren = props.children,
|
||
|
|
restProps = _objectWithoutProperties(props, _excluded);
|
||
|
|
var column = _objectSpread({
|
||
|
|
key: key
|
||
|
|
}, restProps);
|
||
|
|
if (nodeChildren) {
|
||
|
|
column.children = convertChildrenToColumns(nodeChildren);
|
||
|
|
}
|
||
|
|
return column;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
function filterHiddenColumns(columns) {
|
||
|
|
return columns.filter(function (column) {
|
||
|
|
return column && _typeof(column) === 'object' && !column.hidden;
|
||
|
|
}).map(function (column) {
|
||
|
|
var subColumns = column.children;
|
||
|
|
if (subColumns && subColumns.length > 0) {
|
||
|
|
return _objectSpread(_objectSpread({}, column), {}, {
|
||
|
|
children: filterHiddenColumns(subColumns)
|
||
|
|
});
|
||
|
|
}
|
||
|
|
return column;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
function flatColumns(columns) {
|
||
|
|
var parentKey = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'key';
|
||
|
|
return columns.filter(function (column) {
|
||
|
|
return column && _typeof(column) === 'object';
|
||
|
|
}).reduce(function (list, column, index) {
|
||
|
|
var fixed = column.fixed;
|
||
|
|
// Convert `fixed='true'` to `fixed='left'` instead
|
||
|
|
var parsedFixed = fixed === true ? 'left' : fixed;
|
||
|
|
var mergedKey = "".concat(parentKey, "-").concat(index);
|
||
|
|
var subColumns = column.children;
|
||
|
|
if (subColumns && subColumns.length > 0) {
|
||
|
|
return [].concat(_toConsumableArray(list), _toConsumableArray(flatColumns(subColumns, mergedKey).map(function (subColum) {
|
||
|
|
var _subColum$fixed;
|
||
|
|
return _objectSpread(_objectSpread({}, subColum), {}, {
|
||
|
|
fixed: (_subColum$fixed = subColum.fixed) !== null && _subColum$fixed !== void 0 ? _subColum$fixed : parsedFixed
|
||
|
|
});
|
||
|
|
})));
|
||
|
|
}
|
||
|
|
return [].concat(_toConsumableArray(list), [_objectSpread(_objectSpread({
|
||
|
|
key: mergedKey
|
||
|
|
}, column), {}, {
|
||
|
|
fixed: parsedFixed
|
||
|
|
})]);
|
||
|
|
}, []);
|
||
|
|
}
|
||
|
|
function revertForRtl(columns) {
|
||
|
|
return columns.map(function (column) {
|
||
|
|
var fixed = column.fixed,
|
||
|
|
restProps = _objectWithoutProperties(column, _excluded2);
|
||
|
|
|
||
|
|
// Convert `fixed='left'` to `fixed='right'` instead
|
||
|
|
var parsedFixed = fixed;
|
||
|
|
if (fixed === 'left') {
|
||
|
|
parsedFixed = 'right';
|
||
|
|
} else if (fixed === 'right') {
|
||
|
|
parsedFixed = 'left';
|
||
|
|
}
|
||
|
|
return _objectSpread({
|
||
|
|
fixed: parsedFixed
|
||
|
|
}, restProps);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Parse `columns` & `children` into `columns`.
|
||
|
|
*/
|
||
|
|
function useColumns(_ref2, transformColumns) {
|
||
|
|
var prefixCls = _ref2.prefixCls,
|
||
|
|
columns = _ref2.columns,
|
||
|
|
children = _ref2.children,
|
||
|
|
expandable = _ref2.expandable,
|
||
|
|
expandedKeys = _ref2.expandedKeys,
|
||
|
|
columnTitle = _ref2.columnTitle,
|
||
|
|
getRowKey = _ref2.getRowKey,
|
||
|
|
onTriggerExpand = _ref2.onTriggerExpand,
|
||
|
|
expandIcon = _ref2.expandIcon,
|
||
|
|
rowExpandable = _ref2.rowExpandable,
|
||
|
|
expandIconColumnIndex = _ref2.expandIconColumnIndex,
|
||
|
|
_ref2$expandedRowOffs = _ref2.expandedRowOffset,
|
||
|
|
expandedRowOffset = _ref2$expandedRowOffs === void 0 ? 0 : _ref2$expandedRowOffs,
|
||
|
|
direction = _ref2.direction,
|
||
|
|
expandRowByClick = _ref2.expandRowByClick,
|
||
|
|
columnWidth = _ref2.columnWidth,
|
||
|
|
fixed = _ref2.fixed,
|
||
|
|
scrollWidth = _ref2.scrollWidth,
|
||
|
|
clientWidth = _ref2.clientWidth;
|
||
|
|
var baseColumns = React.useMemo(function () {
|
||
|
|
var newColumns = columns || convertChildrenToColumns(children) || [];
|
||
|
|
return filterHiddenColumns(newColumns.slice());
|
||
|
|
}, [columns, children]);
|
||
|
|
|
||
|
|
// ========================== Expand ==========================
|
||
|
|
var withExpandColumns = React.useMemo(function () {
|
||
|
|
if (expandable) {
|
||
|
|
var cloneColumns = baseColumns.slice();
|
||
|
|
|
||
|
|
// >>> Warning if use `expandIconColumnIndex`
|
||
|
|
if (process.env.NODE_ENV !== 'production' && expandIconColumnIndex >= 0) {
|
||
|
|
warning(false, '`expandIconColumnIndex` is deprecated. Please use `Table.EXPAND_COLUMN` in `columns` instead.');
|
||
|
|
}
|
||
|
|
|
||
|
|
// >>> Insert expand column if not exist
|
||
|
|
if (!cloneColumns.includes(EXPAND_COLUMN)) {
|
||
|
|
var expandColIndex = expandIconColumnIndex || 0;
|
||
|
|
var insertIndex = expandColIndex === 0 && fixed === 'right' ? baseColumns.length : expandColIndex;
|
||
|
|
if (insertIndex >= 0) {
|
||
|
|
cloneColumns.splice(insertIndex, 0, EXPAND_COLUMN);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// >>> Deduplicate additional expand column
|
||
|
|
if (process.env.NODE_ENV !== 'production' && cloneColumns.filter(function (c) {
|
||
|
|
return c === EXPAND_COLUMN;
|
||
|
|
}).length > 1) {
|
||
|
|
warning(false, 'There exist more than one `EXPAND_COLUMN` in `columns`.');
|
||
|
|
}
|
||
|
|
var expandColumnIndex = cloneColumns.indexOf(EXPAND_COLUMN);
|
||
|
|
cloneColumns = cloneColumns.filter(function (column, index) {
|
||
|
|
return column !== EXPAND_COLUMN || index === expandColumnIndex;
|
||
|
|
});
|
||
|
|
|
||
|
|
// >>> Check if expand column need to fixed
|
||
|
|
var prevColumn = baseColumns[expandColumnIndex];
|
||
|
|
var fixedColumn;
|
||
|
|
if (fixed) {
|
||
|
|
fixedColumn = fixed;
|
||
|
|
} else {
|
||
|
|
fixedColumn = prevColumn ? prevColumn.fixed : null;
|
||
|
|
}
|
||
|
|
|
||
|
|
// >>> Create expandable column
|
||
|
|
var expandColumn = _defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty(_defineProperty({}, INTERNAL_COL_DEFINE, {
|
||
|
|
className: "".concat(prefixCls, "-expand-icon-col"),
|
||
|
|
columnType: 'EXPAND_COLUMN'
|
||
|
|
}), "title", columnTitle), "fixed", fixedColumn), "className", "".concat(prefixCls, "-row-expand-icon-cell")), "width", columnWidth), "render", function render(_, record, index) {
|
||
|
|
var rowKey = getRowKey(record, index);
|
||
|
|
var expanded = expandedKeys.has(rowKey);
|
||
|
|
var recordExpandable = rowExpandable ? rowExpandable(record) : true;
|
||
|
|
var icon = expandIcon({
|
||
|
|
prefixCls: prefixCls,
|
||
|
|
expanded: expanded,
|
||
|
|
expandable: recordExpandable,
|
||
|
|
record: record,
|
||
|
|
onExpand: onTriggerExpand
|
||
|
|
});
|
||
|
|
if (expandRowByClick) {
|
||
|
|
return /*#__PURE__*/React.createElement("span", {
|
||
|
|
onClick: function onClick(e) {
|
||
|
|
return e.stopPropagation();
|
||
|
|
}
|
||
|
|
}, icon);
|
||
|
|
}
|
||
|
|
return icon;
|
||
|
|
});
|
||
|
|
return cloneColumns.map(function (col, index) {
|
||
|
|
var column = col === EXPAND_COLUMN ? expandColumn : col;
|
||
|
|
if (index < expandedRowOffset) {
|
||
|
|
return _objectSpread(_objectSpread({}, column), {}, {
|
||
|
|
fixed: column.fixed || 'left'
|
||
|
|
});
|
||
|
|
}
|
||
|
|
return column;
|
||
|
|
});
|
||
|
|
}
|
||
|
|
if (process.env.NODE_ENV !== 'production' && baseColumns.includes(EXPAND_COLUMN)) {
|
||
|
|
warning(false, '`expandable` is not config but there exist `EXPAND_COLUMN` in `columns`.');
|
||
|
|
}
|
||
|
|
return baseColumns.filter(function (col) {
|
||
|
|
return col !== EXPAND_COLUMN;
|
||
|
|
});
|
||
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||
|
|
}, [expandable, baseColumns, getRowKey, expandedKeys, expandIcon, direction, expandedRowOffset]);
|
||
|
|
|
||
|
|
// ========================= Transform ========================
|
||
|
|
var mergedColumns = React.useMemo(function () {
|
||
|
|
var finalColumns = withExpandColumns;
|
||
|
|
if (transformColumns) {
|
||
|
|
finalColumns = transformColumns(finalColumns);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Always provides at least one column for table display
|
||
|
|
if (!finalColumns.length) {
|
||
|
|
finalColumns = [{
|
||
|
|
render: function render() {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
}];
|
||
|
|
}
|
||
|
|
return finalColumns;
|
||
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||
|
|
}, [transformColumns, withExpandColumns, direction]);
|
||
|
|
|
||
|
|
// ========================== Flatten =========================
|
||
|
|
var flattenColumns = React.useMemo(function () {
|
||
|
|
if (direction === 'rtl') {
|
||
|
|
return revertForRtl(flatColumns(mergedColumns));
|
||
|
|
}
|
||
|
|
return flatColumns(mergedColumns);
|
||
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||
|
|
}, [mergedColumns, direction, scrollWidth]);
|
||
|
|
|
||
|
|
// ========================= Gap Fixed ========================
|
||
|
|
var hasGapFixed = React.useMemo(function () {
|
||
|
|
// Fixed: left, since old browser not support `findLastIndex`, we should use reverse loop
|
||
|
|
var lastLeftIndex = -1;
|
||
|
|
for (var i = flattenColumns.length - 1; i >= 0; i -= 1) {
|
||
|
|
var colFixed = flattenColumns[i].fixed;
|
||
|
|
if (colFixed === 'left' || colFixed === true) {
|
||
|
|
lastLeftIndex = i;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (lastLeftIndex >= 0) {
|
||
|
|
for (var _i = 0; _i <= lastLeftIndex; _i += 1) {
|
||
|
|
var _colFixed = flattenColumns[_i].fixed;
|
||
|
|
if (_colFixed !== 'left' && _colFixed !== true) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fixed: right
|
||
|
|
var firstRightIndex = flattenColumns.findIndex(function (_ref3) {
|
||
|
|
var colFixed = _ref3.fixed;
|
||
|
|
return colFixed === 'right';
|
||
|
|
});
|
||
|
|
if (firstRightIndex >= 0) {
|
||
|
|
for (var _i2 = firstRightIndex; _i2 < flattenColumns.length; _i2 += 1) {
|
||
|
|
var _colFixed2 = flattenColumns[_i2].fixed;
|
||
|
|
if (_colFixed2 !== 'right') {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}, [flattenColumns]);
|
||
|
|
|
||
|
|
// ========================= FillWidth ========================
|
||
|
|
var _useWidthColumns = useWidthColumns(flattenColumns, scrollWidth, clientWidth),
|
||
|
|
_useWidthColumns2 = _slicedToArray(_useWidthColumns, 2),
|
||
|
|
filledColumns = _useWidthColumns2[0],
|
||
|
|
realScrollWidth = _useWidthColumns2[1];
|
||
|
|
return [mergedColumns, filledColumns, realScrollWidth, hasGapFixed];
|
||
|
|
}
|
||
|
|
export default useColumns;
|