Fixed strong delegate bug.

tLewisII · Nov 4, 2013 · 1020ef7 · 1020ef7
1 parent 9cb5503
commit 1020ef7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/TJLBarButtonMenu/Source/TJLBarButtonMenu.h b/TJLBarButtonMenu/Source/TJLBarButtonMenu.h
@@ -60,4 +60,6 @@ typedef void (^TJLButtonTappedBlock)(TJLBarButtonMenu *buttonView, NSString *tit
  */
 - (void)setButtonTappedBlock:(TJLButtonTappedBlock)block;
 
+///The delegate
+@property(weak, nonatomic) id <TJLButtonViewDelegate> delegate;
 @end
diff --git a/TJLBarButtonMenu/Source/TJLBarButtonMenu.m b/TJLBarButtonMenu/Source/TJLBarButtonMenu.m
@@ -16,7 +16,6 @@ @interface TJLBarButtonMenu () {
 @property(strong, nonatomic) NSMutableArray *firstConstraints;
 @property(strong, nonatomic) UIViewController *parentView;
 @property(strong, nonatomic) NSMutableArray *buttonArray;
-@property(strong, nonatomic) id <TJLButtonViewDelegate> delegate;
 @property(nonatomic) NSLayoutAttribute rightLeftPosition;
 @property(nonatomic) NSLayoutAttribute initialButtonConstant;
 @property(nonatomic) NSLayoutAttribute finalButtonConstant;
@@ -252,8 +251,9 @@ - (void)hide:(NSString *)buttonTitle {
                     [self layoutIfNeeded];
                 }
             }                completion:^(BOOL final) {
+                id<TJLButtonViewDelegate>strongDelegate = self.delegate;
                 if(buttonTappedBlock) buttonTappedBlock(self, buttonTitle);
-                if([self.delegate respondsToSelector:@selector(buttonMenu:titleForTappedButton:)]) [self.delegate buttonMenu:self titleForTappedButton:buttonTitle];
+                if([strongDelegate respondsToSelector:@selector(buttonMenu:titleForTappedButton:)]) [strongDelegate buttonMenu:self titleForTappedButton:buttonTitle];
                 [self removeFromSuperview];
             }];
         }];