mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-06 02:15:35 +08:00
32 lines
911 B
TypeScript
32 lines
911 B
TypeScript
import * as React from 'react';
|
|
import type { ColumnsType, ColumnTitleProps, TransformColumns } from '../interface';
|
|
import { renderColumnTitle } from '../util';
|
|
|
|
function fillTitle<RecordType>(
|
|
columns: ColumnsType<RecordType>,
|
|
columnTitleProps: ColumnTitleProps<RecordType>,
|
|
) {
|
|
return columns.map(column => {
|
|
const cloneColumn = { ...column };
|
|
|
|
cloneColumn.title = renderColumnTitle(column.title, columnTitleProps);
|
|
|
|
if ('children' in cloneColumn) {
|
|
cloneColumn.children = fillTitle(cloneColumn.children, columnTitleProps);
|
|
}
|
|
|
|
return cloneColumn;
|
|
});
|
|
}
|
|
|
|
export default function useTitleColumns<RecordType>(
|
|
columnTitleProps: ColumnTitleProps<RecordType>,
|
|
): [TransformColumns<RecordType>] {
|
|
const filledColumns = React.useCallback(
|
|
(columns: ColumnsType<RecordType>) => fillTitle(columns, columnTitleProps),
|
|
[columnTitleProps],
|
|
);
|
|
|
|
return [filledColumns];
|
|
}
|