Swiftgram/submodules/AsyncDisplayKit/examples/ASDKgram/Sample/PhotoFeedSectionController.m

//
//  PhotoFeedSectionController.m
//  Texture
//
//  Copyright (c) Pinterest, Inc.  All rights reserved.
//  Licensed under Apache 2.0: http://www.apache.org/licenses/LICENSE-2.0
//

#import "PhotoFeedSectionController.h"
#import "PhotoFeedModel.h"
#import "PhotoModel.h"
#import "PhotoCellNode.h"
#import "TailLoadingNode.h"
#import "FeedHeaderNode.h"

@interface PhotoFeedSectionController () <ASSupplementaryNodeSource, IGListSupplementaryViewSource>
@property (nonatomic, strong) NSString *paginatingSpinner;
@end

@implementation PhotoFeedSectionController

- (instancetype)init
{
  if (self = [super init]) {
    _paginatingSpinner = @"Paginating Spinner";
    self.supplementaryViewSource = self;
  }
  return self;
}

#pragma mark - IGListSectionType

- (void)didUpdateToObject:(id)object
{
  _photoFeed = object;
  [self setItems:_photoFeed.photos animated:NO completion:nil];
}

- (__kindof UICollectionViewCell *)cellForItemAtIndex:(NSInteger)index
{
  return [ASIGListSectionControllerMethods cellForItemAtIndex:index sectionController:self];
}

- (CGSize)sizeForItemAtIndex:(NSInteger)index
{
  return [ASIGListSectionControllerMethods sizeForItemAtIndex:index];
}

- (void)didSelectItemAtIndex:(NSInteger)index
{
  // nop
}

#pragma mark - ASSectionController

- (ASCellNodeBlock)nodeBlockForItemAtIndex:(NSInteger)index
{
  id object = self.items[index];
  // this will be executed on a background thread - important to make sure it's thread safe
  ASCellNode *(^nodeBlock)() = nil;
  if (object == _paginatingSpinner) {
    nodeBlock = ^{
      return [[TailLoadingNode alloc] init];
    };
  } else if ([object isKindOfClass:[PhotoModel class]]) {
    PhotoModel *photoModel = object;
    nodeBlock = ^{
      PhotoCellNode *cellNode = [[PhotoCellNode alloc] initWithPhotoObject:photoModel];
      return cellNode;
    };
  }

  return nodeBlock;
}

- (void)beginBatchFetchWithContext:(ASBatchContext *)context
{
  dispatch_async(dispatch_get_main_queue(), ^{
    // Immediately add the loading spinner if needed.
    if (self.items.count > 0) {
      NSArray *newItems = [self.items arrayByAddingObject:_paginatingSpinner];
      [self setItems:newItems animated:NO completion:nil];
    }

    // Push to next runloop to give time to insert the spinner
    dispatch_async(dispatch_get_main_queue(), ^{
      dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(2.0 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        // Start the fetch, then update the items (removing the spinner) when they are loaded.
        [_photoFeed requestPageWithCompletionBlock:^(NSArray *newPhotos){
          [self setItems:_photoFeed.photos animated:NO completion:^{
            [context completeBatchFetching:YES];
          }];
        } numResultsToReturn:20];
      });
    });
  });
}

#pragma mark - RefreshingSectionControllerType

- (void)refreshContentWithCompletion:(void(^)())completion
{
  [_photoFeed refreshFeedWithCompletionBlock:^(NSArray *addedItems) {
    [self setItems:_photoFeed.photos animated:YES completion:completion];
  } numResultsToReturn:4];
}

#pragma mark - ASSupplementaryNodeSource

- (ASCellNodeBlock)nodeBlockForSupplementaryElementOfKind:(NSString *)elementKind atIndex:(NSInteger)index
{
  ASDisplayNodeAssert([elementKind isEqualToString:UICollectionElementKindSectionHeader], nil);
  return ^{
    return [[FeedHeaderNode alloc] init];
  };
}

- (ASSizeRange)sizeRangeForSupplementaryElementOfKind:(NSString *)elementKind atIndex:(NSInteger)index
{
  if ([elementKind isEqualToString:UICollectionElementKindSectionHeader]) {
    return ASSizeRangeUnconstrained;
  } else {
    return ASSizeRangeZero;
  }
}

#pragma mark - IGListSupplementaryViewSource

- (NSArray<NSString *> *)supportedElementKinds
{
  return @[ UICollectionElementKindSectionHeader ];
}

- (__kindof UICollectionReusableView *)viewForSupplementaryElementOfKind:(NSString *)elementKind atIndex:(NSInteger)index
{
  return [ASIGListSupplementaryViewSourceMethods viewForSupplementaryElementOfKind:elementKind atIndex:index sectionController:self];
}

- (CGSize)sizeForSupplementaryViewOfKind:(NSString *)elementKind atIndex:(NSInteger)index
{
  return [ASIGListSupplementaryViewSourceMethods sizeForSupplementaryViewOfKind:elementKind atIndex:index];
}

@end