/*
 bookmarks and active session view controller
 
 Copyright 2013 Thincast Technologies GmbH, Author: Martin Fleisz
 
 This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0. 
 If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

#import "BookmarkListController.h"
#import "Utils.h"
#import "BookmarkEditorController.h"
#import "RDPSessionViewController.h"
#import "Toast+UIView.h"
#import "Reachability.h"
#import "GlobalDefaults.h"
#import "BlockAlertView.h"

#define SECTION_SESSIONS    0
#define SECTION_BOOKMARKS   1
#define NUM_SECTIONS        2

@interface BookmarkListController (Private)
#pragma mark misc functions
- (UIButton*)disclosureButtonWithImage:(UIImage*)image;
- (void)performSearch:(NSString*)searchText;
#pragma mark Persisting bookmarks
- (void)scheduleWriteBookmarksToDataStore;
- (void)writeBookmarksToDataStore;
- (void)scheduleWriteManualBookmarksToDataStore;
- (void)writeManualBookmarksToDataStore;
- (void)readManualBookmarksFromDataStore;
- (void)writeArray:(NSArray*)bookmarks toDataStoreURL:(NSURL*)url;
- (NSMutableArray*)arrayFromDataStoreURL:(NSURL*)url;
- (NSURL*)manualBookmarksDataStoreURL;
- (NSURL*)connectionHistoryDataStoreURL;
@end


@implementation BookmarkListController

@synthesize searchBar = _searchBar, tableView = _tableView, bmTableCell = _bmTableCell, sessTableCell = _sessTableCell;

 // The designated initializer.  Override if you create the controller programmatically and want to perform customization that is not appropriate for viewDidLoad.
- (id)initWithNibName:(NSString *)nibNameOrNil bundle:(NSBundle *)nibBundleOrNil {
    if ((self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil])) 
	{   
        // load bookmarks
        [self readManualBookmarksFromDataStore];

        // load connection history
        [self readConnectionHistoryFromDataStore];
        
		// init search result array
		_manual_search_result = nil;		
        
        // register for session notifications
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(sessionDisconnected:) name:TSXSessionDidDisconnectNotification object:nil];
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(sessionFailedToConnect:) name:TSXSessionDidFailToConnectNotification object:nil];
                
        // set title and tabbar controller image
        [self setTitle:NSLocalizedString(@"Connections", @"'Connections': bookmark controller title")];
        [self setTabBarItem:[[[UITabBarItem alloc] initWithTabBarSystemItem:UITabBarSystemItemBookmarks tag:0] autorelease]];

        // load images
        _star_on_img = [[UIImage imageWithContentsOfFile:[[NSBundle mainBundle] pathForResource:@"icon_accessory_star_on" ofType:@"png"]] retain];
        _star_off_img = [[UIImage imageWithContentsOfFile:[[NSBundle mainBundle] pathForResource:@"icon_accessory_star_off" ofType:@"png"]] retain];

        // init reachability detection
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(reachabilityChanged:) name:kReachabilityChangedNotification object:nil];
        
        // init other properties
        _active_sessions = [[NSMutableArray alloc] init];
        _temporary_bookmark = nil;
    }
    return self;
}

- (void)loadView
{
    [super loadView];    
}

// Implement viewDidLoad to do additional setup after loading the view, typically from a nib.
- (void)viewDidLoad {
    [super viewDidLoad];        
    
    // set edit button to allow bookmark list editing
    [[self navigationItem] setRightBarButtonItem:[self editButtonItem]];
}


- (void)viewWillAppear:(BOOL)animated 
{
	[super viewWillAppear:animated];
    
    // in case we had a search - search again cause the bookmark searchable items could have changed
    if ([[_searchBar text] length] > 0)
        [self performSearch:[_searchBar text]];

    // to reflect any bookmark changes - reload table
    [_tableView reloadData];
}

- (void)viewWillDisappear:(BOOL)animated
{
    [super viewWillDisappear:animated];
    
    // clear any search
    [_searchBar setText:@""];
    [_searchBar resignFirstResponder];
    [self performSearch:@""];
}

// Override to allow orientations other than the default portrait orientation.
- (BOOL)shouldAutorotateToInterfaceOrientation:(UIInterfaceOrientation)interfaceOrientation {
    // Return YES for supported orientations
    return YES;
}


- (void)didReceiveMemoryWarning {
    // Releases the view if it doesn't have a superview.
    [super didReceiveMemoryWarning];
    
    // Release any cached data, images, etc that aren't in use.
}

- (void)viewDidUnload {
    [super viewDidUnload];
    // Release any retained subviews of the main view.
    // e.g. self.myOutlet = nil;
}


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

    [_temporary_bookmark release];
    [_connection_history release];
    [_active_sessions release];
	[_manual_search_result release];
    [_manual_bookmarks release];
    
    [_star_on_img release];
    [_star_off_img release];

    [super dealloc];
}


#pragma mark -
#pragma mark Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
    // Return the number of sections.
	return NUM_SECTIONS;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
	
	switch(section)
	{
        case SECTION_SESSIONS:
            return 0;
            break;
            
		case SECTION_BOOKMARKS:
			{
				// (+1 for Add Bookmark entry)
				if(_manual_search_result != nil)
					return ([_manual_search_result count] + [_history_search_result count] + 1);
				return ([_manual_bookmarks count] + 1);
			}
			break;	
			
		default:
			break;
	}
	return 0;
}

- (UITableViewCell*)cellForGenericListEntry
{
    static NSString *CellIdentifier = @"BookmarkListCell";
    UITableViewCell *cell = [[self tableView] dequeueReusableCellWithIdentifier:CellIdentifier];
    if(cell == nil)
    {
        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:CellIdentifier];
        [cell setSelectionStyle:UITableViewCellSelectionStyleNone];
        [cell setAccessoryView:[self disclosureButtonWithImage:_star_off_img]];                        
    }

    return cell;
}

- (BookmarkTableCell*)cellForBookmark
{
    static NSString *BookmarkCellIdentifier = @"BookmarkCell";   
    BookmarkTableCell *cell = (BookmarkTableCell*)[[self tableView] dequeueReusableCellWithIdentifier:BookmarkCellIdentifier];
    if(cell == nil) 
    {
        [[NSBundle mainBundle] loadNibNamed:@"BookmarkTableViewCell" owner:self options:nil];
        [_bmTableCell setAccessoryView:[self disclosureButtonWithImage:_star_on_img]];                        
        cell = _bmTableCell;
        _bmTableCell = nil;
    }
    
    return cell;
}

// Customize the appearance of table view cells.
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {


    switch ([indexPath section])
    {
        case SECTION_SESSIONS:
        {
            // get custom session cell
            static NSString *SessionCellIdentifier = @"SessionCell";
            SessionTableCell *cell = (SessionTableCell*)[tableView dequeueReusableCellWithIdentifier:SessionCellIdentifier];
            if(cell == nil) 
            {
                [[NSBundle mainBundle] loadNibNamed:@"SessionTableViewCell" owner:self options:nil];
                cell = _sessTableCell;
                _sessTableCell = nil;
            }
            
            // set cell data
            RDPSession* session = [_active_sessions objectAtIndex:[indexPath row]];        
            [[cell title] setText:[session sessionName]];
            [[cell server] setText:[[session params] StringForKey:@"hostname"]];
            [[cell username] setText:[[session params] StringForKey:@"username"]];
            [[cell screenshot] setImage:[session getScreenshotWithSize:[[cell screenshot] bounds].size]];
            [[cell disconnectButton] setTag:[indexPath row]];        
            return cell;
        }
            
        case SECTION_BOOKMARKS:
        {
            // special handling for first cell - quick connect/quick create Bookmark cell
            if([indexPath row] == 0)
            {                
                // if a search text is entered the cell becomes a quick connect/quick create bookmark cell - otherwise it's just an add bookmark cell
                UITableViewCell* cell = [self cellForGenericListEntry];
                if ([[_searchBar text] length] == 0)
                {
                    [[cell textLabel] setText:[@"  " stringByAppendingString:NSLocalizedString(@"Add Connection", @"'Add Connection': button label")]];
                    [((UIButton*)[cell accessoryView]) setHidden:YES];
                }
                else
                {
                    [[cell textLabel] setText:[@"  " stringByAppendingString:[_searchBar text]]];            
                    [((UIButton*)[cell accessoryView]) setHidden:NO];
                }
                
                return cell;
            }
            else            
            {
                // do we have a history cell or bookmark cell?
                if ([self isIndexPathToHistoryItem:indexPath])
                {
                    UITableViewCell* cell = [self cellForGenericListEntry];
                    [[cell textLabel] setText:[@"  " stringByAppendingString:[_history_search_result objectAtIndex:[self historyIndexFromIndexPath:indexPath]]]];
                    [((UIButton*)[cell accessoryView]) setHidden:NO];                    
                    return cell;
                }
                else
                {
                    // set cell properties
                    ComputerBookmark* entry;
                    BookmarkTableCell* cell = [self cellForBookmark];
                    if(_manual_search_result == nil)
                        entry = [_manual_bookmarks objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]];
                    else
                        entry = [[_manual_search_result objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]] valueForKey:@"bookmark"];
                    
                    [[cell title] setText:[entry label]];
                    [[cell subTitle] setText:[[entry params] StringForKey:@"hostname"]];
                    return cell;                
                }                
            }            
        }
            
        default:
            break;
    }
    
    NSAssert(0, @"Failed to create cell");
    return nil;    	
}



// Override to support conditional editing of the table view.
- (BOOL)tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *)indexPath {
	// dont allow to edit Add Bookmark item
    if([indexPath section] == SECTION_SESSIONS)
        return NO;
    if([indexPath section] == SECTION_BOOKMARKS && [indexPath row] == 0)
		return NO;
    return YES;
}



 // Override to support editing the table view.
 - (void)tableView:(UITableView *)tableView commitEditingStyle:(UITableViewCellEditingStyle)editingStyle forRowAtIndexPath:(NSIndexPath *)indexPath 
{
	if(editingStyle == UITableViewCellEditingStyleDelete) 
	{
		// Delete the row from the data source
		switch([indexPath section])
		{
            case SECTION_BOOKMARKS:
            {
                if (_manual_search_result == nil)
                    [_manual_bookmarks removeObjectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]];
                else
                {
                    // history item or bookmark?
                    if ([self isIndexPathToHistoryItem:indexPath])
                    {
                        [_connection_history removeObject:[_history_search_result objectAtIndex:[self historyIndexFromIndexPath:indexPath]]];
                        [_history_search_result removeObjectAtIndex:[self historyIndexFromIndexPath:indexPath]];
                    }                    
                    else
                    {
                        [_manual_bookmarks removeObject:[[_manual_search_result objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]] valueForKey:@"bookmark"]];
                        [_manual_search_result removeObjectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]];
                    }                    
                }
                [self scheduleWriteManualBookmarksToDataStore];
                break;
            }                
		}

		[tableView reloadSections:[NSIndexSet indexSetWithIndex:[indexPath section]] withRowAnimation:UITableViewRowAnimationNone];
	}   
}


// Override to support rearranging the table view.
- (void)tableView:(UITableView *)tableView moveRowAtIndexPath:(NSIndexPath *)fromIndexPath toIndexPath:(NSIndexPath *)toIndexPath 
{
	if([fromIndexPath compare:toIndexPath] != NSOrderedSame)
	{
		switch([fromIndexPath section])
		{
            case SECTION_BOOKMARKS:
            {
                int fromIdx = [self bookmarkIndexFromIndexPath:fromIndexPath];
                int toIdx = [self bookmarkIndexFromIndexPath:toIndexPath];
                ComputerBookmark* temp_bookmark = [[_manual_bookmarks objectAtIndex:fromIdx] retain];
                [_manual_bookmarks removeObjectAtIndex:fromIdx];
                if (toIdx >= [_manual_bookmarks count])
                    [_manual_bookmarks addObject:temp_bookmark];
                else
                    [_manual_bookmarks insertObject:temp_bookmark atIndex:toIdx];
                [temp_bookmark release];
                
                [self scheduleWriteManualBookmarksToDataStore];
                break;
            }
		}
	}
}


// prevent that an item is moved befoer the Add Bookmark item
-(NSIndexPath*)tableView:(UITableView *)tableView targetIndexPathForMoveFromRowAtIndexPath:(NSIndexPath *)sourceIndexPath toProposedIndexPath:(NSIndexPath *)proposedDestinationIndexPath
{
    // don't allow to move:
    //  - items between sections
    //  - the quick connect/quick create bookmark cell
    //  - any item while a search is applied
    if([proposedDestinationIndexPath row] == 0 || ([sourceIndexPath section] != [proposedDestinationIndexPath section]) || 
       _manual_search_result != nil)
    {
        return sourceIndexPath;
    }
    else
    {
        return proposedDestinationIndexPath;
    }
}


// Override to support conditional rearranging of the table view.
- (BOOL)tableView:(UITableView *)tableView canMoveRowAtIndexPath:(NSIndexPath *)indexPath {
	// dont allow to reorder Add Bookmark item
    if([indexPath section] == SECTION_BOOKMARKS && [indexPath row] == 0)
		return NO;
    return YES;
}

- (NSString*)tableView:(UITableView*)tableView titleForHeaderInSection:(NSInteger)section
{
	if(section == SECTION_SESSIONS && [_active_sessions count] > 0)
		return NSLocalizedString(@"My Sessions", @"'My Session': section sessions header");
	if(section == SECTION_BOOKMARKS)
		return NSLocalizedString(@"Manual Connections", @"'Manual Connections': section manual bookmarks header");
	return nil;
}

- (NSString*)tableView:(UITableView*)tableView titleForFooterInSection:(NSInteger)section
{
	return nil;
}

- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath
{
    if([indexPath section] == SECTION_SESSIONS)
        return 72;
    return [tableView rowHeight];
}

#pragma mark -
#pragma mark Table view delegate

- (void)setEditing:(BOOL)editing animated:(BOOL)animated
{
    [super setEditing:editing animated:animated];
    [[self tableView] setEditing:editing animated:animated];
}

- (void)accessoryButtonTapped:(UIControl*)button withEvent:(UIEvent*)event
{
    // forward a tap on our custom accessory button to the real accessory button handler
    NSIndexPath* indexPath = [[self tableView] indexPathForRowAtPoint:[[[event touchesForView:button] anyObject] locationInView:[self tableView]]];
    if (indexPath == nil)
        return;
    
    [[[self tableView] delegate] tableView:[self tableView] accessoryButtonTappedForRowWithIndexPath:indexPath];
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath 
{    
    if([indexPath section] == SECTION_SESSIONS)
    {
        // resume session
        RDPSession* session = [_active_sessions objectAtIndex:[indexPath row]];
        UIViewController* ctrl = [[[RDPSessionViewController alloc] initWithNibName:@"RDPSessionView" bundle:nil session:session] autorelease];
        [ctrl setHidesBottomBarWhenPushed:YES];
        [[self navigationController] pushViewController:ctrl animated:YES];
    }
	else
	{
		ComputerBookmark* bookmark = nil;
		if([indexPath section] == SECTION_BOOKMARKS)
		{
            // first row has either quick connect or add bookmark item
            if([indexPath row] == 0)
            {
                if ([[_searchBar text] length] == 0)
                {                    
                    // show add bookmark controller
                    ComputerBookmark *bookmark = [[[ComputerBookmark alloc] initWithBaseDefaultParameters] autorelease];
                    BookmarkEditorController* bookmarkEditorController = [[[BookmarkEditorController alloc] initWithBookmark:bookmark] autorelease];
                    [bookmarkEditorController setTitle:NSLocalizedString(@"Add Connection", @"Add Connection title")];
                    [bookmarkEditorController setDelegate:self];
                    [bookmarkEditorController setHidesBottomBarWhenPushed:YES];
                    [[self navigationController] pushViewController:bookmarkEditorController animated:YES];
                }
                else
                {
                    // create a quick connect bookmark and add an entry to the quick connect history (if not already in the history)
                    bookmark = [self bookmarkForQuickConnectTo:[_searchBar text]];     
                    if (![_connection_history containsObject:[_searchBar text]])
                    {
                        [_connection_history addObject:[_searchBar text]];
                        [self scheduleWriteConnectionHistoryToDataStore];                        
                    }
                }
            }
            else
            {
                if(_manual_search_result != nil)
                {
                    if ([self isIndexPathToHistoryItem:indexPath])
                    {
                        // create a quick connect bookmark for a history item
                        NSString* item = [_history_search_result objectAtIndex:[self historyIndexFromIndexPath:indexPath]];
                        bookmark = [self bookmarkForQuickConnectTo:item];
                    }
                    else
                        bookmark = [[_manual_search_result objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]] valueForKey:@"bookmark"];                    
                }
                else
                    bookmark = [_manual_bookmarks objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]]; // -1 because of ADD BOOKMARK entry
            }
            
            // set reachability status
            WakeUpWWAN();
            [bookmark setConntectedViaWLAN:[[Reachability reachabilityWithHostName:[[bookmark params] StringForKey:@"hostname"]] currentReachabilityStatus] == ReachableViaWiFi];            
		}
		
		if(bookmark != nil)
		{	            
			// create rdp session
            RDPSession* session = [[[RDPSession alloc] initWithBookmark:bookmark] autorelease];
            UIViewController* ctrl = [[[RDPSessionViewController alloc] initWithNibName:@"RDPSessionView" bundle:nil session:session] autorelease];
            [ctrl setHidesBottomBarWhenPushed:YES];
            [[self navigationController] pushViewController:ctrl animated:YES];
            [_active_sessions addObject:session];
        }
	}
}


- (void)tableView:(UITableView*)tableView accessoryButtonTappedForRowWithIndexPath:(NSIndexPath*)indexPath
{    
	// get the bookmark
    NSString* bookmark_editor_title = NSLocalizedString(@"Edit Connection", @"Edit Connection title");
	ComputerBookmark* bookmark = nil;
	if ([indexPath section] == SECTION_BOOKMARKS)
	{
        if ([indexPath row] == 0)
        {
            // create a new bookmark and init hostname and label
            bookmark = [self bookmarkForQuickConnectTo:[_searchBar text]];     
            bookmark_editor_title = NSLocalizedString(@"Add Connection", @"Add Connection title");
        }
        else
        {
            if (_manual_search_result != nil)
            {
                if ([self isIndexPathToHistoryItem:indexPath])
                {
                    // create a new bookmark and init hostname and label
                    NSString* item = [_history_search_result objectAtIndex:[self historyIndexFromIndexPath:indexPath]];
                    bookmark = [self bookmarkForQuickConnectTo:item];     
                    bookmark_editor_title = NSLocalizedString(@"Add Connection", @"Add Connection title");
                }
                else
                    bookmark = [[_manual_search_result objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]] valueForKey:@"bookmark"];
            }
            else
                bookmark = [_manual_bookmarks objectAtIndex:[self bookmarkIndexFromIndexPath:indexPath]];	// -1 because of ADD BOOKMARK entry
        }
	}

    // bookmark found? - start the editor
	if (bookmark != nil)
	{
		BookmarkEditorController* editBookmarkController = [[[BookmarkEditorController alloc] initWithBookmark:bookmark] autorelease];
        [editBookmarkController setHidesBottomBarWhenPushed:YES];
        [editBookmarkController setTitle:bookmark_editor_title];
        [editBookmarkController setDelegate:self];
		[[self navigationController] pushViewController:editBookmarkController animated:YES];	
	}
}

#pragma mark -
#pragma mark Search Bar Delegates

- (BOOL)searchBarShouldBeginEditing:(UISearchBar*)searchBar
{
	// show cancel button
	[searchBar setShowsCancelButton:YES animated:YES];
	return YES;
}

- (void)searchBarCancelButtonClicked:(UISearchBar *)searchBar
{
	// clear search result
	[_manual_search_result release];
	_manual_search_result = nil;
	
	// clear text and remove cancel button
	[searchBar setText:@""];
	[searchBar resignFirstResponder];
}

- (BOOL)searchBarShouldEndEditing:(UISearchBar*)searchBar
{	
	[searchBar setShowsCancelButton:NO animated:YES];
	
	// re-enable table selection
	[_tableView setAllowsSelection:YES];
	
	return YES;
}

- (void)searchBarSearchButtonClicked:(UISearchBar *)searchBar
{
	[_searchBar resignFirstResponder];
}

- (void)searchBar:(UISearchBar*)searchBar textDidChange:(NSString*)searchText
{
    [self performSearch:searchText];
	[_tableView reloadData];
}

#pragma mark - Session handling

// session was added
- (void)sessionDisconnected:(NSNotification*)notification
{
    // remove session from active sessions
    RDPSession* session = (RDPSession*)[notification object];
    [_active_sessions removeObject:session];

    // if this view is currently active refresh entries
    if([[self navigationController] visibleViewController] == self)
        [_tableView reloadSections:[NSIndexSet indexSetWithIndex:SECTION_SESSIONS] withRowAnimation:UITableViewRowAnimationNone];	
    
    // if session's bookmark is not in the bookmark list ask the user if he wants to add it
    // (this happens if the session is created using the quick connect feature)
    if (![_manual_bookmarks containsObject:[session bookmark]])
    {
        // retain the bookmark in case we want to save it later
        _temporary_bookmark = [[session bookmark] retain];
        
        // ask the user if he wants to save the bookmark
        NSString* title = NSLocalizedString(@"Save Connection Settings?", @"Save connection settings title");
        NSString* message = NSLocalizedString(@"Your Connection Settings have not been saved. Do you want to save them?", @"Save connection settings message");
        BlockAlertView* alert = [BlockAlertView alertWithTitle:title message:message];
        [alert setCancelButtonWithTitle:NSLocalizedString(@"No", @"No Button") block:nil];
        [alert addButtonWithTitle:NSLocalizedString(@"Yes", @"Yes Button") block:^{
            if (_temporary_bookmark)
            {
                [_manual_bookmarks addObject:_temporary_bookmark];
                [_tableView reloadSections:[NSIndexSet indexSetWithIndex:SECTION_BOOKMARKS] withRowAnimation:UITableViewRowAnimationNone];
                [_temporary_bookmark autorelease];
                _temporary_bookmark = nil;
            }
        }];
        [alert show];
    }
}

- (void)sessionFailedToConnect:(NSNotification*)notification
{
    // remove session from active sessions
    RDPSession* session = (RDPSession*)[notification object];
    [_active_sessions removeObject:session];    
    
    // display error toast
    [[self view] makeToast:NSLocalizedString(@"Failed to connect to session!", @"Failed to connect error message") duration:ToastDurationNormal position:@"center"];
}

#pragma mark - Reachability notification
- (void)reachabilityChanged:(NSNotification*)notification
{
    // no matter how the network changed - we will disconnect
    // disconnect session (if there is any)
    if ([_active_sessions count] > 0)
    {
        RDPSession* session = [_active_sessions objectAtIndex:0];
        [session disconnect];
    }
}

#pragma mark - BookmarkEditorController delegate

- (void)commitBookmark:(ComputerBookmark *)bookmark
{
    // if we got a manual bookmark that is not in the list yet - add it otherwise replace it
    BOOL found = NO;
    for (int idx = 0; idx < [_manual_bookmarks count]; ++idx)
    {
        if ([[bookmark uuid] isEqualToString:[[_manual_bookmarks objectAtIndex:idx] uuid]])
        {
            [_manual_bookmarks replaceObjectAtIndex:idx withObject:bookmark];
            found = YES;
            break;
        }
    }
    if (!found)
        [_manual_bookmarks addObject:bookmark];
    
    // remove any quick connect history entry with the same hostname
    NSString* hostname = [[bookmark params] StringForKey:@"hostname"];
    if ([_connection_history containsObject:hostname])
    {
        [_connection_history removeObject:hostname];
        [self scheduleWriteConnectionHistoryToDataStore];
    }
        
    [self scheduleWriteManualBookmarksToDataStore];
}

- (IBAction)disconnectButtonPressed:(id)sender
{
    // disconnect session and refresh table view
    RDPSession* session = [_active_sessions objectAtIndex:[sender tag]];
    [session disconnect];
}

#pragma mark - Misc functions

- (BOOL)hasNoBookmarks
{
    return ([_manual_bookmarks count] == 0);
}

- (UIButton*)disclosureButtonWithImage:(UIImage*)image
{
    // we make the button a little bit bigger (image widht * 2, height + 10) so that the user doesn't accidentally connect to the bookmark ...
    UIButton* button = [UIButton buttonWithType:UIButtonTypeCustom];  
    [button setFrame:CGRectMake(0, 0, [image size].width * 2, [image size].height + 10)];
    [button setImage:image forState:UIControlStateNormal];
    [button addTarget:self action:@selector(accessoryButtonTapped:withEvent:) forControlEvents:UIControlEventTouchUpInside];    
    [button setUserInteractionEnabled:YES];
    return button;
}

- (void)performSearch:(NSString*)searchText
{
    [_manual_search_result autorelease];
    
	if([searchText length] > 0)
	{
		_manual_search_result = [FilterBookmarks(_manual_bookmarks, [searchText componentsSeparatedByCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]]) retain];
        _history_search_result = [FilterHistory(_connection_history, searchText) retain];
    }
	else
	{
        _history_search_result = nil;
		_manual_search_result = nil;
	}    
}

- (int)bookmarkIndexFromIndexPath:(NSIndexPath*)indexPath
{
    return [indexPath row] - ((_history_search_result != nil) ? [_history_search_result count] : 0) - 1;
}

- (int)historyIndexFromIndexPath:(NSIndexPath*)indexPath
{
    return [indexPath row] - 1;
}

- (BOOL)isIndexPathToHistoryItem:(NSIndexPath*)indexPath
{
    return (([indexPath row] - 1) < [_history_search_result count]);
}

- (ComputerBookmark*)bookmarkForQuickConnectTo:(NSString*)host
{
    ComputerBookmark* bookmark = [[[ComputerBookmark alloc] initWithBaseDefaultParameters] autorelease];
    [bookmark setLabel:host]; 
    [[bookmark params] setValue:host forKey:@"hostname"];                
    return bookmark;
}

#pragma mark - Persisting bookmarks

- (void)scheduleWriteBookmarksToDataStore
{
	[[NSOperationQueue mainQueue] addOperationWithBlock:^{
		[self writeBookmarksToDataStore];
	}];
}

- (void)writeBookmarksToDataStore
{    
    [self writeManualBookmarksToDataStore];
}

- (void)scheduleWriteManualBookmarksToDataStore
{
	[[NSOperationQueue mainQueue] addOperation:[[[NSInvocationOperation alloc] initWithTarget:self selector:@selector(writeManualBookmarksToDataStore) object:nil] autorelease]];
}

- (void)writeManualBookmarksToDataStore
{    
    [self writeArray:_manual_bookmarks toDataStoreURL:[self manualBookmarksDataStoreURL]];
}

- (void)scheduleWriteConnectionHistoryToDataStore
{
	[[NSOperationQueue mainQueue] addOperation:[[[NSInvocationOperation alloc] initWithTarget:self selector:@selector(writeConnectionHistoryToDataStore) object:nil] autorelease]];
}

- (void)writeConnectionHistoryToDataStore
{    
    [self writeArray:_connection_history toDataStoreURL:[self connectionHistoryDataStoreURL]];
}

- (void)writeArray:(NSArray*)bookmarks toDataStoreURL:(NSURL*)url
{
	NSData* archived_data = [NSKeyedArchiver archivedDataWithRootObject:bookmarks];
	[archived_data writeToURL:url atomically:YES];    
}

- (void)readManualBookmarksFromDataStore
{
    [_manual_bookmarks autorelease];
    _manual_bookmarks = [self arrayFromDataStoreURL:[self manualBookmarksDataStoreURL]];

    if(_manual_bookmarks == nil)
    {
        _manual_bookmarks = [[NSMutableArray alloc] init];
        [_manual_bookmarks addObject:[[[GlobalDefaults sharedGlobalDefaults] newTestServerBookmark] autorelease]];
    }
}

- (void)readConnectionHistoryFromDataStore
{
    [_connection_history autorelease];
    _connection_history = [self arrayFromDataStoreURL:[self connectionHistoryDataStoreURL]];
    
    if(_connection_history == nil)
        _connection_history = [[NSMutableArray alloc] init];
}

- (NSMutableArray*)arrayFromDataStoreURL:(NSURL*)url
{
	NSData* archived_data = [NSData dataWithContentsOfURL:url];
	
	if (!archived_data)
		return nil;
    
	return [[NSKeyedUnarchiver unarchiveObjectWithData:archived_data] retain];
}

- (NSURL*)manualBookmarksDataStoreURL
{
	return [NSURL fileURLWithPath:[NSString stringWithFormat:@"%@/%@", [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) lastObject], @"com.freerdp.ifreerdp.bookmarks.plist"]];
}

- (NSURL*)connectionHistoryDataStoreURL
{
	return [NSURL fileURLWithPath:[NSString stringWithFormat:@"%@/%@", [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) lastObject], @"com.freerdp.ifreerdp.connection_history.plist"]];
}

@end

