mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-12-24 07:05:35 +00:00
Shuffle & Cleanup Stuff (#3080)
This commit is contained in:
Adlai Holler
GitHub
320
Source/Private/ASIGListAdapterBasedDataSource.m
Normal file
320
Source/Private/ASIGListAdapterBasedDataSource.m
Normal file
@@ -0,0 +1,320 @@
|
||||
//
|
||||
// ASIGListAdapterBasedDataSource.m
|
||||
// AsyncDisplayKit
|
||||
//
|
||||
// Created by Adlai Holler on 1/19/17.
|
||||
// Copyright © 2017 Facebook. All rights reserved.
|
||||
//
|
||||
|
||||
#if IG_LIST_KIT
|
||||
|
||||
#import "ASIGListAdapterBasedDataSource.h"
|
||||
#import <AsyncDisplayKit/AsyncDisplayKit.h>
|
||||
#import <objc/runtime.h>
|
||||
|
||||
typedef IGListSectionController<IGListSectionType, ASSectionController> ASIGSectionController;
|
||||
|
||||
/// The optional methods that a class implements from ASSectionController.
|
||||
/// Note: Bitfields are not supported by NSValue so we can't use them.
|
||||
typedef struct {
|
||||
BOOL sizeRangeForItem;
|
||||
BOOL shouldBatchFetch;
|
||||
BOOL beginBatchFetchWithContext;
|
||||
} ASSectionControllerOverrides;
|
||||
|
||||
/// The optional methods that a class implements from ASSupplementaryNodeSource.
|
||||
/// Note: Bitfields are not supported by NSValue so we can't use them.
|
||||
typedef struct {
|
||||
BOOL sizeRangeForSupplementary;
|
||||
} ASSupplementarySourceOverrides;
|
||||
|
||||
@protocol ASIGSupplementaryNodeSource <IGListSupplementaryViewSource, ASSupplementaryNodeSource>
|
||||
@end
|
||||
|
||||
@interface ASIGListAdapterBasedDataSource ()
|
||||
@property (nonatomic, weak, readonly) IGListAdapter *listAdapter;
|
||||
@property (nonatomic, readonly) id<UICollectionViewDelegateFlowLayout> delegate;
|
||||
@property (nonatomic, readonly) id<UICollectionViewDataSource> dataSource;
|
||||
|
||||
/**
|
||||
* The section controller that we will forward beginBatchFetchWithContext: to.
|
||||
* Since shouldBatchFetch: is called on main, we capture the last section controller in there,
|
||||
* and then we use it and clear it in beginBatchFetchWithContext: (on default queue).
|
||||
*
|
||||
* It is safe to use it without a lock in this limited way, since those two methods will
|
||||
* never execute in parallel.6
|
||||
*/
|
||||
@property (nonatomic, weak) ASIGSectionController *sectionControllerForBatchFetching;
|
||||
@end
|
||||
|
||||
@implementation ASIGListAdapterBasedDataSource
|
||||
|
||||
- (instancetype)initWithListAdapter:(IGListAdapter *)listAdapter
|
||||
{
|
||||
if (self = [super init]) {
|
||||
[ASIGListAdapterBasedDataSource setASCollectionViewSuperclass];
|
||||
[ASIGListAdapterBasedDataSource configureUpdater:listAdapter.updater];
|
||||
|
||||
ASDisplayNodeAssert([listAdapter conformsToProtocol:@protocol(UICollectionViewDataSource)], @"Expected IGListAdapter to conform to UICollectionViewDataSource.");
|
||||
ASDisplayNodeAssert([listAdapter conformsToProtocol:@protocol(UICollectionViewDelegateFlowLayout)], @"Expected IGListAdapter to conform to UICollectionViewDelegateFlowLayout.");
|
||||
_listAdapter = listAdapter;
|
||||
}
|
||||
return self;
|
||||
}
|
||||
|
||||
- (id<UICollectionViewDataSource>)dataSource
|
||||
{
|
||||
return (id<UICollectionViewDataSource>)_listAdapter;
|
||||
}
|
||||
|
||||
- (id<UICollectionViewDelegateFlowLayout>)delegate
|
||||
{
|
||||
return (id<UICollectionViewDelegateFlowLayout>)_listAdapter;
|
||||
}
|
||||
|
||||
#pragma mark - ASCollectionDelegate
|
||||
|
||||
- (void)collectionNode:(ASCollectionNode *)collectionNode didSelectItemAtIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
[self.delegate collectionView:collectionNode.view didSelectItemAtIndexPath:indexPath];
|
||||
}
|
||||
|
||||
- (void)scrollViewDidScroll:(UIScrollView *)scrollView
|
||||
{
|
||||
[self.delegate scrollViewDidScroll:scrollView];
|
||||
}
|
||||
|
||||
- (void)scrollViewWillBeginDragging:(UIScrollView *)scrollView
|
||||
{
|
||||
[self.delegate scrollViewWillBeginDragging:scrollView];
|
||||
}
|
||||
|
||||
- (void)scrollViewDidEndDragging:(UIScrollView *)scrollView willDecelerate:(BOOL)decelerate
|
||||
{
|
||||
[self.delegate scrollViewDidEndDragging:scrollView willDecelerate:decelerate];
|
||||
}
|
||||
|
||||
- (BOOL)shouldBatchFetchForCollectionNode:(ASCollectionNode *)collectionNode
|
||||
{
|
||||
NSInteger sectionCount = [self numberOfSectionsInCollectionNode:collectionNode];
|
||||
if (sectionCount == 0) {
|
||||
return NO;
|
||||
}
|
||||
|
||||
// If they implement shouldBatchFetch, call it. Otherwise, just say YES if they implement beginBatchFetch.
|
||||
ASIGSectionController *ctrl = [self sectionControllerForSection:sectionCount - 1];
|
||||
ASSectionControllerOverrides o = [ASIGListAdapterBasedDataSource overridesForSectionControllerClass:ctrl.class];
|
||||
BOOL result = (o.shouldBatchFetch ? [ctrl shouldBatchFetch] : o.beginBatchFetchWithContext);
|
||||
if (result) {
|
||||
self.sectionControllerForBatchFetching = ctrl;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
- (void)collectionNode:(ASCollectionNode *)collectionNode willBeginBatchFetchWithContext:(ASBatchContext *)context
|
||||
{
|
||||
ASIGSectionController *ctrl = self.sectionControllerForBatchFetching;
|
||||
self.sectionControllerForBatchFetching = nil;
|
||||
[ctrl beginBatchFetchWithContext:context];
|
||||
}
|
||||
|
||||
/**
|
||||
* Note: It is not documented that ASCollectionNode will forward these UIKit delegate calls if they are implemented.
|
||||
* It is not considered harmful to do so, and adding them to documentation will confuse most users, who should
|
||||
* instead using the ASCollectionDelegate callbacks.
|
||||
*/
|
||||
#pragma mark - ASCollectionDelegateInterop
|
||||
|
||||
- (void)collectionView:(UICollectionView *)collectionView willDisplayCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
[self.delegate collectionView:collectionView willDisplayCell:cell forItemAtIndexPath:indexPath];
|
||||
}
|
||||
|
||||
- (void)collectionView:(UICollectionView *)collectionView didEndDisplayingCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
[self.delegate collectionView:collectionView didEndDisplayingCell:cell forItemAtIndexPath:indexPath];
|
||||
}
|
||||
|
||||
#pragma mark - ASCollectionDelegateFlowLayout
|
||||
|
||||
- (ASSizeRange)collectionNode:(ASCollectionNode *)collectionNode sizeRangeForHeaderInSection:(NSInteger)section
|
||||
{
|
||||
id<ASIGSupplementaryNodeSource> src = [self supplementaryElementSourceForSection:section];
|
||||
if ([ASIGListAdapterBasedDataSource overridesForSupplementarySourceClass:[src class]].sizeRangeForSupplementary) {
|
||||
return [src sizeRangeForSupplementaryElementOfKind:UICollectionElementKindSectionHeader atIndex:0];
|
||||
} else {
|
||||
return ASSizeRangeZero;
|
||||
}
|
||||
}
|
||||
|
||||
- (ASSizeRange)collectionNode:(ASCollectionNode *)collectionNode sizeRangeForFooterInSection:(NSInteger)section
|
||||
{
|
||||
id<ASIGSupplementaryNodeSource> src = [self supplementaryElementSourceForSection:section];
|
||||
if ([ASIGListAdapterBasedDataSource overridesForSupplementarySourceClass:[src class]].sizeRangeForSupplementary) {
|
||||
return [src sizeRangeForSupplementaryElementOfKind:UICollectionElementKindSectionFooter atIndex:0];
|
||||
} else {
|
||||
return ASSizeRangeZero;
|
||||
}
|
||||
}
|
||||
|
||||
- (UIEdgeInsets)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout insetForSectionAtIndex:(NSInteger)section
|
||||
{
|
||||
return [self.delegate collectionView:collectionView layout:collectionViewLayout insetForSectionAtIndex:section];
|
||||
}
|
||||
|
||||
- (CGFloat)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout minimumLineSpacingForSectionAtIndex:(NSInteger)section
|
||||
{
|
||||
return [self.delegate collectionView:collectionView layout:collectionViewLayout minimumLineSpacingForSectionAtIndex:section];
|
||||
}
|
||||
|
||||
- (CGFloat)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout minimumInteritemSpacingForSectionAtIndex:(NSInteger)section
|
||||
{
|
||||
return [self.delegate collectionView:collectionView layout:collectionViewLayout minimumInteritemSpacingForSectionAtIndex:section];
|
||||
}
|
||||
|
||||
#pragma mark - ASCollectionDataSource
|
||||
|
||||
- (NSInteger)collectionNode:(ASCollectionNode *)collectionNode numberOfItemsInSection:(NSInteger)section
|
||||
{
|
||||
return [self.dataSource collectionView:collectionNode.view numberOfItemsInSection:section];
|
||||
}
|
||||
|
||||
- (NSInteger)numberOfSectionsInCollectionNode:(ASCollectionNode *)collectionNode
|
||||
{
|
||||
return [self.dataSource numberOfSectionsInCollectionView:collectionNode.view];
|
||||
}
|
||||
|
||||
- (ASCellNodeBlock)collectionNode:(ASCollectionNode *)collectionNode nodeBlockForItemAtIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
return [[self sectionControllerForSection:indexPath.section] nodeBlockForItemAtIndex:indexPath.item];
|
||||
}
|
||||
|
||||
- (ASSizeRange)collectionNode:(ASCollectionNode *)collectionNode constrainedSizeForItemAtIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
ASIGSectionController *ctrl = [self sectionControllerForSection:indexPath.section];
|
||||
if ([ASIGListAdapterBasedDataSource overridesForSectionControllerClass:ctrl.class].sizeRangeForItem) {
|
||||
return [ctrl sizeRangeForItemAtIndex:indexPath.item];
|
||||
} else {
|
||||
return ASSizeRangeUnconstrained;
|
||||
}
|
||||
}
|
||||
|
||||
- (ASCellNodeBlock)collectionNode:(ASCollectionNode *)collectionNode nodeBlockForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
return [[self supplementaryElementSourceForSection:indexPath.section] nodeBlockForSupplementaryElementOfKind:kind atIndex:indexPath.item];
|
||||
}
|
||||
|
||||
- (NSArray<NSString *> *)collectionNode:(ASCollectionNode *)collectionNode supplementaryElementKindsInSection:(NSInteger)section
|
||||
{
|
||||
return [[self supplementaryElementSourceForSection:section] supportedElementKinds];
|
||||
}
|
||||
|
||||
#pragma mark - ASCollectionDataSourceInterop
|
||||
|
||||
- (__kindof UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(nonnull NSIndexPath *)indexPath
|
||||
{
|
||||
return [self.dataSource collectionView:collectionView cellForItemAtIndexPath:indexPath];
|
||||
}
|
||||
|
||||
- (UICollectionReusableView *)collectionView:(UICollectionView *)collectionView viewForSupplementaryElementOfKind:(NSString *)kind atIndexPath:(NSIndexPath *)indexPath
|
||||
{
|
||||
return [self.dataSource collectionView:collectionView viewForSupplementaryElementOfKind:kind atIndexPath:indexPath];
|
||||
}
|
||||
|
||||
+ (BOOL)dequeuesCellsForNodeBackedItems
|
||||
{
|
||||
return YES;
|
||||
}
|
||||
|
||||
#pragma mark - Helpers
|
||||
|
||||
- (id<ASIGSupplementaryNodeSource>)supplementaryElementSourceForSection:(NSInteger)section
|
||||
{
|
||||
ASIGSectionController *ctrl = [self sectionControllerForSection:section];
|
||||
id<ASIGSupplementaryNodeSource> src = (id<ASIGSupplementaryNodeSource>)ctrl.supplementaryViewSource;
|
||||
ASDisplayNodeAssert(src == nil || [src conformsToProtocol:@protocol(ASSupplementaryNodeSource)], @"Supplementary view source should conform to %@", NSStringFromProtocol(@protocol(ASSupplementaryNodeSource)));
|
||||
return src;
|
||||
}
|
||||
|
||||
- (ASIGSectionController *)sectionControllerForSection:(NSInteger)section
|
||||
{
|
||||
id object = [_listAdapter objectAtSection:section];
|
||||
ASIGSectionController *ctrl = (ASIGSectionController *)[_listAdapter sectionControllerForObject:object];
|
||||
ASDisplayNodeAssert([ctrl conformsToProtocol:@protocol(ASSectionController)], @"Expected section controller to conform to %@. Controller: %@", NSStringFromProtocol(@protocol(ASSectionController)), ctrl);
|
||||
return ctrl;
|
||||
}
|
||||
|
||||
/**
|
||||
* Set ASCollectionView's superclass to IGListCollectionView.
|
||||
* Scary! If IGListKit removed the subclassing restriction, we could
|
||||
* use #if in the @interface to choose the superclass based on
|
||||
* whether we have IGListKit available.
|
||||
*/
|
||||
+ (void)setASCollectionViewSuperclass
|
||||
{
|
||||
#pragma clang diagnostic push
|
||||
#pragma clang diagnostic ignored "-Wdeprecated-declarations"
|
||||
static dispatch_once_t onceToken;
|
||||
dispatch_once(&onceToken, ^{
|
||||
class_setSuperclass([ASCollectionView class], [IGListCollectionView class]);
|
||||
});
|
||||
#pragma clang diagnostic pop
|
||||
}
|
||||
|
||||
/// Ensure updater won't call reloadData on us.
|
||||
+ (void)configureUpdater:(id<IGListUpdatingDelegate>)updater
|
||||
{
|
||||
// Cast to NSObject will be removed after https://github.com/Instagram/IGListKit/pull/435
|
||||
if ([(id<NSObject>)updater isKindOfClass:[IGListAdapterUpdater class]]) {
|
||||
[(IGListAdapterUpdater *)updater setAllowsBackgroundReloading:NO];
|
||||
} else {
|
||||
static dispatch_once_t onceToken;
|
||||
dispatch_once(&onceToken, ^{
|
||||
NSLog(@"WARNING: Use of non-%@ updater with AsyncDisplayKit is discouraged. Updater: %@", NSStringFromClass([IGListAdapterUpdater class]), updater);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
+ (ASSupplementarySourceOverrides)overridesForSupplementarySourceClass:(Class)c
|
||||
{
|
||||
static NSCache<Class, NSValue *> *cache;
|
||||
static dispatch_once_t onceToken;
|
||||
dispatch_once(&onceToken, ^{
|
||||
cache = [[NSCache alloc] init];
|
||||
});
|
||||
NSValue *obj = [cache objectForKey:c];
|
||||
ASSupplementarySourceOverrides o;
|
||||
if (obj == nil) {
|
||||
o.sizeRangeForSupplementary = [c instancesRespondToSelector:@selector(sizeRangeForSupplementaryElementOfKind:atIndex:)];
|
||||
obj = [NSValue valueWithBytes:&o objCType:@encode(ASSupplementarySourceOverrides)];
|
||||
[cache setObject:obj forKey:c];
|
||||
} else {
|
||||
[obj getValue:&o];
|
||||
}
|
||||
return o;
|
||||
}
|
||||
|
||||
+ (ASSectionControllerOverrides)overridesForSectionControllerClass:(Class)c
|
||||
{
|
||||
static NSCache<Class, NSValue *> *cache;
|
||||
static dispatch_once_t onceToken;
|
||||
dispatch_once(&onceToken, ^{
|
||||
cache = [[NSCache alloc] init];
|
||||
});
|
||||
NSValue *obj = [cache objectForKey:c];
|
||||
ASSectionControllerOverrides o;
|
||||
if (obj == nil) {
|
||||
o.sizeRangeForItem = [c instancesRespondToSelector:@selector(sizeRangeForItemAtIndex:)];
|
||||
o.beginBatchFetchWithContext = [c instancesRespondToSelector:@selector(beginBatchFetchWithContext:)];
|
||||
o.shouldBatchFetch = [c instancesRespondToSelector:@selector(shouldBatchFetch)];
|
||||
obj = [NSValue valueWithBytes:&o objCType:@encode(ASSectionControllerOverrides)];
|
||||
[cache setObject:obj forKey:c];
|
||||
} else {
|
||||
[obj getValue:&o];
|
||||
}
|
||||
return o;
|
||||
}
|
||||
|
||||
@end
|
||||
|
||||
#endif // IG_LIST_KIT
|
||||
Reference in New Issue
Block a user