Ability to pass icon component

xgfe · May 9, 2017 · 6c02759 · 6c02759
1 parent 95c7d64
commit 6c02759
Showing 1 changed file with 25 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -257,6 +257,29 @@ class DatePicker extends Component {
       this.props.onOpenModal();
     }
   }
+
+  _renderIcon() {
+    const {
+      showIcon,
+      iconSource,
+      iconComponent,
+      customStyles
+    } = this.props;
+
+    if (showIcon) {
+      if (!!iconComponent) {
+        return iconComponent;
+      }
+      return (
+        <Image
+          style={[Style.dateIcon, customStyles.dateIcon]}
+          source={iconSource}
+        />
+      );
+    }
+
+    return null;
+  }
 
   render() {
     const {
@@ -290,10 +313,7 @@ class DatePicker extends Component {
           <View style={dateInputStyle}>
             {this.getTitleElement()}
           </View>
-          {showIcon && <Image
-            style={[Style.dateIcon, customStyles.dateIcon]}
-            source={iconSource}
-          />}
+          {this._renderIcon()}
           {Platform.OS === 'ios' && <Modal
             transparent={true}
             animationType="none"
@@ -389,6 +409,7 @@ DatePicker.propTypes = {
   confirmBtnText: React.PropTypes.string,
   cancelBtnText: React.PropTypes.string,
   iconSource: React.PropTypes.oneOfType([React.PropTypes.number, React.PropTypes.object]),
+  iconComponent: React.PropTypes.element,
   customStyles: React.PropTypes.object,
   showIcon: React.PropTypes.bool,
   disabled: React.PropTypes.bool,