@@ -31,17 +31,19 @@ export default class Step extends React.Component {
3131 PropTypes . func ,
3232 ] ) ,
3333 tailContent : PropTypes . any ,
34+ finishIcon : PropTypes . node ,
35+ errorIcon : PropTypes . node ,
3436 } ;
3537 renderIconNode ( ) {
3638 const {
3739 prefixCls, progressDot, stepNumber, status, title, description, icon,
38- iconPrefix,
40+ iconPrefix, finishIcon , errorIcon ,
3941 } = this . props ;
4042 let iconNode ;
4143 const iconClassName = classNames ( `${ prefixCls } -icon` , `${ iconPrefix } icon` , {
4244 [ `${ iconPrefix } icon-${ icon } ` ] : icon && isString ( icon ) ,
43- [ `${ iconPrefix } icon-check` ] : ! icon && status === 'finish' ,
44- [ `${ iconPrefix } icon-cross ` ] : ! icon && status === 'error' ,
45+ [ `${ iconPrefix } icon-check` ] : ! icon && status === 'finish' && ! finishIcon ,
46+ [ `${ iconPrefix } icon-close ` ] : ! icon && status === 'error' && ! errorIcon ,
4547 } ) ;
4648 const iconDot = < span className = { `${ prefixCls } -icon-dot` } > </ span > ;
4749 // `progressDot` enjoy the highest priority
@@ -57,11 +59,16 @@ export default class Step extends React.Component {
5759 }
5860 } else if ( icon && ! isString ( icon ) ) {
5961 iconNode = < span className = { `${ prefixCls } -icon` } > { icon } </ span > ;
62+ } else if ( finishIcon && status === 'finish' ) {
63+ iconNode = < span className = { `${ prefixCls } -icon` } > { finishIcon } </ span > ;
64+ } else if ( errorIcon && status === 'error' ) {
65+ iconNode = < span className = { `${ prefixCls } -icon` } > { errorIcon } </ span > ;
6066 } else if ( icon || status === 'finish' || status === 'error' ) {
6167 iconNode = < span className = { iconClassName } /> ;
6268 } else {
6369 iconNode = < span className = { `${ prefixCls } -icon` } > { stepNumber } </ span > ;
6470 }
71+
6572 return iconNode ;
6673 }
6774 render ( ) {
@@ -70,6 +77,7 @@ export default class Step extends React.Component {
7077 status = 'wait' , iconPrefix, icon, wrapperStyle,
7178 adjustMarginRight, stepNumber,
7279 description, title, progressDot, tailContent,
80+ finishIcon, errorIcon,
7381 ...restProps ,
7482 } = this . props ;
7583
0 commit comments