add notification support

kyoji2 · kyoji2 · commit 8028fb182d9e · 2013-05-03T16:35:46.000+08:00
diff --git a/SIAlertView/SIAlertView.h b/SIAlertView/SIAlertView.h
@@ -8,6 +8,11 @@
 
 #import <Foundation/Foundation.h>
 
+extern NSString *const SIAlertViewWillShowNotification;
+extern NSString *const SIAlertViewDidShowNotification;
+extern NSString *const SIAlertViewWillDismissNotification;
+extern NSString *const SIAlertViewDidDismissNotification;
+
 typedef NS_ENUM(NSInteger, SIAlertViewButtonType) {
     SIAlertViewButtonTypeDefault = 0,
     SIAlertViewButtonTypeDestructive,
diff --git a/SIAlertView/SIAlertView.m b/SIAlertView/SIAlertView.m
@@ -9,6 +9,11 @@
 #import "SIAlertView.h"
 #import <QuartzCore/QuartzCore.h>
 
+NSString *const SIAlertViewWillShowNotification = @"SIAlertViewWillShowNotification";
+NSString *const SIAlertViewDidShowNotification = @"SIAlertViewDidShowNotification";
+NSString *const SIAlertViewWillDismissNotification = @"SIAlertViewWillDismissNotification";
+NSString *const SIAlertViewDidDismissNotification = @"SIAlertViewDidDismissNotification";
+
 #define DEBUG_LAYOUT 0
 
 #define MESSAGE_MIN_LINE_COUNT 3
@@ -300,6 +305,7 @@ - (void)show
     if (self.willShowHandler) {
         self.willShowHandler(self);
     }
+    [[NSNotificationCenter defaultCenter] postNotificationName:SIAlertViewWillShowNotification object:self userInfo:nil];
     
     self.visible = YES;
     
@@ -328,6 +334,7 @@ - (void)show
         if (self.didShowHandler) {
             self.didShowHandler(self);
         }
+        [[NSNotificationCenter defaultCenter] postNotificationName:SIAlertViewDidShowNotification object:self userInfo:nil];
         
         [SIAlertView setAnimating:NO];
         
@@ -347,8 +354,11 @@ - (void)dismissAnimated:(BOOL)animated cleanup:(BOOL)cleanup
 {
     BOOL isVisible = self.isVisible;
     
-    if (isVisible && self.willDismissHandler) {
-        self.willDismissHandler(self);
+    if (isVisible) {
+        if (self.willDismissHandler) {
+            self.willDismissHandler(self);
+        }
+        [[NSNotificationCenter defaultCenter] postNotificationName:SIAlertViewWillDismissNotification object:self userInfo:nil];
     }
     
     void (^dismissComplete)(void) = ^{
@@ -370,8 +380,11 @@ - (void)dismissAnimated:(BOOL)animated cleanup:(BOOL)cleanup
         
         [SIAlertView setAnimating:NO];
         
-        if (isVisible && self.didDismissHandler) {
-            self.didDismissHandler(self);
+        if (isVisible) {
+            if (self.didDismissHandler) {
+                self.didDismissHandler(self);
+            }
+            [[NSNotificationCenter defaultCenter] postNotificationName:SIAlertViewDidDismissNotification object:self userInfo:nil];
         }
         
         // check if we should show next alert
@@ -867,7 +880,6 @@ - (void)animationDidStop:(CAAnimation *)anim finished:(BOOL)flag
     }
 }
 
-
 #pragma mark - UIAppearance setters
 
 - (void)setTitleFont:(UIFont *)titleFont
diff --git a/SIAlertViewExample/SIAlertViewExample/ViewController.m b/SIAlertViewExample/SIAlertViewExample/ViewController.m
@@ -126,6 +126,8 @@ - (IBAction)alert2:(id)sender
     [alertView show];
 }
 
+id observer1,observer2,observer3,observer4;
+
 - (IBAction)alert3:(id)sender
 {
     SIAlertView *alertView = [[SIAlertView alloc] initWithTitle:nil andMessage:@"Message3"];
@@ -155,6 +157,38 @@ - (IBAction)alert3:(id)sender
         NSLog(@"%@, didDismissHandler3", alertView);
     };
     
+    observer1 = [[NSNotificationCenter defaultCenter] addObserverForName:SIAlertViewWillShowNotification
+                                                                  object:alertView
+                                                                   queue:[NSOperationQueue mainQueue]
+                                                              usingBlock:^(NSNotification *note) {
+                                                                  NSLog(@"%@, -willShowHandler3", alertView);
+                                                              }];
+    observer2 =[[NSNotificationCenter defaultCenter] addObserverForName:SIAlertViewDidShowNotification
+                                                                 object:alertView
+                                                                  queue:[NSOperationQueue mainQueue]
+                                                             usingBlock:^(NSNotification *note) {
+                                                                 NSLog(@"%@, -didShowHandler3", alertView);
+                                                             }];
+    observer3 =[[NSNotificationCenter defaultCenter] addObserverForName:SIAlertViewWillDismissNotification
+                                                                 object:alertView
+                                                                  queue:[NSOperationQueue mainQueue]
+                                                             usingBlock:^(NSNotification *note) {
+                                                                 NSLog(@"%@, -willDismissHandler3", alertView);
+                                                             }];
+    observer4 =[[NSNotificationCenter defaultCenter] addObserverForName:SIAlertViewDidDismissNotification
+                                                                 object:alertView
+                                                                  queue:[NSOperationQueue mainQueue]
+                                                             usingBlock:^(NSNotification *note) {
+                                                                 NSLog(@"%@, -didDismissHandler3", alertView);
+                                                                 
+                                                                 [[NSNotificationCenter defaultCenter] removeObserver:observer1];
+                                                                 [[NSNotificationCenter defaultCenter] removeObserver:observer2];
+                                                                 [[NSNotificationCenter defaultCenter] removeObserver:observer3];
+                                                                 [[NSNotificationCenter defaultCenter] removeObserver:observer4];
+                                                                 
+                                                                 observer1 = observer2 = observer3 = observer4 = nil;
+                                                             }];
+    
     [alertView show];
 }
 
