1

2026-05-25 17:08:18 +08:00
parent df501f6151
commit f1259b71b5
9178 changed files with 1626125 additions and 0 deletions
@@ -0,0 +1,105 @@
+import * as React from 'react';
+import { clsx } from 'clsx';
+import { useContext } from '@rc-component/context';
+import TableContext, { responseImmutable } from "../context/TableContext";
+import devRenderTimes from "../hooks/useRenderTimes";
+import HeaderRow from "./HeaderRow";
+function parseHeaderRows(rootColumns, classNames, styles) {
+  const rows = [];
+  function fillRowCells(columns, colIndex, rowIndex = 0) {
+    // Init rows
+    rows[rowIndex] = rows[rowIndex] || [];
+    let currentColIndex = colIndex;
+    const colSpans = columns.filter(Boolean).map(column => {
+      const cell = {
+        key: column.key,
+        className: clsx(column.className, classNames.cell) || '',
+        style: styles.cell,
+        children: column.title,
+        column,
+        colStart: currentColIndex
+      };
+      let colSpan = 1;
+      const subColumns = column.children;
+      if (subColumns && subColumns.length > 0) {
+        colSpan = fillRowCells(subColumns, currentColIndex, rowIndex + 1).reduce((total, count) => total + count, 0);
+        cell.hasSubColumns = true;
+      }
+      if ('colSpan' in column) {
+        ({
+          colSpan
+        } = column);
+      }
+      if ('rowSpan' in column) {
+        cell.rowSpan = column.rowSpan;
+      }
+      cell.colSpan = colSpan;
+      cell.colEnd = cell.colStart + colSpan - 1;
+      rows[rowIndex].push(cell);
+      currentColIndex += colSpan;
+      return colSpan;
+    });
+    return colSpans;
+  }
+
+  // Generate `rows` cell data
+  fillRowCells(rootColumns, 0);
+
+  // Handle `rowSpan`
+  const rowCount = rows.length;
+  for (let rowIndex = 0; rowIndex < rowCount; rowIndex += 1) {
+    rows[rowIndex].forEach(cell => {
+      if (!('rowSpan' in cell) && !cell.hasSubColumns) {
+        // eslint-disable-next-line no-param-reassign
+        cell.rowSpan = rowCount - rowIndex;
+      }
+    });
+  }
+  return rows;
+}
+const Header = props => {
+  if (process.env.NODE_ENV !== 'production') {
+    devRenderTimes(props);
+  }
+  const {
+    stickyOffsets,
+    columns,
+    flattenColumns,
+    onHeaderRow
+  } = props;
+  const {
+    prefixCls,
+    getComponent,
+    classNames,
+    styles
+  } = useContext(TableContext, ['prefixCls', 'getComponent', 'classNames', 'styles']);
+  const {
+    header: headerCls = {}
+  } = classNames || {};
+  const {
+    header: headerStyles = {}
+  } = styles || {};
+  const rows = React.useMemo(() => parseHeaderRows(columns, headerCls, headerStyles), [columns, headerCls, headerStyles]);
+  const WrapperComponent = getComponent(['header', 'wrapper'], 'thead');
+  const trComponent = getComponent(['header', 'row'], 'tr');
+  const thComponent = getComponent(['header', 'cell'], 'th');
+  return /*#__PURE__*/React.createElement(WrapperComponent, {
+    className: clsx(`${prefixCls}-thead`, headerCls.wrapper),
+    style: headerStyles.wrapper
+  }, rows.map((row, rowIndex) => {
+    const rowNode = /*#__PURE__*/React.createElement(HeaderRow, {
+      classNames: headerCls,
+      styles: headerStyles,
+      key: rowIndex,
+      flattenColumns: flattenColumns,
+      cells: row,
+      stickyOffsets: stickyOffsets,
+      rowComponent: trComponent,
+      cellComponent: thComponent,
+      onHeaderRow: onHeaderRow,
+      index: rowIndex
+    });
+    return rowNode;
+  }));
+};
+export default responseImmutable(Header);