Swiftgram/Source/IGListAdapter+AsyncDisplayKit.m
2017-02-26 18:14:13 -08:00

48 lines
1.5 KiB
Objective-C

//
// IGListAdapter+AsyncDisplayKit.m
// AsyncDisplayKit
//
// Created by Adlai Holler on 1/19/17.
// Copyright © 2017 Facebook. All rights reserved.
//
#if IG_LIST_KIT
#import "IGListAdapter+AsyncDisplayKit.h"
#import "ASIGListAdapterBasedDataSource.h"
#import "ASAssert.h"
#import <objc/runtime.h>
@implementation IGListAdapter (AsyncDisplayKit)
- (void)setASDKCollectionNode:(ASCollectionNode *)collectionNode
{
ASDisplayNodeAssertMainThread();
// Attempt to retrieve previous data source.
ASIGListAdapterBasedDataSource *dataSource = objc_getAssociatedObject(self, _cmd);
// Bomb if we already made one.
if (dataSource != nil) {
ASDisplayNodeFailAssert(@"Attempt to call %@ multiple times on the same list adapter. Not currently allowed!", NSStringFromSelector(_cmd));
return;
}
// Make a data source and retain it.
dataSource = [[ASIGListAdapterBasedDataSource alloc] initWithListAdapter:self];
objc_setAssociatedObject(self, _cmd, dataSource, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
// Attach the data source to the collection node.
collectionNode.dataSource = dataSource;
collectionNode.delegate = dataSource;
__weak IGListAdapter *weakSelf = self;
[collectionNode onDidLoad:^(__kindof ASCollectionNode * _Nonnull collectionNode) {
// We manually set the superclass of ASCollectionView to IGListCollectionView at runtime.
// Issue tracked at https://github.com/Instagram/IGListKit/issues/409
weakSelf.collectionView = (IGListCollectionView *)collectionNode.view;
}];
}
@end
#endif // IG_LIST_KIT