index.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. var _typeof3 = require("@babel/runtime/helpers/typeof");
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.convertChildrenToColumns = convertChildrenToColumns;
  8. exports.default = void 0;
  9. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  10. var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
  11. var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
  12. var _typeof2 = _interopRequireDefault(require("@babel/runtime/helpers/typeof"));
  13. var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
  14. var _objectWithoutProperties2 = _interopRequireDefault(require("@babel/runtime/helpers/objectWithoutProperties"));
  15. var _toArray = _interopRequireDefault(require("rc-util/lib/Children/toArray"));
  16. var _warning = _interopRequireDefault(require("rc-util/lib/warning"));
  17. var React = _interopRequireWildcard(require("react"));
  18. var _constant = require("../../constant");
  19. var _legacyUtil = require("../../utils/legacyUtil");
  20. var _useWidthColumns3 = _interopRequireDefault(require("./useWidthColumns"));
  21. var _excluded = ["children"],
  22. _excluded2 = ["fixed"];
  23. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
  24. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof3(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  25. function convertChildrenToColumns(children) {
  26. return (0, _toArray.default)(children).filter(function (node) {
  27. return /*#__PURE__*/React.isValidElement(node);
  28. }).map(function (_ref) {
  29. var key = _ref.key,
  30. props = _ref.props;
  31. var nodeChildren = props.children,
  32. restProps = (0, _objectWithoutProperties2.default)(props, _excluded);
  33. var column = (0, _objectSpread2.default)({
  34. key: key
  35. }, restProps);
  36. if (nodeChildren) {
  37. column.children = convertChildrenToColumns(nodeChildren);
  38. }
  39. return column;
  40. });
  41. }
  42. function filterHiddenColumns(columns) {
  43. return columns.filter(function (column) {
  44. return column && (0, _typeof2.default)(column) === 'object' && !column.hidden;
  45. }).map(function (column) {
  46. var subColumns = column.children;
  47. if (subColumns && subColumns.length > 0) {
  48. return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, column), {}, {
  49. children: filterHiddenColumns(subColumns)
  50. });
  51. }
  52. return column;
  53. });
  54. }
  55. function flatColumns(columns) {
  56. var parentKey = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'key';
  57. return columns.filter(function (column) {
  58. return column && (0, _typeof2.default)(column) === 'object';
  59. }).reduce(function (list, column, index) {
  60. var fixed = column.fixed;
  61. // Convert `fixed='true'` to `fixed='left'` instead
  62. var parsedFixed = fixed === true ? 'left' : fixed;
  63. var mergedKey = "".concat(parentKey, "-").concat(index);
  64. var subColumns = column.children;
  65. if (subColumns && subColumns.length > 0) {
  66. return [].concat((0, _toConsumableArray2.default)(list), (0, _toConsumableArray2.default)(flatColumns(subColumns, mergedKey).map(function (subColum) {
  67. var _subColum$fixed;
  68. return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, subColum), {}, {
  69. fixed: (_subColum$fixed = subColum.fixed) !== null && _subColum$fixed !== void 0 ? _subColum$fixed : parsedFixed
  70. });
  71. })));
  72. }
  73. return [].concat((0, _toConsumableArray2.default)(list), [(0, _objectSpread2.default)((0, _objectSpread2.default)({
  74. key: mergedKey
  75. }, column), {}, {
  76. fixed: parsedFixed
  77. })]);
  78. }, []);
  79. }
  80. function revertForRtl(columns) {
  81. return columns.map(function (column) {
  82. var fixed = column.fixed,
  83. restProps = (0, _objectWithoutProperties2.default)(column, _excluded2);
  84. // Convert `fixed='left'` to `fixed='right'` instead
  85. var parsedFixed = fixed;
  86. if (fixed === 'left') {
  87. parsedFixed = 'right';
  88. } else if (fixed === 'right') {
  89. parsedFixed = 'left';
  90. }
  91. return (0, _objectSpread2.default)({
  92. fixed: parsedFixed
  93. }, restProps);
  94. });
  95. }
  96. /**
  97. * Parse `columns` & `children` into `columns`.
  98. */
  99. function useColumns(_ref2, transformColumns) {
  100. var prefixCls = _ref2.prefixCls,
  101. columns = _ref2.columns,
  102. children = _ref2.children,
  103. expandable = _ref2.expandable,
  104. expandedKeys = _ref2.expandedKeys,
  105. columnTitle = _ref2.columnTitle,
  106. getRowKey = _ref2.getRowKey,
  107. onTriggerExpand = _ref2.onTriggerExpand,
  108. expandIcon = _ref2.expandIcon,
  109. rowExpandable = _ref2.rowExpandable,
  110. expandIconColumnIndex = _ref2.expandIconColumnIndex,
  111. _ref2$expandedRowOffs = _ref2.expandedRowOffset,
  112. expandedRowOffset = _ref2$expandedRowOffs === void 0 ? 0 : _ref2$expandedRowOffs,
  113. direction = _ref2.direction,
  114. expandRowByClick = _ref2.expandRowByClick,
  115. columnWidth = _ref2.columnWidth,
  116. fixed = _ref2.fixed,
  117. scrollWidth = _ref2.scrollWidth,
  118. clientWidth = _ref2.clientWidth;
  119. var baseColumns = React.useMemo(function () {
  120. var newColumns = columns || convertChildrenToColumns(children) || [];
  121. return filterHiddenColumns(newColumns.slice());
  122. }, [columns, children]);
  123. // ========================== Expand ==========================
  124. var withExpandColumns = React.useMemo(function () {
  125. if (expandable) {
  126. var cloneColumns = baseColumns.slice();
  127. // >>> Warning if use `expandIconColumnIndex`
  128. if (process.env.NODE_ENV !== 'production' && expandIconColumnIndex >= 0) {
  129. (0, _warning.default)(false, '`expandIconColumnIndex` is deprecated. Please use `Table.EXPAND_COLUMN` in `columns` instead.');
  130. }
  131. // >>> Insert expand column if not exist
  132. if (!cloneColumns.includes(_constant.EXPAND_COLUMN)) {
  133. var expandColIndex = expandIconColumnIndex || 0;
  134. var insertIndex = expandColIndex === 0 && fixed === 'right' ? baseColumns.length : expandColIndex;
  135. if (insertIndex >= 0) {
  136. cloneColumns.splice(insertIndex, 0, _constant.EXPAND_COLUMN);
  137. }
  138. }
  139. // >>> Deduplicate additional expand column
  140. if (process.env.NODE_ENV !== 'production' && cloneColumns.filter(function (c) {
  141. return c === _constant.EXPAND_COLUMN;
  142. }).length > 1) {
  143. (0, _warning.default)(false, 'There exist more than one `EXPAND_COLUMN` in `columns`.');
  144. }
  145. var expandColumnIndex = cloneColumns.indexOf(_constant.EXPAND_COLUMN);
  146. cloneColumns = cloneColumns.filter(function (column, index) {
  147. return column !== _constant.EXPAND_COLUMN || index === expandColumnIndex;
  148. });
  149. // >>> Check if expand column need to fixed
  150. var prevColumn = baseColumns[expandColumnIndex];
  151. var fixedColumn;
  152. if (fixed) {
  153. fixedColumn = fixed;
  154. } else {
  155. fixedColumn = prevColumn ? prevColumn.fixed : null;
  156. }
  157. // >>> Create expandable column
  158. var expandColumn = (0, _defineProperty2.default)((0, _defineProperty2.default)((0, _defineProperty2.default)((0, _defineProperty2.default)((0, _defineProperty2.default)((0, _defineProperty2.default)({}, _legacyUtil.INTERNAL_COL_DEFINE, {
  159. className: "".concat(prefixCls, "-expand-icon-col"),
  160. columnType: 'EXPAND_COLUMN'
  161. }), "title", columnTitle), "fixed", fixedColumn), "className", "".concat(prefixCls, "-row-expand-icon-cell")), "width", columnWidth), "render", function render(_, record, index) {
  162. var rowKey = getRowKey(record, index);
  163. var expanded = expandedKeys.has(rowKey);
  164. var recordExpandable = rowExpandable ? rowExpandable(record) : true;
  165. var icon = expandIcon({
  166. prefixCls: prefixCls,
  167. expanded: expanded,
  168. expandable: recordExpandable,
  169. record: record,
  170. onExpand: onTriggerExpand
  171. });
  172. if (expandRowByClick) {
  173. return /*#__PURE__*/React.createElement("span", {
  174. onClick: function onClick(e) {
  175. return e.stopPropagation();
  176. }
  177. }, icon);
  178. }
  179. return icon;
  180. });
  181. return cloneColumns.map(function (col, index) {
  182. var column = col === _constant.EXPAND_COLUMN ? expandColumn : col;
  183. if (index < expandedRowOffset) {
  184. return (0, _objectSpread2.default)((0, _objectSpread2.default)({}, column), {}, {
  185. fixed: column.fixed || 'left'
  186. });
  187. }
  188. return column;
  189. });
  190. }
  191. if (process.env.NODE_ENV !== 'production' && baseColumns.includes(_constant.EXPAND_COLUMN)) {
  192. (0, _warning.default)(false, '`expandable` is not config but there exist `EXPAND_COLUMN` in `columns`.');
  193. }
  194. return baseColumns.filter(function (col) {
  195. return col !== _constant.EXPAND_COLUMN;
  196. });
  197. // eslint-disable-next-line react-hooks/exhaustive-deps
  198. }, [expandable, baseColumns, getRowKey, expandedKeys, expandIcon, direction, expandedRowOffset]);
  199. // ========================= Transform ========================
  200. var mergedColumns = React.useMemo(function () {
  201. var finalColumns = withExpandColumns;
  202. if (transformColumns) {
  203. finalColumns = transformColumns(finalColumns);
  204. }
  205. // Always provides at least one column for table display
  206. if (!finalColumns.length) {
  207. finalColumns = [{
  208. render: function render() {
  209. return null;
  210. }
  211. }];
  212. }
  213. return finalColumns;
  214. // eslint-disable-next-line react-hooks/exhaustive-deps
  215. }, [transformColumns, withExpandColumns, direction]);
  216. // ========================== Flatten =========================
  217. var flattenColumns = React.useMemo(function () {
  218. if (direction === 'rtl') {
  219. return revertForRtl(flatColumns(mergedColumns));
  220. }
  221. return flatColumns(mergedColumns);
  222. // eslint-disable-next-line react-hooks/exhaustive-deps
  223. }, [mergedColumns, direction, scrollWidth]);
  224. // ========================= Gap Fixed ========================
  225. var hasGapFixed = React.useMemo(function () {
  226. // Fixed: left, since old browser not support `findLastIndex`, we should use reverse loop
  227. var lastLeftIndex = -1;
  228. for (var i = flattenColumns.length - 1; i >= 0; i -= 1) {
  229. var colFixed = flattenColumns[i].fixed;
  230. if (colFixed === 'left' || colFixed === true) {
  231. lastLeftIndex = i;
  232. break;
  233. }
  234. }
  235. if (lastLeftIndex >= 0) {
  236. for (var _i = 0; _i <= lastLeftIndex; _i += 1) {
  237. var _colFixed = flattenColumns[_i].fixed;
  238. if (_colFixed !== 'left' && _colFixed !== true) {
  239. return true;
  240. }
  241. }
  242. }
  243. // Fixed: right
  244. var firstRightIndex = flattenColumns.findIndex(function (_ref3) {
  245. var colFixed = _ref3.fixed;
  246. return colFixed === 'right';
  247. });
  248. if (firstRightIndex >= 0) {
  249. for (var _i2 = firstRightIndex; _i2 < flattenColumns.length; _i2 += 1) {
  250. var _colFixed2 = flattenColumns[_i2].fixed;
  251. if (_colFixed2 !== 'right') {
  252. return true;
  253. }
  254. }
  255. }
  256. return false;
  257. }, [flattenColumns]);
  258. // ========================= FillWidth ========================
  259. var _useWidthColumns = (0, _useWidthColumns3.default)(flattenColumns, scrollWidth, clientWidth),
  260. _useWidthColumns2 = (0, _slicedToArray2.default)(_useWidthColumns, 2),
  261. filledColumns = _useWidthColumns2[0],
  262. realScrollWidth = _useWidthColumns2[1];
  263. return [mergedColumns, filledColumns, realScrollWidth, hasGapFixed];
  264. }
  265. var _default = exports.default = useColumns;