From ba2d73f472e171b543b1187f2fee69c6f5f4f015 Mon Sep 17 00:00:00 2001
From: ZeroToOne <ydreammer@gmail.com>
Date: Fri, 4 Dec 2020 14:22:01 +0800
Subject: [PATCH] feat: add columnWidth prop for expandable config

---
 src/hooks/useColumns.tsx | 3 +++
 src/interface.ts         | 1 +
 2 files changed, 4 insertions(+)

diff --git a/src/hooks/useColumns.tsx b/src/hooks/useColumns.tsx
index 4f170b6a8..c1f33329a 100644
--- a/src/hooks/useColumns.tsx
+++ b/src/hooks/useColumns.tsx
@@ -119,6 +119,7 @@ function useColumns<RecordType>(
     expandIconColumnIndex,
     direction,
     expandRowByClick,
+    columnWidth,
   }: {
     prefixCls?: string;
     columns?: ColumnsType<RecordType>;
@@ -132,6 +133,7 @@ function useColumns<RecordType>(
     expandIconColumnIndex?: number;
     direction?: 'ltr' | 'rtl';
     expandRowByClick?: boolean;
+    columnWidth?: number | string;
   },
   transformColumns: (columns: ColumnsType<RecordType>) => ColumnsType<RecordType>,
 ): [ColumnsType<RecordType>, ColumnType<RecordType>[]] {
@@ -153,6 +155,7 @@ function useColumns<RecordType>(
         title: '',
         fixed: prevColumn ? prevColumn.fixed : null,
         className: `${prefixCls}-row-expand-icon-cell`,
+        width: columnWidth,
         render: (_, record, index) => {
           const rowKey = getRowKey(record, index);
           const expanded = expandedKeys.has(rowKey);
diff --git a/src/interface.ts b/src/interface.ts
index 0b4064a2f..c70ba1013 100644
--- a/src/interface.ts
+++ b/src/interface.ts
@@ -210,6 +210,7 @@ export interface ExpandableConfig<RecordType> {
   expandedRowClassName?: RowClassName<RecordType>;
   childrenColumnName?: string;
   rowExpandable?: (record: RecordType) => boolean;
+  columnWidth?: number | string;
 }
 
 // =================== Render ===================