#import "ASCollectionInternal.h"

#import "ASCollectionViewLayoutFacilitatorProtocol.h"

#import "ASCollectionNode.h"

#import "ASDisplayNode+Subclasses.h"

#import "ASEnvironmentInternal.h"

#import "ASInternalHelpers.h"

#import "ASCellNode+Internal.h"

#pragma mark - _ASCollectionPendingState

@interface _ASCollectionPendingState : NSObject

@property (weak, nonatomic) id <ASCollectionDelegate> delegate;

@property (weak, nonatomic) id <ASCollectionDataSource> dataSource;

@property (assign, nonatomic) ASLayoutRangeMode rangeMode;

@implementation _ASCollectionPendingState

- (instancetype)init

{

self = [super init];

if (self) {

_rangeMode = ASLayoutRangeModeCount;

}

return self;

}

#if 0 // This is not used yet, but will provide a way to avoid creating the view to set range values.

@implementation _ASCollectionPendingState {

std::vector<std::vector<ASRangeTuningParameters>> _tuningParameters;

}

- (instancetype)init

{

self = [super init];

if (self) {

_tuningParameters = std::vector<std::vector<ASRangeTuningParameters>> (ASLayoutRangeModeCount, std::vector<ASRangeTuningParameters> (ASLayoutRangeTypeCount));

_rangeMode = ASLayoutRangeModeCount;

}

return self;

}

- (ASRangeTuningParameters)tuningParametersForRangeType:(ASLayoutRangeType)rangeType

{

return [self tuningParametersForRangeMode:ASLayoutRangeModeFull rangeType:rangeType];

}

- (void)setTuningParameters:(ASRangeTuningParameters)tuningParameters forRangeType:(ASLayoutRangeType)rangeType

{

return [self setTuningParameters:tuningParameters forRangeMode:ASLayoutRangeModeFull rangeType:rangeType];

}

- (ASRangeTuningParameters)tuningParametersForRangeMode:(ASLayoutRangeMode)rangeMode rangeType:(ASLayoutRangeType)rangeType

{

ASDisplayNodeAssert(rangeMode < _tuningParameters.size() && rangeType < _tuningParameters[rangeMode].size(), @"Requesting a range that is OOB for the configured tuning parameters");

return _tuningParameters[rangeMode][rangeType];

}

- (void)setTuningParameters:(ASRangeTuningParameters)tuningParameters forRangeMode:(ASLayoutRangeMode)rangeMode rangeType:(ASLayoutRangeType)rangeType

{

ASDisplayNodeAssert(rangeMode < _tuningParameters.size() && rangeType < _tuningParameters[rangeMode].size(), @"Setting a range that is OOB for the configured tuning parameters");

_tuningParameters[rangeMode][rangeType] = tuningParameters;

}

@end

#endif

#pragma mark - ASCollectionNode

@interface ASCollectionNode ()

{

ASDN::RecursiveMutex _environmentStateLock;

}

@property (nonatomic) _ASCollectionPendingState *pendingState;

@implementation ASCollectionNode

#pragma mark Lifecycle

- (instancetype)init

{

ASDISPLAYNODE_NOT_DESIGNATED_INITIALIZER();

UICollectionViewLayout *nilLayout = nil;

self = [self initWithCollectionViewLayout:nilLayout]; // Will throw an exception for lacking a UICV Layout.

return nil;

}

- (instancetype)initWithCollectionViewLayout:(UICollectionViewLayout *)layout

{

return [self initWithFrame:CGRectZero collectionViewLayout:layout];

}

- (instancetype)_initWithCollectionView:(ASCollectionView *)collectionView

{

ASDisplayNodeViewBlock collectionViewBlock = ^UIView *{ return collectionView; };

if (self = [super initWithViewBlock:collectionViewBlock]) {

// ASCollectionView created directly by the app. Trigger -loadView to set up collectionNode pointer.

__unused ASCollectionView *collectionView = [self view];

return self;

}

return nil;

}

- (instancetype)initWithFrame:(CGRect)frame collectionViewLayout:(UICollectionViewLayout *)layout

{

return [self initWithFrame:frame collectionViewLayout:layout layoutFacilitator:nil];

}

- (instancetype)initWithFrame:(CGRect)frame collectionViewLayout:(UICollectionViewLayout *)layout layoutFacilitator:(id<ASCollectionViewLayoutFacilitatorProtocol>)layoutFacilitator

{

ASDisplayNodeViewBlock collectionViewBlock = ^UIView *{

return [[ASCollectionView alloc] _initWithFrame:frame collectionViewLayout:layout layoutFacilitator:layoutFacilitator ownedByNode:YES];

};

if (self = [super initWithViewBlock:collectionViewBlock]) {

return self;

}

return nil;

}

#pragma mark ASDisplayNode

- (void)didLoad

{

[super didLoad];

ASCollectionView *view = self.view;

view.collectionNode = self;

if (_pendingState) {

_ASCollectionPendingState *pendingState = _pendingState;

self.pendingState = nil;

view.asyncDelegate = pendingState.delegate;

view.asyncDataSource = pendingState.dataSource;

if (pendingState.rangeMode != ASLayoutRangeModeCount) {

[view.rangeController updateCurrentRangeWithMode:pendingState.rangeMode];

}

}

}

- (ASCollectionView *)view

{

return (ASCollectionView *)[super view];

}

- (void)clearContents

{

[super clearContents];

[self.view clearContents];

}

- (void)clearFetchedData

{

[super clearFetchedData];

[self.view clearFetchedData];

}

#if ASRangeControllerLoggingEnabled

- (void)visibleStateDidChange:(BOOL)isVisible

{

[super visibleStateDidChange:isVisible];

NSLog(@"%@ - visible: %d", self, isVisible);

}

#endif

#pragma mark Setter / Getter

- (_ASCollectionPendingState *)pendingState

{

if (!_pendingState && ![self isNodeLoaded]) {

self.pendingState = [[_ASCollectionPendingState alloc] init];

}

ASDisplayNodeAssert(![self isNodeLoaded] || !_pendingState, @"ASCollectionNode should not have a pendingState once it is loaded");

return _pendingState;

}

- (void)setDelegate:(id <ASCollectionDelegate>)delegate

{

if ([self pendingState]) {

_pendingState.delegate = delegate;

} else {

ASDisplayNodeAssert([self isNodeLoaded], @"ASCollectionNode should be loaded if pendingState doesn't exist");

self.view.asyncDelegate = delegate;

}

}

- (id <ASCollectionDelegate>)delegate

{

if ([self pendingState]) {

return _pendingState.delegate;

} else {

return self.view.asyncDelegate;

}

}

- (void)setDataSource:(id <ASCollectionDataSource>)dataSource

{

if ([self pendingState]) {

_pendingState.dataSource = dataSource;

} else {

ASDisplayNodeAssert([self isNodeLoaded], @"ASCollectionNode should be loaded if pendingState doesn't exist");

self.view.asyncDataSource = dataSource;

}

}

- (id <ASCollectionDataSource>)dataSource

{

if ([self pendingState]) {

return _pendingState.dataSource;

} else {

return self.view.asyncDataSource;

}

}

#pragma mark ASCollectionView Forwards

- (ASRangeTuningParameters)tuningParametersForRangeType:(ASLayoutRangeType)rangeType

{

return [self.view.rangeController tuningParametersForRangeMode:ASLayoutRangeModeFull rangeType:rangeType];

}

- (void)setTuningParameters:(ASRangeTuningParameters)tuningParameters forRangeType:(ASLayoutRangeType)rangeType

{

[self.view.rangeController setTuningParameters:tuningParameters forRangeMode:ASLayoutRangeModeFull rangeType:rangeType];

}

- (ASRangeTuningParameters)tuningParametersForRangeMode:(ASLayoutRangeMode)rangeMode rangeType:(ASLayoutRangeType)rangeType

{

return [self.view.rangeController tuningParametersForRangeMode:rangeMode rangeType:rangeType];

}

- (void)setTuningParameters:(ASRangeTuningParameters)tuningParameters forRangeMode:(ASLayoutRangeMode)rangeMode rangeType:(ASLayoutRangeType)rangeType

{

return [self.view.rangeController setTuningParameters:tuningParameters forRangeMode:rangeMode rangeType:rangeType];

}

- (void)reloadDataWithCompletion:(void (^)())completion

{

[self.view reloadDataWithCompletion:completion];

}

- (void)reloadData

{

[self.view reloadData];

}

- (void)reloadDataImmediately

{

[self.view reloadDataImmediately];

}

- (void)beginUpdates

{

[self.view.dataController beginUpdates];

}

- (void)endUpdatesAnimated:(BOOL)animated

{

[self endUpdatesAnimated:animated completion:nil];

}

- (void)endUpdatesAnimated:(BOOL)animated completion:(void (^)(BOOL))completion

{

[self.view.dataController endUpdatesAnimated:animated completion:completion];

}

#pragma mark - ASRangeControllerUpdateRangeProtocol

- (void)updateCurrentRangeWithMode:(ASLayoutRangeMode)rangeMode;

{

if ([self pendingState]) {

_pendingState.rangeMode = rangeMode;

} else {

[self.view.rangeController updateCurrentRangeWithMode:rangeMode];

}

}

#pragma mark ASEnvironment

ASEnvironmentCollectionTableSetEnvironmentState(_environmentStateLock)

@end