153 lines
4.2 KiB
Plaintext
153 lines
4.2 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 AppKit;
|
|
|
|
#import "FileListNode.h"
|
|
|
|
@interface FileListNode ()
|
|
|
|
@property(nonatomic, readonly) NSMutableIndexSet* indexesInternal;
|
|
@property(nonatomic) NSImage* iconInternal;
|
|
|
|
- (instancetype)initWithFolder:(BOOL)isFolder name:(NSString*)name path:(NSString*)path torrent:(Torrent*)torrent;
|
|
|
|
@end
|
|
|
|
@implementation FileListNode
|
|
|
|
- (instancetype)initWithFolderName:(NSString*)name path:(NSString*)path torrent:(Torrent*)torrent
|
|
{
|
|
if ((self = [self initWithFolder:YES name:name path:path torrent:torrent]))
|
|
{
|
|
_children = [[NSMutableArray alloc] init];
|
|
_size = 0;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithFileName:(NSString*)name
|
|
path:(NSString*)path
|
|
size:(uint64_t)size
|
|
index:(NSUInteger)index
|
|
torrent:(Torrent*)torrent
|
|
{
|
|
if ((self = [self initWithFolder:NO name:name path:path torrent:torrent]))
|
|
{
|
|
_size = size;
|
|
[_indexesInternal addIndex:index];
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)insertChild:(FileListNode*)child
|
|
{
|
|
NSAssert(_isFolder, @"method can only be invoked on folders");
|
|
|
|
[_children addObject:child];
|
|
}
|
|
|
|
- (void)insertIndex:(NSUInteger)index withSize:(uint64_t)size
|
|
{
|
|
NSAssert(_isFolder, @"method can only be invoked on folders");
|
|
|
|
[self.indexesInternal addIndex:index];
|
|
_size += size;
|
|
}
|
|
|
|
- (id)copyWithZone:(NSZone*)zone
|
|
{
|
|
//this object is essentially immutable after initial setup
|
|
return self;
|
|
}
|
|
|
|
- (NSString*)description
|
|
{
|
|
if (!_isFolder)
|
|
{
|
|
return [NSString stringWithFormat:@"%@ (%ld)", _name, _indexesInternal.firstIndex];
|
|
}
|
|
else
|
|
{
|
|
return [NSString stringWithFormat:@"%@ (folder: %@)", _name, _indexesInternal];
|
|
}
|
|
}
|
|
|
|
- (NSImage*)icon
|
|
{
|
|
if (!_iconInternal)
|
|
{
|
|
_iconInternal = [NSWorkspace.sharedWorkspace
|
|
iconForFileType:_isFolder ? NSFileTypeForHFSTypeCode(kGenericFolderIcon) : _name.pathExtension];
|
|
}
|
|
return _iconInternal;
|
|
}
|
|
|
|
- (NSIndexSet*)indexes
|
|
{
|
|
return _indexesInternal;
|
|
}
|
|
|
|
- (BOOL)updateFromOldName:(NSString*)oldName toNewName:(NSString*)newName inPath:(NSString*)path
|
|
{
|
|
NSParameterAssert(oldName != nil);
|
|
NSParameterAssert(newName != nil);
|
|
NSParameterAssert(path != nil);
|
|
|
|
NSArray* lookupPathComponents = path.pathComponents;
|
|
NSArray* thesePathComponents = self.path.pathComponents;
|
|
|
|
if ([lookupPathComponents isEqualToArray:thesePathComponents]) //this node represents what's being renamed
|
|
{
|
|
if ([oldName isEqualToString:self.name])
|
|
{
|
|
_name = [newName copy];
|
|
_iconInternal = nil;
|
|
return YES;
|
|
}
|
|
}
|
|
else if (lookupPathComponents.count < thesePathComponents.count) //what's being renamed is part of this node's path
|
|
{
|
|
lookupPathComponents = [lookupPathComponents arrayByAddingObject:oldName];
|
|
BOOL const allSame = NSNotFound ==
|
|
[lookupPathComponents indexOfObjectWithOptions:NSEnumerationConcurrent
|
|
passingTest:^BOOL(NSString* name, NSUInteger idx, BOOL* stop) {
|
|
return ![name isEqualToString:thesePathComponents[idx]];
|
|
}];
|
|
|
|
if (allSame)
|
|
{
|
|
NSString* oldPathPrefix = [path stringByAppendingPathComponent:oldName];
|
|
NSString* newPathPrefix = [path stringByAppendingPathComponent:newName];
|
|
|
|
_path = [_path stringByReplacingCharactersInRange:NSMakeRange(0, oldPathPrefix.length) withString:newPathPrefix];
|
|
return YES;
|
|
}
|
|
}
|
|
|
|
return NO;
|
|
}
|
|
|
|
#pragma mark - Private
|
|
|
|
- (instancetype)initWithFolder:(BOOL)isFolder name:(NSString*)name path:(NSString*)path torrent:(Torrent*)torrent
|
|
{
|
|
if ((self = [super init]))
|
|
{
|
|
_isFolder = isFolder;
|
|
_name = [name copy];
|
|
_path = [path copy];
|
|
|
|
_indexesInternal = [[NSMutableIndexSet alloc] init];
|
|
|
|
_torrent = torrent;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
@end
|