Allow executing a block while temporarily overriding default animations

Author: Ernesto Rivera

Source/AMBTableViewController.h

@@ -80,6 +80,22 @@
 /// Insert section/row animation. By default `UITableViewRowAnimationAutomatic`.
 @property (nonatomic) UITableViewRowAnimation removeAnimation;
 
+/// Execute a block while temporarily overriding reload, insert and remove animations.
+/// @param changes The block to be executed.
+/// @param animation The animation to be used while executing the changes.
+- (void)applyChanges:(void (^)(void))changes
+       withAnimation:(UITableViewRowAnimation)animation;
+
+/// Execute a block while temporarily overriding reload, insert and remove animations.
+/// @param changes The block to be executed.
+/// @param reloadAnimation The animation to be used while reloading sections/rows.
+/// @param insertAnimation The animation to be used while inserting sections/rows.
+/// @param removeAnimation The animation to be used while removing sections/rows.
+- (void)applyChanges:(void (^)(void))changes
+ withReloadAnimation:(UITableViewRowAnimation)reloadAnimation
+     insertAnimation:(UITableViewRowAnimation)insertAnimation
+     removeAnimation:(UITableViewRowAnimation)removeAnimation;
+
 /// @name Convenience Methods
 
 /// Trigger [AMBTableViewSection update] on all sections.

Source/AMBTableViewController.m

@@ -23,6 +23,10 @@
 @implementation AMBTableViewController
 {
     NSMutableArray * _mutableSections;
+    BOOL _useCustomAnimations;
+    UITableViewRowAnimation _customReloadAnimation;
+    UITableViewRowAnimation _customInsertAnimation;
+    UITableViewRowAnimation _customRemoveAnimation;
 }
 
 @dynamic sections;
@@ -126,6 +130,46 @@ - (void)replaceSectionAtIndex:(NSUInteger)index
     [section reload];
 }
 
+#pragma mark - Configuring animations
+
+- (UITableViewRowAnimation)reloadAnimation
+{
+    return _useCustomAnimations ? _customReloadAnimation : _reloadAnimation;
+}
+
+- (UITableViewRowAnimation)insertAnimation
+{
+    return _useCustomAnimations ? _customInsertAnimation : _insertAnimation;
+}
+
+- (UITableViewRowAnimation)removeAnimation
+{
+    return _useCustomAnimations ? _customRemoveAnimation : _removeAnimation;
+}
+
+- (void)applyChanges:(void (^)(void))changes
+       withAnimation:(UITableViewRowAnimation)animation
+{
+    [self applyChanges:changes
+   withReloadAnimation:animation
+       insertAnimation:animation
+       removeAnimation:animation];
+}
+
+- (void)applyChanges:(void (^)(void))changes
+ withReloadAnimation:(UITableViewRowAnimation)reloadAnimation
+     insertAnimation:(UITableViewRowAnimation)insertAnimation
+     removeAnimation:(UITableViewRowAnimation)removeAnimation
+{
+    _useCustomAnimations = YES;
+    _customReloadAnimation = reloadAnimation;
+    _customInsertAnimation = insertAnimation;
+    _customRemoveAnimation = removeAnimation;
+    
+    changes();
+    
+    _useCustomAnimations = NO;
+}
 
 #pragma mark - Convenience methods