transmission/macosx/FileOutlineController.mm

726 lines
24 KiB
Plaintext

// This file Copyright © 2008-2022 Transmission authors and contributors.
// It may be used under the MIT (SPDX: MIT) license.
// License text can be found in the licenses/ folder.
#import <Quartz/Quartz.h>
#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 ()
@property(nonatomic) NSMutableArray<FileListNode*>* fFileList;
@property(nonatomic) IBOutlet FileOutlineView* fOutline;
@property(nonatomic, readonly) NSMenu* menu;
- (NSUInteger)findFileNode:(FileListNode*)node
inList:(NSArray<FileListNode*>*)list
atIndexes:(NSIndexSet*)range
currentParent:(FileListNode*)currentParent
finalParent:(FileListNode**)parent;
@end
@implementation FileOutlineController
- (void)awakeFromNib
{
self.fFileList = [[NSMutableArray alloc] init];
self.fOutline.doubleAction = @selector(revealFile:);
self.fOutline.target = self;
//set table header tool tips
[self.fOutline tableColumnWithIdentifier:@"Check"].headerToolTip = NSLocalizedString(@"Download", "file table -> header tool tip");
[self.fOutline tableColumnWithIdentifier:@"Priority"].headerToolTip = NSLocalizedString(@"Priority", "file table -> header tool tip");
self.fOutline.menu = self.menu;
self.torrent = nil;
}
- (FileOutlineView*)outlineView
{
return _fOutline;
}
- (void)setTorrent:(Torrent*)torrent
{
_torrent = torrent;
[self.fFileList setArray:_torrent.fileList];
self.filterText = nil;
[self.fOutline reloadData];
[self.fOutline deselectAll:nil]; //do this after reloading the data #4575
}
- (void)setFilterText:(NSString*)text
{
NSArray* components = [text nonEmptyComponentsSeparatedByCharactersInSet:NSCharacterSet.whitespaceAndNewlineCharacterSet];
if (!components || components.count == 0)
{
text = nil;
components = nil;
}
if ((!text && !_filterText) || (text && _filterText && [text isEqualToString:_filterText]))
{
return;
}
[self.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 ? self.torrent.fileList : self.torrent.flatFileList;
for (FileListNode* item in tempList)
{
__block BOOL filter = NO;
if (components)
{
[components enumerateObjectsWithOptions:NSEnumerationConcurrent usingBlock:^(NSString* obj, NSUInteger idx, BOOL* stop) {
if ([item.name rangeOfString:obj options:(NSCaseInsensitiveSearch | NSDiacriticInsensitiveSearch)].location == NSNotFound)
{
filter = YES;
*stop = YES;
}
}];
}
if (!filter)
{
FileListNode* parent = nil;
NSUInteger previousIndex = !item.isFolder ?
[self findFileNode:item inList:self.fFileList
atIndexes:[NSIndexSet indexSetWithIndexesInRange:NSMakeRange(currentIndex, self.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)
{
[self.fFileList moveObjectAtIndex:previousIndex toIndex:currentIndex];
}
else
{
move = NO;
}
}
else
{
[self.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)
{
[self.fOutline moveItemAtIndex:previousIndex inParent:parent toIndex:currentIndex inParent:nil];
}
++currentIndex;
}
++totalCount;
}
}
//remove trailing items - those are the unused
if (currentIndex < self.fFileList.count)
{
NSRange const removeRange = NSMakeRange(currentIndex, self.fFileList.count - currentIndex);
[self.fFileList removeObjectsInRange:removeRange];
[self.fOutline removeItemsAtIndexes:[NSIndexSet indexSetWithIndexesInRange:removeRange] inParent:nil
withAnimation:NSTableViewAnimationSlideDown];
}
//add new items
[self.fFileList insertObjects:itemsToAdd atIndexes:itemsToAddIndexes];
[self.fOutline insertItemsAtIndexes:itemsToAddIndexes inParent:nil withAnimation:NSTableViewAnimationSlideUp];
[self.fOutline endUpdates];
_filterText = text;
}
- (void)refresh
{
self.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 self.fFileList ? self.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 : self.fFileList)[index];
}
- (id)outlineView:(NSOutlineView*)outlineView objectValueForTableColumn:(NSTableColumn*)tableColumn byItem:(id)item
{
if ([tableColumn.identifier isEqualToString:@"Check"])
{
return @([self.torrent 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:[self.torrent canChangeDownloadCheckForFiles:((FileListNode*)item).indexes]];
}
else if ([identifier isEqualToString:@"Priority"])
{
[cell setRepresentedObject:item];
NSInteger hoveredRow = self.fOutline.hoveredRow;
((FilePriorityCell*)cell).hovered = hoveredRow != -1 && hoveredRow == [self.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 & NSEventModifierFlagOption)
{
indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, self.torrent.fileCount)];
}
else
{
indexSet = ((FileListNode*)item).indexes;
}
[self.torrent setFileCheckState:[object intValue] != NSControlStateValueOff ? NSControlStateValueOn : NSControlStateValueOff
forIndexes:indexSet];
self.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 = [self.torrent 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 NSControlStateValueOff:
return NSLocalizedString(@"Don't Download", "files tab -> tooltip");
case NSControlStateValueOn:
return NSLocalizedString(@"Download", "files tab -> tooltip");
case NSControlStateValueMixed:
return NSLocalizedString(@"Download Some", "files tab -> tooltip");
}
}
else if ([ident isEqualToString:@"Priority"])
{
NSSet* priorities = [self.torrent 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 ? NSControlStateValueOff : NSControlStateValueOn;
NSIndexSet* indexSet = self.fOutline.selectedRowIndexes;
NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet];
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
FileListNode* item = [self.fOutline itemAtRow:i];
[itemIndexes addIndexes:item.indexes];
}
[self.torrent setFileCheckState:state forIndexes:itemIndexes];
self.fOutline.needsDisplay = YES;
}
- (void)setOnlySelectedCheck:(id)sender
{
NSIndexSet* indexSet = self.fOutline.selectedRowIndexes;
NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet];
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
FileListNode* item = [self.fOutline itemAtRow:i];
[itemIndexes addIndexes:item.indexes];
}
[self.torrent setFileCheckState:NSControlStateValueOn forIndexes:itemIndexes];
NSMutableIndexSet* remainingItemIndexes = [NSMutableIndexSet indexSetWithIndexesInRange:NSMakeRange(0, self.torrent.fileCount)];
[remainingItemIndexes removeIndexes:itemIndexes];
[self.torrent setFileCheckState:NSControlStateValueOff forIndexes:remainingItemIndexes];
self.fOutline.needsDisplay = YES;
}
- (void)checkAll
{
NSIndexSet* indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, self.torrent.fileCount)];
[self.torrent setFileCheckState:NSControlStateValueOn forIndexes:indexSet];
self.fOutline.needsDisplay = YES;
}
- (void)uncheckAll
{
NSIndexSet* indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, self.torrent.fileCount)];
[self.torrent setFileCheckState:NSControlStateValueOff forIndexes:indexSet];
self.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 = self.fOutline.selectedRowIndexes;
NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet];
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
FileListNode* item = [self.fOutline itemAtRow:i];
[itemIndexes addIndexes:item.indexes];
}
[self.torrent setFilePriority:priority forIndexes:itemIndexes];
self.fOutline.needsDisplay = YES;
}
- (void)revealFile:(id)sender
{
NSIndexSet* indexes = self.fOutline.selectedRowIndexes;
NSMutableArray* paths = [NSMutableArray arrayWithCapacity:indexes.count];
for (NSUInteger i = indexes.firstIndex; i != NSNotFound; i = [indexes indexGreaterThanIndex:i])
{
NSString* path = [self.torrent fileLocation:[self.fOutline itemAtRow:i]];
if (path)
{
[paths addObject:[NSURL fileURLWithPath:path]];
}
}
if (paths.count > 0)
{
[NSWorkspace.sharedWorkspace activateFileViewerSelectingURLs:paths];
}
}
- (void)renameSelected:(id)sender
{
NSIndexSet* indexes = self.fOutline.selectedRowIndexes;
NSAssert(indexes.count == 1, @"1 file needs to be selected to rename, but %ld are selected", indexes.count);
FileListNode* node = [self.fOutline itemAtRow:indexes.firstIndex];
Torrent* torrent = node.torrent;
if (!torrent.folder)
{
[FileRenameSheetController presentSheetForTorrent:torrent modalForWindow:self.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:self.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 (!self.torrent)
{
return NO;
}
SEL action = menuItem.action;
if (action == @selector(revealFile:))
{
NSIndexSet* indexSet = self.fOutline.selectedRowIndexes;
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
if ([self.torrent fileLocation:[self.fOutline itemAtRow:i]] != nil)
{
return YES;
}
}
return NO;
}
if (action == @selector(setCheck:))
{
if (self.fOutline.numberOfSelectedRows == 0)
{
return NO;
}
NSIndexSet* indexSet = self.fOutline.selectedRowIndexes;
NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet];
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
FileListNode* node = [self.fOutline itemAtRow:i];
[itemIndexes addIndexes:node.indexes];
}
NSInteger state = (menuItem.tag == FILE_CHECK_TAG) ? NSControlStateValueOn : NSControlStateValueOff;
return [self.torrent checkForFiles:itemIndexes] != state && [self.torrent canChangeDownloadCheckForFiles:itemIndexes];
}
if (action == @selector(setOnlySelectedCheck:))
{
if (self.fOutline.numberOfSelectedRows == 0)
{
return NO;
}
NSIndexSet* indexSet = self.fOutline.selectedRowIndexes;
NSMutableIndexSet* itemIndexes = [NSMutableIndexSet indexSet];
for (NSInteger i = indexSet.firstIndex; i != NSNotFound; i = [indexSet indexGreaterThanIndex:i])
{
FileListNode* node = [self.fOutline itemAtRow:i];
[itemIndexes addIndexes:node.indexes];
}
return [self.torrent canChangeDownloadCheckForFiles:itemIndexes];
}
if (action == @selector(setPriority:))
{
if (self.fOutline.numberOfSelectedRows == 0)
{
menuItem.state = NSControlStateValueOff;
return NO;
}
//determine which priorities are checked
NSIndexSet* indexSet = self.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 = [self.fOutline itemAtRow:i];
NSIndexSet* fileIndexSet = node.indexes;
if (![self.torrent canChangeDownloadCheckForFiles:fileIndexSet])
{
continue;
}
canChange = YES;
if ([self.torrent hasFilePriority:priority forIndexes:fileIndexSet])
{
current = YES;
break;
}
}
menuItem.state = current ? NSControlStateValueOn : NSControlStateValueOff;
return canChange;
}
if (action == @selector(renameSelected:))
{
return self.fOutline.numberOfSelectedRows == 1;
}
return YES;
}
#pragma mark - 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<FileListNode*>*)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