Check for when actionSheetDismissedBlock() is nil

TimOliver · Jan 12, 2017 · 71422d2 · 71422d2
1 parent f286b38
commit 71422d2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/TOActionSheet/TOActionSheet.m b/TOActionSheet/TOActionSheet.m
@@ -814,7 +814,8 @@ - (void)dismissViewWithCompactAnimation
         self.backgroundColor = [UIColor clearColor];
     } completion:^(BOOL complete) {
         [self removeFromSuperview];
-        self.actionSheetDismissedBlock();
+        if (self.actionSheetDismissedBlock)
+            self.actionSheetDismissedBlock();
     }];
 }
 
@@ -863,7 +864,8 @@ - (void)dismissViewWithRegularAnimation
         self.backgroundColor = [UIColor clearColor];
      } completion:^(BOOL complete) {
          [self removeFromSuperview];
-         self.actionSheetDismissedBlock();
+         if (self.actionSheetDismissedBlock)
+             self.actionSheetDismissedBlock();
      }];
 }