/****************************************************************************** * Copyright (c) 2008-2012 Transmission authors and contributors * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER * DEALINGS IN THE SOFTWARE. *****************************************************************************/ #import #import "FileOutlineController.h" #import "Torrent.h" #import "FileListNode.h" #import "FileOutlineView.h" #import "FilePriorityCell.h" #import "FileRenameSheetController.h" #import "NSApplicationAdditions.h" #import "NSMutableArrayAdditions.h" #import "NSStringAdditions.h" #define ROW_SMALL_HEIGHT 18.0 typedef NS_ENUM(unsigned int, fileCheckMenuTag) { // FILE_CHECK_TAG, FILE_UNCHECK_TAG }; typedef NS_ENUM(unsigned int, filePriorityMenuTag) { // FILE_PRIORITY_HIGH_TAG, FILE_PRIORITY_NORMAL_TAG, FILE_PRIORITY_LOW_TAG }; @interface FileOutlineController (Private) @property(nonatomic, readonly) NSMenu* menu; - (NSUInteger)findFileNode:(FileListNode*)node inList:(NSArray*)list atIndexes:(NSIndexSet*)range currentParent:(FileListNode*)currentParent finalParent:(FileListNode**)parent; @end @implementation FileOutlineController - (void)awakeFromNib { fFileList = [[NSMutableArray alloc] init]; fOutline.doubleAction = @selector(revealFile:); fOutline.target = self; //set table header tool tips [fOutline tableColumnWithIdentifier:@"Check"].headerToolTip = NSLocalizedString(@"Download", "file table -> header tool tip"); [fOutline tableColumnWithIdentifier:@"Priority"].headerToolTip = NSLocalizedString(@"Priority", "file table -> header tool tip"); fOutline.menu = self.menu; [self setTorrent:nil]; } - (FileOutlineView*)outlineView { return fOutline; } - (void)setTorrent:(Torrent*)torrent { fTorrent = torrent; [fFileList setArray:fTorrent.fileList]; fFilterText = nil; [fOutline reloadData]; [fOutline deselectAll:nil]; //do this after reloading the data #4575 } - (void)setFilterText:(NSString*)text { NSArray* components = [text betterComponentsSeparatedByCharactersInSet:NSCharacterSet.whitespaceAndNewlineCharacterSet]; if (!components || components.count == 0) { text = nil; components = nil; } if ((!text && !fFilterText) || (text && fFilterText && [text isEqualToString:fFilterText])) { return; } [fOutline beginUpdates]; NSUInteger currentIndex = 0, totalCount = 0; NSMutableArray* itemsToAdd = [NSMutableArray array]; NSMutableIndexSet* itemsToAddIndexes = [NSMutableIndexSet indexSet]; NSMutableDictionary* removedIndexesForParents = nil; //ugly, but we can't modify the actual file nodes NSArray* tempList = !text ? fTorrent.fileList : fTorrent.flatFileList; for (FileListNode* item in tempList) { __block BOOL filter = NO; if (components) { [components enumerateObjectsWithOptions:NSEnumerationConcurrent usingBlock:^(id obj, NSUInteger idx, BOOL* stop) { if ([item.name rangeOfString:(NSString*)obj options:(NSCaseInsensitiveSearch | NSDiacriticInsensitiveSearch)].location == NSNotFound) { filter = YES; *stop = YES; } }]; } if (!filter) { FileListNode* parent = nil; NSUInteger previousIndex = !item.isFolder ? [self findFileNode:item inList:fFileList atIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(currentIndex, fFileList.count - currentIndex)] currentParent:nil finalParent:&parent] : NSNotFound; if (previousIndex == NSNotFound) { [itemsToAdd addObject:item]; [itemsToAddIndexes addIndex:totalCount]; } else { BOOL move = YES; if (!parent) { if (previousIndex != currentIndex) { [fFileList moveObjectAtIndex:previousIndex toIndex:currentIndex]; } else { move = NO; } } else { [fFileList insertObject:item atIndex:currentIndex]; //figure out the index within the semi-edited table - UGLY if (!removedIndexesForParents) { removedIndexesForParents = [NSMutableDictionary dictionary]; } NSMutableIndexSet* removedIndexes = removedIndexesForParents[parent]; if (!removedIndexes) { removedIndexes = [NSMutableIndexSet indexSetWithIndex:previousIndex]; removedIndexesForParents[parent] = removedIndexes; } else { [removedIndexes addIndex:previousIndex]; previousIndex -= [removedIndexes countOfIndexesInRange:NSMakeRange(0, previousIndex)]; } } if (move) { [fOutline moveItemAtIndex:previousIndex inParent:parent toIndex:currentIndex inParent:nil]; } ++currentIndex; } ++totalCount; } } //remove trailing items - those are the unused if (currentIndex < fFileList.count) { NSRange const removeRange = NSMakeRange(currentIndex, fFileList.count - currentIndex); [fFileList removeObjectsInRange:removeRange]; [fOutline removeItemsAtIndexes:[NSIndexSet indexSetWithIndexesInRange:removeRange] inParent:nil withAnimation:NSTableViewAnimationSlideDown]; } //add new items [fFileList insertObjects:itemsToAdd atIndexes:itemsToAddIndexes]; [fOutline insertItemsAtIndexes:itemsToAddIndexes inParent:nil withAnimation:NSTableViewAnimationSlideUp]; [fOutline endUpdates]; fFilterText = text; } - (void)refresh { [fTorrent updateFileStat]; fOutline.needsDisplay = YES; } - (void)outlineViewSelectionDidChange:(NSNotification*)notification { if ([QLPreviewPanel sharedPreviewPanelExists] && [QLPreviewPanel sharedPreviewPanel].visible) { [[QLPreviewPanel sharedPreviewPanel] reloadData]; } } - (NSInteger)outlineView:(NSOutlineView*)outlineView numberOfChildrenOfItem:(id)item { if (!item) { return fFileList ? fFileList.count : 0; } else { FileListNode* node = (FileListNode*)item; return node.isFolder ? node.children.count : 0; } } - (BOOL)outlineView:(NSOutlineView*)outlineView isItemExpandable:(id)item { return ((FileListNode*)item).isFolder; } - (id)outlineView:(NSOutlineView*)outlineView child:(NSInteger)index ofItem:(id)item { return (item ? ((FileListNode*)item).children : fFileList)[index]; } - (id)outlineView:(NSOutlineView*)outlineView objectValueForTableColumn:(NSTableColumn*)tableColumn byItem:(id)item { if ([tableColumn.identifier isEqualToString:@"Check"]) { return @([fTorrent checkForFiles:((FileListNode*)item).indexes]); } else { return item; } } - (void)outlineView:(NSOutlineView*)outlineView willDisplayCell:(id)cell forTableColumn:(NSTableColumn*)tableColumn item:(id)item { NSString* identifier = tableColumn.identifier; if ([identifier isEqualToString:@"Check"]) { [cell setEnabled:[fTorrent canChangeDownloadCheckForFiles:((FileListNode*)item).indexes]]; } else if ([identifier isEqualToString:@"Priority"]) { [cell setRepresentedObject:item]; NSInteger hoveredRow = fOutline.hoveredRow; [(FilePriorityCell*)cell setHovered:hoveredRow != -1 && hoveredRow == [fOutline rowForItem:item]]; } } - (void)outlineView:(NSOutlineView*)outlineView setObjectValue:(id)object forTableColumn:(NSTableColumn*)tableColumn byItem:(id)item { NSString* identifier = tableColumn.identifier; if ([identifier isEqualToString:@"Check"]) { NSIndexSet* indexSet; if (NSEvent.modifierFlags & NSAlternateKeyMask) { indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, fTorrent.fileCount)]; } else { indexSet = ((FileListNode*)item).indexes; } [fTorrent setFileCheckState:[object intValue] != NSOffState ? NSOnState : NSOffState forIndexes:indexSet]; fOutline.needsDisplay = YES; [NSNotificationCenter.defaultCenter postNotificationName:@"UpdateUI" object:nil]; } } - (NSString*)outlineView:(NSOutlineView*)outlineView typeSelectStringForTableColumn:(NSTableColumn*)tableColumn item:(id)item { return ((FileListNode*)item).name; } - (NSString*)outlineView:(NSOutlineView*)outlineView toolTipForCell:(NSCell*)cell rect:(NSRectPointer)rect tableColumn:(NSTableColumn*)tableColumn item:(id)item mouseLocation:(NSPoint)mouseLocation { NSString* ident = tableColumn.identifier; if ([ident isEqualToString:@"Name"]) { NSString* path = [fTorrent fileLocation:item]; if (!path) { FileListNode* node = (FileListNode*)item; path = [node.path stringByAppendingPathComponent:node.name]; } return path; } else if ([ident isEqualToString:@"Check"]) { switch (cell.state) { case NSOffState: return NSLocalizedString(@"Don't Download", "files tab -> tooltip"); case NSOnState: return NSLocalizedString(@"Download", "files tab -> tooltip"); case NSMixedState: return NSLocalizedString(@"Download Some", "files tab -> tooltip"); } } else if ([ident isEqualToString:@"Priority"]) { NSSet* priorities = [fTorrent filePrioritiesForIndexes:((FileListNode*)item).indexes]; switch (priorities.count) { case 0: return NSLocalizedString(@"Priority Not Available", "files tab -> tooltip"); case 1: switch ([[priorities anyObject] intValue]) { case TR_PRI_LOW: return NSLocalizedString(@"Low Priority", "files tab -> tooltip"); case TR_PRI_HIGH: return NSLocalizedString(@"High Priority", "files tab -> tooltip"); case TR_PRI_NORMAL: return NSLocalizedString(@"Normal Priority", "files tab -> tooltip"); } break; default: return NSLocalizedString(@"Multiple Priorities", "files tab -> tooltip"); } } return nil; } - (CGFloat)outlineView:(NSOutlineView*)outlineView heightOfRowByItem:(id)item { if (((FileListNode*)item).isFolder) { return ROW_SMALL_HEIGHT; } else { return outlineView.rowHeight; } } - (void)setCheck:(id)sender { NSInteger state = [sender tag] == FILE_UNCHECK_TAG ? NSOffState : NSOnState; NSIndexSet* indexSet = fOutline.selectedRowIndexes; NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet]; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* item = [fOutline itemAtRow:i]; [itemIndexes addIndexes:item.indexes]; } [fTorrent setFileCheckState:state forIndexes:itemIndexes]; fOutline.needsDisplay = YES; } - (void)setOnlySelectedCheck:(id)sender { NSIndexSet* indexSet = fOutline.selectedRowIndexes; NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet]; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* item = [fOutline itemAtRow:i]; [itemIndexes addIndexes:item.indexes]; } [fTorrent setFileCheckState:NSOnState forIndexes:itemIndexes]; NSMutableIndexSet* remainingItemIndexes = [NSMutableIndexSet indexSetWithIndexesInRange:NSMakeRange(0, fTorrent.fileCount)]; [remainingItemIndexes removeIndexes:itemIndexes]; [fTorrent setFileCheckState:NSOffState forIndexes:remainingItemIndexes]; fOutline.needsDisplay = YES; } - (void)checkAll { NSIndexSet* indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, fTorrent.fileCount)]; [fTorrent setFileCheckState:NSOnState forIndexes:indexSet]; fOutline.needsDisplay = YES; } - (void)uncheckAll { NSIndexSet* indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, fTorrent.fileCount)]; [fTorrent setFileCheckState:NSOffState forIndexes:indexSet]; fOutline.needsDisplay = YES; } - (void)setPriority:(id)sender { tr_priority_t priority; switch ([sender tag]) { case FILE_PRIORITY_HIGH_TAG: priority = TR_PRI_HIGH; break; case FILE_PRIORITY_NORMAL_TAG: priority = TR_PRI_NORMAL; break; case FILE_PRIORITY_LOW_TAG: priority = TR_PRI_LOW; } NSIndexSet* indexSet = fOutline.selectedRowIndexes; NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet]; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* item = [fOutline itemAtRow:i]; [itemIndexes addIndexes:item.indexes]; } [fTorrent setFilePriority:priority forIndexes:itemIndexes]; fOutline.needsDisplay = YES; } - (void)revealFile:(id)sender { NSIndexSet* indexes = fOutline.selectedRowIndexes; NSMutableArray* paths = [NSMutableArray arrayWithCapacity:indexes.count]; for (NSUInteger i = indexes.firstIndex; i != NSNotFound; i = [indexes indexGreaterThanIndex:i]) { NSString* path = [fTorrent fileLocation:[fOutline itemAtRow:i]]; if (path) { [paths addObject:[NSURL fileURLWithPath:path]]; } } if (paths.count > 0) { [NSWorkspace.sharedWorkspace activateFileViewerSelectingURLs:paths]; } } - (void)renameSelected:(id)sender { NSIndexSet* indexes = fOutline.selectedRowIndexes; NSAssert(indexes.count == 1, @"1 file needs to be selected to rename, but %ld are selected", indexes.count); FileListNode* node = [fOutline itemAtRow:indexes.firstIndex]; Torrent* torrent = node.torrent; if (!torrent.folder) { [FileRenameSheetController presentSheetForTorrent:torrent modalForWindow:fOutline.window completionHandler:^(BOOL didRename) { if (didRename) { [NSNotificationCenter.defaultCenter postNotificationName:@"UpdateQueue" object:self]; [NSNotificationCenter.defaultCenter postNotificationName:@"ResetInspector" object:self userInfo:@{ @"Torrent" : torrent }]; } }]; } else { [FileRenameSheetController presentSheetForFileListNode:node modalForWindow:fOutline.window completionHandler:^(BOOL didRename) { #warning instead of calling reset inspector, just resort? if (didRename) [NSNotificationCenter.defaultCenter postNotificationName:@"ResetInspector" object:self userInfo:@{ @"Torrent" : torrent }]; }]; } } #warning make real view controller (Leopard-only) so that Command-R will work - (BOOL)validateMenuItem:(NSMenuItem*)menuItem { if (!fTorrent) { return NO; } SEL action = menuItem.action; if (action == @selector(revealFile:)) { NSIndexSet* indexSet = fOutline.selectedRowIndexes; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { if ([fTorrent fileLocation:[fOutline itemAtRow:i]] != nil) { return YES; } } return NO; } if (action == @selector(setCheck:)) { if (fOutline.numberOfSelectedRows == 0) { return NO; } NSIndexSet* indexSet = fOutline.selectedRowIndexes; NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet]; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* node = [fOutline itemAtRow:i]; [itemIndexes addIndexes:node.indexes]; } NSInteger state = (menuItem.tag == FILE_CHECK_TAG) ? NSOnState : NSOffState; return [fTorrent checkForFiles:itemIndexes] != state && [fTorrent canChangeDownloadCheckForFiles:itemIndexes]; } if (action == @selector(setOnlySelectedCheck:)) { if (fOutline.numberOfSelectedRows == 0) { return NO; } NSIndexSet* indexSet = fOutline.selectedRowIndexes; NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet]; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* node = [fOutline itemAtRow:i]; [itemIndexes addIndexes:node.indexes]; } return [fTorrent canChangeDownloadCheckForFiles:itemIndexes]; } if (action == @selector(setPriority:)) { if (fOutline.numberOfSelectedRows == 0) { menuItem.state = NSOffState; return NO; } //determine which priorities are checked NSIndexSet* indexSet = fOutline.selectedRowIndexes; tr_priority_t priority; switch (menuItem.tag) { case FILE_PRIORITY_HIGH_TAG: priority = TR_PRI_HIGH; break; case FILE_PRIORITY_NORMAL_TAG: priority = TR_PRI_NORMAL; break; case FILE_PRIORITY_LOW_TAG: priority = TR_PRI_LOW; break; } BOOL current = NO, canChange = NO; for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i]) { FileListNode* node = [fOutline itemAtRow:i]; NSIndexSet* fileIndexSet = node.indexes; if (![fTorrent canChangeDownloadCheckForFiles:fileIndexSet]) { continue; } canChange = YES; if ([fTorrent hasFilePriority:priority forIndexes:fileIndexSet]) { current = YES; break; } } menuItem.state = current ? NSOnState : NSOffState; return canChange; } if (action == @selector(renameSelected:)) { return fOutline.numberOfSelectedRows == 1; } return YES; } @end @implementation FileOutlineController (Private) - (NSMenu*)menu { NSMenu* menu = [[NSMenu alloc] initWithTitle:@"File Outline Menu"]; //check and uncheck NSMenuItem* item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Check Selected", "File Outline -> Menu") action:@selector(setCheck:) keyEquivalent:@""]; item.target = self; item.tag = FILE_CHECK_TAG; [menu addItem:item]; item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Uncheck Selected", "File Outline -> Menu") action:@selector(setCheck:) keyEquivalent:@""]; item.target = self; item.tag = FILE_UNCHECK_TAG; [menu addItem:item]; //only check selected item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Only Check Selected", "File Outline -> Menu") action:@selector(setOnlySelectedCheck:) keyEquivalent:@""]; item.target = self; [menu addItem:item]; [menu addItem:[NSMenuItem separatorItem]]; //priority item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Priority", "File Outline -> Menu") action:NULL keyEquivalent:@""]; NSMenu* priorityMenu = [[NSMenu alloc] initWithTitle:@"File Priority Menu"]; item.submenu = priorityMenu; [menu addItem:item]; item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"High", "File Outline -> Priority Menu") action:@selector(setPriority:) keyEquivalent:@""]; item.target = self; item.tag = FILE_PRIORITY_HIGH_TAG; item.image = [NSImage imageNamed:@"PriorityHighTemplate"]; [priorityMenu addItem:item]; item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Normal", "File Outline -> Priority Menu") action:@selector(setPriority:) keyEquivalent:@""]; item.target = self; item.tag = FILE_PRIORITY_NORMAL_TAG; item.image = [NSImage imageNamed:@"PriorityNormalTemplate"]; [priorityMenu addItem:item]; item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Low", "File Outline -> Priority Menu") action:@selector(setPriority:) keyEquivalent:@""]; item.target = self; item.tag = FILE_PRIORITY_LOW_TAG; item.image = [NSImage imageNamed:@"PriorityLowTemplate"]; [priorityMenu addItem:item]; [menu addItem:[NSMenuItem separatorItem]]; //reveal in finder item = [[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Show in Finder", "File Outline -> Menu") action:@selector(revealFile:) keyEquivalent:@""]; item.target = self; [menu addItem:item]; [menu addItem:[NSMenuItem separatorItem]]; //rename item = [[NSMenuItem alloc] initWithTitle:[NSLocalizedString(@"Rename File", "File Outline -> Menu") stringByAppendingEllipsis] action:@selector(renameSelected:) keyEquivalent:@""]; item.target = self; [menu addItem:item]; return menu; } - (NSUInteger)findFileNode:(FileListNode*)node inList:(NSArray*)list atIndexes:(NSIndexSet*)indexes currentParent:(FileListNode*)currentParent finalParent:(FileListNode* __autoreleasing*)parent { NSAssert(!node.isFolder, @"Looking up folder node!"); __block NSUInteger retIndex = NSNotFound; [list enumerateObjectsAtIndexes:indexes options:NSEnumerationConcurrent usingBlock:^(FileListNode* checkNode, NSUInteger index, BOOL* stop) { if ([checkNode.indexes containsIndex:node.indexes.firstIndex]) { if (!checkNode.isFolder) { NSAssert2([checkNode isEqualTo:node], @"Expected file nodes to be equal: %@ %@", checkNode, node); *parent = currentParent; retIndex = index; } else { NSUInteger const subIndex = [self findFileNode:node inList:checkNode.children atIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, checkNode.children.count)] currentParent:checkNode finalParent:parent]; NSAssert(subIndex != NSNotFound, @"We didn't find an expected file node."); retIndex = subIndex; } *stop = YES; } }]; return retIndex; } @end