transmission/macosx/Torrent.m

1851 lines
58 KiB
Mathematica
Raw Normal View History

2007-09-16 01:02:06 +00:00
/******************************************************************************
* $Id$
*
2010-01-01 21:12:04 +00:00
* Copyright (c) 2006-2010 Transmission authors and contributors
2007-09-16 01:02:06 +00:00
*
* 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 "Torrent.h"
#import "GroupsController.h"
#import "FileListNode.h"
2007-09-16 01:02:06 +00:00
#import "NSStringAdditions.h"
#import "TrackerNode.h"
#import "transmission.h" // required by utils.h
#import "utils.h" // tr_new()
2007-09-16 01:02:06 +00:00
#define ETA_IDLE_DISPLAY_MIN 2
2007-09-16 01:02:06 +00:00
@interface Torrent (Private)
2009-12-13 01:36:22 +00:00
- (id) initWithPath: (NSString *) path hash: (NSString *) hashString torrentStruct: (tr_torrent *) torrentStruct
magnetAddress: (NSString *) magnetAddress lib: (tr_session *) lib
2007-12-17 16:06:20 +00:00
waitToStart: (NSNumber *) waitToStart
2009-10-22 00:10:41 +00:00
groupValue: (NSNumber *) groupValue
2009-12-11 04:01:47 +00:00
downloadFolder: (NSString *) downloadFolder legacyIncompleteFolder: (NSString *) incompleteFolder;
2007-09-16 01:02:06 +00:00
- (void) createFileList;
- (void) insertPath: (NSMutableArray *) components forParent: (FileListNode *) parent fileSize: (uint64_t) size
index: (NSInteger) index flatList: (NSMutableArray *) flatFileList;
2007-09-16 01:02:06 +00:00
- (void) completenessChange: (NSDictionary *) statusInfo;
- (void) ratioLimitHit;
- (void) idleLimitHit;
- (void) metadataRetrieved;
- (BOOL) shouldShowEta;
- (NSString *) etaString;
2008-04-07 02:55:33 +00:00
- (void) setTimeMachineExclude: (BOOL) exclude forPath: (NSString *) path;
2007-09-16 01:02:06 +00:00
@end
void completenessChangeCallback(tr_torrent * torrent, tr_completeness status, tr_bool wasRunning, void * torrentData)
{
NSDictionary * dict = [[NSDictionary alloc] initWithObjectsAndKeys: [NSNumber numberWithInt: status], @"Status",
[NSNumber numberWithBool: wasRunning], @"WasRunning", nil];
[(Torrent *)torrentData performSelectorOnMainThread: @selector(completenessChange:) withObject: dict waitUntilDone: NO];
}
void ratioLimitHitCallback(tr_torrent * torrent, void * torrentData)
{
[(Torrent *)torrentData performSelectorOnMainThread: @selector(ratioLimitHit) withObject: nil waitUntilDone: NO];
}
void idleLimitHitCallback(tr_torrent * torrent, void * torrentData)
{
[(Torrent *)torrentData performSelectorOnMainThread: @selector(idleLimitHit) withObject: nil waitUntilDone: NO];
}
void metadataCallback(tr_torrent * torrent, void * torrentData)
{
[(Torrent *)torrentData performSelectorOnMainThread: @selector(metadataRetrieved) withObject: nil waitUntilDone: NO];
}
int trashDataFile(const char * filename)
{
if (filename != NULL)
[Torrent trashFile: [NSString stringWithUTF8String: filename]];
return 0;
}
2007-09-16 01:02:06 +00:00
@implementation Torrent
- (id) initWithPath: (NSString *) path location: (NSString *) location deleteTorrentFile: (BOOL) torrentDelete
lib: (tr_session *) lib
2007-09-16 01:02:06 +00:00
{
2009-12-13 01:36:22 +00:00
self = [self initWithPath: path hash: nil torrentStruct: NULL magnetAddress: nil lib: lib
2009-10-22 00:10:41 +00:00
waitToStart: nil groupValue: nil
2009-12-11 04:01:47 +00:00
downloadFolder: location legacyIncompleteFolder: nil];
2007-09-16 01:02:06 +00:00
if (self)
{
if (torrentDelete && ![[self torrentLocation] isEqualToString: path])
[Torrent trashFile: path];
2007-09-16 01:02:06 +00:00
}
return self;
}
- (id) initWithTorrentStruct: (tr_torrent *) torrentStruct location: (NSString *) location lib: (tr_session *) lib
{
2009-12-13 01:36:22 +00:00
self = [self initWithPath: nil hash: nil torrentStruct: torrentStruct magnetAddress: nil lib: lib
2009-10-22 00:10:41 +00:00
waitToStart: nil groupValue: nil
2009-12-11 04:01:47 +00:00
downloadFolder: location legacyIncompleteFolder: nil];
return self;
}
- (id) initWithMagnetAddress: (NSString *) address location: (NSString *) location lib: (tr_session *) lib
{
2009-12-13 01:36:22 +00:00
self = [self initWithPath: nil hash: nil torrentStruct: nil magnetAddress: address
lib: lib waitToStart: nil groupValue: nil
downloadFolder: location legacyIncompleteFolder: nil];
return self;
}
- (id) initWithHistory: (NSDictionary *) history lib: (tr_session *) lib forcePause: (BOOL) pause
2007-09-16 01:02:06 +00:00
{
self = [self initWithPath: [history objectForKey: @"InternalTorrentPath"]
hash: [history objectForKey: @"TorrentHash"]
2009-12-13 01:36:22 +00:00
torrentStruct: NULL
magnetAddress: nil
lib: lib
2007-09-16 01:02:06 +00:00
waitToStart: [history objectForKey: @"WaitToStart"]
2009-10-22 00:10:41 +00:00
groupValue: [history objectForKey: @"GroupValue"]
2009-12-11 04:01:47 +00:00
downloadFolder: [history objectForKey: @"DownloadFolder"] //upgrading from versions < 1.80
2009-10-22 00:10:41 +00:00
legacyIncompleteFolder: [[history objectForKey: @"UseIncompleteFolder"] boolValue] //upgrading from versions < 1.80
? [history objectForKey: @"IncompleteFolder"] : nil];
2007-09-16 01:02:06 +00:00
if (self)
{
//start transfer
NSNumber * active;
if (!pause && (active = [history objectForKey: @"Active"]) && [active boolValue])
2007-09-16 01:02:06 +00:00
{
fStat = tr_torrentStat(fHandle);
[self startTransfer];
}
2008-08-09 20:20:21 +00:00
//upgrading from versions < 1.30: get old added, activity, and done dates
NSDate * date;
if ((date = [history objectForKey: @"Date"]))
tr_torrentSetAddedDate(fHandle, [date timeIntervalSince1970]);
if ((date = [history objectForKey: @"DateActivity"]))
tr_torrentSetActivityDate(fHandle, [date timeIntervalSince1970]);
if ((date = [history objectForKey: @"DateCompleted"]))
tr_torrentSetDoneDate(fHandle, [date timeIntervalSince1970]);
//upgrading from versions < 1.60: get old stop ratio settings
NSNumber * ratioSetting;
if ((ratioSetting = [history objectForKey: @"RatioSetting"]))
{
switch ([ratioSetting intValue])
{
case NSOnState: [self setRatioSetting: TR_RATIOLIMIT_SINGLE]; break;
case NSOffState: [self setRatioSetting: TR_RATIOLIMIT_UNLIMITED]; break;
case NSMixedState: [self setRatioSetting: TR_RATIOLIMIT_GLOBAL]; break;
}
}
NSNumber * ratioLimit;
if ((ratioLimit = [history objectForKey: @"RatioLimit"]))
[self setRatioLimit: [ratioLimit floatValue]];
2007-09-16 01:02:06 +00:00
}
return self;
}
- (NSDictionary *) history
{
return [NSDictionary dictionaryWithObjectsAndKeys:
[self torrentLocation], @"InternalTorrentPath",
[self hashString], @"TorrentHash",
[NSNumber numberWithBool: [self isActive]], @"Active",
[NSNumber numberWithBool: fWaitToStart], @"WaitToStart",
[NSNumber numberWithInt: fGroupValue], @"GroupValue", nil];
2007-09-16 01:02:06 +00:00
}
- (void) dealloc
{
2007-12-17 16:06:20 +00:00
[[NSNotificationCenter defaultCenter] removeObserver: self];
if (fFileStat)
tr_torrentFilesFree(fFileStat, [self fileCount]);
2007-09-16 01:02:06 +00:00
[fPreviousFinishedIndexes release];
[fPreviousFinishedIndexesDate release];
[fHashString release];
2007-09-16 01:02:06 +00:00
[fIcon release];
[fFileList release];
[fFlatFileList release];
2007-09-16 01:02:06 +00:00
[fTimeMachineExclude release];
2007-09-16 01:02:06 +00:00
[super dealloc];
}
2008-05-12 01:32:33 +00:00
- (NSString *) description
{
return [@"Torrent: " stringByAppendingString: [self name]];
}
- (id) copyWithZone: (NSZone *) zone
{
return [self retain];
}
- (void) closeRemoveTorrent
2007-09-16 01:02:06 +00:00
{
//allow the file to be index by Time Machine
if (fTimeMachineExclude)
2009-10-21 13:03:37 +00:00
{
[self setTimeMachineExclude: NO forPath: fTimeMachineExclude];
2009-10-21 13:03:37 +00:00
[fTimeMachineExclude release];
fTimeMachineExclude = nil;
}
tr_torrentRemove(fHandle);
2007-09-16 01:02:06 +00:00
}
- (void) changeDownloadFolderBeforeUsing: (NSString *) folder
2007-09-16 01:02:06 +00:00
{
tr_torrentSetDownloadDir(fHandle, [folder UTF8String]);
[self updateTimeMachineExclude];
2007-09-16 01:02:06 +00:00
}
- (NSString *) currentDirectory
2007-09-16 01:02:06 +00:00
{
return [NSString stringWithUTF8String: tr_torrentGetCurrentDir(fHandle)];
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (void) getAvailability: (int8_t *) tab size: (NSInteger) size
2007-09-16 01:02:06 +00:00
{
tr_torrentAvailability(fHandle, tab, size);
}
2008-11-02 01:07:01 +00:00
- (void) getAmountFinished: (float *) tab size: (NSInteger) size
2007-09-16 01:02:06 +00:00
{
tr_torrentAmountFinished(fHandle, tab, size);
}
- (NSIndexSet *) previousFinishedPieces
{
//if the torrent hasn't been seen in a bit, and therefore hasn't been refreshed, return nil
if (fPreviousFinishedIndexesDate && [fPreviousFinishedIndexesDate timeIntervalSinceNow] > -2.0)
return fPreviousFinishedIndexes;
else
return nil;
}
-(void) setPreviousFinishedPieces: (NSIndexSet *) indexes
{
[fPreviousFinishedIndexes release];
fPreviousFinishedIndexes = [indexes retain];
[fPreviousFinishedIndexesDate release];
fPreviousFinishedIndexesDate = indexes != nil ? [[NSDate alloc] init] : nil;
}
2007-09-16 01:02:06 +00:00
- (void) update
{
2008-01-14 04:50:19 +00:00
//get previous status values before update
BOOL wasChecking = NO, wasError = NO, wasStalled = NO;
if (fStat != NULL)
{
wasChecking = [self isChecking];
wasError = [self isError];
wasStalled = fStalled;
}
2008-01-14 04:50:19 +00:00
2007-09-16 01:02:06 +00:00
fStat = tr_torrentStat(fHandle);
2008-01-14 04:50:19 +00:00
//check if stalled (stored because based on time and needs to check if it was previously stalled)
2007-09-16 01:02:06 +00:00
fStalled = [self isActive] && [fDefaults boolForKey: @"CheckStalled"]
&& [self stalledMinutes] > [fDefaults integerForKey: @"StalledMinutes"];
2007-09-16 01:02:06 +00:00
//update queue for checking (from downloading to seeding), stalled, or error
2008-01-14 04:50:19 +00:00
if ((wasChecking && ![self isChecking]) || (wasStalled != fStalled) || (!wasError && [self isError] && [self isActive]))
2007-09-16 01:02:06 +00:00
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateQueue" object: self];
//when the data first appears, update time machine exclusion
if (!fTimeMachineExclude)
[self updateTimeMachineExclude];
2007-09-16 01:02:06 +00:00
}
- (void) startTransfer
{
fWaitToStart = NO;
if (![self isActive] && [self alertForRemainingDiskSpace])
2007-09-16 01:02:06 +00:00
{
2010-04-14 13:54:18 +00:00
tr_ninf( fInfo->name, "restarting via startTransfer" );
2007-09-16 01:02:06 +00:00
tr_torrentStart(fHandle);
[self update];
//capture, specifically, stop-seeding settings changing to unlimited
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateOptions" object: nil];
2007-09-16 01:02:06 +00:00
}
}
- (void) stopTransfer
{
fWaitToStart = NO;
if ([self isActive])
{
tr_torrentStop(fHandle);
[self update];
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateQueue" object: self];
}
}
- (void) sleep
{
if ((fResumeOnWake = [self isActive]))
tr_torrentStop(fHandle);
}
- (void) wakeUp
{
if (fResumeOnWake)
{
2010-04-14 13:54:18 +00:00
tr_ninf( fInfo->name, "restarting because of wakeUp" );
2007-09-16 01:02:06 +00:00
tr_torrentStart(fHandle);
}
2007-09-16 01:02:06 +00:00
}
- (void) manualAnnounce
{
tr_torrentManualUpdate(fHandle);
2007-09-16 01:02:06 +00:00
}
- (BOOL) canManualAnnounce
{
return tr_torrentCanManualUpdate(fHandle);
}
- (void) resetCache
{
2008-02-27 17:38:39 +00:00
tr_torrentVerify(fHandle);
2007-09-16 01:02:06 +00:00
[self update];
}
- (BOOL) isMagnet
{
return !tr_torrentHasMetadata(fHandle);
}
- (NSString *) magnetLink
{
return [NSString stringWithUTF8String: tr_torrentGetMagnetLink(fHandle)];
}
2008-11-02 01:07:01 +00:00
- (CGFloat) ratio
2007-09-16 01:02:06 +00:00
{
return fStat->ratio;
}
- (tr_ratiolimit) ratioSetting
2007-09-16 01:02:06 +00:00
{
return tr_torrentGetRatioMode(fHandle);
2007-09-16 01:02:06 +00:00
}
- (void) setRatioSetting: (tr_ratiolimit) setting
2007-09-16 01:02:06 +00:00
{
tr_torrentSetRatioMode(fHandle, setting);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) ratioLimit
2007-09-16 01:02:06 +00:00
{
return tr_torrentGetRatioLimit(fHandle);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (void) setRatioLimit: (CGFloat) limit
2007-09-16 01:02:06 +00:00
{
NSAssert(limit >= 0, @"Ratio cannot be negative");
tr_torrentSetRatioLimit(fHandle, limit);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) progressStopRatio
2008-06-24 04:00:39 +00:00
{
2010-04-14 00:03:34 +00:00
return fStat->seedRatioPercentDone;
2007-09-16 01:02:06 +00:00
}
- (tr_idlelimit) idleSetting
{
return tr_torrentGetIdleMode(fHandle);
}
- (void) setIdleSetting: (tr_idlelimit) setting
{
tr_torrentSetIdleMode(fHandle, setting);
}
- (NSUInteger) idleLimitMinutes
{
return tr_torrentGetIdleLimit(fHandle);
}
- (void) setIdleLimitMinutes: (NSUInteger) limit
{
NSAssert(limit > 0, @"Idle limit must be greater than zero");
tr_torrentSetIdleLimit(fHandle, limit);
}
- (BOOL) usesSpeedLimit: (BOOL) upload
2007-09-16 01:02:06 +00:00
{
return tr_torrentUsesSpeedLimit(fHandle, upload ? TR_UP : TR_DOWN);
2007-09-16 01:02:06 +00:00
}
- (void) setUseSpeedLimit: (BOOL) use upload: (BOOL) upload
2007-09-16 01:02:06 +00:00
{
tr_torrentUseSpeedLimit(fHandle, upload ? TR_UP : TR_DOWN, use);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (NSInteger) speedLimit: (BOOL) upload
2007-09-16 01:02:06 +00:00
{
2010-07-06 03:31:17 +00:00
return tr_torrentGetSpeedLimit_KBps(fHandle, upload ? TR_UP : TR_DOWN);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (void) setSpeedLimit: (NSInteger) limit upload: (BOOL) upload
2007-09-16 01:02:06 +00:00
{
2010-07-06 03:31:17 +00:00
tr_torrentSetSpeedLimit_KBps(fHandle, upload ? TR_UP : TR_DOWN, limit);
2007-09-16 01:02:06 +00:00
}
- (BOOL) usesGlobalSpeedLimit
{
return tr_torrentUsesSessionLimits(fHandle);
}
- (void) setUseGlobalSpeedLimit: (BOOL) use
{
tr_torrentUseSessionLimits(fHandle, use);
}
- (void) setMaxPeerConnect: (uint16_t) count
{
2008-06-17 17:17:15 +00:00
NSAssert(count > 0, @"max peer count must be greater than 0");
tr_torrentSetPeerLimit(fHandle, count);
}
- (uint16_t) maxPeerConnect
{
2008-05-12 16:39:32 +00:00
return tr_torrentGetPeerLimit(fHandle);
}
2007-09-16 01:02:06 +00:00
- (void) setWaitToStart: (BOOL) wait
{
fWaitToStart = wait;
}
- (BOOL) waitingToStart
{
return fWaitToStart;
}
- (tr_priority_t) priority
{
return tr_torrentGetPriority(fHandle);
}
- (void) setPriority: (tr_priority_t) priority
{
return tr_torrentSetPriority(fHandle, priority);
}
#warning when 10.6-only use recycleURLs:completionHandler:
+ (void) trashFile: (NSString *) path
2007-09-16 01:02:06 +00:00
{
//attempt to move to trash
if (![[NSWorkspace sharedWorkspace] performFileOperation: NSWorkspaceRecycleOperation
source: [path stringByDeletingLastPathComponent] destination: @""
files: [NSArray arrayWithObject: [path lastPathComponent]] tag: nil])
{
//if cannot trash, just delete it (will work if it's on a remote volume)
NSError * error;
if (![[NSFileManager defaultManager] removeItemAtPath: path error: &error])
NSLog(@"Could not trash %@: %@", path, [error localizedDescription]);
}
2007-09-16 01:02:06 +00:00
}
- (void) trashData
{
tr_torrentDeleteLocalData(fHandle, trashDataFile);
2007-09-16 01:02:06 +00:00
}
- (void) moveTorrentDataFileTo: (NSString *) folder
{
NSString * oldFolder = [self currentDirectory];
if ([oldFolder isEqualToString: folder])
return;
//check if moving inside itself
NSArray * oldComponents = [oldFolder pathComponents],
* newComponents = [folder pathComponents];
const NSInteger oldCount = [oldComponents count];
if (oldCount < [newComponents count] && [[newComponents objectAtIndex: oldCount] isEqualToString: [self name]]
&& [folder hasPrefix: oldFolder])
2007-09-16 01:02:06 +00:00
{
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: NSLocalizedString(@"A folder cannot be moved to inside itself.",
"Move inside itself alert -> title")];
[alert setInformativeText: [NSString stringWithFormat:
NSLocalizedString(@"The move operation of \"%@\" cannot be done.",
"Move inside itself alert -> message"), [self name]]];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Move inside itself alert -> button")];
2007-09-16 01:02:06 +00:00
[alert runModal];
[alert release];
2007-09-16 01:02:06 +00:00
return;
}
volatile int status;
tr_torrentSetLocation(fHandle, [folder UTF8String], YES, NULL, &status);
while (status == TR_LOC_MOVING) //block while moving (for now)
[NSThread sleepForTimeInterval: 0.05];
2009-10-21 22:23:08 +00:00
if (status == TR_LOC_DONE)
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateStats" object: nil];
else
{
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: NSLocalizedString(@"There was an error moving the data file.", "Move error alert -> title")];
[alert setInformativeText: [NSString stringWithFormat:
NSLocalizedString(@"The move operation of \"%@\" cannot be done.", "Move error alert -> message"), [self name]]];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Move error alert -> button")];
2007-09-16 01:02:06 +00:00
[alert runModal];
[alert release];
2007-09-16 01:02:06 +00:00
}
[self updateTimeMachineExclude];
2007-09-16 01:02:06 +00:00
}
- (void) copyTorrentFileTo: (NSString *) path
{
[[NSFileManager defaultManager] copyItemAtPath: [self torrentLocation] toPath: path error: NULL];
2007-09-16 01:02:06 +00:00
}
- (BOOL) alertForRemainingDiskSpace
{
if ([self allDownloaded] || ![fDefaults boolForKey: @"WarningRemainingSpace"])
return YES;
NSString * downloadFolder = [self currentDirectory], * volumeName;
2009-10-31 02:49:59 +00:00
if ((volumeName = [[[NSFileManager defaultManager] componentsToDisplayForPath: downloadFolder] objectAtIndex: 0]))
2007-09-16 01:02:06 +00:00
{
NSDictionary * systemAttributes = [[NSFileManager defaultManager] attributesOfFileSystemForPath: downloadFolder error: NULL];
uint64_t remainingSpace = [[systemAttributes objectForKey: NSFileSystemFreeSize] unsignedLongLongValue];
2007-09-16 01:02:06 +00:00
2008-11-06 03:45:56 +00:00
//if the remaining space is greater than the size left, then there is enough space regardless of preallocation
if (remainingSpace < [self sizeLeft] && remainingSpace < tr_torrentGetBytesLeftToAllocate(fHandle))
{
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: [NSString stringWithFormat:
NSLocalizedString(@"Not enough remaining disk space to download \"%@\" completely.",
"Torrent disk space alert -> title"), [self name]]];
[alert setInformativeText: [NSString stringWithFormat: NSLocalizedString(@"The transfer will be paused."
" Clear up space on %@ or deselect files in the torrent inspector to continue.",
"Torrent disk space alert -> message"), volumeName]];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Torrent disk space alert -> button")];
[alert addButtonWithTitle: NSLocalizedString(@"Download Anyway", "Torrent disk space alert -> button")];
2008-12-26 05:57:51 +00:00
[alert setShowsSuppressionButton: YES];
[[alert suppressionButton] setTitle: NSLocalizedString(@"Do not check disk space again",
"Torrent disk space alert -> button")];
2009-10-25 16:31:28 +00:00
const NSInteger result = [alert runModal];
2008-12-26 05:57:51 +00:00
if ([[alert suppressionButton] state] == NSOnState)
[fDefaults setBool: NO forKey: @"WarningRemainingSpace"];
[alert release];
return result != NSAlertFirstButtonReturn;
}
2007-09-16 01:02:06 +00:00
}
return YES;
}
- (NSImage *) icon
{
2009-11-26 02:26:51 +00:00
if ([self isMagnet])
2009-12-20 14:44:56 +00:00
return [NSImage imageNamed: @"Magnet.png"];
2009-11-26 02:26:51 +00:00
#warning replace 'fldr' stuff with NSImageNameFolder on 10.6
2007-09-16 01:02:06 +00:00
if (!fIcon)
fIcon = [[[NSWorkspace sharedWorkspace] iconForFileType: [self isFolder] ? NSFileTypeForHFSTypeCode('fldr')
: [[self name] pathExtension]] retain];
2007-09-16 01:02:06 +00:00
return fIcon;
}
- (NSString *) name
{
return fInfo->name != NULL ? [NSString stringWithUTF8String: fInfo->name] : fHashString;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isFolder
{
return fInfo->isMultifile;
}
2007-09-16 01:02:06 +00:00
- (uint64_t) size
{
return fInfo->totalSize;
}
- (uint64_t) sizeLeft
{
return fStat->leftUntilDone;
}
- (NSMutableArray *) allTrackerStats
{
int count;
tr_tracker_stat * stats = tr_torrentTrackers(fHandle, &count);
NSMutableArray * trackers = [NSMutableArray arrayWithCapacity: (count > 0 ? count + (stats[count-1].tier + 1) : 0)];
int prevTier = -1;
for (int i=0; i < count; ++i)
{
if (stats[i].tier != prevTier)
2008-04-24 06:42:45 +00:00
{
[trackers addObject: [NSDictionary dictionaryWithObjectsAndKeys: [NSNumber numberWithInteger: stats[i].tier + 1], @"Tier",
[self name], @"Name", nil]];
prevTier = stats[i].tier;
2008-04-24 06:42:45 +00:00
}
TrackerNode * tracker = [[TrackerNode alloc] initWithTrackerStat: &stats[i] torrent: self];
[trackers addObject: tracker];
[tracker release];
}
tr_torrentTrackersFree(stats, count);
return trackers;
}
2009-10-11 17:17:29 +00:00
- (NSArray *) allTrackersFlat
2008-12-24 17:41:45 +00:00
{
NSMutableArray * allTrackers = [NSMutableArray arrayWithCapacity: fInfo->trackerCount];
for (NSInteger i=0; i < fInfo->trackerCount; i++)
[allTrackers addObject: [NSString stringWithUTF8String: fInfo->trackers[i].announce]];
return allTrackers;
2008-12-24 17:41:45 +00:00
}
- (BOOL) addTrackerToNewTier: (NSString *) tracker
{
tracker = [tracker stringByTrimmingCharactersInSet: [NSCharacterSet whitespaceAndNewlineCharacterSet]];
if ([tracker rangeOfString: @"://"].location == NSNotFound)
tracker = [@"http://" stringByAppendingString: tracker];
//recreate the tracker structure
const int oldTrackerCount = fInfo->trackerCount;
tr_tracker_info * trackerStructs = tr_new(tr_tracker_info, oldTrackerCount+1);
for (NSUInteger i=0; i < oldTrackerCount; ++i)
trackerStructs[i] = fInfo->trackers[i];
trackerStructs[oldTrackerCount].announce = (char *)[tracker UTF8String];
trackerStructs[oldTrackerCount].tier = trackerStructs[oldTrackerCount-1].tier + 1;
trackerStructs[oldTrackerCount].id = oldTrackerCount;
2010-01-05 04:34:31 +00:00
const BOOL success = tr_torrentSetAnnounceList(fHandle, trackerStructs, oldTrackerCount+1);
tr_free(trackerStructs);
2010-01-05 04:34:31 +00:00
return success;
}
- (void) removeTrackersWithIdentifiers: (NSIndexSet *) removeIdentifiers
{
//recreate the tracker structure
tr_tracker_info * trackerStructs = tr_new(tr_tracker_info, fInfo->trackerCount);
NSUInteger newCount = 0;
for (NSUInteger i = 0; i < fInfo->trackerCount; i++)
{
if (![removeIdentifiers containsIndex: fInfo->trackers[i].id])
trackerStructs[newCount++] = fInfo->trackers[i];
}
2010-01-05 04:34:31 +00:00
const BOOL success = tr_torrentSetAnnounceList(fHandle, trackerStructs, newCount);
NSAssert(success, @"Removing tracker addresses failed");
tr_free(trackerStructs);
}
2007-09-16 01:02:06 +00:00
- (NSString *) comment
{
return fInfo->comment ? [NSString stringWithUTF8String: fInfo->comment] : @"";
2007-09-16 01:02:06 +00:00
}
- (NSString *) creator
{
return fInfo->creator ? [NSString stringWithUTF8String: fInfo->creator] : @"";
2007-09-16 01:02:06 +00:00
}
- (NSDate *) dateCreated
{
2008-11-02 01:07:01 +00:00
NSInteger date = fInfo->dateCreated;
2007-09-16 01:02:06 +00:00
return date > 0 ? [NSDate dateWithTimeIntervalSince1970: date] : nil;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) pieceSize
2007-09-16 01:02:06 +00:00
{
return fInfo->pieceSize;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) pieceCount
2007-09-16 01:02:06 +00:00
{
return fInfo->pieceCount;
}
- (NSString *) hashString
{
return fHashString;
2007-09-16 01:02:06 +00:00
}
- (BOOL) privateTorrent
{
2007-10-15 21:52:51 +00:00
return fInfo->isPrivate;
2007-09-16 01:02:06 +00:00
}
- (NSString *) torrentLocation
{
2009-11-26 02:11:00 +00:00
return fInfo->torrent ? [NSString stringWithUTF8String: fInfo->torrent] : @"";
2007-09-16 01:02:06 +00:00
}
- (NSString *) dataLocation
{
if ([self isMagnet])
return nil;
if ([self isFolder])
{
NSString * dataLocation = [[self currentDirectory] stringByAppendingPathComponent: [self name]];
if (![[NSFileManager defaultManager] fileExistsAtPath: dataLocation])
return nil;
return dataLocation;
}
else
{
char * location = tr_torrentFindFile(fHandle, 0);
if (location == NULL)
return nil;
NSString * dataLocation = [NSString stringWithUTF8String: location];
free(location);
return dataLocation;
}
}
- (NSString *) fileLocation: (FileListNode *) node
{
if ([node isFolder])
{
NSString * basePath = [[node path] stringByAppendingPathComponent: [node name]];
NSString * dataLocation = [[self currentDirectory] stringByAppendingPathComponent: basePath];
if (![[NSFileManager defaultManager] fileExistsAtPath: dataLocation])
return nil;
return dataLocation;
}
else
{
char * location = tr_torrentFindFile(fHandle, [[node indexes] firstIndex]);
if (location == NULL)
return nil;
NSString * dataLocation = [NSString stringWithUTF8String: location];
free(location);
return dataLocation;
}
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) progress
2007-09-16 01:02:06 +00:00
{
2010-05-31 14:07:35 +00:00
return fStat->percentComplete;
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) progressDone
2007-09-16 01:02:06 +00:00
{
return fStat->percentDone;
}
- (CGFloat) progressLeft
{
if ([self size] == 0) //magnet links
return 0.0;
return (CGFloat)[self sizeLeft] / [self size];
}
2008-11-02 01:07:01 +00:00
- (CGFloat) checkingProgress
{
return fStat->recheckProgress;
}
- (CGFloat) availableDesired
2007-10-15 18:44:39 +00:00
{
2009-09-06 21:19:19 +00:00
return (CGFloat)fStat->desiredAvailable / [self sizeLeft];
}
2007-09-16 01:02:06 +00:00
- (BOOL) isActive
{
return fStat->activity != TR_STATUS_STOPPED;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isSeeding
{
return fStat->activity == TR_STATUS_SEED;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isChecking
{
return fStat->activity == TR_STATUS_CHECK || fStat->activity == TR_STATUS_CHECK_WAIT;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isCheckingWaiting
{
return fStat->activity == TR_STATUS_CHECK_WAIT;
}
2007-09-16 01:02:06 +00:00
- (BOOL) allDownloaded
{
return [self sizeLeft] == 0 && ![self isMagnet];
2007-09-16 01:02:06 +00:00
}
- (BOOL) isComplete
{
2007-10-15 18:20:39 +00:00
return [self progress] >= 1.0;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isFinishedSeeding
{
return fStat->finished;
}
2007-09-16 01:02:06 +00:00
- (BOOL) isError
{
return fStat->error == TR_STAT_LOCAL_ERROR;
2009-08-05 01:55:37 +00:00
}
- (BOOL) isAnyErrorOrWarning
2009-08-05 01:55:37 +00:00
{
return fStat->error != TR_STAT_OK;
2007-09-16 01:02:06 +00:00
}
- (NSString *) errorMessage
{
if (![self isAnyErrorOrWarning])
2007-09-16 01:02:06 +00:00
return @"";
NSString * error;
if (!(error = [NSString stringWithUTF8String: fStat->errorString])
&& !(error = [NSString stringWithCString: fStat->errorString encoding: NSISOLatin1StringEncoding]))
error = [NSString stringWithFormat: @"(%@)", NSLocalizedString(@"unreadable error", "Torrent -> error string unreadable")];
2007-09-16 01:02:06 +00:00
//libtransmission uses "Set Location", Mac client uses "Move data file to..." - very hacky!
error = [error stringByReplacingOccurrencesOfString: @"Set Location" withString: [@"Move Data File To" stringByAppendingEllipsis]];
2007-09-16 01:02:06 +00:00
return error;
}
- (NSArray *) peers
{
int totalPeers;
2007-09-20 20:24:33 +00:00
tr_peer_stat * peers = tr_torrentPeers(fHandle, &totalPeers);
2007-09-16 01:02:06 +00:00
NSMutableArray * peerDicts = [NSMutableArray arrayWithCapacity: totalPeers];
2007-09-16 01:02:06 +00:00
for (int i = 0; i < totalPeers; i++)
2007-09-16 01:02:06 +00:00
{
tr_peer_stat * peer = &peers[i];
NSMutableDictionary * dict = [NSMutableDictionary dictionaryWithCapacity: 11];
2007-09-16 01:02:06 +00:00
[dict setObject: [self name] forKey: @"Name"];
[dict setObject: [NSNumber numberWithInt: peer->from] forKey: @"From"];
[dict setObject: [NSString stringWithUTF8String: peer->addr] forKey: @"IP"];
[dict setObject: [NSNumber numberWithInt: peer->port] forKey: @"Port"];
[dict setObject: [NSNumber numberWithFloat: peer->progress] forKey: @"Progress"];
[dict setObject: [NSNumber numberWithBool: peer->isSeed] forKey: @"Seed"];
[dict setObject: [NSNumber numberWithBool: peer->isEncrypted] forKey: @"Encryption"];
[dict setObject: [NSString stringWithUTF8String: peer->client] forKey: @"Client"];
[dict setObject: [NSString stringWithUTF8String: peer->flagStr] forKey: @"Flags"];
2007-09-16 01:02:06 +00:00
2008-01-10 00:52:02 +00:00
if (peer->isUploadingTo)
2010-07-06 03:40:34 +00:00
[dict setObject: [NSNumber numberWithDouble: peer->rateToPeer_KBps] forKey: @"UL To Rate"];
2008-01-10 00:52:02 +00:00
if (peer->isDownloadingFrom)
2010-07-06 03:40:34 +00:00
[dict setObject: [NSNumber numberWithDouble: peer->rateToClient_KBps] forKey: @"DL From Rate"];
2007-09-16 01:02:06 +00:00
[peerDicts addObject: dict];
2007-09-16 01:02:06 +00:00
}
tr_torrentPeersFree(peers, totalPeers);
return peerDicts;
2007-09-16 01:02:06 +00:00
}
- (NSUInteger) webSeedCount
{
return fInfo->webseedCount;
}
2008-06-08 18:02:16 +00:00
- (NSArray *) webSeeds
{
NSMutableArray * webSeeds = [NSMutableArray arrayWithCapacity: fInfo->webseedCount];
2010-07-06 03:31:17 +00:00
double * dlSpeeds = tr_torrentWebSpeeds_KBps(fHandle);
for (NSInteger i = 0; i < fInfo->webseedCount; i++)
{
NSMutableDictionary * dict = [NSMutableDictionary dictionaryWithCapacity: 3];
[dict setObject: [self name] forKey: @"Name"];
[dict setObject: [NSString stringWithUTF8String: fInfo->webseeds[i]] forKey: @"Address"];
if (dlSpeeds[i] != -1.0)
[dict setObject: [NSNumber numberWithFloat: dlSpeeds[i]] forKey: @"DL From Rate"];
[webSeeds addObject: dict];
}
tr_free(dlSpeeds);
2008-06-08 18:02:16 +00:00
return webSeeds;
}
2007-09-16 01:02:06 +00:00
- (NSString *) progressString
{
if ([self isMagnet])
{
NSString * progressString = fStat->metadataPercentComplete > 0.0
? [NSString stringWithFormat: NSLocalizedString(@"%@ of torrent metadata retrieved",
"Torrent -> progress string"), [NSString percentString: fStat->metadataPercentComplete longDecimals: YES]]
: NSLocalizedString(@"torrent metadata needed", "Torrent -> progress string");
return [NSString stringWithFormat: @"%@ - %@", NSLocalizedString(@"Magnetized transfer", "Torrent -> progress string"),
progressString];
}
NSString * string;
if (![self allDownloaded])
{
2008-11-02 01:07:01 +00:00
CGFloat progress;
if ([self isFolder] && [fDefaults boolForKey: @"DisplayStatusProgressSelected"])
{
2008-04-07 13:43:26 +00:00
string = [NSString stringWithFormat: NSLocalizedString(@"%@ of %@ selected", "Torrent -> progress string"),
[NSString stringForFileSize: [self haveTotal]], [NSString stringForFileSize: [self totalSizeSelected]]];
progress = [self progressDone];
}
else
{
2008-04-07 13:43:26 +00:00
string = [NSString stringWithFormat: NSLocalizedString(@"%@ of %@", "Torrent -> progress string"),
[NSString stringForFileSize: [self haveTotal]], [NSString stringForFileSize: [self size]]];
progress = [self progress];
}
2008-04-07 13:43:26 +00:00
string = [string stringByAppendingFormat: @" (%@)", [NSString percentString: progress longDecimals: YES]];
}
2008-03-31 19:36:11 +00:00
else
{
2008-03-31 19:36:11 +00:00
NSString * downloadString;
if (![self isComplete]) //only multifile possible
{
if ([fDefaults boolForKey: @"DisplayStatusProgressSelected"])
downloadString = [NSString stringWithFormat: NSLocalizedString(@"%@ selected", "Torrent -> progress string"),
2008-04-07 13:43:26 +00:00
[NSString stringForFileSize: [self haveTotal]]];
2008-03-31 19:36:11 +00:00
else
downloadString = [NSString stringWithFormat: NSLocalizedString(@"%@ of %@ (%@)", "Torrent -> progress string"),
[NSString stringForFileSize: [self haveTotal]], [NSString stringForFileSize: [self size]],
[NSString percentString: [self progress] longDecimals: YES]];
2008-03-31 19:36:11 +00:00
}
else
2008-03-31 19:36:11 +00:00
downloadString = [NSString stringForFileSize: [self size]];
NSString * uploadString = [NSString stringWithFormat: NSLocalizedString(@"uploaded %@ (Ratio: %@)",
"Torrent -> progress string"), [NSString stringForFileSize: [self uploadedTotal]],
[NSString stringForRatio: [self ratio]]];
2008-04-07 14:09:28 +00:00
string = [downloadString stringByAppendingFormat: @", %@", uploadString];
}
//add time when downloading or seed limit set
if ([self shouldShowEta])
string = [string stringByAppendingFormat: @" - %@", [self etaString]];
return string;
2007-09-16 01:02:06 +00:00
}
- (NSString *) statusString
{
NSString * string;
if ([self isAnyErrorOrWarning])
{
2009-08-05 01:55:37 +00:00
switch (fStat->error)
{
case TR_STAT_LOCAL_ERROR: string = NSLocalizedString(@"Error", "Torrent -> status string"); break;
case TR_STAT_TRACKER_ERROR: string = NSLocalizedString(@"Tracker returned error", "Torrent -> status string"); break;
case TR_STAT_TRACKER_WARNING: string = NSLocalizedString(@"Tracker returned warning", "Torrent -> status string"); break;
2009-08-05 01:55:37 +00:00
default: NSAssert(NO, @"unknown error state");
}
NSString * errorString = [self errorMessage];
2008-02-22 15:39:20 +00:00
if (errorString && ![errorString isEqualToString: @""])
string = [string stringByAppendingFormat: @": %@", errorString];
}
else
{
switch (fStat->activity)
{
case TR_STATUS_STOPPED:
if (fWaitToStart)
{
string = ![self allDownloaded]
? [NSLocalizedString(@"Waiting to download", "Torrent -> status string") stringByAppendingEllipsis]
: [NSLocalizedString(@"Waiting to seed", "Torrent -> status string") stringByAppendingEllipsis];
}
else if ([self isFinishedSeeding])
string = NSLocalizedString(@"Seeding complete", "Torrent -> status string");
else
string = NSLocalizedString(@"Paused", "Torrent -> status string");
break;
case TR_STATUS_CHECK_WAIT:
string = [NSLocalizedString(@"Waiting to check existing data", "Torrent -> status string") stringByAppendingEllipsis];
break;
case TR_STATUS_CHECK:
string = [NSString stringWithFormat: @"%@ (%@)",
NSLocalizedString(@"Checking existing data", "Torrent -> status string"),
[NSString percentString: [self checkingProgress] longDecimals: YES]];
break;
case TR_STATUS_DOWNLOAD:
if ([self totalPeersConnected] != 1)
string = [NSString stringWithFormat: NSLocalizedString(@"Downloading from %d of %d peers",
"Torrent -> status string"), [self peersSendingToUs], [self totalPeersConnected]];
else
string = [NSString stringWithFormat: NSLocalizedString(@"Downloading from %d of 1 peer",
"Torrent -> status string"), [self peersSendingToUs]];
2009-12-26 00:02:20 +00:00
const NSInteger webSeedCount = fStat->webseedsSendingToUs;
if (webSeedCount > 0)
{
NSString * webSeedString;
if (webSeedCount == 1)
webSeedString = NSLocalizedString(@"web seed", "Torrent -> status string");
else
webSeedString = [NSString stringWithFormat: NSLocalizedString(@"%d web seeds", "Torrent -> status string"),
webSeedCount];
string = [string stringByAppendingFormat: @" + %@", webSeedString];
}
break;
case TR_STATUS_SEED:
if ([self totalPeersConnected] != 1)
string = [NSString stringWithFormat: NSLocalizedString(@"Seeding to %d of %d peers", "Torrent -> status string"),
[self peersGettingFromUs], [self totalPeersConnected]];
else
string = [NSString stringWithFormat: NSLocalizedString(@"Seeding to %d of 1 peer", "Torrent -> status string"),
[self peersGettingFromUs]];
}
if (fStalled)
2008-04-22 13:16:39 +00:00
string = [NSLocalizedString(@"Stalled", "Torrent -> status string") stringByAppendingFormat: @", %@", string];
}
//append even if error
if ([self isActive] && ![self isChecking])
{
if (fStat->activity == TR_STATUS_DOWNLOAD)
string = [string stringByAppendingFormat: @" - %@: %@, %@: %@",
NSLocalizedString(@"DL", "Torrent -> status string"), [NSString stringForSpeed: [self downloadRate]],
NSLocalizedString(@"UL", "Torrent -> status string"), [NSString stringForSpeed: [self uploadRate]]];
else
string = [string stringByAppendingFormat: @" - %@: %@",
NSLocalizedString(@"UL", "Torrent -> status string"), [NSString stringForSpeed: [self uploadRate]]];
}
return string;
2007-09-16 01:02:06 +00:00
}
- (NSString *) shortStatusString
{
NSString * string;
switch (fStat->activity)
{
case TR_STATUS_STOPPED:
if (fWaitToStart)
{
string = ![self allDownloaded]
? [NSLocalizedString(@"Waiting to download", "Torrent -> status string") stringByAppendingEllipsis]
: [NSLocalizedString(@"Waiting to seed", "Torrent -> status string") stringByAppendingEllipsis];
}
else if ([self isFinishedSeeding])
string = NSLocalizedString(@"Seeding complete", "Torrent -> status string");
else
string = NSLocalizedString(@"Paused", "Torrent -> status string");
break;
case TR_STATUS_CHECK_WAIT:
string = [NSLocalizedString(@"Waiting to check existing data", "Torrent -> status string") stringByAppendingEllipsis];
break;
case TR_STATUS_CHECK:
string = [NSString stringWithFormat: @"%@ (%@)",
NSLocalizedString(@"Checking existing data", "Torrent -> status string"),
[NSString percentString: [self checkingProgress] longDecimals: YES]];
break;
2007-09-30 13:33:50 +00:00
case TR_STATUS_DOWNLOAD:
string = [NSString stringWithFormat: @"%@: %@, %@: %@",
NSLocalizedString(@"DL", "Torrent -> status string"), [NSString stringForSpeed: [self downloadRate]],
NSLocalizedString(@"UL", "Torrent -> status string"), [NSString stringForSpeed: [self uploadRate]]];
2007-09-30 13:33:50 +00:00
break;
case TR_STATUS_SEED:
string = [NSString stringWithFormat: @"%@: %@, %@: %@",
NSLocalizedString(@"Ratio", "Torrent -> status string"), [NSString stringForRatio: [self ratio]],
NSLocalizedString(@"UL", "Torrent -> status string"), [NSString stringForSpeed: [self uploadRate]]];
}
return string;
2007-09-16 01:02:06 +00:00
}
- (NSString *) remainingTimeString
{
if ([self shouldShowEta])
return [self etaString];
else
2008-02-16 19:32:22 +00:00
return [self shortStatusString];
2007-09-16 01:02:06 +00:00
}
- (NSString *) stateString
{
switch (fStat->activity)
{
case TR_STATUS_STOPPED:
{
NSString * string = NSLocalizedString(@"Paused", "Torrent -> status string");
NSString * extra = nil;
if (fWaitToStart)
{
extra = ![self allDownloaded]
? NSLocalizedString(@"Waiting to download", "Torrent -> status string")
: NSLocalizedString(@"Waiting to seed", "Torrent -> status string");
}
else if ([self isFinishedSeeding])
extra = NSLocalizedString(@"Seeding complete", "Torrent -> status string");
else;
return extra ? [string stringByAppendingFormat: @" (%@)", extra] : string;
}
case TR_STATUS_CHECK_WAIT:
return [NSLocalizedString(@"Waiting to check existing data", "Torrent -> status string") stringByAppendingEllipsis];
case TR_STATUS_CHECK:
return [NSString stringWithFormat: @"%@ (%@)",
NSLocalizedString(@"Checking existing data", "Torrent -> status string"),
[NSString percentString: [self checkingProgress] longDecimals: YES]];
case TR_STATUS_DOWNLOAD:
return NSLocalizedString(@"Downloading", "Torrent -> status string");
case TR_STATUS_SEED:
return NSLocalizedString(@"Seeding", "Torrent -> status string");
}
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersConnected
2007-09-16 01:02:06 +00:00
{
return fStat->peersConnected;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersTracker
2007-09-16 01:02:06 +00:00
{
return fStat->peersFrom[TR_PEER_FROM_TRACKER];
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersIncoming
2007-09-16 01:02:06 +00:00
{
return fStat->peersFrom[TR_PEER_FROM_INCOMING];
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersCache
2007-09-16 01:02:06 +00:00
{
2009-11-26 05:40:27 +00:00
return fStat->peersFrom[TR_PEER_FROM_RESUME];
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersPex
2007-09-16 01:02:06 +00:00
{
return fStat->peersFrom[TR_PEER_FROM_PEX];
}
- (NSInteger) totalPeersDHT
{
return fStat->peersFrom[TR_PEER_FROM_DHT];
}
- (NSInteger) totalPeersLocal
{
2010-05-08 08:54:45 +00:00
return fStat->peersFrom[TR_PEER_FROM_LPD];
}
- (NSInteger) totalPeersLTEP
{
return fStat->peersFrom[TR_PEER_FROM_LTEP];
}
2008-11-02 01:07:01 +00:00
- (NSInteger) totalPeersKnown
2007-10-01 16:23:00 +00:00
{
return fStat->peersKnown;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) peersSendingToUs
2007-09-16 01:02:06 +00:00
{
return fStat->peersSendingToUs;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) peersGettingFromUs
2007-09-16 01:02:06 +00:00
{
return fStat->peersGettingFromUs;
}
2008-11-02 01:07:01 +00:00
- (CGFloat) downloadRate
2007-09-16 01:02:06 +00:00
{
2010-07-06 03:31:17 +00:00
return fStat->pieceDownloadSpeed_KBps;
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) uploadRate
2007-09-16 01:02:06 +00:00
{
2010-07-06 03:31:17 +00:00
return fStat->pieceUploadSpeed_KBps;
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) totalRate
2007-11-23 16:03:49 +00:00
{
return [self downloadRate] + [self uploadRate];
}
- (uint64_t) haveVerified
2007-09-16 01:02:06 +00:00
{
return fStat->haveValid;
}
- (uint64_t) haveTotal
{
return [self haveVerified] + fStat->haveUnchecked;
2007-09-16 01:02:06 +00:00
}
2008-02-27 19:34:55 +00:00
- (uint64_t) totalSizeSelected
{
return fStat->sizeWhenDone;
2008-02-27 19:34:55 +00:00
}
2007-09-26 18:53:11 +00:00
- (uint64_t) downloadedTotal
{
2007-09-26 18:53:11 +00:00
return fStat->downloadedEver;
}
2007-09-26 18:53:11 +00:00
- (uint64_t) uploadedTotal
2007-09-16 01:02:06 +00:00
{
2007-09-26 18:53:11 +00:00
return fStat->uploadedEver;
2007-09-16 01:02:06 +00:00
}
- (uint64_t) failedHash
2007-09-16 01:02:06 +00:00
{
return fStat->corruptEver;
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (NSInteger) groupValue
2007-12-17 16:06:20 +00:00
{
return fGroupValue;
}
2008-11-02 01:07:01 +00:00
- (void) setGroupValue: (NSInteger) goupValue
2007-12-17 16:06:20 +00:00
{
fGroupValue = goupValue;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) groupOrderValue
{
return [[GroupsController groups] rowValueForIndex: fGroupValue];
}
- (void) checkGroupValueForRemoval: (NSNotification *) notification
2007-12-17 16:06:20 +00:00
{
if (fGroupValue != -1 && [[[notification userInfo] objectForKey: @"Index"] integerValue] == fGroupValue)
2007-12-17 16:06:20 +00:00
fGroupValue = -1;
}
2007-09-16 01:02:06 +00:00
- (NSArray *) fileList
{
return fFileList;
}
2010-01-02 02:50:22 +00:00
- (NSArray *) flatFileList
{
return fFlatFileList;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) fileCount
2007-09-16 01:02:06 +00:00
{
return fInfo->fileCount;
}
- (void) updateFileStat
{
if (fFileStat)
tr_torrentFilesFree(fFileStat, [self fileCount]);
2007-09-16 01:02:06 +00:00
fFileStat = tr_torrentFiles(fHandle, NULL);
2007-09-16 01:02:06 +00:00
}
2008-11-02 01:07:01 +00:00
- (CGFloat) fileProgress: (FileListNode *) node
2007-09-16 01:02:06 +00:00
{
2009-12-26 00:02:20 +00:00
if ([self fileCount] == 1 || [self isComplete])
return [self progress];
if (!fFileStat)
2007-09-16 01:02:06 +00:00
[self updateFileStat];
NSIndexSet * indexSet = [node indexes];
if ([indexSet count] == 1)
return fFileStat[[indexSet firstIndex]].progress;
uint64_t have = 0;
2008-11-02 01:07:01 +00:00
for (NSInteger index = [indexSet firstIndex]; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index])
have += fFileStat[index].bytesCompleted;
NSAssert([node size], @"directory in torrent file has size 0");
2008-11-02 01:07:01 +00:00
return (CGFloat)have / [node size];
2007-09-16 01:02:06 +00:00
}
2009-12-26 00:02:20 +00:00
- (BOOL) canChangeDownloadCheckForFile: (NSUInteger) index
2007-09-16 01:02:06 +00:00
{
NSAssert2(index < [self fileCount], @"Index %d is greater than file count %d", index, [self fileCount]);
if ([self fileCount] == 1 || [self isComplete])
return NO;
if (!fFileStat)
2007-09-16 01:02:06 +00:00
[self updateFileStat];
return fFileStat[index].progress < 1.0;
2007-09-16 01:02:06 +00:00
}
- (BOOL) canChangeDownloadCheckForFiles: (NSIndexSet *) indexSet
{
if ([self fileCount] == 1 || [self isComplete])
2007-09-16 01:02:06 +00:00
return NO;
if (!fFileStat)
2007-09-16 01:02:06 +00:00
[self updateFileStat];
2009-12-26 00:02:20 +00:00
for (NSUInteger index = [indexSet firstIndex]; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index])
if (fFileStat[index].progress < 1.0)
2007-09-16 01:02:06 +00:00
return YES;
return NO;
}
2008-11-02 01:07:01 +00:00
- (NSInteger) checkForFiles: (NSIndexSet *) indexSet
2007-09-16 01:02:06 +00:00
{
BOOL onState = NO, offState = NO;
2009-12-26 00:02:20 +00:00
for (NSUInteger index = [indexSet firstIndex]; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index])
2007-09-16 01:02:06 +00:00
{
if (!fInfo->files[index].dnd || ![self canChangeDownloadCheckForFile: index])
2007-09-16 01:02:06 +00:00
onState = YES;
else
offState = YES;
if (onState && offState)
return NSMixedState;
}
return onState ? NSOnState : NSOffState;
}
2008-11-02 01:07:01 +00:00
- (void) setFileCheckState: (NSInteger) state forIndexes: (NSIndexSet *) indexSet
2007-09-16 01:02:06 +00:00
{
NSUInteger count = [indexSet count];
2008-03-22 18:22:10 +00:00
tr_file_index_t * files = malloc(count * sizeof(tr_file_index_t));
for (NSUInteger index = [indexSet firstIndex], i = 0; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index], i++)
2007-09-16 01:02:06 +00:00
files[i] = index;
tr_torrentSetFileDLs(fHandle, files, count, state != NSOffState);
free(files);
[self update];
2008-02-27 19:34:55 +00:00
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentFileCheckChange" object: self];
2007-09-16 01:02:06 +00:00
}
- (void) setFilePriority: (tr_priority_t) priority forIndexes: (NSIndexSet *) indexSet
2007-09-16 01:02:06 +00:00
{
const NSUInteger count = [indexSet count];
2009-12-26 00:02:20 +00:00
tr_file_index_t * files = tr_malloc(count * sizeof(tr_file_index_t));
for (NSUInteger index = [indexSet firstIndex], i = 0; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index], i++)
2007-09-16 01:02:06 +00:00
files[i] = index;
2007-09-16 01:02:06 +00:00
tr_torrentSetFilePriorities(fHandle, files, count, priority);
2009-12-26 00:02:20 +00:00
tr_free(files);
2007-09-16 01:02:06 +00:00
}
- (BOOL) hasFilePriority: (tr_priority_t) priority forIndexes: (NSIndexSet *) indexSet
2007-09-16 01:02:06 +00:00
{
2009-12-26 00:02:20 +00:00
for (NSUInteger index = [indexSet firstIndex]; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index])
if (priority == fInfo->files[index].priority && [self canChangeDownloadCheckForFile: index])
2007-09-16 01:02:06 +00:00
return YES;
return NO;
}
- (NSSet *) filePrioritiesForIndexes: (NSIndexSet *) indexSet
{
BOOL low = NO, normal = NO, high = NO;
2009-12-26 00:02:20 +00:00
NSMutableSet * priorities = [NSMutableSet setWithCapacity: MIN([indexSet count], 3)];
2007-09-16 01:02:06 +00:00
2009-12-26 00:02:20 +00:00
for (NSUInteger index = [indexSet firstIndex]; index != NSNotFound; index = [indexSet indexGreaterThanIndex: index])
2007-09-16 01:02:06 +00:00
{
if (![self canChangeDownloadCheckForFile: index])
continue;
const tr_priority_t priority = fInfo->files[index].priority;
switch (priority)
2007-09-16 01:02:06 +00:00
{
case TR_PRI_LOW:
if (low)
continue;
low = YES;
break;
case TR_PRI_NORMAL:
if (normal)
continue;
normal = YES;
break;
case TR_PRI_HIGH:
if (high)
continue;
high = YES;
break;
default:
NSAssert2(NO, @"Unknown priority %d for file index %d", priority, index);
2007-09-16 01:02:06 +00:00
}
[priorities addObject: [NSNumber numberWithInteger: priority]];
2007-09-16 01:02:06 +00:00
if (low && normal && high)
break;
}
return priorities;
}
- (NSDate *) dateAdded
{
2009-02-17 04:00:53 +00:00
const time_t date = fStat->addedDate;
return [NSDate dateWithTimeIntervalSince1970: date];
2007-09-16 01:02:06 +00:00
}
- (NSDate *) dateCompleted
{
2009-02-17 04:00:53 +00:00
const time_t date = fStat->doneDate;
return date != 0 ? [NSDate dateWithTimeIntervalSince1970: date] : nil;
2007-09-16 01:02:06 +00:00
}
- (NSDate *) dateActivity
{
2009-02-17 04:00:53 +00:00
const time_t date = fStat->activityDate;
return date != 0 ? [NSDate dateWithTimeIntervalSince1970: date] : nil;
2007-09-16 01:02:06 +00:00
}
- (NSDate *) dateActivityOrAdd
{
NSDate * date = [self dateActivity];
return date ? date : [self dateAdded];
}
2008-11-02 01:07:01 +00:00
- (NSInteger) stalledMinutes
2007-09-16 01:02:06 +00:00
{
2010-07-11 21:06:28 +00:00
if (fStat->idleSecs == -1)
return -1;
return fStat->idleSecs / 60;
2007-09-16 01:02:06 +00:00
}
- (BOOL) isStalled
{
return fStalled;
}
- (void) updateTimeMachineExclude
{
NSString * currentLocation = ![self allDownloaded] ? [self dataLocation] : nil;
//return if the locations are the same
if (fTimeMachineExclude && currentLocation && [fTimeMachineExclude isEqualToString: currentLocation])
return;
//remove old location...
if (fTimeMachineExclude)
{
[self setTimeMachineExclude: NO forPath: fTimeMachineExclude];
[fTimeMachineExclude release];
fTimeMachineExclude = nil;
}
//...set new location
if (currentLocation)
{
[self setTimeMachineExclude: YES forPath: currentLocation];
fTimeMachineExclude = [currentLocation retain];
}
}
- (NSInteger) stateSortKey
2007-09-16 01:02:06 +00:00
{
if (![self isActive]) //paused
{
if (fWaitToStart)
return 1;
else
return 0;
}
else if ([self isSeeding]) //seeding
return 10;
else //downloading
return 20;
2007-09-16 01:02:06 +00:00
}
- (NSString *) trackerSortKey
{
int count;
tr_tracker_stat * stats = tr_torrentTrackers(fHandle, &count);
NSString * best = nil;
for (int i=0; i < count; ++i)
{
NSString * tracker = [NSString stringWithUTF8String: stats[i].host];
if (!best || [tracker localizedCaseInsensitiveCompare: best] == NSOrderedAscending)
best = tracker;
}
tr_torrentTrackersFree(stats, count);
return best;
}
- (tr_torrent *) torrentStruct
2007-09-16 01:02:06 +00:00
{
return fHandle;
2007-09-16 01:02:06 +00:00
}
- (NSURL *) previewItemURL
{
NSString * location = [self dataLocation];
return location ? [NSURL fileURLWithPath: location] : nil;
}
2007-09-16 01:02:06 +00:00
@end
@implementation Torrent (Private)
2009-12-13 01:36:22 +00:00
- (id) initWithPath: (NSString *) path hash: (NSString *) hashString torrentStruct: (tr_torrent *) torrentStruct
magnetAddress: (NSString *) magnetAddress lib: (tr_session *) lib
2007-12-17 16:06:20 +00:00
waitToStart: (NSNumber *) waitToStart
groupValue: (NSNumber *) groupValue
2009-12-11 04:01:47 +00:00
downloadFolder: (NSString *) downloadFolder legacyIncompleteFolder: (NSString *) incompleteFolder
2007-09-16 01:02:06 +00:00
{
if (!(self = [super init]))
return nil;
fDefaults = [NSUserDefaults standardUserDefaults];
2008-05-26 23:23:07 +00:00
if (torrentStruct)
fHandle = torrentStruct;
else
2007-09-16 01:02:06 +00:00
{
2008-05-26 23:23:07 +00:00
//set libtransmission settings for initialization
tr_ctor * ctor = tr_ctorNew(lib);
2009-12-13 01:36:22 +00:00
2008-05-26 23:23:07 +00:00
tr_ctorSetPaused(ctor, TR_FORCE, YES);
2009-12-13 01:36:22 +00:00
if (downloadFolder)
tr_ctorSetDownloadDir(ctor, TR_FORCE, [downloadFolder UTF8String]);
if (incompleteFolder)
tr_ctorSetIncompleteDir(ctor, [incompleteFolder UTF8String]);
2008-05-26 23:23:07 +00:00
tr_parse_result result = TR_PARSE_ERR;
if (path)
2009-05-22 02:52:28 +00:00
result = tr_ctorSetMetainfoFromFile(ctor, [path UTF8String]);
2009-12-13 01:40:53 +00:00
if (result != TR_PARSE_OK && magnetAddress)
result = tr_ctorSetMetainfoFromMagnetLink(ctor, [magnetAddress UTF8String]);
2009-12-13 01:40:53 +00:00
//backup - shouldn't be needed after upgrade to 1.70
2009-08-05 02:13:45 +00:00
if (result != TR_PARSE_OK && hashString)
2009-05-22 12:11:45 +00:00
result = tr_ctorSetMetainfoFromHash(ctor, [hashString UTF8String]);
2008-05-26 23:23:07 +00:00
2009-08-05 02:13:45 +00:00
if (result == TR_PARSE_OK)
fHandle = tr_torrentNew(ctor, NULL);
2008-05-26 23:23:07 +00:00
tr_ctorFree(ctor);
if (!fHandle)
{
[self release];
return nil;
}
2007-09-16 01:02:06 +00:00
}
fInfo = tr_torrentInfo(fHandle);
2007-09-16 01:02:06 +00:00
tr_torrentSetCompletenessCallback(fHandle, completenessChangeCallback, self);
tr_torrentSetRatioLimitHitCallback(fHandle, ratioLimitHitCallback, self);
tr_torrentSetIdleLimitHitCallback(fHandle, idleLimitHitCallback, self);
tr_torrentSetMetadataCallback(fHandle, metadataCallback, self);
fHashString = [[NSString alloc] initWithUTF8String: fInfo->hashString];
2007-09-16 01:02:06 +00:00
2008-01-04 04:54:26 +00:00
fWaitToStart = waitToStart && [waitToStart boolValue];
fResumeOnWake = NO;
//don't do after this point - it messes with auto-group functionality
if (![self isMagnet])
[self createFileList];
fGroupValue = groupValue ? [groupValue intValue] : [[GroupsController groups] groupIndexForTorrent: self];
2007-09-16 01:02:06 +00:00
[[NSNotificationCenter defaultCenter] addObserver: self selector: @selector(checkGroupValueForRemoval:)
2007-12-17 16:06:20 +00:00
name: @"GroupValueRemoved" object: nil];
fTimeMachineExclude = nil;
2007-09-16 01:02:06 +00:00
[self update];
2007-09-16 01:02:06 +00:00
return self;
}
- (void) createFileList
{
NSAssert(![self isMagnet], @"Cannot create a file list until the torrent is demagnetized");
if ([self isFolder])
2007-09-16 01:02:06 +00:00
{
2008-11-02 01:07:01 +00:00
NSInteger count = [self fileCount];
2010-01-01 21:12:04 +00:00
NSMutableArray * fileList = [NSMutableArray arrayWithCapacity: count],
* flatFileList = [NSMutableArray arrayWithCapacity: count];
2007-09-16 01:02:06 +00:00
2008-11-02 01:07:01 +00:00
for (NSInteger i = 0; i < count; i++)
2007-09-16 01:02:06 +00:00
{
tr_file * file = &fInfo->files[i];
NSString * fullPath = [NSString stringWithUTF8String: file->name];
NSArray * pathComponents = [fullPath pathComponents];
NSAssert1([pathComponents count] >= 2, @"Not enough components in path %@", fullPath);
NSString * path = [pathComponents objectAtIndex: 0];
NSString * name = [pathComponents objectAtIndex: 1];
if ([pathComponents count] > 2)
{
//determine if folder node already exists
2008-12-26 15:11:37 +00:00
FileListNode * node;
2008-12-26 07:25:17 +00:00
for (node in fileList)
2010-01-01 21:12:04 +00:00
if ([[node name] isEqualToString: name] && [node isFolder])
break;
if (!node)
{
node = [[FileListNode alloc] initWithFolderName: name path: path];
[fileList addObject: node];
[node release];
}
NSMutableArray * trimmedComponents = [NSMutableArray arrayWithArray: [pathComponents subarrayWithRange:
NSMakeRange(2, [pathComponents count]-2)]];
[node insertIndex: i withSize: file->length];
[self insertPath: trimmedComponents forParent: node fileSize: file->length index: i flatList: flatFileList];
}
else
{
FileListNode * node = [[FileListNode alloc] initWithFileName: name path: path size: file->length index: i];
[fileList addObject: node];
[flatFileList addObject: node];
[node release];
}
2007-09-16 01:02:06 +00:00
}
fFileList = [[NSArray alloc] initWithArray: fileList];
fFlatFileList = [[NSArray alloc] initWithArray: flatFileList];
}
else
{
FileListNode * node = [[FileListNode alloc] initWithFileName: [self name] path: @"" size: [self size] index: 0];
fFileList = [[NSArray arrayWithObject: node] retain];
fFlatFileList = [fFileList retain];
[node release];
2007-09-16 01:02:06 +00:00
}
}
- (void) insertPath: (NSMutableArray *) components forParent: (FileListNode *) parent fileSize: (uint64_t) size
index: (NSInteger) index flatList: (NSMutableArray *) flatFileList
2007-09-16 01:02:06 +00:00
{
NSString * name = [components objectAtIndex: 0];
2009-10-19 01:38:13 +00:00
const BOOL isFolder = [components count] > 1;
2007-09-16 01:02:06 +00:00
FileListNode * node = nil;
2007-09-16 01:02:06 +00:00
if (isFolder)
{
2008-12-26 07:25:17 +00:00
for (node in [parent children])
if ([[node name] isEqualToString: name] && [node isFolder])
2007-09-16 01:02:06 +00:00
break;
}
//create new folder or file if it doesn't already exist
if (!node)
2007-09-16 01:02:06 +00:00
{
NSString * path = [[parent path] stringByAppendingPathComponent: [parent name]];
2007-09-16 01:02:06 +00:00
if (isFolder)
node = [[FileListNode alloc] initWithFolderName: name path: path];
2007-09-16 01:02:06 +00:00
else
{
node = [[FileListNode alloc] initWithFileName: name path: path size: size index: index];
[flatFileList addObject: node];
}
[parent insertChild: node];
[node release];
2007-09-16 01:02:06 +00:00
}
if (isFolder)
{
[node insertIndex: index withSize: size];
2007-09-16 01:02:06 +00:00
[components removeObjectAtIndex: 0];
[self insertPath: components forParent: node fileSize: size index: index flatList: flatFileList];
2007-09-16 01:02:06 +00:00
}
}
//status has been retained
- (void) completenessChange: (NSDictionary *) statusInfo
{
fStat = tr_torrentStat(fHandle); //don't call update yet to avoid auto-stop
switch ([[statusInfo objectForKey: @"Status"] intValue])
{
case TR_SEED:
case TR_PARTIAL_SEED:
//simpler to create a new dictionary than to use statusInfo - avoids retention chicanery
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentFinishedDownloading" object: self
userInfo: [NSDictionary dictionaryWithObject: [statusInfo objectForKey: @"WasRunning"] forKey: @"WasRunning"]];
break;
case TR_LEECH:
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentRestartedDownloading" object: self];
break;
}
[statusInfo release];
[self update];
[self updateTimeMachineExclude];
}
- (void) ratioLimitHit
{
fStat = tr_torrentStat(fHandle);
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentStoppedForRatio" object: self];
}
- (void) idleLimitHit
{
fStat = tr_torrentStat(fHandle);
#warning fix
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentStoppedForRatio" object: self];
}
- (void) metadataRetrieved
{
fStat = tr_torrentStat(fHandle);
[self createFileList];
[[NSNotificationCenter defaultCenter] postNotificationName: @"ResetInspector" object: self];
}
- (BOOL) shouldShowEta
{
if (fStat->activity == TR_STATUS_DOWNLOAD)
return YES;
else if ([self isSeeding])
{
if (tr_torrentGetSeedRatio(fHandle, NULL))
return YES;
if (tr_torrentGetSeedIdle(fHandle, NULL))
return (fStat->etaIdle != TR_ETA_NOT_AVAIL && fStat->etaIdle != TR_ETA_UNKNOWN) && fStat->etaIdle < ETA_IDLE_DISPLAY_MIN * 60;
}
return NO;
}
#warning don't show idle minutes when very large
- (NSString *) etaString
2008-04-07 02:55:33 +00:00
{
const BOOL etaReg = fStat->eta != TR_ETA_NOT_AVAIL && fStat->eta != TR_ETA_UNKNOWN;
const BOOL etaIdelSeed = fStat->etaIdle != TR_ETA_NOT_AVAIL && fStat->etaIdle != TR_ETA_UNKNOWN;
NSInteger eta;
if (etaReg && etaIdelSeed)
eta = MIN(fStat->eta, fStat->etaIdle);
else if (etaReg)
eta = fStat->eta;
else if (etaIdelSeed)
eta = fStat->etaIdle;
else
return NSLocalizedString(@"remaining time unknown", "Torrent -> eta string");
return [NSString stringWithFormat: NSLocalizedString(@"%@ remaining", "Torrent -> eta string"),
[NSString timeString: eta showSeconds: YES maxFields: 2]];
2008-04-07 02:55:33 +00:00
}
- (void) setTimeMachineExclude: (BOOL) exclude forPath: (NSString *) path
{
2008-12-26 05:57:51 +00:00
CSBackupSetItemExcluded((CFURLRef)[NSURL fileURLWithPath: path], exclude, true);
}
2007-09-16 01:02:06 +00:00
@end