Added ‘setDelegateForPushNote’.

furu8ma · Aug 16, 2014 · 5f17085 · 5f17085
1 parent 4fa916e
commit 5f17085
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/AGPushNote/AGPushNoteView.h b/AGPushNote/AGPushNoteView.h
@@ -11,8 +11,8 @@
 @optional
 - (void)pushNoteDidAppear; // Called after the view has been fully transitioned onto the screen. (equel to completion block).
 - (void)pushNoteWillDisappear; // Called before the view is hidden, after the message action block.
-
 @end
+
 @interface AGPushNoteView : UIToolbar
 + (void)showWithNotificationMessage:(NSString *)message;
 + (void)showWithNotificationMessage:(NSString *)message completion:(void (^)(void))completion;
@@ -21,6 +21,7 @@
 + (void)awake;
 
 + (void)setMessageAction:(void (^)(NSString *message))action;
++ (void)setDelegateForPushNote:(id<AGPushNoteViewDelegate>)delegate;
 
 @property (nonatomic, weak) id<AGPushNoteViewDelegate> pushNoteDelegate;
 @end
diff --git a/AGPushNote/AGPushNoteView.m b/AGPushNote/AGPushNoteView.m
@@ -55,6 +55,10 @@ + (instancetype)sharedPushView
  return nil;
 }
 
++ (void)setDelegateForPushNote:(id<AGPushNoteViewDelegate>)delegate {
+ [PUSH_VIEW setPushNoteDelegate:delegate];
+}
+
 #pragma mark - Lifecycle (of sort)
 - (id)initWithFrame:(CGRect)frame
 {
@@ -143,7 +147,7 @@ + (void)showWithNotificationMessage:(NSString *)message completion:(void (^)(voi
  }];
 
  //Start timer (Currently not used to make sure user see & read the push...)
-// PUSH_VIEW.closeTimer = [NSTimer scheduledTimerWithTimeInterval:CLOSE_PUSH_SEC target:[AGPushNoteView class] selector:@selector(close) userInfo:nil repeats:NO];
+// PUSH_VIEW.closeTimer = [NSTimer scheduledTimerWithTimeInterval:CLOSE_PUSH_SEC target:[IAAPushNoteView class] selector:@selector(close) userInfo:nil repeats:NO];
  }
 }
 + (void)closeWitCompletion:(void (^)(void))completion {