/*
 Copyright 2015 OpenMarket 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 "HomeViewController.h"

#import "MatrixSDKHandler.h"
#import "AppDelegate.h"
#import "PublicRoomTableCell.h"

#import "RageShakeManager.h"

@interface HomeViewController () {
    NSArray *publicRooms;
    
    // List of public room names to highlight in displayed list
    NSArray* highlightedPublicRooms;
    
    // Search in public room
    UISearchBar     *publicRoomsSearchBar;
    NSMutableArray  *filteredPublicRooms;
    BOOL             searchBarShouldEndEditing;
    UIView          *savedTableHeaderView;
    
    NSString *homeServerSuffix;
}

@property (weak, nonatomic) IBOutlet UITableView *publicRoomsTable;

@property (weak, nonatomic) IBOutlet UILabel *roomCreationSectionLabel;
@property (weak, nonatomic) IBOutlet UIView *roomCreationSectionView;
@property (weak, nonatomic) IBOutlet UILabel *roomNameLabel;
@property (weak, nonatomic) IBOutlet UILabel *roomAliasLabel;
@property (weak, nonatomic) IBOutlet UILabel *participantsLabel;
@property (weak, nonatomic) IBOutlet UITextField *roomNameTextField;
@property (weak, nonatomic) IBOutlet UITextField *roomAliasTextField;
@property (weak, nonatomic) IBOutlet UITextField *participantsTextField;
@property (weak, nonatomic) IBOutlet UISegmentedControl *roomVisibilityControl;
@property (weak, nonatomic) IBOutlet UIButton *createRoomBtn;

@property (weak, nonatomic) IBOutlet UILabel *joinRoomSectionLabel;
@property (weak, nonatomic) IBOutlet UITextField *joinRoomAliasTextField;
@property (weak, nonatomic) IBOutlet UIButton *joinRoomBtn;

- (IBAction)onButtonPressed:(id)sender;

@end

@implementation HomeViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    // Do any additional setup after loading the view, typically from a nib.
    _roomCreationSectionLabel.backgroundColor = [UIColor colorWithRed:0.9 green:0.9 blue:0.9 alpha:1.0];
    _createRoomBtn.enabled = NO;
    
    _joinRoomSectionLabel.backgroundColor = [UIColor colorWithRed:0.9 green:0.9 blue:0.9 alpha:1.0];
    _joinRoomBtn.enabled = NO;
    
    // Set rageShake handler
    self.rageShakeManager = [RageShakeManager sharedManager];
    
    // Init
    publicRooms = nil;
    highlightedPublicRooms = @[@"#matrix:matrix.org", @"#matrix-dev:matrix.org", @"#matrix-fr:matrix.org"]; // Add here a room name to highlight its display in public room list
}

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

- (void)dealloc{
    publicRooms = nil;
    highlightedPublicRooms = nil;
    
    publicRoomsSearchBar = nil;
    filteredPublicRooms = nil;
    savedTableHeaderView = nil;
}

- (void)viewWillAppear:(BOOL)animated {
    [super viewWillAppear:animated];
    
    // Ensure to display room creation section
    [self.tableView scrollRectToVisible:_roomCreationSectionLabel.frame animated:NO];
    
    if ([MatrixSDKHandler sharedHandler].status != MatrixSDKHandlerStatusLoggedOut) {
        homeServerSuffix = [NSString stringWithFormat:@":%@",[MatrixSDKHandler sharedHandler].homeServer];
        // Update alias placeholder
        _roomAliasTextField.placeholder = [NSString stringWithFormat:@"(e.g. #foo%@)", homeServerSuffix];
        // Refresh listed public rooms
        [self refreshPublicRooms];
    }
    
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(onTextFieldChange:) name:UITextFieldTextDidChangeNotification object:nil];
}

- (void)viewWillDisappear:(BOOL)animated {
    [super viewWillDisappear:animated];
    // Leave potential search session
    if (publicRoomsSearchBar) {
        [self searchBarCancelButtonClicked:publicRoomsSearchBar];
    }
    
    [[NSNotificationCenter defaultCenter] removeObserver:self name:UITextFieldTextDidChangeNotification object:nil];
}

#pragma mark - Internals

- (void)refreshPublicRooms {
    // Retrieve public rooms
    [[MatrixSDKHandler sharedHandler].mxRestClient publicRooms:^(NSArray *rooms){
        publicRooms = [rooms sortedArrayUsingComparator:^NSComparisonResult(id a, id b) {
            
            MXPublicRoom *firstRoom =  (MXPublicRoom*)a;
            MXPublicRoom *secondRoom = (MXPublicRoom*)b;
            
            // Compare member count
            if (firstRoom.numJoinedMembers < secondRoom.numJoinedMembers) {
                return NSOrderedDescending;
            } else if (firstRoom.numJoinedMembers > secondRoom.numJoinedMembers) {
                return NSOrderedAscending;
            } else {
                // Alphabetic order
                return [firstRoom.displayname compare:secondRoom.displayname options:NSCaseInsensitiveSearch];
            }
        }];
        [_publicRoomsTable reloadData];
    }
                                                    failure:^(NSError *error){
                                                        NSLog(@"[HomeVC] Failed to get public rooms: %@", error);
                                                        //Alert user
                                                        [[AppDelegate theDelegate] showErrorAsAlert:error];
                                                    }];
    
}

- (void)search:(id)sender {
    if (!publicRoomsSearchBar) {
        // Check whether there are data in which search
        if (publicRooms.count) {
            // Create search bar
            publicRoomsSearchBar = [[UISearchBar alloc] initWithFrame:CGRectMake(0, 0, self.view.frame.size.width, 44)];
            publicRoomsSearchBar.showsCancelButton = YES;
            publicRoomsSearchBar.returnKeyType = UIReturnKeyDone;
            publicRoomsSearchBar.delegate = self;
            [publicRoomsSearchBar becomeFirstResponder];
            publicRoomsSearchBar.autoresizingMask = UIViewAutoresizingFlexibleWidth;
            // Hide table header during search session
            savedTableHeaderView = self.tableView.tableHeaderView;
            self.tableView.tableHeaderView = nil;
            // Reload table in order to display search bar as section header
            [self.tableView scrollToRowAtIndexPath:[NSIndexPath indexPathForRow:0 inSection:0] atScrollPosition:UITableViewScrollPositionTop animated:YES];
            [self.tableView reloadData];
            
        }
    } else {
        [self searchBarCancelButtonClicked: publicRoomsSearchBar];
    }
}

- (void)dismissKeyboard {
    // Hide the keyboard
    [_roomNameTextField resignFirstResponder];
    [_roomAliasTextField resignFirstResponder];
    [_participantsTextField resignFirstResponder];
    [_joinRoomAliasTextField resignFirstResponder];
}

- (NSString*)alias {
    // Extract alias name from alias text field
    NSString *alias = _roomAliasTextField.text;
    if (alias.length > 1) {
        // Remove '#' character
        alias = [alias substringFromIndex:1];
        // Remove homeserver
        NSRange range = [alias rangeOfString:homeServerSuffix];
        if (range.location == NSNotFound) {
            NSLog(@"[HomeVC] Wrong room alias has been set (%@)", _roomAliasTextField.text);
            alias = nil;
        } else {
            alias = [alias stringByReplacingCharactersInRange:range withString:@""];
        }
    }
    
    if (! alias.length) {
        alias = nil;
    }
    
    return alias;
}

- (NSArray*)participantsList {
    NSMutableArray *participants = [NSMutableArray array];
    
    if (_participantsTextField.text.length) {
        NSArray *components = [_participantsTextField.text componentsSeparatedByString:@";"];
        
        for (NSString *component in components) {
            // Remove white space from both ends
            NSString *user = [component stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];
            if (user.length > 1 && [user hasPrefix:@"@"]) {
                [participants addObject:user];
            }
        }
    }
    
    if (participants.count == 0) {
        participants = nil;
    }
    
    return participants;
}

#pragma mark - UITextField delegate

- (void)onTextFieldChange:(NSNotification *)notif {
    // Update Create Room button
    NSString *roomName = _roomNameTextField.text;
    NSString *roomAlias = _roomAliasTextField.text;
    NSString *participants = _participantsTextField.text;
    
    if (roomName.length || roomAlias.length || participants.length) {
        _createRoomBtn.enabled = YES;
    } else {
        _createRoomBtn.enabled = NO;
    }
    
    // Update Join Room button
    _joinRoomBtn.enabled = (_joinRoomAliasTextField.text.length != 0);
}

- (void)textFieldDidBeginEditing:(UITextField *)textField {
    if (textField == _participantsTextField) {
        if (textField.text.length == 0) {
            textField.text = @"@";
        }
    } else if (textField == _roomAliasTextField || textField == _joinRoomAliasTextField) {
        if (textField.text.length == 0) {
            textField.text = @"#";
        }
    }
}

- (void)textFieldDidEndEditing:(UITextField *)textField {
    if (textField == _roomAliasTextField) {
        // Check whether homeserver suffix should be added
        NSRange range = [textField.text rangeOfString:@":"];
        if (range.location == NSNotFound) {
            textField.text = [textField.text stringByAppendingString:homeServerSuffix];
        }
        // Check whether the alias is valid
        if (!self.alias) {
            // reset text field
            textField.text = nil;
            [self onTextFieldChange:nil];
        }
    } else if (textField == _participantsTextField) {
        NSArray *participants = self.participantsList;
        textField.text = [participants componentsJoinedByString:@"; "];
    } else if (textField == _joinRoomAliasTextField) {
        if (textField.text.length > 1) {
            // Add homeserver suffix if none
            NSRange range = [textField.text rangeOfString:@":"];
            if (range.location == NSNotFound) {
                textField.text = [textField.text stringByAppendingString:homeServerSuffix];
            }
        } else {
            // reset text field
            textField.text = nil;
            [self onTextFieldChange:nil];
        }
    }
}

- (BOOL)textField:(UITextField *)textField shouldChangeCharactersInRange:(NSRange)range replacementString:(NSString *)string {
    // Auto complete participant IDs
    if (textField == _participantsTextField) {
        // Add @ if none
        if (!textField.text.length || textField.text.length == range.length) {
            if ([string hasPrefix:@"@"] == NO) {
                textField.text = [NSString stringWithFormat:@"@%@",string];
                // Update Create button status
                [self onTextFieldChange:nil];
                return NO;
            }
        } else if (range.location == textField.text.length) {
            if ([string isEqualToString:@";"]) {
                // Add '@' character
                textField.text = [textField.text stringByAppendingString:@"; @"];
                // Update Create button status
                [self onTextFieldChange:nil];
                return NO;
            }
        }
    } else if (textField == _roomAliasTextField) {
        // Add # if none
        if (!textField.text.length || textField.text.length == range.length) {
            if ([string hasPrefix:@"#"] == NO) {
                if ([string isEqualToString:@":"]) {
                    textField.text = [NSString stringWithFormat:@"#%@",homeServerSuffix];
                } else {
                    textField.text = [NSString stringWithFormat:@"#%@",string];
                }
                // Update Create button status
                [self onTextFieldChange:nil];
                return NO;
            }
        } else {
            // Add homeserver automatically when user adds ':' at the end
            if (range.location == textField.text.length && [string isEqualToString:@":"]) {
                textField.text = [textField.text stringByAppendingString:homeServerSuffix];
                // Update Create button status
                [self onTextFieldChange:nil];
                return NO;
            }
        }
    } else if (textField == _joinRoomAliasTextField) {
        // Add # if none
        if (!textField.text.length || textField.text.length == range.length) {
            if ([string hasPrefix:@"#"] == NO) {
                textField.text = [NSString stringWithFormat:@"#%@",string];
                // Update Create button status
                [self onTextFieldChange:nil];
                return NO;
            }
        }
    }
    return YES;
}

- (BOOL)textFieldShouldReturn:(UITextField*) textField {
    // "Done" key has been pressed
    [textField resignFirstResponder];
    return YES;
}

#pragma mark - Actions

- (IBAction)onButtonPressed:(id)sender {
    [self dismissKeyboard];
    
    if (sender == _createRoomBtn) {
        // Disable button to prevent multiple request
        _createRoomBtn.enabled = NO;
        
        NSString *roomName = _roomNameTextField.text;
        if (! roomName.length) {
            roomName = nil;
        }
        
        // Create new room
        MatrixSDKHandler *mxHandler = [MatrixSDKHandler sharedHandler];
        [mxHandler.mxRestClient createRoom:roomName
         visibility:(_roomVisibilityControl.selectedSegmentIndex == 0) ? kMXRoomVisibilityPublic : kMXRoomVisibilityPrivate
         roomAlias:self.alias
         topic:nil
         success:^(MXCreateRoomResponse *response) {
             // Check whether some users must be invited
             NSArray *invitedUsers = self.participantsList;
             for (NSString *userId in invitedUsers) {
                 [mxHandler.mxRestClient inviteUser:userId toRoom:response.roomId success:^{
                     NSLog(@"[HomeVC] %@ has been invited (roomId: %@)", userId, response.roomId);
                 } failure:^(NSError *error) {
                     NSLog(@"[HomeVC] %@ invitation failed (roomId: %@): %@", userId, response.roomId, error);
                     //Alert user
                     [[AppDelegate theDelegate] showErrorAsAlert:error];
                 }];
             }
             
             // Reset text fields
             _roomNameTextField.text = nil;
             _roomAliasTextField.text = nil;
             _participantsTextField.text = nil;
             // Open created room
             [[AppDelegate theDelegate].masterTabBarController showRoom:response.roomId];
         } failure:^(NSError *error) {
             _createRoomBtn.enabled = YES;
             NSLog(@"[HomeVC] Create room (%@ %@ (%@)) failed: %@", _roomNameTextField.text, self.alias, (_roomVisibilityControl.selectedSegmentIndex == 0) ? @"Public":@"Private", error);
             //Alert user
             [[AppDelegate theDelegate] showErrorAsAlert:error];
         }];
    } else if (sender == _joinRoomBtn) {
        // Disable button to prevent multiple request
        _joinRoomBtn.enabled = NO;
        
        NSString *roomAlias = _joinRoomAliasTextField.text;
        // Remove white space from both ends
        roomAlias = [roomAlias stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceCharacterSet]];
        
        // Check
        if (roomAlias.length) {
            [[MatrixSDKHandler sharedHandler].mxSession joinRoom:roomAlias success:^(MXRoom *room) {
                // Reset text fields
                _joinRoomAliasTextField.text = nil;
                // Show the room
                [[AppDelegate theDelegate].masterTabBarController showRoom:room.state.roomId];
            } failure:^(NSError *error) {
                _joinRoomBtn.enabled = YES;
                NSLog(@"[HomeVC] Failed to join room alias (%@): %@", roomAlias, error);
                //Alert user
                [[AppDelegate theDelegate] showErrorAsAlert:error];
            }];
        } else {
            // Reset text fields
            _joinRoomAliasTextField.text = nil;
        }
    }
}

#pragma mark - Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
    return 1;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
    if (filteredPublicRooms) {
        return filteredPublicRooms.count;
    }
    return publicRooms.count;
}

- (CGFloat)tableView:(UITableView *)tableView heightForHeaderInSection:(NSInteger)section {
    if (publicRoomsSearchBar) {
        return (publicRoomsSearchBar.frame.size.height + 40);
    }
    return 40;
}

- (UIView *)tableView:(UITableView *)tableView viewForHeaderInSection:(NSInteger)section {
    UIView *sectionHeader = [[UIView alloc] initWithFrame:[tableView rectForHeaderInSection:section]];
    sectionHeader.backgroundColor = [UIColor colorWithRed:0.9 green:0.9 blue:0.9 alpha:1.0];
    UILabel *sectionLabel = [[UILabel alloc] initWithFrame:CGRectMake(0, 0, sectionHeader.frame.size.width, 40)];
    sectionLabel.font = [UIFont boldSystemFontOfSize:16];
    sectionLabel.backgroundColor = [UIColor clearColor];
    [sectionHeader addSubview:sectionLabel];
    
    if (publicRooms) {
        NSString *homeserver = [MatrixSDKHandler sharedHandler].homeServerURL;
        if (homeserver.length) {
            sectionLabel.text = [NSString stringWithFormat:@" Public Rooms (at %@):", homeserver];
        } else {
            sectionLabel.text = @" Public Rooms:";
        }
        
        UIButton *searchButton = [UIButton buttonWithType:UIButtonTypeCustom];
        [searchButton setImage:[UIImage imageNamed:@"icon_search"] forState:UIControlStateNormal];
        [searchButton setImage:[UIImage imageNamed:@"icon_search"] forState:UIControlStateHighlighted];
        [searchButton addTarget:self action:@selector(search:) forControlEvents:UIControlEventTouchUpInside];
        searchButton.frame = CGRectMake(sectionLabel.frame.size.width - 45, 0, 40, 40);
        searchButton.autoresizingMask = (UIViewAutoresizingFlexibleLeftMargin | UIViewAutoresizingFlexibleRightMargin);
        [sectionHeader addSubview:searchButton];
        sectionHeader.userInteractionEnabled = YES;
        
        if (publicRoomsSearchBar) {
            CGRect frame = publicRoomsSearchBar.frame;
            frame.origin.y = 40;
            publicRoomsSearchBar.frame = frame;
            [sectionHeader addSubview:publicRoomsSearchBar];
        }
    } else {
        sectionLabel.text = @" No Public Rooms";
    }
    
    return sectionHeader;
}

- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {
    // Cell is larger for public room with topic
    MXPublicRoom *publicRoom;
    if (filteredPublicRooms) {
        publicRoom = [filteredPublicRooms objectAtIndex:indexPath.row];
    } else {
        publicRoom = [publicRooms objectAtIndex:indexPath.row];
    }
    
    if (publicRoom.topic) {
        return 60;
    }
    return 44;
}

- (UITableViewCell *)tableView:(UITableView *)aTableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
    PublicRoomTableCell *cell;
    PublicRoomWithTopicTableCell *cellWithTopic = nil;
    
    MXPublicRoom *publicRoom;
    if (filteredPublicRooms) {
        publicRoom = [filteredPublicRooms objectAtIndex:indexPath.row];
    } else {
        publicRoom = [publicRooms objectAtIndex:indexPath.row];
    }
    
    // Check whether this public room has topic
    if (publicRoom.topic) {
        cellWithTopic = [_publicRoomsTable dequeueReusableCellWithIdentifier:@"PublicRoomWithTopicCell" forIndexPath:indexPath];
        cellWithTopic.roomTopic.text = publicRoom.topic;
        cell = cellWithTopic;
    } else {
        cell = [_publicRoomsTable dequeueReusableCellWithIdentifier:@"PublicRoomCell" forIndexPath:indexPath];
    }
    
    // Set room display name
    cell.roomDisplayName.text = [publicRoom displayname];
    
    // Set member count
    if (publicRoom.numJoinedMembers > 1) {
        cell.memberCount.text = [NSString stringWithFormat:@"%lu users", (unsigned long)publicRoom.numJoinedMembers];
    } else if (publicRoom.numJoinedMembers == 1) {
        cell.memberCount.text = @"1 user";
    } else {
        cell.memberCount.text = nil;
    }
    
    // Highlight?
    if (cell.roomDisplayName.text && [highlightedPublicRooms indexOfObject:cell.roomDisplayName.text] != NSNotFound) {
        cell.roomDisplayName.font = [UIFont boldSystemFontOfSize:20];
        if (cellWithTopic) {
            cellWithTopic.roomTopic.font = [UIFont boldSystemFontOfSize:17];
        }
        cell.backgroundColor = [UIColor colorWithRed:1.0 green:1.0 blue:0.9 alpha:1.0];
    } else {
        cell.roomDisplayName.font = [UIFont systemFontOfSize:19];
        if (cellWithTopic) {
            cellWithTopic.roomTopic.font = [UIFont systemFontOfSize:16];
        }
        cell.backgroundColor = [UIColor clearColor];
    }
    
    return cell;
}

#pragma mark - Table view delegate

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {
    MXPublicRoom *publicRoom;
    if (filteredPublicRooms) {
        publicRoom = [filteredPublicRooms objectAtIndex:indexPath.row];
    } else {
        publicRoom = [publicRooms objectAtIndex:indexPath.row];
    }
    
    // Check whether the user has already joined the selected public room
    MatrixSDKHandler *mxHandler = [MatrixSDKHandler sharedHandler];
    if ([mxHandler.mxSession roomWithRoomId:publicRoom.roomId]) {
        // Open selected room
        [[AppDelegate theDelegate].masterTabBarController showRoom:publicRoom.roomId];
    } else {
        // Join the selected room
        UIActivityIndicatorView *loadingWheel = [[UIActivityIndicatorView alloc] initWithActivityIndicatorStyle:UIActivityIndicatorViewStyleGray];
        UITableViewCell *selectedCell = [tableView cellForRowAtIndexPath:indexPath];
        if (selectedCell) {
            CGPoint center = CGPointMake(selectedCell.frame.size.width / 2, selectedCell.frame.size.height / 2);
            loadingWheel.center = center;
            [selectedCell addSubview:loadingWheel];
        }
        [loadingWheel startAnimating];
        [mxHandler.mxSession joinRoom:publicRoom.roomId success:^(MXRoom *room) {
            // Show joined room
            [loadingWheel stopAnimating];
            [loadingWheel removeFromSuperview];
            [[AppDelegate theDelegate].masterTabBarController showRoom:publicRoom.roomId];
        } failure:^(NSError *error) {
            NSLog(@"[HomeVC] Failed to join public room (%@): %@", publicRoom.displayname, error);
            //Alert user
            [loadingWheel stopAnimating];
            [loadingWheel removeFromSuperview];
            [[AppDelegate theDelegate] showErrorAsAlert:error];
        }];
    }
    
    [tableView deselectRowAtIndexPath:indexPath animated:YES];
}

#pragma mark - UISearchBarDelegate

- (BOOL)searchBarShouldBeginEditing:(UISearchBar *)searchBar {
    searchBarShouldEndEditing = NO;
    return YES;
}

- (BOOL)searchBarShouldEndEditing:(UISearchBar *)searchBar {
    return searchBarShouldEndEditing;
}

- (void)searchBar:(UISearchBar *)searchBar textDidChange:(NSString *)searchText {
    // Update filtered list
    if (searchText.length) {
        if (filteredPublicRooms) {
            [filteredPublicRooms removeAllObjects];
        } else {
            filteredPublicRooms = [NSMutableArray arrayWithCapacity:publicRooms.count];
        }
        for (MXPublicRoom *publicRoom in publicRooms) {
            if ([[publicRoom displayname] rangeOfString:searchText options:NSCaseInsensitiveSearch].location != NSNotFound) {
                [filteredPublicRooms addObject:publicRoom];
            }
        }
    } else {
        filteredPublicRooms = nil;
    }
    // Refresh display
    [self.tableView reloadData];
    if (filteredPublicRooms.count) {
        [self.tableView scrollToRowAtIndexPath:[NSIndexPath indexPathForRow:0 inSection:0] atScrollPosition:UITableViewScrollPositionTop animated:NO];
    }
}

- (void)searchBarSearchButtonClicked:(UISearchBar *)searchBar {
    // "Done" key has been pressed
    searchBarShouldEndEditing = YES;
    [searchBar resignFirstResponder];
}

- (void)searchBarCancelButtonClicked:(UISearchBar *)searchBar {
    // Leave search
    searchBarShouldEndEditing = YES;
    [searchBar resignFirstResponder];
    publicRoomsSearchBar = nil;
    filteredPublicRooms = nil;
    // Restore table header and refresh table display
    self.tableView.tableHeaderView = savedTableHeaderView;
    [self.tableView reloadData];
}

@end