UIImagePickerController+BlocksKit.m 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. //
  2. // UIImagePickerController+BlocksKit.m
  3. // BlocksKit
  4. //
  5. #import "A2DynamicDelegate.h"
  6. #import "NSObject+A2BlockDelegate.h"
  7. #import "UIImagePickerController+BlocksKit.h"
  8. #pragma mark Custom delegate
  9. @interface A2DynamicUIImagePickerControllerDelegate : A2DynamicDelegate <UIImagePickerControllerDelegate>
  10. @end
  11. @implementation A2DynamicUIImagePickerControllerDelegate
  12. - (void)imagePickerController:(UIImagePickerController *)picker didFinishPickingMediaWithInfo:(NSDictionary *)info {
  13. id realDelegate = self.realDelegate;
  14. if (realDelegate && [realDelegate respondsToSelector:@selector(imagePickerController:didFinishPickingMediaWithInfo:)])
  15. [realDelegate imagePickerController:picker didFinishPickingMediaWithInfo:info];
  16. void (^block)(UIImagePickerController *, NSDictionary *) = [self blockImplementationForMethod:_cmd];
  17. if (block) block(picker, info);
  18. }
  19. - (void)imagePickerControllerDidCancel:(UIImagePickerController *)picker {
  20. id realDelegate = self.realDelegate;
  21. if (realDelegate && [realDelegate respondsToSelector:@selector(imagePickerControllerDidCancel:)])
  22. [realDelegate imagePickerControllerDidCancel:picker];
  23. void (^block)(UIImagePickerController *) = [self blockImplementationForMethod:_cmd];
  24. if (block) block(picker);
  25. }
  26. @end
  27. #pragma mark Category
  28. @implementation UIImagePickerController (BlocksKit)
  29. @dynamic bk_didFinishPickingMediaBlock;
  30. @dynamic bk_didCancelBlock;
  31. + (void)load
  32. {
  33. @autoreleasepool {
  34. [self bk_registerDynamicDelegate];
  35. [self bk_linkDelegateMethods:@{ @"bk_didFinishPickingMediaBlock": @"imagePickerController:didFinishPickingMediaWithInfo:",
  36. @"bk_didCancelBlock": @"imagePickerControllerDidCancel:" }];
  37. }
  38. }
  39. @end