Adds closePicker function (#602)

Co-authored-by: Henning Hall <[email protected]>

Author: aureosouza

android/src/main/java/com/henninghall/date_picker/DatePickerModule.java

@@ -19,6 +19,8 @@
 
 public class DatePickerModule extends ReactContextBaseJavaModule {
 
+    private AlertDialog dialog;
+
     DatePickerModule(ReactApplicationContext context) {
         super(context);
         ApplicationStarter.initialize(context, false); // false = no need to prefetch on time data background tread
@@ -37,10 +39,15 @@ public void removeListeners(Integer count) {
     @ReactMethod
     public void openPicker(ReadableMap props, Callback onConfirm, Callback onCancel){
         PickerView picker = createPicker(props);
-        AlertDialog dialog = createDialog(props, picker, onConfirm, onCancel);
+        dialog = createDialog(props, picker, onConfirm, onCancel);
         dialog.show();
     }
 
+    @ReactMethod
+    public void closePicker(){
+        dialog.dismiss();
+    }
+
     private AlertDialog createDialog(
             ReadableMap props, final PickerView picker, final Callback onConfirm, final Callback onCancel) {
         String title = props.getString("title");

ios/RNDatePicker/RNDatePickerManager.h

@@ -10,4 +10,6 @@
 
 @interface RNDatePickerManager : RCTViewManager
 
+@property (strong, nonatomic) UIViewController *topViewController;
+
 @end

ios/RNDatePicker/RNDatePickerManager.m

@@ -156,16 +156,22 @@ - (UIView *)view
         }
 
         // Finding the top view controller which is neccessary to be able to show the picker from within modal
-        UIViewController *topViewController = rootViewController;
-        while (topViewController.presentedViewController){
-            topViewController = topViewController.presentedViewController;
+        _topViewController = rootViewController;
+        while (_topViewController.presentedViewController){
+            _topViewController = _topViewController.presentedViewController;
         }
-        
-        [topViewController presentViewController:alertController animated:YES completion:nil];
+        [_topViewController presentViewController:alertController animated:YES completion:nil];
     });
 
 }
 
+RCT_EXPORT_METHOD(closePicker)
+{
+    dispatch_async(dispatch_get_main_queue(), ^{
+        [_topViewController dismissViewControllerAnimated:YES completion:nil];
+    });
+}
+
 - (double) getPickerWidth :(UIView *) alertView
 {
     bool iPad = UI_USER_INTERFACE_IDIOM() == UIUserInterfaceIdiomPad;

src/DatePickerAndroid.js

@@ -20,14 +20,16 @@ class DatePickerAndroid extends React.PureComponent {
     const props = this.getProps()
     const isClosed = this._isCurrentlyClosed();
 
-    this.previousProps = props; 
+    this.previousProps = props;
     if (props.modal) {
       if (props.open && isClosed) {
         NativeModules.RNDatePicker.openPicker(
           props,
           this._onConfirm,
           this.props.onCancel
         )
+      } else if (!props.open && !isClosed) {
+        NativeModules.RNDatePicker.closePicker()
       }
       return null
     }

src/DatePickerIOS.js

@@ -47,14 +47,16 @@ export default class DatePickerIOS extends React.Component {
     const props = this._toIosProps(this.props)
     const isClosed = this._isCurrentlyClosed();
 
-    this.previousProps = props; 
+    this.previousProps = props;
     if (props.modal) {
       if (props.open && isClosed) {
         NativeModules.RNDatePickerManager.openPicker(
           props,
           this._onConfirm,
           props.onCancel
         )
+      } else if (!props.open && !isClosed) {
+        NativeModules.RNDatePickerManager.closePicker()
       }
       return null
     }
@@ -72,7 +74,7 @@ export default class DatePickerIOS extends React.Component {
       />
     )
   }
-  
+
   _isCurrentlyClosed = () => !this.previousProps || !this.previousProps.open
 }