//
//  WLNativeAdsCollectionMode.m
//  DemoApp
//
//  Created by Nick Xirotyris on 19/10/15.
//  Copyright © 2015 YC. All rights reserved.
//

#import "WLNativeAdsCollectionMode.h"
#import "WLNativeAdCollectionViewCell.h"
#import "Warply.h"


@interface WLNativeAdsCollectionMode () <UICollectionViewDelegate, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout>

@property (weak, nonatomic) id<UICollectionViewDataSource> originalDatasource;
@property (weak, nonatomic) id<UICollectionViewDelegate> originalDelegate;
@property (strong, nonatomic) id<UICollectionViewDelegateFlowLayout> originalFlowLayout;
@property (weak, nonatomic) UICollectionView *nativeCollectionView;
@property (nonatomic, weak) WLCustomNativeCollectionViewCell *customCell;
@property (nonatomic, strong) NSString *customCellIdentifier;
@property (nonatomic, strong) NSString *customCellNibName;

@property (weak, nonatomic) UIViewController <UICollectionViewDataSource,UICollectionViewDelegate,UICollectionViewDelegateFlowLayout> *originalVC;


@end


@implementation WLNativeAdsCollectionMode {
    NSInteger adsForCollectionView;
    NSInteger startAd;
    NSInteger frequencyAd;
    NSInteger originalNumberOfsections;
    NSInteger currentRows;
    NSInteger totalExtraRows;
    NSMutableArray *indexArrayOfAds;
    NSArray *offersArray;
    CGSize cellSize;
    NSString *display_type;
    BOOL repeatableAds;
}


-(id)initWithViewController:(UIViewController<UICollectionViewDataSource,UICollectionViewDelegate,UICollectionViewDelegateFlowLayout> *)viewController CollectionView:(UICollectionView *)collectionView withStartPosition:(NSInteger)startPosition Frequency:(NSInteger)frequency CellSize:(CGSize)size andDisplayType:(NSString *)displayType{
    
    self = [super init];
    if (self) {
        
        [self getInbox];
        [self.nativeCollectionView registerClass:[WLNativeAdCollectionViewCell class] forCellWithReuseIdentifier:@"adItem"];
        [collectionView registerClass:[WLNativeAdCollectionViewCell class] forCellWithReuseIdentifier:@"adItem"];
        
        
        self.originalDatasource = collectionView.dataSource;
        self.originalDelegate = collectionView.delegate;
        self.nativeCollectionView = collectionView;
        self.originalVC = viewController;
        self.originalFlowLayout = (id<UICollectionViewDelegateFlowLayout>)self.originalDelegate;
        startAd = startPosition;
        frequencyAd = frequency;
        display_type = displayType;

        if ([self.originalDatasource respondsToSelector:@selector(numberOfSectionsInCollectionView:)]) {
            originalNumberOfsections = [self.originalDatasource numberOfSectionsInCollectionView:collectionView];
        } else {
            originalNumberOfsections = 1;
        }
        
        repeatableAds = NO;
        currentRows = 0;
        totalExtraRows = 0;
        cellSize = size;
        
    }
    
    return self;
}


- (void)refreshAdsForCollectionView:(UICollectionView *)collectionView {
    self.originalDatasource = nil;
    self.originalDelegate = nil;
    self.nativeCollectionView = nil;
    collectionView.dataSource = nil;
    collectionView.delegate = nil;
    collectionView.dataSource = self.originalVC;
    collectionView.delegate = self.originalVC;
    
    self.originalDatasource = collectionView.dataSource;
    self.originalDelegate = collectionView.delegate;
    self.nativeCollectionView = collectionView;
    
    if ([self.originalDatasource respondsToSelector:@selector(numberOfSectionsInCollectionView:)]) {
        originalNumberOfsections = [self.originalDatasource numberOfSectionsInCollectionView:collectionView];
    } else {
        originalNumberOfsections = 1;
    }
    
    currentRows = 0;
    totalExtraRows = 0;
    
    [self getInbox];
}


-(void)setRepeatable:(BOOL)repeatable {
    
    repeatableAds = repeatable;
    
    if (!repeatable) {
        int counter = 0;
        
        for (int i = 0; i < originalNumberOfsections; i++) {
            
            for (int j = 0; j < [[indexArrayOfAds objectAtIndex:i] count]; j++) {
                if ([[[indexArrayOfAds objectAtIndex:i] objectAtIndex:j] isEqual:@(YES)]) {
                    counter++;
                    if (counter > adsForCollectionView) {
                        [[indexArrayOfAds objectAtIndex:i] removeObjectAtIndex:j];
                    }
                }
                
            }
        }
        
        [self.nativeCollectionView reloadData];
        
    } else {
        NSLog(@"Nothing to remove");
    }
}


-(void)setCustomCellForNativeAd:(WLCustomNativeCollectionViewCell *)cell withIdentifier:(NSString *)cellIdentifier andNibName:(NSString *)nibName {
    self.customCell = cell;
    self.customCellIdentifier = cellIdentifier;
    self.customCellNibName = nibName;
}


-(void)getInbox {
    
    
    [[Warply sharedService] getInboxWithSuccessBlock:^(NSArray *list) {
        
        NSPredicate *predicate = [NSPredicate predicateWithFormat:[NSString stringWithFormat:@"(display_type == '%@')", display_type]];
        offersArray = [list filteredArrayUsingPredicate:predicate];
        
        if (offersArray.count > 0) {
            [self setupNewRowsPerSection];
            [self setRepeatable:repeatableAds];
            self.nativeCollectionView.dataSource = self;
            self.nativeCollectionView.delegate = self;
            self.nativeCollectionView.allowsSelection = YES;
            [self.nativeCollectionView reloadData];
        }
    } failureBlock:^(NSError *error) {
        //
    }];
}


#pragma mark - Original IndexPath & Helpers


- (void)setupNewRowsPerSection {
    
    indexArrayOfAds = [NSMutableArray array];
    int count = 0;
    
    for (int i = 0; i < originalNumberOfsections; i++) {
        NSMutableArray *newRows = [NSMutableArray array];
        NSInteger total = [self totalRowsWithAdsForSection:i];
        
        for (int j = 0; j < total; j++) {
            
            if (count == startAd || (count > startAd && (count - startAd)%(frequencyAd + 1) == 0)) {
                [newRows addObject:@(YES)];
            } else {
                [newRows addObject:@(NO)];
            }
            count++;
        }
        [indexArrayOfAds addObject:newRows];
    }
    
    NSLog(@"%@", indexArrayOfAds);
}



-(NSInteger)totalRowsWithAdsForSection:(int)section {
    
    NSInteger originalRows = [self.originalDatasource collectionView:self.nativeCollectionView numberOfItemsInSection:section];
    currentRows += originalRows;
    
    if (currentRows < startAd) {
        return originalRows;
    } else {
        
        if (startAd + 1 >= currentRows - originalRows && startAd + 1 <= currentRows) {
            NSInteger extraRows = (currentRows - startAd)/frequencyAd + 1;
            totalExtraRows += extraRows;
            return extraRows + originalRows;
        } else {
            
            NSInteger extraRowsSoFar = (currentRows - startAd)/frequencyAd + 1;
            NSInteger extraRowsForSection = extraRowsSoFar - totalExtraRows;
            totalExtraRows += extraRowsForSection;
            
            return originalRows + extraRowsForSection;
        }
    }
}


-(NSIndexPath *)getOriginalIndexPath:(NSIndexPath *)indexPath {
    int extraRows = 0;
    
    for (int i = 0; i < indexPath.row; i++) {
        if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:i]  isEqual:@(YES)]) {
            extraRows++;
        }
    }
    
    //    NSLog(@"row:    %ld", (long)indexPath.row - extraRows);
    return [NSIndexPath indexPathForRow:indexPath.row - extraRows inSection:indexPath.section];
}


-(int)getIndexOfAd:(NSIndexPath *)indexPath {
    
    int adsCount = 0;
    
    for (int i = 0; i < indexPath.section + 1; i++) {
        
        int rows = 0;
        
        if (indexPath.section > i) {
            rows = (int)[[indexArrayOfAds objectAtIndex:i] count];
        } else {
            rows = (int)indexPath.row + 1;
        }
        
        for (int j = 0; j < rows; j++) {
            if ([[[indexArrayOfAds objectAtIndex:i] objectAtIndex:j]  isEqual:@(YES)]) {
                adsCount++;
            }
        }
        
    }
    
    if (adsCount == offersArray.count) {
        return adsCount-1;
    } else {
        return adsCount%offersArray.count - 1;
    }
    
}



#pragma mark - CollectionView Datasource


-(NSInteger)numberOfSectionsInCollectionView:(UICollectionView *)collectionView {
    return originalNumberOfsections;
}

-(NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section {
    return [[indexArrayOfAds objectAtIndex:section] count];
}

-(UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath {

        
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        
        
        if (self.customCell) {
            
            NSString *customIdentifier = [NSString stringWithFormat:@"%@", self.customCellIdentifier];
            
            WLCustomNativeCollectionViewCell *customCell = (WLCustomNativeCollectionViewCell *)[collectionView dequeueReusableCellWithReuseIdentifier:customIdentifier forIndexPath:indexPath];
            
            if (self.customCellNibName) {
                if (customCell == nil) {
                    customCell = [[[NSBundle mainBundle] loadNibNamed:[NSString stringWithFormat:@"%@", self.customCellNibName] owner:self options:nil] objectAtIndex:0];
                }
            }
            
            if (offersArray.count > 0) {
                WLInboxItem *item = (WLInboxItem *)offersArray[[self getIndexOfAd:indexPath]];
                [customCell setupCustomCell:item];
            }
            
            return customCell;
            
        } else {
            WLNativeAdCollectionViewCell *cell = (WLNativeAdCollectionViewCell *)[collectionView dequeueReusableCellWithReuseIdentifier:@"adItem" forIndexPath:indexPath];
            
            if (cell == nil) {
                cell = [[[NSBundle mainBundle] loadNibNamed:@"WLNativeAdCollectionViewCell" owner:self options:nil] objectAtIndex:0];
            }
            
            cell.backgroundColor = [UIColor whiteColor];
            
            
            WLInboxItem *item = (WLInboxItem *)offersArray[[self getIndexOfAd:indexPath]];
            [cell loadContentwithURL:[item getPageURL]];
            
            return cell;
        }
        
        
    } else {
        return [self.originalDatasource collectionView:self.nativeCollectionView cellForItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
    }
}





#pragma mark - CollectionView Delegate


-(BOOL)collectionView:(UICollectionView *)collectionView canPerformAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return NO;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:canPerformAction:forItemAtIndexPath:withSender:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView canPerformAction:(SEL)action forItemAtIndexPath:[self getOriginalIndexPath:indexPath] withSender:sender];
        }
        
        return NO;
    }

}


-(void)collectionView:(UICollectionView *)collectionView performAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:performAction:forItemAtIndexPath:withSender:)]) {
            [self.originalDelegate collectionView:self.nativeCollectionView performAction:(SEL)action forItemAtIndexPath:[self getOriginalIndexPath:indexPath] withSender:sender];
        }
    }
}


-(void)collectionView:(UICollectionView *)collectionView didDeselectItemAtIndexPath:(NSIndexPath *)indexPath {
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didDeselectItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView didDeselectItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}


-(void)collectionView:(UICollectionView *)collectionView didSelectItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {

        if (offersArray.count > 0) {
            
            WLInboxItem *item = (WLInboxItem *)offersArray[[self getIndexOfAd:indexPath]];
            
            if ([[self delegate] respondsToSelector:@selector(nativeAdDidSelectWithItem:)]) {
                [[self delegate] nativeAdDidSelectWithItem:item];
            }
        }
        
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didSelectItemAtIndexPath:)]) {
            [self.originalDelegate collectionView:self.nativeCollectionView didSelectItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}



-(void)collectionView:(UICollectionView *)collectionView didHighlightItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didHighlightItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView didHighlightItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}




-(void)collectionView:(UICollectionView *)collectionView didUnhighlightItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didUnhighlightItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView didUnhighlightItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}


-(BOOL)collectionView:(UICollectionView *)collectionView shouldSelectItemAtIndexPath:(NSIndexPath *)indexPath {
    
    //FIXME: wrong conditions
    
//    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
//        return NO;
//    } else {
//        if ([self.originalDelegate respondsToSelector:@selector(collectionView:shouldSelectItemAtIndexPath:)]) {
//            return [self.originalDelegate collectionView:self.nativeCollectionView shouldSelectItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
//        }
//    }
    
    return YES;
}


-(BOOL)collectionView:(UICollectionView *)collectionView shouldDeselectItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
//    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
//        return NO;
//    } else {
//        if ([self.originalDelegate respondsToSelector:@selector(collectionView:shouldDeselectItemAtIndexPath:)]) {
//            return [self.originalDelegate collectionView:self.nativeCollectionView shouldDeselectItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
//        }
//    }
    
    return YES;
}


-(BOOL)collectionView:(UICollectionView *)collectionView shouldHighlightItemAtIndexPath:(NSIndexPath *)indexPath {
    
    //FIXME: wrong conditions
    
//    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
//        return NO;
//    } else {
//        if ([self.originalDelegate respondsToSelector:@selector(collectionView:shouldHighlightItemAtIndexPath:)]) {
//            return [self.originalDelegate collectionView:self.nativeCollectionView shouldHighlightItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
//        }
//    }
//    
//    return NO;
    
    return YES;
}


- (BOOL)collectionView:(UICollectionView *)collectionView shouldShowMenuForItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return NO;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:shouldShowMenuForItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView shouldShowMenuForItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
    
    return NO;
}


-(void)collectionView:(UICollectionView *)collectionView willDisplayCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:willDisplayCell:forItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView willDisplayCell:cell forItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}

-(void)collectionView:(UICollectionView *)collectionView didEndDisplayingCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didEndDisplayingCell:forItemAtIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView didEndDisplayingCell:cell forItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}

-(void)collectionView:(UICollectionView *)collectionView willDisplaySupplementaryView:(UICollectionReusableView *)view forElementKind:(NSString *)elementKind atIndexPath:(NSIndexPath *)indexPath {
    
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:willDisplaySupplementaryView:forElementKind:atIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView willDisplaySupplementaryView:view forElementKind:elementKind atIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}


-(void)collectionView:(UICollectionView *)collectionView didEndDisplayingSupplementaryView:(UICollectionReusableView *)view forElementOfKind:(NSString *)elementKind atIndexPath:(NSIndexPath *)indexPath {
    
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:didEndDisplayingSupplementaryView:forElementOfKind:atIndexPath:)]) {
            return [self.originalDelegate collectionView:self.nativeCollectionView didEndDisplayingSupplementaryView:view forElementOfKind:elementKind atIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
}





#pragma mark - CollectionView Flow Layout

-(UIEdgeInsets)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout insetForSectionAtIndex:(NSInteger)section {
    if ([self.originalFlowLayout respondsToSelector:@selector(collectionView:layout:insetForSectionAtIndex:)]) {
        return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout insetForSectionAtIndex:section];
    }
    return UIEdgeInsetsZero;
}



-(CGFloat)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout minimumInteritemSpacingForSectionAtIndex:(NSInteger)section {
    if ([self.originalFlowLayout respondsToSelector:@selector(collectionView:layout:minimumInteritemSpacingForSectionAtIndex:)]) {
        return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout minimumInteritemSpacingForSectionAtIndex:section];
    }
    return 10;
}


-(CGFloat)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout minimumLineSpacingForSectionAtIndex:(NSInteger)section {
    if ([self.originalFlowLayout respondsToSelector:@selector(collectionView:layout:minimumLineSpacingForSectionAtIndex:)]) {
        return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout minimumLineSpacingForSectionAtIndex:section];
    }
    return 10;
}


-(CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout referenceSizeForFooterInSection:(NSInteger)section {
    if ([self.originalFlowLayout respondsToSelector:@selector(collectionView:layout:referenceSizeForFooterInSection:)]) {
        return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout referenceSizeForFooterInSection:section];
    }
    return CGSizeZero;
}


-(CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout referenceSizeForHeaderInSection:(NSInteger)section {
    if ([self.originalFlowLayout respondsToSelector:@selector(collectionView:layout:referenceSizeForHeaderInSection:)]) {
        return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout referenceSizeForHeaderInSection:section];
    }
    return CGSizeZero;
}


-(CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout sizeForItemAtIndexPath:(NSIndexPath *)indexPath {
    if ([[[indexArrayOfAds objectAtIndex:indexPath.section] objectAtIndex:indexPath.row]  isEqual:@(YES)]) {
        return cellSize;
    } else {
        if ([self.originalDelegate respondsToSelector:@selector(collectionView:layout:sizeForItemAtIndexPath:)]) {
            return [self.originalFlowLayout collectionView:self.nativeCollectionView layout:collectionViewLayout sizeForItemAtIndexPath:[self getOriginalIndexPath:indexPath]];
        }
    }
    
    UICollectionViewFlowLayout *flow = (UICollectionViewFlowLayout *)collectionView.collectionViewLayout;
    
    return flow.itemSize;
}

@end