141 lines
4.8 KiB
Objective-C
141 lines
4.8 KiB
Objective-C
#import <UIKit/UINavigationController.h>
|
|
#import "UIViewController+AnyPromise.h"
|
|
#import <PromiseKit/PromiseKit.h>
|
|
|
|
#if PMKImagePickerController
|
|
#import <UIKit/UIImagePickerController.h>
|
|
#endif
|
|
|
|
@interface PMKGenericDelegate : NSObject <UINavigationControllerDelegate> {
|
|
@public
|
|
PMKResolver resolve;
|
|
}
|
|
+ (instancetype)delegateWithPromise:(AnyPromise **)promise;
|
|
@end
|
|
|
|
@interface UIViewController ()
|
|
- (AnyPromise*) promise;
|
|
@end
|
|
|
|
@implementation UIViewController (PromiseKit)
|
|
|
|
- (AnyPromise *)promiseViewController:(UIViewController *)vc animated:(BOOL)animated completion:(void (^)(void))block {
|
|
__kindof UIViewController *vc2present = vc;
|
|
AnyPromise *promise = nil;
|
|
|
|
if ([vc isKindOfClass:NSClassFromString(@"MFMailComposeViewController")]) {
|
|
PMKGenericDelegate *delegate = [PMKGenericDelegate delegateWithPromise:&promise];
|
|
[vc setValue:delegate forKey:@"mailComposeDelegate"];
|
|
}
|
|
else if ([vc isKindOfClass:NSClassFromString(@"MFMessageComposeViewController")]) {
|
|
PMKGenericDelegate *delegate = [PMKGenericDelegate delegateWithPromise:&promise];
|
|
[vc setValue:delegate forKey:@"messageComposeDelegate"];
|
|
}
|
|
#ifdef PMKImagePickerController
|
|
else if ([vc isKindOfClass:[UIImagePickerController class]]) {
|
|
PMKGenericDelegate *delegate = [PMKGenericDelegate delegateWithPromise:&promise];
|
|
[vc setValue:delegate forKey:@"delegate"];
|
|
}
|
|
#endif
|
|
else if ([vc isKindOfClass:NSClassFromString(@"SLComposeViewController")]) {
|
|
PMKResolver resolve;
|
|
promise = [[AnyPromise alloc] initWithResolver:&resolve];
|
|
[vc setValue:^(NSInteger result){
|
|
if (result == 0) {
|
|
resolve([NSError errorWithDomain:NSCocoaErrorDomain code:NSUserCancelledError userInfo:nil]);
|
|
} else {
|
|
resolve(@(result));
|
|
}
|
|
} forKey:@"completionHandler"];
|
|
}
|
|
else if ([vc isKindOfClass:[UINavigationController class]])
|
|
vc = [(id)vc viewControllers].firstObject;
|
|
|
|
if (!vc) {
|
|
id userInfo = @{NSLocalizedDescriptionKey: @"nil or effective nil passed to promiseViewController"};
|
|
id err = [NSError errorWithDomain:PMKErrorDomain code:PMKInvalidUsageError userInfo:userInfo];
|
|
return [AnyPromise promiseWithValue:err];
|
|
}
|
|
|
|
if (!promise) {
|
|
if (![vc respondsToSelector:@selector(promise)]) {
|
|
id userInfo = @{NSLocalizedDescriptionKey: @"ViewController is not promisable"};
|
|
id err = [NSError errorWithDomain:PMKErrorDomain code:PMKInvalidUsageError userInfo:userInfo];
|
|
return [AnyPromise promiseWithValue:err];
|
|
}
|
|
|
|
promise = [vc valueForKey:@"promise"];
|
|
|
|
if (![promise isKindOfClass:[AnyPromise class]]) {
|
|
id userInfo = @{NSLocalizedDescriptionKey: @"The promise property is nil or not of type AnyPromise"};
|
|
id err = [NSError errorWithDomain:PMKErrorDomain code:PMKInvalidUsageError userInfo:userInfo];
|
|
return [AnyPromise promiseWithValue:err];
|
|
}
|
|
}
|
|
|
|
if (!promise.pending)
|
|
return promise;
|
|
|
|
[self presentViewController:vc2present animated:animated completion:block];
|
|
|
|
promise.ensure(^{
|
|
[vc2present.presentingViewController dismissViewControllerAnimated:animated completion:nil];
|
|
});
|
|
|
|
return promise;
|
|
}
|
|
|
|
@end
|
|
|
|
|
|
|
|
@implementation PMKGenericDelegate {
|
|
id retainCycle;
|
|
}
|
|
|
|
+ (instancetype)delegateWithPromise:(AnyPromise **)promise; {
|
|
PMKGenericDelegate *d = [PMKGenericDelegate new];
|
|
d->retainCycle = d;
|
|
*promise = [[AnyPromise alloc] initWithResolver:&d->resolve];
|
|
return d;
|
|
}
|
|
|
|
- (void)mailComposeController:(id)controller didFinishWithResult:(int)result error:(NSError *)error {
|
|
if (error != nil) {
|
|
resolve(error);
|
|
} else if (result == 0) {
|
|
resolve([NSError errorWithDomain:NSCocoaErrorDomain code:NSUserCancelledError userInfo:nil]);
|
|
} else {
|
|
resolve(@(result));
|
|
}
|
|
retainCycle = nil;
|
|
}
|
|
|
|
- (void)messageComposeViewController:(id)controller didFinishWithResult:(int)result {
|
|
if (result == 2) {
|
|
id userInfo = @{NSLocalizedDescriptionKey: @"The attempt to save or send the message was unsuccessful."};
|
|
id error = [NSError errorWithDomain:PMKErrorDomain code:PMKOperationFailed userInfo:userInfo];
|
|
resolve(error);
|
|
} else {
|
|
resolve(@(result));
|
|
}
|
|
retainCycle = nil;
|
|
}
|
|
|
|
#ifdef PMKImagePickerController
|
|
|
|
- (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary *)info {
|
|
id img = info[UIImagePickerControllerEditedImage] ?: info[UIImagePickerControllerOriginalImage];
|
|
resolve(PMKManifold(img, info));
|
|
retainCycle = nil;
|
|
}
|
|
|
|
- (void)imagePickerControllerDidCancel:(UIImagePickerController *)picker {
|
|
resolve([NSError errorWithDomain:NSCocoaErrorDomain code:NSUserCancelledError userInfo:nil]);
|
|
retainCycle = nil;
|
|
}
|
|
|
|
#endif
|
|
|
|
@end
|