HomeViewController.m

//
//  HomeViewController.m
//  unnamed
//
//  Created by Casing Chu on 2/18/15.
//  Copyright (c) 2015 com.yahoo. All rights reserved.
//

#import "HomeViewController.h"
#import "SurveyViewController.h"
#import "ProfileViewController.h"
#import "ComposeViewController.h"
#import "SurveyViewCell.h"
#import "PhotoAnswerCell.h"
#import "ParseClient.h"
#import "UIColor+AppColor.h"
#import "HomeProfileAnimation.h"

NSString * const kSurveyViewCell = @"SurveyViewCell";
NSString * const kPhotoViewCell = @"PhotoAnswerCell";

@interface HomeViewController () <UITableViewDelegate, UITableViewDataSource, SurveyViewCellDelegate, PhotoAnswerCellDelegate>
@property (weak, nonatomic) IBOutlet UITableView *tableView;
@property (nonatomic, strong) UIRefreshControl *tableRefreshControl;
@property (nonatomic, strong) NSMutableArray *surveys;
@property (nonatomic, assign) NSInteger pageIndex;
@property (nonatomic, assign) BOOL isUpdating;
@property (nonatomic, assign) BOOL isInsertingNewPost;
@property (nonatomic, strong) SurveyViewCell * prototypeSurveyCell;
@property (nonatomic, strong) PhotoAnswerCell *prototypePhotoCell;
@property (nonatomic, strong) HomeProfileAnimation *transitionAnimation;

@end

@implementation HomeViewController

- (void) setupTestData {
    for (int i=0; i<20; i++) {
        Survey *survey = [[Survey alloc] init];
        survey.question = [[Question alloc] init];
        survey.question.text = [NSString stringWithFormat:@"Question %d?", i];
        survey.question.anonymous = (i%2==0)?YES:NO;
        survey.question.complete = (i%2==0)?NO:YES;
        
        survey.user = [[User alloc] init];
        survey.user.name = [NSString stringWithFormat:@"Name %d", i];
        
        NSInteger total = 0;
        NSMutableArray *answers = [[NSMutableArray alloc] init];
        for (int j=0; j < ((i % 4) + 2); j++) {
            Answer *answer = [[Answer alloc] init];
            answer.count = i * j;
            total += answer.count;
            answer.text = [NSString stringWithFormat:@"Answer %d - %d", i, j];
            [answers addObject:answer];
        }
        survey.answers = [NSArray arrayWithArray:answers];
        survey.voted = (i%2 == 0)?YES:NO;
        survey.totalVotes = total;
        [self.surveys addObject:survey];
    }
}

- (id)initWithNibName:(NSString *)nibNameOrNil bundle:(NSBundle *)nibBundleOrNil {
    self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil];
    if (self) {
        self.title = @"Home";
        self.tabBarItem.image = [UIImage imageNamed:@"Home"];
        UIBarButtonItem *leftBarButtonItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"User Male"] style:UIBarButtonItemStylePlain target:self action:@selector(onProfileTap)];
        self.navigationItem.leftBarButtonItem = leftBarButtonItem;
        UIBarButtonItem *rightBarButtonItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"Poll Topic"] style:UIBarButtonItemStylePlain target:self action:@selector(onCreateTap)];
        self.navigationItem.rightBarButtonItem = rightBarButtonItem;
    }
    return self;
}

- (void)viewDidLoad {
    [super viewDidLoad];
    self.view.backgroundColor = [UIColor appBgColor];
    self.transitionAnimation = [[HomeProfileAnimation alloc] init];
    self.navigationController.delegate = self.transitionAnimation;
    self.navigationController.modalPresentationStyle = UIModalPresentationCustom;
    
    //Setup Notification listener
    self.isInsertingNewPost = NO;
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(onNewPost:) name:UserDidPostNewSurveyNotification object:nil];
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(onUpdatePost) name:UserDidPostUpdateSurveyNotification object:nil];
    
    // Setup Objects
    self.pageIndex = 0;
    self.surveys = [[NSMutableArray alloc] init];
    
    // Table Refresh control
    self.tableRefreshControl = [[UIRefreshControl alloc] init];
    [self.tableRefreshControl addTarget:self action:@selector(onTableRefresh) forControlEvents:UIControlEventValueChanged];
    [self.tableView insertSubview:self.tableRefreshControl atIndex:0];
    
    // TableView Setup
    self.tableView.dataSource = self;
    self.tableView.delegate = self;
    [self.tableView registerNib:[UINib nibWithNibName:kSurveyViewCell bundle:nil] forCellReuseIdentifier:kSurveyViewCell];
    [self.tableView registerNib:[UINib nibWithNibName:kPhotoViewCell bundle:nil] forCellReuseIdentifier:kPhotoViewCell];
    self.tableView.rowHeight = UITableViewAutomaticDimension;
    self.tableView.backgroundColor = [UIColor appBgColor];

    [self onTableRefresh];
}

- (void)viewWillAppear:(BOOL)animated {
    if (self.isInsertingNewPost) {
        self.isInsertingNewPost = NO;
        NSIndexPath *indexPath = [NSIndexPath indexPathForRow:0 inSection:0];
            [UIView animateWithDuration:1 animations:^{
                [self.tableView beginUpdates];
                [self.tableView insertRowsAtIndexPaths:[NSArray arrayWithObject:indexPath] withRowAnimation:UITableViewRowAnimationTop];
                [self.tableView endUpdates];
            }];
    }
}

- (void)viewDidAppear:(BOOL)animated {
    self.transitionAnimation.selectedCell = nil;
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

#pragma mark - RefreshControl
- (void)onTableRefresh {
    self.pageIndex = 0;
    [self fetchSurveys];
}

- (void)fetchSurveys {
    if (!self.isUpdating) {
        self.isUpdating = YES;
        [ParseClient getHomeSurveysOnPage:self.pageIndex withCompletion:^(NSArray *surveys, NSError *error) {
            if (error == nil) {
                if (self.pageIndex == 0) {
                    self.surveys = [NSMutableArray arrayWithArray:surveys];
                } else {
                    [self.surveys addObjectsFromArray:surveys];
                }
                [self.tableView reloadData];
                self.isUpdating = NO;
            } else {
                NSLog(@"%@", error);
                [[[UIAlertView alloc] initWithTitle:@"Network Error" message:@"Unable to retrieve surveys. Please try again." delegate:self cancelButtonTitle:@"OK" otherButtonTitles:nil] show];
                self.isUpdating = NO;
            }
            [self.tableRefreshControl endRefreshing];
        }];
//        [self setupTestData];
//        [self.tableView reloadData];
//        [self.tableRefreshControl endRefreshing];
    }
}

#pragma mark - TableViewDelegate Methods
- (SurveyViewCell *)prototypeSurveyCell {
    if (!_prototypeSurveyCell) {
        _prototypeSurveyCell = [self.tableView dequeueReusableCellWithIdentifier:kSurveyViewCell];
    }
    return _prototypeSurveyCell;
}

- (PhotoAnswerCell *)prototypePhotoCell {
    if (!_prototypePhotoCell) {
        _prototypePhotoCell = [self.tableView dequeueReusableCellWithIdentifier:kPhotoViewCell];
    }
    return _prototypePhotoCell;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
    return self.surveys.count;
}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
    if (indexPath.row == self.surveys.count - 1 && !self.isUpdating) {
        self.pageIndex++;
        if (self.surveys.count == (self.pageIndex * ResultCount)) {
            [self fetchSurveys];
        }
    }
    Survey *survey = self.surveys[indexPath.row];
    if (survey.question.isTextSurvey) {
        SurveyViewCell *cell = [self.tableView dequeueReusableCellWithIdentifier:kSurveyViewCell];
        [self configureCell:cell forRowAtIndexPath:indexPath];
        return cell;
    } else {
        PhotoAnswerCell *cell = [self.tableView dequeueReusableCellWithIdentifier:kPhotoViewCell];
        [self configureCell:cell forRowAtIndexPath:indexPath];
        return cell;
    }
}

- (CGFloat)tableView:(UITableView *)tableView estimatedHeightForRowAtIndexPath:(NSIndexPath *)indexPath {
    return UITableViewAutomaticDimension;
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
    [tableView deselectRowAtIndexPath:indexPath animated:YES];
    SurveyViewController *vc = [[SurveyViewController alloc] init];
    vc.survey = self.surveys[indexPath.row];
    vc.view.frame = self.view.frame;
    self.transitionAnimation.selectedCell = (SurveyViewCell *)[self.tableView cellForRowAtIndexPath:indexPath];
    [self.navigationController pushViewController:vc animated:YES];
}

- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {
    Survey *survey = self.surveys[indexPath.row];
    if (survey.question.isTextSurvey) {
        [self configureCell:self.prototypeSurveyCell forRowAtIndexPath:indexPath];
        [self.prototypeSurveyCell layoutIfNeeded];
        CGSize size = [self.prototypeSurveyCell.contentView systemLayoutSizeFittingSize:UILayoutFittingCompressedSize];
        //    NSLog(@"Row %ld has height %f", indexPath.row, size.height);
        return size.height + 1;
    } else {
        [self configureCell:self.prototypePhotoCell forRowAtIndexPath:indexPath];
        [self.prototypePhotoCell layoutIfNeeded];
        CGSize size = [self.prototypePhotoCell.contentView systemLayoutSizeFittingSize:UILayoutFittingCompressedSize];
        //    NSLog(@"Row %ld has height %f", indexPath.row, size.height);
        return size.height + 1;
    }
}

- (void)configureCell:(UITableViewCell *)pCell forRowAtIndexPath:(NSIndexPath *)indexPath {
    if ([pCell isKindOfClass:[SurveyViewCell class]]) {
        SurveyViewCell *cell = (SurveyViewCell *)pCell;
        cell.survey = self.surveys[indexPath.row];
        cell.delegate = self;
    } else if ([pCell isKindOfClass:[PhotoAnswerCell class]]) {
        PhotoAnswerCell *cell = (PhotoAnswerCell *)pCell;
        cell.survey = self.surveys[indexPath.row];
        cell.delegate = self;
    }
}

#pragma mark - SurveyViewCellDelegate methods
- (void)surveyViewCell:(SurveyViewCell *)cell didClickOnUser:(User *)user {
    [self clickedOnCell:cell userPhoto:user];
}

#pragma mark - PhotoAnswerCellDelegate methods
- (void)photoAnswerCell:(PhotoAnswerCell *)cell didClickOnUser:(User *)user{
    [self clickedOnCell:cell userPhoto:user];
}

- (void)clickedOnCell:(UITableViewCell *)cell userPhoto:(User *)user {
    if ([user isEqualUser:[User currentUser]]) {
        [self onProfileTap];
    } else {
        ProfileViewController *vc = [[ProfileViewController alloc] init];
        vc.user = user;
        vc.view.frame = self.view.frame;
        self.transitionAnimation.selectedCell = cell;
        [self.navigationController pushViewController:vc animated:YES];
    }
}


#pragma mark - Private Methods
- (void)onNewPost:(NSNotification *)notification {
    Survey *survey = notification.userInfo[@"survey"];
    if (survey) {
        NSMutableArray *newSurveys = [NSMutableArray arrayWithObject:survey];
        [newSurveys addObjectsFromArray:self.surveys];
        self.surveys = newSurveys;
        self.isInsertingNewPost = YES;
    }
}

- (void)onProfileTap {
    ProfileViewController *vc = [[ProfileViewController alloc] init];
    vc.user = [User currentUser];
    vc.view.frame = self.view.frame;
    self.transitionAnimation.selectedCell = nil;
    [self.navigationController pushViewController:vc animated:YES];
}

- (void)onCreateTap{
    ComposeViewController *vc = [[ComposeViewController alloc] init];
    [self.navigationController pushViewController:vc animated:YES];
}

- (void)onUpdatePost {
    [self.tableView reloadData];
}

@end