Added ‘setDelegateForPushNote’.

waynett · Aug 16, 2014 · 4fa916e · 4fa916e
1 parent 4d38c7b
commit 4fa916e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/AGPushNote_Example/AGPushNoteView.h b/AGPushNote_Example/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_Example/AGPushNoteView.m b/AGPushNote_Example/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
 {