Enable infitite scroll view and pull to refresh

Author: m1entus

Example/INSPullToRefresh/INSPullToRefreshViewController.m

@@ -33,7 +33,6 @@ - (void)viewDidLoad {
         dispatch_time_t popTime = dispatch_time(DISPATCH_TIME_NOW, delayInSeconds * NSEC_PER_SEC);
         dispatch_after(popTime, dispatch_get_main_queue(), ^(void){
             [scrollView ins_endPullToRefresh];
-
         });
     }];
 
@@ -72,6 +71,11 @@ - (void)viewDidLoad {
             [weakSelf.tableView endUpdates];
 
             [scrollView ins_endInfinityScroll];
+
+            if (weakSelf.numberOfRows > 30) {
+                // Disable infinite scroll after 45 rows
+                scrollView.ins_infiniteScrollBackgroundView.enabled = NO;
+            }
         });
     }];
 

INSPullToRefresh/INSInfiniteScrollBackgroundView.h

@@ -48,6 +48,8 @@ typedef NS_ENUM(NSUInteger, INSInfiniteScrollBackgroundViewState) {
 @property (nonatomic, readonly) INSInfiniteScrollBackgroundViewState state;
 @property (nonatomic, assign) BOOL preserveContentInset;
 
+@property (nonatomic, assign) BOOL enabled;
+
 @property (nonatomic, assign) CGFloat additionalBottomOffsetForInfinityScrollTrigger;
 
 - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scrollView;

INSPullToRefresh/INSInfiniteScrollBackgroundView.m

@@ -47,6 +47,19 @@ - (void)setPreserveContentInset:(BOOL)preserveContentInset {
     }
 }
 
+- (void)setEnabled:(BOOL)enabled {
+    if (_enabled != enabled) {
+        _enabled = enabled;
+
+        if (_enabled) {
+            [self resetFrame];
+        } else {
+            [self stopInfiniteScroll];
+        }
+
+        self.hidden = !_enabled;
+    }
+}
 
 #pragma mark - Initializers
 
@@ -66,6 +79,7 @@ - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scroll
         self.autoresizingMask = UIViewAutoresizingFlexibleWidth;
         _state = INSInfiniteScrollBackgroundViewStateNone;
         _preserveContentInset = NO;
+        _enabled = YES;
         self.hidden = YES;
 
         [self resetFrame];
@@ -77,6 +91,9 @@ - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scroll
 #pragma mark - Observing
 
 - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context {
+    if (!self.enabled) {
+        return;
+    }
     if ([keyPath isEqualToString:@"contentOffset"]) {
         [self scrollViewDidScroll:[[change valueForKey:NSKeyValueChangeNewKey] CGPointValue]];
     }
@@ -119,6 +136,10 @@ - (void)scrollViewDidScroll:(CGPoint)contentOffset {
 #pragma mark - Public
 
 - (void)beginInfiniteScrolling {
+    if (!self.enabled) {
+        return;
+    }
+
     if (self.state == INSInfiniteScrollBackgroundViewStateNone) {
         [self startInfiniteScroll];
     }

INSPullToRefresh/INSPullToRefreshBackgroundView.h

@@ -59,6 +59,8 @@ typedef NS_ENUM(NSUInteger, INSPullToRefreshBackgroundViewState) {
 
 @property (nonatomic, assign) CGFloat dragToTriggerOffset;
 
+@property (nonatomic, assign) BOOL enabled;
+
 - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scrollView;
 
 - (void)beginRefreshing;

INSPullToRefresh/INSPullToRefreshBackgroundView.m

@@ -109,6 +109,18 @@ - (void)setPreserveContentInset:(BOOL)preserveContentInset {
     }
 }
 
+- (void)setEnabled:(BOOL)enabled {
+    if (_enabled != enabled) {
+        _enabled = enabled;
+        if (_enabled) {
+            [self resetFrame];
+        } else {
+            [self endRefreshing];
+        }
+        self.hidden = !_enabled;
+    }
+}
+
 #pragma mark - Initializers
 
 - (instancetype)initWithFrame:(CGRect)frame {
@@ -128,6 +140,7 @@ - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scroll
         _state = INSPullToRefreshBackgroundViewStateNone;
         _preserveContentInset = NO;
         _scrollToTopAfterEndRefreshing = YES;
+        _enabled = YES;
 
         [self resetFrame];
     }
@@ -138,6 +151,10 @@ - (instancetype)initWithHeight:(CGFloat)height scrollView:(UIScrollView *)scroll
 #pragma mark - Public
 
 - (void)beginRefreshing {
+    if (!self.enabled) {
+        return;
+    }
+    
     if (self.state == INSPullToRefreshBackgroundViewStateNone) {
         [self changeState:INSPullToRefreshBackgroundViewStateTriggered];
 
@@ -173,6 +190,11 @@ - (void)changeState:(INSPullToRefreshBackgroundViewState)state {
 #pragma mark - Observing
 
 - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context {
+
+    if (!self.enabled) {
+        return;
+    }
+
     if ([keyPath isEqualToString:@"contentOffset"]) {
         [self scrollViewDidScroll:[[change valueForKey:NSKeyValueChangeNewKey] CGPointValue]];
     }

INSPullToRefresh/UIScrollView+INSPullToRefresh.h

@@ -37,6 +37,8 @@
 - (void)ins_addPullToRefreshWithHeight:(CGFloat)height handler:(INSPullToRefreshActionHandler)actionHandler;
 - (void)ins_removePullToRefresh;
 
+- (void)ins_setPullToRefreshEnabled:(BOOL)enabled;
+
 - (void)ins_beginPullToRefresh;
 - (void)ins_endPullToRefresh;
 
@@ -45,6 +47,8 @@
 - (void)ins_addInfinityScrollWithHeight:(CGFloat)height handler:(INSInfinityScrollActionHandler)actionHandler;
 - (void)ins_removeInfinityScroll;
 
+- (void)ins_setInfinityScrollEnabled:(BOOL)enabled;
+
 - (void)ins_beginInfinityScroll;
 - (void)ins_endInfinityScroll;
 

INSPullToRefresh/UIScrollView+INSPullToRefresh.m

@@ -64,6 +64,10 @@ - (void)ins_endPullToRefresh {
     [self.ins_pullToRefreshBackgroundView endRefreshing];
 }
 
+- (void)ins_setPullToRefreshEnabled:(BOOL)enabled {
+    self.ins_pullToRefreshBackgroundView.enabled = enabled;
+}
+
 - (void)ins_addPullToRefreshWithHeight:(CGFloat)height handler:(INSPullToRefreshActionHandler)actionHandler {
 
     [self ins_removePullToRefresh];
@@ -112,6 +116,10 @@ - (void)ins_removeInfinityScroll {
     }
 }
 
+- (void)ins_setInfinityScrollEnabled:(BOOL)enabled {
+    self.ins_infiniteScrollBackgroundView.enabled = enabled;
+}
+
 - (void)ins_beginInfinityScroll {
     [self.ins_infiniteScrollBackgroundView beginInfiniteScrolling];
 }