transmission/macosx/Torrent.m

1213 lines
38 KiB
Objective-C

/******************************************************************************
* $Id$
*
* Copyright (c) 2006 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 "Torrent.h"
#import "StringAdditions.h"
#define BAR_HEIGHT 12.0
#define MAX_PIECES 324
#define BLANK_PIECE -99
@interface Torrent (Private)
- (id) initWithHash: (NSString *) hashString path: (NSString *) path lib: (tr_handle_t *) lib
publicTorrent: (NSNumber *) publicTorrent
date: (NSDate *) date stopRatioSetting: (NSNumber *) stopRatioSetting
ratioLimit: (NSNumber *) ratioLimit waitToStart: (NSNumber *) waitToStart
orderValue: (NSNumber *) orderValue;
- (NSImage *) advancedBar;
- (void) trashFile: (NSString *) path;
@end
@implementation Torrent
// Used to optimize drawing. They contain packed RGBA pixels for every color needed.
#define BE OSSwapBigToHostConstInt32
static uint32_t kRed = BE(0xFF6450FF), //255, 100, 80
kBlue = BE(0x50A0FFFF), //80, 160, 255
kBlue2 = BE(0x1E46B4FF), //30, 70, 180
kGray = BE(0x969696FF), //150, 150, 150
kGreen1 = BE(0x99FFCCFF), //153, 255, 204
kGreen2 = BE(0x66FF99FF), //102, 255, 153
kGreen3 = BE(0x00FF66FF), //0, 255, 102
kWhite = BE(0xFFFFFFFF); //255, 255, 255
- (id) initWithPath: (NSString *) path forceDeleteTorrent: (BOOL) delete lib: (tr_handle_t *) lib
{
self = [self initWithHash: nil path: path lib: lib
publicTorrent: delete ? [NSNumber numberWithBool: NO] : nil
date: nil stopRatioSetting: nil ratioLimit: nil waitToStart: nil orderValue: nil];
if (self)
{
fUseIncompleteFolder = [fDefaults boolForKey: @"UseIncompleteDownloadFolder"];
fIncompleteFolder = [[fDefaults stringForKey: @"IncompleteDownloadFolder"] copy];
if (!fPublicTorrent)
[self trashFile: path];
}
return self;
}
- (id) initWithHistory: (NSDictionary *) history lib: (tr_handle_t *) lib
{
self = [self initWithHash: [history objectForKey: @"TorrentHash"]
path: [history objectForKey: @"TorrentPath"] lib: lib
publicTorrent: [history objectForKey: @"PublicCopy"]
date: [history objectForKey: @"Date"]
stopRatioSetting: [history objectForKey: @"StopRatioSetting"]
ratioLimit: [history objectForKey: @"RatioLimit"]
waitToStart: [history objectForKey: @"WaitToStart"]
orderValue: [history objectForKey: @"OrderValue"]];
if (self)
{
//download folders
NSString * downloadFolder;
if (!(downloadFolder = [history objectForKey: @"DownloadFolder"]))
downloadFolder = [[fDefaults stringForKey: @"DownloadFolder"] stringByExpandingTildeInPath];
NSNumber * useIncompleteFolder;
if ((useIncompleteFolder = [history objectForKey: @"UseIncompleteFolder"]))
{
if ((fUseIncompleteFolder = [useIncompleteFolder boolValue]))
{
NSString * incompleteFolder;
if (incompleteFolder = [history objectForKey: @"IncompleteFolder"])
fIncompleteFolder = [incompleteFolder copy];
else
fIncompleteFolder = [[[fDefaults stringForKey: @"IncompleteDownloadFolder"]
stringByExpandingTildeInPath] copy];
}
}
else
fUseIncompleteFolder = NO;
[self setDownloadFolder: downloadFolder];
NSString * paused;
if (!(paused = [history objectForKey: @"Paused"]) || [paused isEqualToString: @"NO"])
{
fStat = tr_torrentStat(fHandle);
[self startTransfer];
}
}
return self;
}
- (NSDictionary *) history
{
NSMutableDictionary * history = [NSMutableDictionary dictionaryWithObjectsAndKeys:
[NSNumber numberWithBool: fPublicTorrent], @"PublicCopy",
[self hashString], @"TorrentHash",
fDownloadFolder, @"DownloadFolder",
[NSNumber numberWithBool: fUseIncompleteFolder], @"UseIncompleteFolder",
[self isActive] ? @"NO" : @"YES", @"Paused",
[self date], @"Date",
[NSNumber numberWithInt: fStopRatioSetting], @"StopRatioSetting",
[NSNumber numberWithFloat: fRatioLimit], @"RatioLimit",
[NSNumber numberWithBool: fWaitToStart], @"WaitToStart",
[self orderValue], @"OrderValue", nil];
if (fUseIncompleteFolder)
[history setObject: fIncompleteFolder forKey: @"IncompleteFolder"];
if (fPublicTorrent)
[history setObject: [self publicTorrentLocation] forKey: @"TorrentPath"];
return history;
}
- (void) dealloc
{
if (fHandle)
{
tr_torrentClose(fLib, fHandle);
if (fDownloadFolder)
[fDownloadFolder release];
if (fIncompleteFolder)
[fIncompleteFolder release];
if (fPublicTorrentLocation)
[fPublicTorrentLocation release];
[fDate release];
[fIcon release];
[fIconFlipped release];
[fIconSmall release];
[fProgressString release];
[fStatusString release];
[fShortStatusString release];
[fRemainingTimeString release];
[fBitmap release];
free(fPieces);
}
[super dealloc];
}
- (void) setDownloadFolder: (NSString *) path
{
if (path)
fDownloadFolder = [path copy];
if (!fUseIncompleteFolder || [[NSFileManager defaultManager] fileExistsAtPath:
[path stringByAppendingPathComponent: [self name]]])
tr_torrentSetFolder(fHandle, [path UTF8String]);
else
tr_torrentSetFolder(fHandle, [fIncompleteFolder UTF8String]);
}
- (NSString *) downloadFolder
{
return [NSString stringWithUTF8String: tr_torrentGetFolder(fHandle)];
}
- (void) getAvailability: (int8_t *) tab size: (int) size
{
tr_torrentAvailability(fHandle, tab, size);
}
- (void) getAmountFinished: (float *) tab size: (int) size
{
tr_torrentAmountFinished(fHandle, tab, size);
}
- (void) update
{
fStat = tr_torrentStat(fHandle);
//notification when downloading finished
if ([self justFinished])
{
BOOL canMove = YES;
//move file from incomplete folder to download folder
if (fUseIncompleteFolder && ![[self downloadFolder] isEqualToString: fDownloadFolder]
&& (canMove = [self alertForMoveVolumeAvailable]))
{
tr_torrentStop(fHandle);
if ([[NSFileManager defaultManager] movePath: [[self downloadFolder] stringByAppendingPathComponent: [self name]]
toPath: [fDownloadFolder stringByAppendingPathComponent: [self name]] handler: nil])
tr_torrentSetFolder(fHandle, [fDownloadFolder UTF8String]);
tr_torrentStart(fHandle);
}
if (!canMove)
{
fUseIncompleteFolder = NO;
[fDownloadFolder release];
fDownloadFolder = fIncompleteFolder;
fIncompleteFolder = nil;
}
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentFinishedDownloading" object: self];
}
//check to stop for ratio
if ([self isSeeding] && ((fStopRatioSetting == RATIO_CHECK && [self ratio] >= fRatioLimit)
|| (fStopRatioSetting == RATIO_GLOBAL && [fDefaults boolForKey: @"RatioCheck"]
&& [self ratio] >= [fDefaults floatForKey: @"RatioLimit"])))
{
[self stopTransfer];
[self setStopRatioSetting: RATIO_NO_CHECK];
fFinishedSeeding = YES;
fStat = tr_torrentStat(fHandle);
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentStoppedForRatio" object: self];
}
[fProgressString setString: @""];
if ([self progress] < 1.0)
[fProgressString appendFormat: NSLocalizedString(@"%@ of %@ (%.2f%%)", "Torrent -> progress string"),
[NSString stringForFileSize: [self downloadedValid]],
[NSString stringForFileSize: [self size]], 100.0 * [self progress]];
else
[fProgressString appendFormat: NSLocalizedString(@"%@, uploaded %@ (Ratio: %@)", "Torrent -> progress string"),
[NSString stringForFileSize: [self size]], [NSString stringForFileSize: [self uploadedTotal]],
[NSString stringForRatioWithDownload: [self downloadedTotal] upload: [self uploadedTotal]]];
switch (fStat->status)
{
NSString * tempString;
case TR_STATUS_PAUSE:
if (fFinishedSeeding)
tempString = NSLocalizedString(@"Seeding complete", "Torrent -> status string");
else if (fWaitToStart)
tempString = [NSLocalizedString(@"Waiting to start", "Torrent -> status string") stringByAppendingEllipsis];
else
tempString = NSLocalizedString(@"Paused", "Torrent -> status string");
[fStatusString setString: tempString];
[fShortStatusString setString: tempString];
break;
case TR_STATUS_CHECK:
tempString = [NSLocalizedString(@"Checking existing files", "Torrent -> status string") stringByAppendingEllipsis];
[fStatusString setString: tempString];
[fShortStatusString setString: tempString];
[fRemainingTimeString setString: tempString];
break;
case TR_STATUS_DOWNLOAD:
[fStatusString setString: @""];
if ([self totalPeers] > 1)
[fStatusString appendFormat: NSLocalizedString(@"Downloading from %d of %d peers",
"Torrent -> status string"), [self peersUploading], [self totalPeers]];
else
[fStatusString appendFormat: NSLocalizedString(@"Downloading from %d of %d peer",
"Torrent -> status string"), [self peersUploading], [self totalPeers]];
[fRemainingTimeString setString: @""];
int eta = [self eta];
if (eta < 0)
{
[fRemainingTimeString setString: NSLocalizedString(@"Unknown", "Torrent -> remaining time")];
[fProgressString appendString: NSLocalizedString(@" - remaining time unknown", "Torrent -> progress string")];
}
else
{
if (eta < 60)
[fRemainingTimeString appendFormat: NSLocalizedString(@"%d sec", "Torrent -> remaining time"), eta];
else if (eta < 3600) //60 * 60
[fRemainingTimeString appendFormat: NSLocalizedString(@"%d min %02d sec", "Torrent -> remaining time"),
eta / 60, eta % 60];
else if (eta < 86400) //24 * 60 * 60
[fRemainingTimeString appendFormat: NSLocalizedString(@"%d hr %02d min", "Torrent -> remaining time"),
eta / 3600, (eta / 60) % 60];
else
{
if (eta / 86400 > 1)
[fRemainingTimeString appendFormat: NSLocalizedString(@"%d days %d hr", "Torrent -> remaining time"),
eta / 86400, (eta / 3600) % 24];
else
[fRemainingTimeString appendFormat: NSLocalizedString(@"%d day %d hr", "Torrent -> remaining time"),
eta / 86400, (eta / 3600) % 24];
}
[fProgressString appendFormat: NSLocalizedString(@" - %@ remaining", "Torrent -> progress string"),
fRemainingTimeString];
}
break;
case TR_STATUS_SEED:
[fStatusString setString: @""];
if ([self totalPeers] > 1)
[fStatusString appendFormat: NSLocalizedString(@"Seeding to %d of %d peers", "Torrent -> status string"),
[self peersDownloading], [self totalPeers]];
else
[fStatusString appendFormat: NSLocalizedString(@"Seeding to %d of %d peer", "Torrent -> status string"),
[self peersDownloading], [self totalPeers]];
break;
case TR_STATUS_STOPPING:
tempString = [NSLocalizedString(@"Stopping", "Torrent -> status string") stringByAppendingEllipsis];
[fStatusString setString: tempString];
[fShortStatusString setString: tempString];
break;
}
if (fStat->error & TR_ETRACKER)
{
[fStatusString setString: [NSLocalizedString(@"Error: ", "Torrent -> status string") stringByAppendingString:
[self errorMessage]]];
}
BOOL wasError = fError;
if ((fError = fStat->cannotConnect))
{
if (!wasError && [self isActive] && ![self isSeeding])
[[NSNotificationCenter defaultCenter] postNotificationName: @"StoppedDownloading" object: self];
}
if ([self isActive] && fStat->status != TR_STATUS_CHECK )
{
NSString * stringToAppend = @"";
if ([self progress] < 1.0)
{
stringToAppend = [NSString stringWithFormat: NSLocalizedString(@"DL: %@, ", "Torrent -> status string"),
[NSString stringForSpeed: [self downloadRate]]];
[fShortStatusString setString: @""];
}
else
{
NSString * ratioString = [NSString stringForRatioWithDownload: [self downloadedTotal]
upload: [self uploadedTotal]];
[fShortStatusString setString: [NSString stringWithFormat: NSLocalizedString(@"Ratio: %@, ",
"Torrent -> status string"), ratioString]];
[fRemainingTimeString setString: [NSLocalizedString(@"Ratio: ", "Torrent -> status string")
stringByAppendingString: ratioString]];
}
stringToAppend = [stringToAppend stringByAppendingString: [NSLocalizedString(@"UL: ", "Torrent -> status string")
stringByAppendingString: [NSString stringForSpeed: [self uploadRate]]]];
[fStatusString appendFormat: @" - %@", stringToAppend];
[fShortStatusString appendString: stringToAppend];
}
}
- (NSDictionary *) infoForCurrentView
{
NSMutableDictionary * info = [NSMutableDictionary dictionaryWithObjectsAndKeys:
[self name], @"Name",
[NSNumber numberWithBool: [self isSeeding]], @"Seeding",
[NSNumber numberWithFloat: [self progress]], @"Progress",
[NSNumber numberWithBool: [self isActive]], @"Active",
[NSNumber numberWithBool: [self isError]], @"Error", nil];
if (![fDefaults boolForKey: @"SmallView"])
{
[info setObject: fIconFlipped forKey: @"Icon"];
[info setObject: [self progressString] forKey: @"ProgressString"];
[info setObject: [self statusString] forKey: @"StatusString"];
}
else
{
[info setObject: fIconSmall forKey: @"Icon"];
[info setObject: [self remainingTimeString] forKey: @"RemainingTimeString"];
[info setObject: [self shortStatusString] forKey: @"ShortStatusString"];
}
if ([fDefaults boolForKey: @"UseAdvancedBar"])
[info setObject: [self advancedBar] forKey: @"AdvancedBar"];
return info;
}
- (void) startTransfer
{
fWaitToStart = NO;
fFinishedSeeding = NO;
if (![self isActive] && [self alertForVolumeAvailable] && [self alertForRemainingDiskSpace])
tr_torrentStart(fHandle);
}
- (void) stopTransfer
{
fError = NO;
if ([self isActive])
{
BOOL wasSeeding = [self isSeeding];
tr_torrentStop(fHandle);
if (!wasSeeding)
[[NSNotificationCenter defaultCenter] postNotificationName: @"StoppedDownloading" object: self];
}
}
- (void) stopTransferForQuit
{
if ([self isActive])
tr_torrentStop(fHandle);
}
- (void) removeForever
{
tr_torrentRemoveSaved(fHandle);
}
- (void) sleep
{
if ((fResumeOnWake = [self isActive]))
tr_torrentStop(fHandle);
}
- (void) wakeUp
{
if (fResumeOnWake)
tr_torrentStart(fHandle);
}
- (float) ratio
{
float downloaded = [self downloadedTotal];
return downloaded > 0 ? (float)[self uploadedTotal] / downloaded : -1;
}
- (int) stopRatioSetting
{
return fStopRatioSetting;
}
- (void) setStopRatioSetting: (int) setting
{
fStopRatioSetting = setting;
}
- (float) ratioLimit
{
return fRatioLimit;
}
- (void) setRatioLimit: (float) limit
{
if (limit >= 0)
fRatioLimit = limit;
}
- (void) setWaitToStart: (BOOL) wait
{
fWaitToStart = wait;
}
- (BOOL) waitingToStart
{
return fWaitToStart;
}
- (void) revealData
{
[[NSWorkspace sharedWorkspace] selectFile: [self dataLocation] inFileViewerRootedAtPath: nil];
}
- (void) revealPublicTorrent
{
if (fPublicTorrent)
[[NSWorkspace sharedWorkspace] selectFile: fPublicTorrentLocation inFileViewerRootedAtPath: nil];
}
- (void) trashData
{
[self trashFile: [self dataLocation]];
}
- (void) trashTorrent
{
if (fPublicTorrent)
[self trashFile: [self publicTorrentLocation]];
}
- (BOOL) alertForRemainingDiskSpace
{
if ([self progress] >= 1.0)
return YES;
NSString * volumeName = [[[NSFileManager defaultManager] componentsToDisplayForPath: [self downloadFolder]]
objectAtIndex: 0];
NSDictionary * fsAttributes = [[NSFileManager defaultManager] fileSystemAttributesAtPath: [self downloadFolder]];
uint64_t remainingSpace = [[fsAttributes objectForKey: NSFileSystemFreeSize] unsignedLongLongValue],
torrentRemaining = [self size] - (uint64_t)[self downloadedValid];
/*NSLog(@"Volume: %@", volumeName);
NSLog(@"Remaining disk space: %qu (%@)", remainingSpace, [NSString stringForFileSize: remainingSpace]);
NSLog(@"Torrent remaining size: %qu (%@)", torrentRemaining, [NSString stringForFileSize: torrentRemaining]);*/
if (volumeName && remainingSpace <= torrentRemaining)
{
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: [NSString stringWithFormat:
NSLocalizedString(@"Not enough remaining disk space to download \"%@\" completely.",
"Torrent file disk space alert -> title"), [self name]]];
[alert setInformativeText: [NSString stringWithFormat:
NSLocalizedString(@"The transfer will be paused. Clear up space on \"%@\" to continue.",
"Torrent file disk space alert -> message"), volumeName]];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Torrent file disk space alert -> button")];
[alert addButtonWithTitle: NSLocalizedString(@"Download Anyway", "Torrent file disk space alert -> button")];
if ([alert runModal] == NSAlertFirstButtonReturn)
{
[[NSNotificationCenter defaultCenter] postNotificationName: @"StoppedDownloading" object: self];
return NO;
}
else
return YES;
[alert release];
}
return YES;
}
- (BOOL) alertForVolumeAvailable
{
NSArray * pathComponents = [[self downloadFolder] pathComponents];
if ([pathComponents count] < 3)
return YES;
NSString * volume = [[[pathComponents objectAtIndex: 0] stringByAppendingPathComponent:
[pathComponents objectAtIndex: 1]] stringByAppendingPathComponent: [pathComponents objectAtIndex: 2]];
/*NSLog(@"%@", [self downloadFolder]);
NSLog(@"Volume: %@", volume);*/
if (![[NSFileManager defaultManager] fileExistsAtPath: volume])
{
NSString * volumeName = [pathComponents objectAtIndex: 2];
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: [NSString stringWithFormat:
NSLocalizedString(@"The volume for downloading \"%@\" cannot be found.",
"Volume cannot be found alert -> title"), [self name]]];
[alert setInformativeText: [NSString stringWithFormat:
NSLocalizedString(@"The transfer will be paused. Mount the volume \"%@\" to continue.",
"Volume cannot be found alert -> message"), volumeName]];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Volume cannot be found alert -> button")];
[alert addButtonWithTitle: [NSLocalizedString(@"Choose New Directory",
"Volume cannot be found alert -> directory button") stringByAppendingEllipsis]];
if ([alert runModal] == NSAlertFirstButtonReturn)
[[NSNotificationCenter defaultCenter] postNotificationName: @"StoppedDownloading" object: self];
else
{
NSOpenPanel * panel = [NSOpenPanel openPanel];
[panel setPrompt: @"Select"];
[panel setAllowsMultipleSelection: NO];
[panel setCanChooseFiles: NO];
[panel setCanChooseDirectories: YES];
[panel setMessage: [NSString stringWithFormat: NSLocalizedString(@"Select the download folder for \"%@\"",
"Open torrent -> select destination folder"), [self name]]];
[[NSNotificationCenter defaultCenter] postNotificationName: @"MakeWindowKey" object: nil];
[panel beginSheetForDirectory: nil file: nil types: nil modalForWindow: [NSApp keyWindow] modalDelegate: self
didEndSelector: @selector(destinationChoiceClosed:returnCode:contextInfo:) contextInfo: nil];
}
[alert release];
return NO;
}
return YES;
}
- (void) destinationChoiceClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) context
{
NSString * folder = [[openPanel filenames] objectAtIndex: 0];
if (code == NSOKButton)
{
if (fUseIncompleteFolder)
{
[fIncompleteFolder release];
fIncompleteFolder = [folder retain];
[self setDownloadFolder: nil];
}
else
{
[fDownloadFolder release];
fDownloadFolder = folder;
[self setDownloadFolder: fDownloadFolder];
}
[self startTransfer];
[self update];
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateInfoSettings" object: nil];
}
else
[[NSNotificationCenter defaultCenter] postNotificationName: @"StoppedDownloading" object: self];
}
- (BOOL) alertForMoveVolumeAvailable
{
NSArray * pathComponents = [fDownloadFolder pathComponents];
if ([pathComponents count] < 3)
return YES;
NSString * volume = [[[pathComponents objectAtIndex: 0] stringByAppendingPathComponent:
[pathComponents objectAtIndex: 1]] stringByAppendingPathComponent: [pathComponents objectAtIndex: 2]];
/*NSLog(@"%@", [self downloadFolder]);
NSLog(@"Volume: %@", volume);*/
if (![[NSFileManager defaultManager] fileExistsAtPath: volume])
{
NSAlert * alert = [[NSAlert alloc] init];
[alert setMessageText: [NSString stringWithFormat:
NSLocalizedString(@"The volume for moving the completed \"%@\" cannot be found.",
"Move volume cannot be found alert -> title"), [self name]]];
[alert setInformativeText: NSLocalizedString(@"The file will remain in its current location",
"Move volume cannot be found alert -> message")];
[alert addButtonWithTitle: NSLocalizedString(@"OK", "Move volume cannot be found alert -> button")];
[alert runModal];
[alert release];
return NO;
}
return YES;
}
- (NSImage *) icon
{
return fIcon;
}
- (NSImage *) iconFlipped
{
return fIconFlipped;
}
- (NSImage *) iconSmall
{
return fIconSmall;
}
- (NSString *) name
{
return [NSString stringWithUTF8String: fInfo->name];
}
- (uint64_t) size
{
return fInfo->totalSize;
}
- (NSString *) tracker
{
return [NSString stringWithFormat: @"http://%s:%d", fStat->trackerAddress, fStat->trackerPort];
}
- (NSString *) announce
{
return [NSString stringWithUTF8String: fStat->trackerAnnounce];
}
- (NSString *) comment
{
return [NSString stringWithUTF8String: fInfo->comment];
}
- (NSString *) creator
{
return [NSString stringWithUTF8String: fInfo->creator];
}
- (NSDate *) dateCreated
{
int date = fInfo->dateCreated;
return date > 0 ? [NSDate dateWithTimeIntervalSince1970: date] : nil;
}
- (int) pieceSize
{
return fInfo->pieceSize;
}
- (int) pieceCount
{
return fInfo->pieceCount;
}
- (NSString *) hashString
{
return [NSString stringWithUTF8String: fInfo->hashString];
}
- (BOOL) privateTorrent
{
return fInfo->privateTorrent;
}
- (NSString *) torrentLocation
{
return [NSString stringWithUTF8String: fInfo->torrent];
}
- (NSString *) publicTorrentLocation
{
return fPublicTorrentLocation;
}
- (NSString *) dataLocation
{
return [[self downloadFolder] stringByAppendingPathComponent: [self name]];
}
- (BOOL) publicTorrent
{
return fPublicTorrent;
}
- (NSString *) stateString
{
switch( fStat->status )
{
case TR_STATUS_PAUSE:
return NSLocalizedString(@"Paused", "Torrent -> status string");
break;
case TR_STATUS_CHECK:
return [NSLocalizedString(@"Checking existing files", "Torrent -> status string") stringByAppendingEllipsis];
break;
case TR_STATUS_DOWNLOAD:
return NSLocalizedString(@"Downloading", "Torrent -> status string");
break;
case TR_STATUS_SEED:
return NSLocalizedString(@"Seeding", "Torrent -> status string");
break;
case TR_STATUS_STOPPING:
return [NSLocalizedString(@"Stopping", "Torrent -> status string") stringByAppendingEllipsis];
break;
default:
return NSLocalizedString(@"N/A", "Torrent -> status string");
}
}
- (float) progress
{
return fStat->progress;
}
- (int) eta
{
return fStat->eta;
}
- (BOOL) isActive
{
return fStat->status & TR_STATUS_ACTIVE;
}
- (BOOL) isSeeding
{
return fStat->status == TR_STATUS_SEED;
}
- (BOOL) isPaused
{
return fStat->status == TR_STATUS_PAUSE;
}
- (BOOL) isError
{
return fStat->error & TR_ETRACKER;
}
- (NSString *) errorMessage
{
[NSString stringWithUTF8String: fStat->trackerError];
}
- (BOOL) justFinished
{
return tr_getFinished(fHandle);
}
- (NSArray *) peers
{
int totalPeers, i;
tr_peer_stat_t * peers = tr_torrentPeers(fHandle, & totalPeers);
NSMutableArray * peerDics = [NSMutableArray arrayWithCapacity: totalPeers];
NSMutableDictionary * dic;
tr_peer_stat_t * peer;
NSString * client;
for (i = 0; i < totalPeers; i++)
{
peer = &peers[i];
dic = [NSMutableDictionary dictionaryWithObjectsAndKeys:
[NSNumber numberWithBool: peer->isConnected], @"Connected",
[NSNumber numberWithBool: peer->isIncoming], @"Incoming",
[NSString stringWithCString: (char *) peer->addr encoding: NSUTF8StringEncoding], @"IP",
[NSString stringWithCString: (char *) peer->client encoding: NSUTF8StringEncoding], @"Client",
[NSNumber numberWithFloat: peer->progress], @"Progress",
[NSNumber numberWithBool: peer->isDownloading], @"UL To",
[NSNumber numberWithBool: peer->isUploading], @"DL From",
[NSNumber numberWithInt: peer->port], @"Port", nil];
if (peer->isDownloading)
[dic setObject: [NSNumber numberWithFloat: peer->uploadToRate] forKey: @"UL To Rate"];
if (peer->isUploading)
[dic setObject: [NSNumber numberWithFloat: peer->downloadFromRate] forKey: @"DL From Rate"];
[peerDics addObject: dic];
}
tr_torrentPeersFree(peers, totalPeers);
return peerDics;
}
- (NSString *) progressString
{
return fProgressString;
}
- (NSString *) statusString
{
return fStatusString;
}
- (NSString *) shortStatusString
{
return fShortStatusString;
}
- (NSString *) remainingTimeString
{
return fRemainingTimeString;
}
- (int) seeders
{
return fStat->seeders;
}
- (int) leechers
{
return fStat->leechers;
}
- (int) completedFromTracker
{
return fStat->completedFromTracker;
}
- (int) totalPeers
{
return fStat->peersTotal;
}
- (int) totalPeersIncoming
{
return fStat->peersIncoming;
}
- (int) totalPeersOutgoing
{
return [self totalPeers] - [self totalPeersIncoming];
}
//peers uploading to you
- (int) peersUploading
{
return fStat->peersUploading;
}
//peers downloading from you
- (int) peersDownloading
{
return fStat->peersDownloading;
}
- (float) downloadRate
{
return fStat->rateDownload;
}
- (float) uploadRate
{
return fStat->rateUpload;
}
- (float) downloadedValid
{
return [self progress] * [self size];
}
- (uint64_t) downloadedTotal
{
return fStat->downloaded;
}
- (uint64_t) uploadedTotal
{
return fStat->uploaded;
}
- (float) swarmSpeed
{
return fStat->swarmspeed;
}
- (NSNumber *) orderValue
{
return [NSNumber numberWithInt: fOrderValue];
}
- (void) setOrderValue: (int) orderValue
{
fOrderValue = orderValue;
}
- (NSArray *) fileList
{
int count = fInfo->fileCount, i;
tr_file_t file;
NSMutableArray * files = [NSMutableArray arrayWithCapacity: count];
for (i = 0; i < count; i++)
{
file = fInfo->files[i];
[files addObject: [NSDictionary dictionaryWithObjectsAndKeys:
[[self downloadFolder] stringByAppendingPathComponent: [NSString stringWithUTF8String: file.name]], @"Name",
[NSNumber numberWithUnsignedLongLong: file.length], @"Size", nil]];
}
return files;
}
- (NSDate *) date
{
return fDate;
}
- (NSNumber *) stateSortKey
{
if (![self isActive])
return [NSNumber numberWithInt: 0];
else if ([self isSeeding])
return [NSNumber numberWithInt: 1];
else
return [NSNumber numberWithInt: 2];
}
- (NSNumber *) progressSortKey
{
//if finished downloading sort by ratio instead of progress
float progress = [self progress];
return [NSNumber numberWithFloat: progress < 1.0 ? progress : 2.0 + [self ratio]];
}
@end
@implementation Torrent (Private)
//if a hash is given, attempt to load that; otherwise, attempt to open file at path
- (id) initWithHash: (NSString *) hashString path: (NSString *) path lib: (tr_handle_t *) lib
publicTorrent: (NSNumber *) publicTorrent
date: (NSDate *) date stopRatioSetting: (NSNumber *) stopRatioSetting
ratioLimit: (NSNumber *) ratioLimit waitToStart: (NSNumber *) waitToStart
orderValue: (NSNumber *) orderValue
{
if (!(self = [super init]))
return nil;
fLib = lib;
fDefaults = [NSUserDefaults standardUserDefaults];
fPublicTorrent = path && (publicTorrent ? [publicTorrent boolValue] : ![fDefaults boolForKey: @"DeleteOriginalTorrent"]);
if (fPublicTorrent)
fPublicTorrentLocation = [path retain];
int error;
if (hashString)
fHandle = tr_torrentInitSaved(fLib, [hashString UTF8String], TR_FSAVEPRIVATE, & error);
if (!fHandle && path)
fHandle = tr_torrentInit(fLib, [path UTF8String], TR_FSAVEPRIVATE, & error);
if (!fHandle)
{
[self release];
return nil;
}
fInfo = tr_torrentInfo( fHandle );
fDate = date ? [date retain] : [[NSDate alloc] init];
fStopRatioSetting = stopRatioSetting ? [stopRatioSetting intValue] : -1;
fRatioLimit = ratioLimit ? [ratioLimit floatValue] : [fDefaults floatForKey: @"RatioLimit"];
fFinishedSeeding = NO;
fWaitToStart = waitToStart ? [waitToStart boolValue] : [fDefaults boolForKey: @"AutoStartDownload"];
fOrderValue = orderValue ? [orderValue intValue] : tr_torrentCount(fLib) - 1;
fError = NO;
NSString * fileType = fInfo->multifile ? NSFileTypeForHFSTypeCode('fldr') : [[self name] pathExtension];
fIcon = [[NSWorkspace sharedWorkspace] iconForFileType: fileType];
[fIcon retain];
fIconFlipped = [fIcon copy];
[fIconFlipped setFlipped: YES];
fIconSmall = [fIconFlipped copy];
[fIconSmall setScalesWhenResized: YES];
[fIconSmall setSize: NSMakeSize(16.0, 16.0)];
fProgressString = [[NSMutableString alloc] initWithCapacity: 50];
fStatusString = [[NSMutableString alloc] initWithCapacity: 75];
fShortStatusString = [[NSMutableString alloc] initWithCapacity: 30];
fRemainingTimeString = [[NSMutableString alloc] initWithCapacity: 30];
//set up advanced bar
fBitmap = [[NSBitmapImageRep alloc] initWithBitmapDataPlanes: nil
pixelsWide: MAX_PIECES pixelsHigh: BAR_HEIGHT bitsPerSample: 8 samplesPerPixel: 4 hasAlpha: YES
isPlanar: NO colorSpaceName: NSCalibratedRGBColorSpace bytesPerRow: 0 bitsPerPixel: 0];
fPieces = malloc(MAX_PIECES);
int i;
for (i = 0; i < MAX_PIECES; i++)
fPieces[i] = BLANK_PIECE;
[self update];
return self;
}
- (NSImage *) advancedBar
{
int h, w;
uint32_t * p;
uint8_t * bitmapData = [fBitmap bitmapData];
int bytesPerRow = [fBitmap bytesPerRow];
int8_t * pieces = malloc(MAX_PIECES);
[self getAvailability: pieces size: MAX_PIECES];
//lines 2 to 14: blue, green, or gray depending on whether we have the piece or not
int have = 0, avail = 0;
uint32_t color;
BOOL change;
for (w = 0; w < MAX_PIECES; w++)
{
change = NO;
if (pieces[w] < 0)
{
if (fPieces[w] != -1)
{
color = kBlue;
fPieces[w] = -1;
change = YES;
}
have++;
}
else if (pieces[w] == 0)
{
if (fPieces[w] != 0)
{
color = kGray;
fPieces[w] = 0;
change = YES;
}
}
else
{
if (pieces[w] == 1)
{
if (fPieces[w] != 1)
{
color = kGreen1;
fPieces[w] = 1;
change = YES;
}
}
else if (pieces[w] == 2)
{
if (fPieces[w] != 2)
{
color = kGreen2;
fPieces[w] = 2;
change = YES;
}
}
else
{
if (fPieces[w] != 3)
{
color = kGreen3;
fPieces[w] = 3;
change = YES;
}
}
avail++;
}
if (change)
{
//point to pixel (w, 2) and draw "vertically"
p = (uint32_t *)(bitmapData + 2 * bytesPerRow) + w;
for (h = 2; h < BAR_HEIGHT; h++)
{
p[0] = color;
p = (uint32_t *)((uint8_t *)p + bytesPerRow);
}
}
}
//first two lines: dark blue to show progression, green to show available
p = (uint32_t *) bitmapData;
for (w = 0; w < have; w++)
{
p[w] = kBlue2;
p[w + bytesPerRow / 4] = kBlue2;
}
for (; w < avail + have; w++)
{
p[w] = kGreen3;
p[w + bytesPerRow / 4] = kGreen3;
}
for (; w < MAX_PIECES; w++)
{
p[w] = kWhite;
p[w + bytesPerRow / 4] = kWhite;
}
free(pieces);
//actually draw image
NSImage * bar = [[NSImage alloc] initWithSize: [fBitmap size]];
[bar addRepresentation: fBitmap];
[bar setScalesWhenResized: YES];
return [bar autorelease];
}
- (void) trashFile: (NSString *) path
{
//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 is on a remote volume)
if (![[NSFileManager defaultManager] removeFileAtPath: path handler: nil])
NSLog(@"Could not trash %@", path);
}
}
@end