element-ios/Riot/Modules/People/PeopleViewController.m

/*
 Copyright 2017 Vector Creations Ltd
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at
 
 http://www.apache.org/licenses/LICENSE-2.0
 
 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 */

#import <Contacts/Contacts.h>
#import "PeopleViewController.h"

#import "UIViewController+RiotSearch.h"

#import "RageShakeManager.h"

#import "RecentsDataSource.h"
#import "RecentTableViewCell.h"
#import "InviteRecentTableViewCell.h"

#import "ContactTableViewCell.h"

#import "Riot-Swift.h"

@interface PeopleViewController ()
{
    NSInteger          directRoomsSectionNumber;
    
    ContactsDataSource *contactsDataSource;
    NSInteger          contactsSectionNumber;
    
    RecentsDataSource *recentsDataSource;
}

@end

@implementation PeopleViewController

- (void)finalizeInit
{
    [super finalizeInit];
    
    directRoomsSectionNumber = 0;
    contactsSectionNumber = 0;
    
    self.screenName = @"People";
}

- (void)viewDidLoad
{
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
    
    self.view.accessibilityIdentifier = @"PeopleVCView";
    self.recentsTableView.accessibilityIdentifier = @"PeopleVCTableView";
    
    // Tag the recents table with the its recents data source mode.
    // This will be used by the shared RecentsDataSource instance for sanity checks (see UITableViewDataSource methods).
    self.recentsTableView.tag = RecentsDataSourceModePeople;
    
    // Add the (+) button programmatically
    plusButtonImageView = [self vc_addFABWithImage:[UIImage imageNamed:@"people_floating_action"]
                                            target:self
                                            action:@selector(onPlusButtonPressed)];
    
    // Register table view cell for contacts.
    [self.recentsTableView registerClass:ContactTableViewCell.class forCellReuseIdentifier:ContactTableViewCell.defaultReuseIdentifier];
    
    // Change the table data source. It must be the people view controller itself.
    self.recentsTableView.dataSource = self;
    
    self.enableStickyHeaders = YES;
}

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

- (void)destroy
{
    contactsDataSource.delegate = nil;
    [contactsDataSource destroy];
    contactsDataSource = nil;
    
    [super destroy];
}

- (void)viewWillAppear:(BOOL)animated
{
    [super viewWillAppear:animated];

    if (BuildSettings.allowLocalContactsAccess)
    {
        // Check whether the access to the local contacts has not been already asked
        // and check that the user has decided to use or not to use an identity server
        if ([CNContactStore authorizationStatusForEntityType:CNEntityTypeContacts] == CNAuthorizationStatusNotDetermined
            || !contactsDataSource.mxSession.hasAccountDataIdentityServerValue)
        {
            // Allow by default the local contacts sync in order to discover matrix users.
            // This setting change will trigger the loading of the local contacts, which will automatically
            // ask user permission to access their local contacts.
            [MXKAppSettings standardAppSettings].syncLocalContacts = YES;
        }
        
        // Refresh the local contacts list.
        [[MXKContactManager sharedManager] refreshLocalContacts];
    }
    
    [AppDelegate theDelegate].masterTabBarController.navigationItem.title = NSLocalizedStringFromTable(@"title_people", @"Vector", nil);
    [AppDelegate theDelegate].masterTabBarController.tabBar.tintColor = ThemeService.shared.theme.tintColor;
    
    if (recentsDataSource)
    {
        // Take the lead on the shared data source.
        recentsDataSource.areSectionsShrinkable = NO;
        [recentsDataSource setDelegate:self andRecentsDataSourceMode:RecentsDataSourceModePeople];
    }
}

#pragma mark - 

- (void)displayList:(MXKRecentsDataSource *)listDataSource
{
    [super displayList:listDataSource];
    
    // Change the table data source. It must be the people view controller itself.
    self.recentsTableView.dataSource = self;
    
    // Keep a ref on the recents data source
    if ([listDataSource isKindOfClass:RecentsDataSource.class])
    {
        recentsDataSource = (RecentsDataSource*)listDataSource;
    }

    if (BuildSettings.allowLocalContactsAccess)
    {
        if (!contactsDataSource)
        {
            // Prepare its contacts data source
            contactsDataSource = [[ContactsDataSource alloc] initWithMatrixSession:listDataSource.mxSession];
            contactsDataSource.contactCellAccessoryImage = [[UIImage imageNamed: @"disclosure_icon"] vc_tintedImageUsingColor:ThemeService.shared.theme.textSecondaryColor];
            contactsDataSource.delegate = self;
        }
    }
}

#pragma mark - MXKDataSourceDelegate

- (Class<MXKCellRendering>)cellViewClassForCellData:(MXKCellData*)cellData
{
    if ([cellData isKindOfClass:MXKContact.class])
    {
        return ContactTableViewCell.class;
    }
    
    return [super cellViewClassForCellData:cellData];
}

#pragma mark - UITableView data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView
{
    // Retrieve the current number of sections related to the direct rooms.
    // Sanity check: check whether the recents data source is correctly configured.
    directRoomsSectionNumber = 0;
    
    if (recentsDataSource.recentsDataSourceMode == RecentsDataSourceModePeople)
    {
        directRoomsSectionNumber = [self.dataSource numberOfSectionsInTableView:self.recentsTableView];
    }
    
    // Retrieve the current number of sections related to the contacts
    contactsSectionNumber = [contactsDataSource numberOfSectionsInTableView:self.recentsTableView];
    
    return (directRoomsSectionNumber + contactsSectionNumber);
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section
{
    NSInteger count = 0;
    
    if (section < directRoomsSectionNumber)
    {
        count = [self.dataSource tableView:tableView numberOfRowsInSection:section];
    }
    else
    {
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            count = [contactsDataSource tableView:tableView numberOfRowsInSection:section];
        }
    }
    
    return count;
}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
    NSInteger section = indexPath.section;
    
    if (section < directRoomsSectionNumber)
    {
        return [self.dataSource tableView:tableView cellForRowAtIndexPath:indexPath];
    }
    else
    {
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            return [contactsDataSource tableView:tableView cellForRowAtIndexPath:[NSIndexPath indexPathForRow:indexPath.row inSection:section]];
        }
    }
    
    // Return a fake cell to prevent app from crashing.
    return [[UITableViewCell alloc] init];
}

- (BOOL)tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *)indexPath
{
    NSInteger section = indexPath.section;
    
    if (section < directRoomsSectionNumber)
    {
        return [self.dataSource tableView:tableView canEditRowAtIndexPath:indexPath];
    }
    else
    {
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            return [contactsDataSource tableView:tableView canEditRowAtIndexPath:[NSIndexPath indexPathForRow:indexPath.row inSection:section]];
        }
    }
    
    return NO;
}

#pragma mark - UITableView delegate

- (CGFloat)tableView:(UITableView *)tableView heightForHeaderInSection:(NSInteger)section
{
    if (section >= directRoomsSectionNumber)
    {
        // Let the contact dataSource provide the height of the section header.
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            return [contactsDataSource heightForHeaderInSection:section];
        }
        else
        {
            return 0.0;
        }
    }
    
    return [super tableView:tableView heightForHeaderInSection:section];
}

- (UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section
{
    if (section >= directRoomsSectionNumber)
    {
        // Let the contact dataSource provide the section header.
        CGRect frame = [tableView rectForHeaderInSection:section];
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            UIView *sectionHeader = [contactsDataSource viewForHeaderInSection:section withFrame:frame];
            sectionHeader.tag = section + directRoomsSectionNumber;
            
            if (self.enableStickyHeaders)
            {
                while (sectionHeader.gestureRecognizers.count)
                {
                    UIGestureRecognizer *gestureRecognizer = sectionHeader.gestureRecognizers.lastObject;
                    [sectionHeader removeGestureRecognizer:gestureRecognizer];
                }
                
                // Handle tap gesture
                UITapGestureRecognizer *tap = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(didTapOnSectionHeader:)];
                [tap setNumberOfTouchesRequired:1];
                [tap setNumberOfTapsRequired:1];
                [sectionHeader addGestureRecognizer:tap];
            }
            
            return sectionHeader;
        }
        else
        {
            return nil;
        }
    }
    
    return [super tableView:tableView viewForHeaderInSection:section];
}

- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath
{
    NSInteger section = indexPath.section;
    if (section >= directRoomsSectionNumber)
    {
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            if ([contactsDataSource contactAtIndexPath:[NSIndexPath indexPathForRow:indexPath.row inSection:section]])
            {
                // Return the default height of the contact cell
                return 74.0;
            }
            
            return 50;
        }
        else
        {
            return 0.0;
        }
    }
    
    return [super tableView:tableView heightForRowAtIndexPath:indexPath];
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
    NSInteger section = indexPath.section;
    if (section >= directRoomsSectionNumber)
    {
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            MXKContact *mxkContact = [contactsDataSource contactAtIndexPath:[NSIndexPath indexPathForRow:indexPath.row inSection:section]];
            
            if (mxkContact)
            {
                [[AppDelegate theDelegate].masterTabBarController selectContact:mxkContact];
                
                // Keep selected the cell by default.
                return;
            }
        }
        else
        {
            [tableView deselectRowAtIndexPath:indexPath animated:NO];
            return;
        }
    }
    
    return [super tableView:tableView didSelectRowAtIndexPath:indexPath];
}

#pragma mark - Override RecentsViewController

- (UIView *)tableView:(UITableView *)tableView viewForStickyHeaderInSection:(NSInteger)section
{
    CGRect frame = [tableView rectForHeaderInSection:section];
    frame.size.height = self.stickyHeaderHeight;
    
    if (section >= directRoomsSectionNumber)
    {
        // Let the contact dataSource provide this header.
        section -= directRoomsSectionNumber;
        if (section < contactsSectionNumber)
        {
            return [contactsDataSource viewForStickyHeaderInSection:section withFrame:frame];
        }
    }
    else if (recentsDataSource)
    {
        return [recentsDataSource viewForStickyHeaderInSection:section withFrame:frame];
    }
    
    return nil;
}

- (void)refreshCurrentSelectedCell:(BOOL)forceVisible
{
    // Check whether the recents data source is correctly configured.
    if (recentsDataSource.recentsDataSourceMode != RecentsDataSourceModePeople)
    {
        return;
    }
    
    // Update here the index of the current selected cell (if any) - Useful in landscape mode with split view controller.
    NSIndexPath *currentSelectedCellIndexPath = nil;
    MasterTabBarController *masterTabBarController = [AppDelegate theDelegate].masterTabBarController;
    if (masterTabBarController.currentContactDetailViewController)
    {
        // Look for the rank of this selected contact
        currentSelectedCellIndexPath = [contactsDataSource cellIndexPathWithContact:masterTabBarController.selectedContact];
        
        if (currentSelectedCellIndexPath)
        {
            // Select the right row
            currentSelectedCellIndexPath = [NSIndexPath indexPathForRow:currentSelectedCellIndexPath.row inSection:(directRoomsSectionNumber + currentSelectedCellIndexPath.section)];
            [self.recentsTableView selectRowAtIndexPath:currentSelectedCellIndexPath animated:YES scrollPosition:UITableViewScrollPositionNone];
            
            if (forceVisible)
            {
                // Scroll table view to make the selected row appear at second position
                NSInteger topCellIndexPathRow = currentSelectedCellIndexPath.row ? currentSelectedCellIndexPath.row - 1: currentSelectedCellIndexPath.row;
                NSIndexPath* indexPath = [NSIndexPath indexPathForRow:topCellIndexPathRow inSection:currentSelectedCellIndexPath.section];
                [self.recentsTableView scrollToRowAtIndexPath:indexPath atScrollPosition:UITableViewScrollPositionTop animated:NO];
            }
        }
        else
        {
            NSIndexPath *indexPath = [self.recentsTableView indexPathForSelectedRow];
            if (indexPath)
            {
                [self.recentsTableView deselectRowAtIndexPath:indexPath animated:NO];
            }
        }
    }
    else
    {
        [super refreshCurrentSelectedCell:forceVisible];
    }
}

- (void)onPlusButtonPressed
{
    //  directly open the new chat screen
    [self performSegueWithIdentifier:@"presentStartChat" sender:self];
}

#pragma mark -

- (void)scrollToNextRoomWithMissedNotifications
{
    // Check whether the recents data source is correctly configured.
    if (recentsDataSource.recentsDataSourceMode == RecentsDataSourceModePeople)
    {
        [self scrollToTheTopTheNextRoomWithMissedNotificationsInSection:recentsDataSource.conversationSection];
    }
}

#pragma mark - UISearchBarDelegate

- (void)searchBar:(UISearchBar *)searchBar textDidChange:(NSString *)searchText
{
    // Apply filter on contact source
    [contactsDataSource searchWithPattern:searchText forceReset:NO];
    
    [super searchBar:searchBar textDidChange:searchText];
}

- (void)searchBarCancelButtonClicked:(UISearchBar *)searchBar
{
    // Reset filtering
    [contactsDataSource searchWithPattern:nil forceReset:NO];
    
    [super searchBarCancelButtonClicked:searchBar];
}

#pragma mark - Empty view management

- (void)updateEmptyView
{
    [self.emptyView fillWith:[self emptyViewArtwork]
                       title:NSLocalizedStringFromTable(@"people_empty_view_title", @"Vector", nil)
             informationText:NSLocalizedStringFromTable(@"people_empty_view_information", @"Vector", nil)];
}

- (UIImage*)emptyViewArtwork
{
    if (ThemeService.shared.isCurrentThemeDark)
    {
        return [UIImage imageNamed:@"people_empty_screen_artwork_dark"];
    }
    else
    {
        return [UIImage imageNamed:@"people_empty_screen_artwork"];
    }
}

- (BOOL)shouldShowEmptyView
{
    // Do not present empty screen while searching
    if (recentsDataSource.searchPatternsList.count)
    {
        return NO;
    }
    
    return [self totalItemCounts] == 0;
}

// Total items to display on the screen
- (NSUInteger)totalItemCounts
{
    return recentsDataSource.invitesCellDataArray.count
    + recentsDataSource.conversationCellDataArray.count
    + recentsDataSource.peopleCellDataArray.count
    + [self numberOfContactsInContactsDataSource];
}

- (NSUInteger)numberOfContactsInContactsDataSource
{
    BOOL areLocalContactsAccessAuthorized = [CNContactStore authorizationStatusForEntityType:CNEntityTypeContacts] == CNAuthorizationStatusAuthorized;
    
    NSInteger nbOfItemsInContactDataSource = 0;
    
    for (NSInteger i = 0; i < contactsSectionNumber; i++)
    {
        nbOfItemsInContactDataSource += [contactsDataSource tableView:self.recentsTableView numberOfRowsInSection:i];
    }
    
    NSInteger numberOfContactsInContactsDataSource;
    
    // No local contacts to show and no search in directory
    if (!areLocalContactsAccessAuthorized && contactsSectionNumber == 1 && nbOfItemsInContactDataSource <= 1)
    {
        numberOfContactsInContactsDataSource = 0;
    }
    else
    {
        numberOfContactsInContactsDataSource = nbOfItemsInContactDataSource;
    }
    
    return numberOfContactsInContactsDataSource;
}

@end