enhance: Let column.dataIndex support RecordType (#1090)

* chor: keyof check

* chore: fix type

Author: zombieJ

src/Body/BodyRow.tsx

@@ -158,7 +158,7 @@ function BodyRow<RecordType extends { children?: readonly RecordType[] }>(
         );
 
         return (
-          <Cell
+          <Cell<RecordType>
             className={columnClassName}
             ellipsis={column.ellipsis}
             align={column.align}

src/Cell/index.tsx

@@ -24,7 +24,7 @@ export interface CellProps<RecordType extends DefaultRecordType> {
   index?: number;
   /** the index of the record. For the render(value, record, renderIndex) */
   renderIndex?: number;
-  dataIndex?: DataIndex;
+  dataIndex?: DataIndex<RecordType>;
   render?: ColumnType<RecordType>['render'];
   component?: CustomizeComponent;
   children?: React.ReactNode;
@@ -256,4 +256,4 @@ function Cell<RecordType>(props: CellProps<RecordType>) {
   );
 }
 
-export default React.memo(Cell);
+export default React.memo(Cell) as typeof Cell;

src/Cell/useCellRender.ts

@@ -16,7 +16,7 @@ function isRenderCell<RecordType>(
 
 export default function useCellRender<RecordType>(
   record: RecordType,
-  dataIndex: DataIndex,
+  dataIndex: DataIndex<RecordType>,
   renderIndex: number,
   children?: React.ReactNode,
   render?: ColumnType<RecordType>['render'],
@@ -37,8 +37,8 @@ export default function useCellRender<RecordType>(
         dataIndex === null || dataIndex === undefined || dataIndex === ''
           ? []
           : Array.isArray(dataIndex)
-          ? dataIndex
-          : [dataIndex];
+            ? dataIndex
+            : [dataIndex];
 
       const value: React.ReactNode = getValue(record, path);
 

src/FixedHolder/index.tsx

@@ -39,7 +39,7 @@ export interface FixedHeaderProps<RecordType> extends HeaderProps<RecordType> {
   children: (info: HeaderProps<RecordType>) => React.ReactNode;
 }
 
-const FixedHolder = React.forwardRef<HTMLDivElement, FixedHeaderProps<unknown>>((props, ref) => {
+const FixedHolder = React.forwardRef<HTMLDivElement, FixedHeaderProps<any>>((props, ref) => {
   if (process.env.NODE_ENV !== 'production') {
     devRenderTimes(props);
   }

src/VirtualTable/VirtualCell.tsx

@@ -8,7 +8,7 @@ import type { ColumnType, CustomizeComponent } from '../interface';
 import { GridContext } from './context';
 
 export interface VirtualCellProps<RecordType> {
-  rowInfo: ReturnType<typeof useRowInfo>;
+  rowInfo: ReturnType<typeof useRowInfo<RecordType>>;
   column: ColumnType<RecordType>;
   colIndex: number;
   indent: number;

src/interface.ts

@@ -66,7 +66,11 @@ export interface RenderedCell<RecordType> {
 
 export type Direction = 'ltr' | 'rtl';
 
-export type DataIndex = string | number | readonly (string | number)[];
+export type DataIndex<RecordType = any> =
+  | keyof RecordType
+  | string
+  | number
+  | readonly (string | number)[];
 
 export type CellEllipsisType = { showTitle?: boolean } | boolean;
 
@@ -96,7 +100,7 @@ export type AlignType = 'start' | 'end' | 'left' | 'right' | 'center' | 'justify
 
 export interface ColumnType<RecordType> extends ColumnSharedType<RecordType> {
   colSpan?: number;
-  dataIndex?: DataIndex;
+  dataIndex?: DataIndex<RecordType>;
   render?: (
     value: any,
     record: RecordType,