From 0e973a1ffd1e073e9bb2005584b91d9fb4fbed85 Mon Sep 17 00:00:00 2001
From: Harman Goei <harman.goei@openx.com>
Date: Thu, 15 Feb 2018 21:50:04 -0800
Subject: [PATCH] feat(expandIconRender): Add expandIconRender prop for
 overriding expandIcon

---
 src/ExpandableRow.js   | 14 +++++++++++++-
 src/ExpandableTable.js |  1 +
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/ExpandableRow.js b/src/ExpandableRow.js
index a8452fd3a..d0b2dfd90 100644
--- a/src/ExpandableRow.js
+++ b/src/ExpandableRow.js
@@ -23,6 +23,7 @@ class ExpandableRow extends React.Component {
     expandIconColumnIndex: PropTypes.number,
     childrenColumnName: PropTypes.string,
     expandedRowRender: PropTypes.func,
+    expandIconRender: PropTypes.func,
     onExpandedChange: PropTypes.func.isRequired,
     onRowClick: PropTypes.func,
     children: PropTypes.func.isRequired,
@@ -64,7 +65,18 @@ class ExpandableRow extends React.Component {
   }
 
   renderExpandIcon = () => {
-    const { prefixCls, expanded, record, needIndentSpaced } = this.props;
+    const { prefixCls, expanded, record, needIndentSpaced, expandIconRender } = this.props;
+
+    if (expandIconRender) {
+      return expandIconRender({
+        prefixCls,
+        expanded,
+        record,
+        needIndentSpaced,
+        expandable: this.expandable,
+        onExpand: this.handleExpandChange,
+      });
+    }
 
     return (
       <ExpandIcon
diff --git a/src/ExpandableTable.js b/src/ExpandableTable.js
index 1c7e7d7c0..96fec861a 100644
--- a/src/ExpandableTable.js
+++ b/src/ExpandableTable.js
@@ -13,6 +13,7 @@ class ExpandableTable extends React.Component {
     defaultExpandedRowKeys: PropTypes.array,
     expandIconColumnIndex: PropTypes.number,
     expandedRowRender: PropTypes.func,
+    expandIconRender: PropTypes.func,
     childrenColumnName: PropTypes.string,
     indentSize: PropTypes.number,
     onExpand: PropTypes.func,