/* 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