//
//  FeedChooserViewController.m
//  NewsBlur
//
//  Created by David Sinclair on 2016-01-22.
//  Copyright © 2016 NewsBlur. All rights reserved.
//

#import "FeedChooserViewController.h"
#import "NewsBlurAppDelegate.h"
#import "NSObject+SBJSON.h"
#import "MenuViewController.h"
#import "FeedChooserTitleView.h"
#import "FeedChooserViewCell.h"
#import "FeedChooserItem.h"

static const CGFloat kTableViewRowHeight = 31.0;
static const CGFloat kFolderTitleHeight = 36.0;

@interface FeedChooserViewController () <FeedChooserTitleDelegate>

@property (nonatomic, strong) UIBarButtonItem *optionsItem;
@property (nonatomic, strong) UIBarButtonItem *moveItem;
@property (nonatomic, strong) UIBarButtonItem *deleteItem;
@property (nonatomic, strong) NSArray *sections;
@property (nonatomic, strong) NSArray *indexTitles;
@property (nonatomic, strong) NSArray *folders;
@property (nonatomic, strong) NSDictionary *dictFolders;
@property (nonatomic, strong) NSDictionary *inactiveFeeds;
@property (nonatomic) FeedChooserSort sort;
@property (nonatomic) BOOL ascending;
@property (nonatomic) BOOL flat;
@property (nonatomic, readonly) NewsBlurAppDelegate *appDelegate;

@end

@implementation FeedChooserViewController

- (void)dealloc {
    [[NSNotificationCenter defaultCenter] removeObserver:self];
}

- (void)viewDidLoad {
    [super viewDidLoad];
    
    UIBarButtonItem *doneItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemDone target:self action:@selector(done)];
    self.optionsItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"nav_icn_settings.png"] style:UIBarButtonItemStylePlain target:self action:@selector(showOptionsMenu)];
    
    if (self.operation == FeedChooserOperationOrganizeSites) {
        self.moveItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"menu_icn_move.png"] style:UIBarButtonItemStylePlain target:self action:@selector(showMoveMenu)];
        self.deleteItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"menu_icn_delete.png"] style:UIBarButtonItemStylePlain target:self action:@selector(deleteFeeds)];
        
        self.navigationItem.leftBarButtonItems = @[self.moveItem, self.deleteItem];
        self.navigationItem.rightBarButtonItems = @[doneItem, self.optionsItem];
        self.navigationItem.title = @"Organize Sites";
        
        self.tableView.editing = YES;
    } else {
        UIBarButtonItem *cancelItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemCancel target:self action:@selector(cancel)];
        
        self.navigationItem.leftBarButtonItem = cancelItem;
        self.navigationItem.rightBarButtonItems = @[doneItem, self.optionsItem];
        self.navigationItem.title = @"Mute Sites";
        
        self.tableView.editing = NO;
    }
    
    self.tableView.backgroundColor = UIColorFromRGB(0xECEEEA);
    self.tableView.separatorColor = UIColorFromRGB(0x909090);
    self.tableView.sectionIndexColor = UIColorFromRGB(0x303030);
    self.tableView.sectionIndexBackgroundColor = UIColorFromRGB(0xDCDFD6);
    
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(finishedLoadingFeedsNotification:) name:@"FinishedLoadingFeedsNotification" object:nil];
    
    if (self.operation == FeedChooserOperationMuteSites) {
        [self performGetInactiveFeeds];
    } else {
        [self updateDictFolders];
        [self rebuildItemsAnimated:NO];
    }
}

- (void)performGetInactiveFeeds {
    [MBProgressHUD hideHUDForView:self.view animated:YES];
    MBProgressHUD *HUD = [MBProgressHUD showHUDAddedTo:self.view animated:YES];
    HUD.labelText = @"Loading...";
    
    NSArray *selection = self.tableView.indexPathsForSelectedRows;
    NSMutableArray *feedsByFolder = [NSMutableArray arrayWithCapacity:selection.count];
    
    for (NSIndexPath *indexPath in selection) {
        FeedChooserItem *fromFolder = self.sections[indexPath.section];
        FeedChooserItem *item = fromFolder.contents[indexPath.row];
        
        [feedsByFolder addObject:@[item.identifier, fromFolder.identifier]];
    }
    
    NSString *urlString = [NSString stringWithFormat:@"%@/reader/feeds?flat=true&update_counts=false&include_inactive=true", self.appDelegate.url];
    NSURL *url = [NSURL URLWithString:urlString];
    ASIHTTPRequest *request = [ASIHTTPRequest requestWithURL:url];
    request.delegate = self;
    request.didFinishSelector = @selector(finishLoadingInactiveFeeds:);
    request.didFailSelector = @selector(finishedWithError:);
    request.timeOutSeconds = 30;
    [request startAsynchronous];
}

- (void)finishLoadingInactiveFeeds:(ASIHTTPRequest *)request {
    NSString *responseString = request.responseString;
    NSData *responseData = [responseString dataUsingEncoding:NSUTF8StringEncoding];
    NSError *error = nil;
    NSDictionary *results = [NSJSONSerialization JSONObjectWithData:responseData options:kNilOptions error:&error];
    
    self.dictFolders = results[@"flat_folders_with_inactive"];
    self.inactiveFeeds = results[@"inactive_feeds"];
    
    [self rebuildItemsAnimated:NO];
    
    [self enumerateAllRowsUsingBlock:^(NSIndexPath *indexPath, FeedChooserItem *item) {
        if (![item.info[@"active"] boolValue]) {
            [self.tableView selectRowAtIndexPath:indexPath animated:NO scrollPosition:UITableViewScrollPositionNone];
        }
    }];
    
    [MBProgressHUD hideHUDForView:self.view animated:YES];
}

- (void)finishedWithError:(ASIHTTPRequest *)request {
    NSError *error = request.error;
    NSLog(@"informError: %@", error);
    NSString *errorMessage = [error localizedDescription];
    
    [MBProgressHUD hideHUDForView:self.view animated:YES];
    MBProgressHUD *HUD = [MBProgressHUD showHUDAddedTo:self.view animated:YES];
    [HUD setCustomView:[[UIImageView alloc]
                        initWithImage:[UIImage imageNamed:@"warning.gif"]]];
    [HUD setMode:MBProgressHUDModeCustomView];
    HUD.labelText = errorMessage;
    [HUD hide:YES afterDelay:1];
    [self rebuildItemsAnimated:YES];
}

- (void)rebuildItemsAnimated:(BOOL)animated {
    NewsBlurAppDelegate *appDelegate = self.appDelegate;
    FeedChooserItem *section = nil;
    
    NSMutableArray *sections = [NSMutableArray array];
    NSMutableArray *indexTitles = [NSMutableArray array];
    NSMutableArray *folders = [NSMutableArray array];
    
    if (self.flat) {
        section = [FeedChooserItem makeFolderWithTitle:@""];
        [sections addObject:section];
    }
    
    NSArray *folderArray = [self.dictFolders.allKeys sortedArrayUsingSelector:@selector(caseInsensitiveCompare:)];
    
    for (NSString *folderName in folderArray) {
        if (![folderName hasPrefix:@"river_"] && ![folderName isEqualToString:@"read_stories"] && ![folderName isEqualToString:@"saved_stories"]) {
            FeedChooserItem *folder = [FeedChooserItem makeFolderWithTitle:folderName];
            [folders addObject:folder];
            
            if (!self.flat) {
                section = folder;
                [sections addObject:section];
                [indexTitles addObject:section.title.length ? [section.title substringToIndex:1] : @"-"];
            }
            
            for (id feedId in self.dictFolders[folderName]) {
                NSString *feedIdStr = [NSString stringWithFormat:@"%@", feedId];
                NSDictionary *info = appDelegate.dictFeeds[feedIdStr];
                
                if (!info) {
                    info = self.inactiveFeeds[feedIdStr];
                }
                
                if (![appDelegate isSocialFeed:feedIdStr] && ![appDelegate isSavedFeed:feedIdStr]) {
                    [section addItemWithInfo:info];
                }
            }
        }
    }
    
    self.sections = sections;
    self.indexTitles = indexTitles;
    self.folders = folders;
    
    [self.tableView reloadData];
    [self sortItemsAnimated:animated];
    [self updateControls];
}

- (void)sortItemsAnimated:(BOOL)animated {
    [self.tableView beginUpdates];
    
    NSString *key = [FeedChooserItem keyForSort:self.sort];
    
    [self enumerateSectionsUsingBlock:^(NSUInteger section, FeedChooserItem *folder) {
        NSArray *oldItems = [NSArray arrayWithArray:folder.contents];
        
        if (self.sort == FeedChooserSortName) {
            [folder.contents sortUsingDescriptors:@[[NSSortDescriptor sortDescriptorWithKey:key ascending:self.ascending selector:@selector(caseInsensitiveCompare:)], [NSSortDescriptor sortDescriptorWithKey:@"info.feed_title" ascending:self.ascending selector:@selector(caseInsensitiveCompare:)]]];
        } else {
            [folder.contents sortUsingDescriptors:@[[NSSortDescriptor sortDescriptorWithKey:key ascending:self.ascending], [NSSortDescriptor sortDescriptorWithKey:@"info.feed_title" ascending:self.ascending selector:@selector(caseInsensitiveCompare:)]]];
        }
        
        if (animated) {
            [self enumerateRowsInSection:section usingBlock:^(NSUInteger row, FeedChooserItem *item) {
                NSUInteger newRow = [folder.contents indexOfObject:oldItems[row]];
                
                [self.tableView moveRowAtIndexPath:[NSIndexPath indexPathForRow:row inSection:section] toIndexPath:[NSIndexPath indexPathForRow:newRow inSection:section]];
            }];
        }
    }];
    
    [self.tableView endUpdates];
    
    if (!animated) {
        [self.tableView reloadData];
    }
}

- (FeedChooserItem *)itemForIndexPath:(NSIndexPath *)indexPath {
    FeedChooserItem *folder = self.sections[indexPath.section];
    
    return folder.contents[indexPath.row];
}

- (NSIndexPath *)indexPathForItem:(FeedChooserItem *)item {
    __block NSIndexPath *indexPath = nil;
    
    [self enumerateAllRowsUsingBlock:^(NSIndexPath *localIndexPath, FeedChooserItem *localItem) {
        if ([item.identifierString isEqualToString:localItem.identifierString]) {
            indexPath = localIndexPath;
        }
    }];
    
    return indexPath;
}

- (NewsBlurAppDelegate *)appDelegate {
    return [NewsBlurAppDelegate sharedAppDelegate];
}

- (void)enumerateSectionsUsingBlock:(void (^)(NSUInteger section, FeedChooserItem *folder))block {
    if (!block) {
        return;
    }
    
    for (NSUInteger section = 0; section < self.sections.count; section++) {
        FeedChooserItem *folder = self.sections[section];
        
        block(section, folder);
    }
}

- (void)enumerateRowsInSection:(NSUInteger)section usingBlock:(void (^)(NSUInteger row, FeedChooserItem *item))block {
    if (!block) {
        return;
    }
    
    FeedChooserItem *folder = self.sections[section];
    
    for (NSUInteger row = 0; row < folder.contents.count; row++) {
        FeedChooserItem *item = folder.contents[row];
        
        block(row, item);
    }
}

- (void)enumerateAllRowsUsingBlock:(void (^)(NSIndexPath *indexPath, FeedChooserItem *item))block {
    if (!block) {
        return;
    }
    
    [self enumerateSectionsUsingBlock:^(NSUInteger section, FeedChooserItem *folder) {
        [self enumerateRowsInSection:section usingBlock:^(NSUInteger row, FeedChooserItem *item) {
            NSIndexPath *indexPath = [NSIndexPath indexPathForRow:row inSection:section];
            
            block(indexPath, item);
        }];
    }];
}

- (BOOL)isSelectionInSection:(NSUInteger)section {
    NSArray *selection = self.tableView.indexPathsForSelectedRows;
    
    for (NSIndexPath *indexPath in selection) {
        if (indexPath.section == section) {
            return YES;
        }
    }
    
    return NO;
}

- (NSArray *)selectedItemIdentifiers {
    NSMutableArray *identifiers = [NSMutableArray array];
    NSArray *selection = self.tableView.indexPathsForSelectedRows;
    
    for (NSIndexPath *indexPath in selection) {
        [identifiers addObject:[self itemForIndexPath:indexPath].identifier];
    }
    
    return identifiers;
}

- (void)selectItemsWithIdentifiers:(NSArray *)itemIdentifiers animated:(BOOL)animated {
    [self enumerateAllRowsUsingBlock:^(NSIndexPath *indexPath, FeedChooserItem *item) {
        if ([itemIdentifiers containsObject:item.identifier]) {
            [self.tableView selectRowAtIndexPath:indexPath animated:animated scrollPosition:UITableViewScrollPositionNone];
        } else {
            [self.tableView deselectRowAtIndexPath:indexPath animated:animated];
        }
    }];
    
    [self updateControls];
}

- (void)select:(BOOL)select section:(NSUInteger)section {
    [self enumerateRowsInSection:section usingBlock:^(NSUInteger row, FeedChooserItem *item) {
        if (select) {
            [self.tableView selectRowAtIndexPath:[NSIndexPath indexPathForRow:row inSection:section] animated:YES scrollPosition:UITableViewScrollPositionNone];
        } else {
            [self.tableView deselectRowAtIndexPath:[NSIndexPath indexPathForRow:row inSection:section] animated:YES];
        }
    }];
    
    [self updateControls];
}

- (void)updateControls {
    BOOL hasSelection = self.tableView.indexPathsForSelectedRows.count > 0;
    
    self.moveItem.enabled = hasSelection;
    self.deleteItem.enabled = hasSelection;
}

#pragma mark - Title delegate methods

- (void)didSelectTitleView:(UIButton *)sender {
    NSUInteger section = sender.tag;
    BOOL select = ![self isSelectionInSection:section];
    
    [self enumerateRowsInSection:section usingBlock:^(NSUInteger row, FeedChooserItem *item) {
        [self select:select section:section];
    }];
}

#pragma mark - Target/action methods

- (NSString *)sortIconName {
    if (self.ascending) {
        return @"barbutton_sort_asc.png";
    } else {
        return @"barbutton_sort_desc.png";
    }
}

- (void)sort:(FeedChooserSort)sort {
    NSArray *identifiers = [self selectedItemIdentifiers];
    
    // Rebuild to get the new info values (which also quietly sorts in the old order), then sort with animation:
    [self rebuildItemsAnimated:NO];
    self.sort = sort;
    [self sortItemsAnimated:YES];
    
    [self selectItemsWithIdentifiers:identifiers animated:NO];
}

- (void)showOptionsMenu {
    MenuViewController *viewController = [MenuViewController new];
    BOOL isMute = self.operation == FeedChooserOperationMuteSites;
    
    [viewController addTitle:@"Name" iconName:[self sortIconName] selectionShouldDismiss:YES handler:^{
        [self sort:FeedChooserSortName];
    }];
    
    [viewController addTitle:@"Subscribers" iconName:[self sortIconName] selectionShouldDismiss:YES handler:^{
        [self sort:FeedChooserSortSubscribers];
    }];
    
    [viewController addTitle:@"Stories per Month" iconName:[self sortIconName] selectionShouldDismiss:YES handler:^{
        [self sort:FeedChooserSortFrequency];
    }];
    
    [viewController addTitle:@"Most Recent Story" iconName:[self sortIconName] selectionShouldDismiss:YES handler:^{
        [self sort:FeedChooserSortRecency];
    }];
    
    [viewController addTitle:@"Number of Opens" iconName:[self sortIconName] selectionShouldDismiss:YES handler:^{
        [self sort:FeedChooserSortOpens];
    }];
    
    viewController.checkedRow = self.sort;
    
    [viewController addSegmentedControlWithTitles:@[@"Ascending", @"Descending"] selectIndex:self.ascending ? 0 : 1 selectionShouldDismiss:YES handler:^(NSUInteger selectedIndex) {
        NSArray *identifiers = [self selectedItemIdentifiers];
        self.ascending = selectedIndex == 0;
        [self sortItemsAnimated:YES];
        [self selectItemsWithIdentifiers:identifiers animated:NO];
    }];
    
    [viewController addSegmentedControlWithTitles:@[@"Nested", @"Flat"] selectIndex:self.flat ? 1 : 0 selectionShouldDismiss:YES handler:^(NSUInteger selectedIndex) {
        NSArray *identifiers = [self selectedItemIdentifiers];
        self.flat = selectedIndex == 1;
        [self rebuildItemsAnimated:YES];
        [self selectItemsWithIdentifiers:identifiers animated:NO];
    }];
    
    [viewController addTitle:isMute ? @"Mute All" : @"Select All" iconName:isMute ? @"mute_feed_off.png" : @"barbutton_selection.png" selectionShouldDismiss:YES handler:^{
        [self enumerateSectionsUsingBlock:^(NSUInteger section, FeedChooserItem *folder) {
            [self select:YES section:section];
        }];
    }];
    
    [viewController addTitle:isMute ? @"Unmute All" : @"Select None" iconName:isMute ? @"mute_feed_on.png" : @"barbutton_selection_off.png" selectionShouldDismiss:YES handler:^{
        [self enumerateSectionsUsingBlock:^(NSUInteger section, FeedChooserItem *folder) {
            [self select:NO section:section];
        }];
    }];
    
    [viewController showFromNavigationController:self.navigationController barButtonItem:self.optionsItem];
}

- (void)performMoveToFolder:(FeedChooserItem *)toFolder {
    [MBProgressHUD hideHUDForView:self.view animated:YES];
    MBProgressHUD *HUD = [MBProgressHUD showHUDAddedTo:self.view animated:YES];
    HUD.labelText = @"Moving...";
    
    NSArray *selection = self.tableView.indexPathsForSelectedRows;
    NSMutableArray *feedsByFolder = [NSMutableArray arrayWithCapacity:selection.count];
    
    for (NSIndexPath *indexPath in selection) {
        FeedChooserItem *fromFolder = self.sections[indexPath.section];
        FeedChooserItem *item = fromFolder.contents[indexPath.row];
        
        [feedsByFolder addObject:@[item.identifier, fromFolder.identifier]];
    }
    
    NSString *urlString = [NSString stringWithFormat:@"%@/reader/move_feeds_by_folder_to_folder", self.appDelegate.url];
    NSURL *url = [NSURL URLWithString:urlString];
    ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
    [request setPostValue:feedsByFolder.JSONRepresentation forKey:@"feeds_by_folder"];
    [request setPostValue:toFolder.identifier forKey:@"to_folder"];
    [request setCompletionBlock:^(void) {
        HUD.labelText = @"Reloading...";
        [self.appDelegate reloadFeedsView:YES];
    }];
    request.didFailSelector = @selector(finishedWithError:);
    request.timeOutSeconds = 30;
    [request startAsynchronous];
}

- (void)showMoveMenu {
    MenuViewController *viewController = [MenuViewController new];
    
    for (FeedChooserItem *folder in self.folders) {
        NSString *title = folder.title;
        NSString *iconName = @"menu_icn_move.png";
        
        if (!title.length) {
            title = @"Top Level";
            iconName = @"menu_icn_all.png";
        } else {
            NSArray *components = [title componentsSeparatedByString:@" - "];
            title = components.lastObject;
            for (NSUInteger idx = 0; idx < components.count; idx++) {
                title = [@"\t" stringByAppendingString:title];
            }
        }
        
        [viewController addTitle:title iconName:iconName selectionShouldDismiss:YES handler:^{
            [self performMoveToFolder:folder];
        }];
    }
    
    [viewController showFromNavigationController:self.navigationController barButtonItem:self.moveItem];
}

- (void)performDeleteFeeds {
    [MBProgressHUD hideHUDForView:self.view animated:YES];
    MBProgressHUD *HUD = [MBProgressHUD showHUDAddedTo:self.view animated:YES];
    HUD.labelText = @"Deleting...";
    
    NSArray *selection = self.tableView.indexPathsForSelectedRows;
    NSMutableArray *feedsByFolder = [NSMutableArray arrayWithCapacity:selection.count];
    
    for (NSIndexPath *indexPath in selection) {
        FeedChooserItem *fromFolder = self.sections[indexPath.section];
        FeedChooserItem *item = fromFolder.contents[indexPath.row];
        
        [feedsByFolder addObject:@[item.identifier, fromFolder.identifier]];
    }
    
    NSString *urlString = [NSString stringWithFormat:@"%@/reader/delete_feeds_by_folder", self.appDelegate.url];
    NSURL *url = [NSURL URLWithString:urlString];
    ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
    [request setPostValue:feedsByFolder.JSONRepresentation forKey:@"feeds_by_folder"];
    [request setCompletionBlock:^(void) {
        HUD.labelText = @"Reloading...";
        [self.appDelegate reloadFeedsView:YES];
    }];
    request.didFailSelector = @selector(finishedWithError:);
    request.timeOutSeconds = 30;
    [request startAsynchronous];
}

- (void)deleteFeeds {
    MenuViewController *viewController = [MenuViewController new];
    NSUInteger count = self.tableView.indexPathsForSelectedRows.count;
    NSString *title = count == 1 ? @"Delete selected site?" : [NSString stringWithFormat:@"Delete %@ sites?", @(count)];
    
    [viewController addTitle:title iconName:@"menu_icn_delete.png" selectionShouldDismiss:YES handler:^{
        [self performDeleteFeeds];
    }];
    
    [viewController showFromNavigationController:self.navigationController barButtonItem:self.deleteItem];
}

- (void)finishedLoadingFeedsNotification:(NSNotification *)note {
    if (self.operation != FeedChooserOperationMuteSites) {
        [self updateDictFolders];
    }
    
    [self rebuildItemsAnimated:YES];
    [MBProgressHUD hideHUDForView:self.view animated:YES];
}

- (void)updateDictFolders {
    NSMutableDictionary *folders = [self.appDelegate.dictFolders mutableCopy];
    NSDictionary *everything = folders[@"everything"];
    
    [folders removeObjectForKey:@"everything"];
    [folders setObject:everything forKey:@" "];
    
    self.dictFolders = folders;
}

- (void)performSaveActiveFeeds {
    [MBProgressHUD hideHUDForView:self.view animated:YES];
    MBProgressHUD *HUD = [MBProgressHUD showHUDAddedTo:self.view animated:YES];
    HUD.labelText = @"Updating...";
    
    NSString *urlString = [NSString stringWithFormat:@"%@/reader/save_feed_chooser", self.appDelegate.url];
    NSURL *url = [NSURL URLWithString:urlString];
    ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
    NSArray *mutedIndexPaths = self.tableView.indexPathsForSelectedRows;
    
    [self enumerateAllRowsUsingBlock:^(NSIndexPath *indexPath, FeedChooserItem *item) {
        if (![mutedIndexPaths containsObject:indexPath]) {
            [request addPostValue:item.identifier forKey:@"approved_feeds"];
        }
    }];
    
    [request setCompletionBlock:^(void) {
        [self.appDelegate reloadFeedsView:YES];
        [self dismissViewControllerAnimated:YES completion:nil];
    }];
    request.didFailSelector = @selector(finishedWithError:);
    request.timeOutSeconds = 30;
    [request startAsynchronous];
}

- (BOOL)didChangeActiveFeeds {
    __block BOOL didChange = NO;
    NSArray *inactiveFeedsIdentifiers = self.inactiveFeeds.allKeys;
    NSArray *selectedIndexPaths = self.tableView.indexPathsForSelectedRows;
    
    [self enumerateAllRowsUsingBlock:^(NSIndexPath *indexPath, FeedChooserItem *item) {
        BOOL wasInactive = [inactiveFeedsIdentifiers containsObject:item.identifierString];
        BOOL isInactive = ![selectedIndexPaths containsObject:indexPath];
        
        if (wasInactive != isInactive) {
            didChange = YES;
        }
    }];
    
    return didChange;
}

- (void)done {
    if (self.operation == FeedChooserOperationMuteSites && [self didChangeActiveFeeds]) {
        [self performSaveActiveFeeds];
    } else {
        [self dismissViewControllerAnimated:YES completion:nil];
    }
}

- (void)cancel {
    [self dismissViewControllerAnimated:YES completion:nil];
}

#pragma mark - Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
    if (self.appDelegate.hasNoSites) {
        return 0;
    } else {
        return self.sections.count;
    }
}

- (NSString *)tableView:(UITableView *)tableView titleForHeaderInSection:(NSInteger)section {
    FeedChooserItem *item = self.sections[section];
    return item.title;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
    FeedChooserItem *item = self.sections[section];
    NSArray *contents = item.contents;
    return contents.count;
}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
    static NSString *CellIndentifier = @"FeedChooserCell";
    FeedChooserViewCell *cell = [tableView dequeueReusableCellWithIdentifier:CellIndentifier];
    
    if (!cell) {
        cell = [[FeedChooserViewCell alloc] initWithStyle:UITableViewCellStyleValue1 reuseIdentifier:CellIndentifier];
    }
    
    FeedChooserItem *item = [self itemForIndexPath:indexPath];
    
    cell.isMuteOperation = self.operation == FeedChooserOperationMuteSites;
    cell.textLabel.text = item.title;
    cell.detailTextLabel.text = [item detailForSort:self.sort];
    cell.imageView.image = item.icon;
    
    if (self.operation == FeedChooserOperationMuteSites) {
        UIImage *image = [UIImage imageNamed:@"mute_feed_on.png"];
        UIImage *highlightedImage = [UIImage imageNamed:@"mute_feed_off.png"];
        
        cell.accessoryView = [[UIImageView alloc] initWithImage:image highlightedImage:highlightedImage];
    } else {
        cell.accessoryView = nil;
    }
    
    return cell;
}

- (UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section {
    CGRect rect = CGRectMake(0.0, 0.0, tableView.frame.size.width, kFolderTitleHeight);
    FeedChooserTitleView *titleView = [[FeedChooserTitleView alloc] initWithFrame:rect];
    FeedChooserItem *item = self.sections[section];
    
    titleView.delegate = self;
    titleView.section = section;
    titleView.title = item.title;
    
    return titleView;
}

- (CGFloat)tableView:(UITableView *)tableView heightForHeaderInSection:(NSInteger)section {
    FeedChooserItem *item = self.sections[section];
    
    return !item.title.length ? 0.0 : kFolderTitleHeight;
}

- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {
    return kTableViewRowHeight;
}

- (NSArray *)sectionIndexTitlesForTableView:(UITableView *)theTableView {
    return self.indexTitles;
}

- (NSInteger)tableView:(UITableView *)theTableView sectionForSectionIndexTitle:(NSString *)indexTitle atIndex:(NSInteger)indexIndex {
    return indexIndex;
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
    [self updateControls];
}

- (void)tableView:(UITableView *)tableView didDeselectRowAtIndexPath:(NSIndexPath *)indexPath {
    [self updateControls];
}

@end