1
0
Fork 0
mirror of https://github.com/transmission/transmission synced 2024-12-25 17:17:31 +00:00
transmission/macosx/Torrent.m

595 lines
15 KiB
Mathematica
Raw Normal View History

2006-03-23 12:39:39 +00:00
/******************************************************************************
* $Id$
*
* Copyright (c) 2006 Transmission authors and contributors
2006-03-23 12:39:39 +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 "StringAdditions.h"
@interface Torrent (Private)
- (id) initWithHash: (NSString *) hashString path: (NSString *) path lib: (tr_handle_t *) lib
2006-06-15 05:30:08 +00:00
privateTorrent: (NSNumber *) privateTorrent publicTorrent: (NSNumber *) publicTorrent
date: (NSDate *) date stopRatioSetting: (NSNumber *) stopRatioSetting
ratioLimit: (NSNumber *) ratioLimit;
- (void) trashFile: (NSString *) path;
2006-03-23 12:39:39 +00:00
@end
@implementation Torrent
- (id) initWithPath: (NSString *) path lib: (tr_handle_t *) lib
2006-03-23 12:39:39 +00:00
{
self = [self initWithHash: nil path: path lib: lib privateTorrent: nil publicTorrent: nil
date: nil stopRatioSetting: nil ratioLimit: nil];
if (self)
{
if (!fPublicTorrent)
[self trashFile: path];
}
2006-06-15 05:30:08 +00:00
return self;
}
2006-03-23 12:39:39 +00:00
- (id) initWithHistory: (NSDictionary *) history lib: (tr_handle_t *) lib
{
self = [self initWithHash: [history objectForKey: @"TorrentHash"]
2006-06-15 05:30:08 +00:00
path: [history objectForKey: @"TorrentPath"] lib: lib
privateTorrent: [history objectForKey: @"PrivateCopy"]
publicTorrent: [history objectForKey: @"PublicCopy"]
date: [history objectForKey: @"Date"]
2006-06-15 05:03:32 +00:00
stopRatioSetting: [history objectForKey: @"StopRatioSetting"]
ratioLimit: [history objectForKey: @"RatioLimit"]];
if (self)
2006-03-23 12:39:39 +00:00
{
NSString * downloadFolder;
if (!(downloadFolder = [history objectForKey: @"DownloadFolder"]))
downloadFolder = [[fDefaults stringForKey: @"DownloadFolder"] stringByExpandingTildeInPath];
[self setDownloadFolder: downloadFolder];
NSString * paused;
if (!(paused = [history objectForKey: @"Paused"]) || [paused isEqualToString: @"NO"])
[self startTransfer];
2006-03-23 12:39:39 +00:00
}
2006-06-15 05:30:08 +00:00
return self;
2006-03-23 12:39:39 +00:00
}
- (NSDictionary *) history
{
NSMutableDictionary * history = [NSMutableDictionary dictionaryWithObjectsAndKeys:
[NSNumber numberWithBool: fPrivateTorrent], @"PrivateCopy",
[NSNumber numberWithBool: fPublicTorrent], @"PublicCopy",
[self downloadFolder], @"DownloadFolder",
[self isActive] ? @"NO" : @"YES", @"Paused",
[self date], @"Date",
[NSNumber numberWithInt: fStopRatioSetting], @"StopRatioSetting",
[NSNumber numberWithFloat: fRatioLimit], @"RatioLimit", nil];
if (fPrivateTorrent)
[history setObject: [self hashString] forKey: @"TorrentHash"];
if (fPublicTorrent)
[history setObject: [self publicTorrentLocation] forKey: @"TorrentPath"];
return history;
}
2006-03-23 12:39:39 +00:00
- (void) dealloc
{
if (fHandle)
2006-03-23 12:39:39 +00:00
{
tr_torrentClose(fLib, fHandle);
if (fPublicTorrentLocation)
[fPublicTorrentLocation release];
[fDate release];
2006-03-23 12:39:39 +00:00
[fIcon release];
[fIconFlipped release];
[fProgressString release];
2006-03-23 12:39:39 +00:00
[fStatusString release];
}
[super dealloc];
}
- (void) setDownloadFolder: (NSString *) path
2006-03-23 12:39:39 +00:00
{
tr_torrentSetFolder(fHandle, [path UTF8String]);
2006-03-23 12:39:39 +00:00
}
- (NSString *) downloadFolder
2006-03-23 12:39:39 +00:00
{
return [NSString stringWithUTF8String: tr_torrentGetFolder(fHandle)];
2006-03-23 12:39:39 +00:00
}
- (void) getAvailability: (int8_t *) tab size: (int) size
{
tr_torrentAvailability(fHandle, tab, size);
2006-03-23 12:39:39 +00:00
}
- (void) update
{
fStat = tr_torrentStat(fHandle);
if ([self isSeeding] && ((fStopRatioSetting == RATIO_CHECK && [self ratio] >= fRatioLimit)
2006-06-17 19:34:08 +00:00
|| (fStopRatioSetting == RATIO_GLOBAL && [fDefaults boolForKey: @"RatioCheck"]
2006-06-16 17:33:16 +00:00
&& [self ratio] >= [fDefaults floatForKey: @"RatioLimit"])))
{
[self stopTransfer];
[self setStopRatioSetting: RATIO_NO_CHECK];
fFinishedSeeding = YES;
fStat = tr_torrentStat(fHandle);
[[NSNotificationCenter defaultCenter] postNotificationName: @"TorrentRatioChanged" object: self];
}
[fProgressString setString: @""];
if ([self progress] < 1.0)
[fProgressString appendFormat: @"%@ of %@ (%.2f%%)", [NSString stringForFileSize:
2006-06-16 00:35:11 +00:00
[self downloadedValid]], [NSString stringForFileSize: [self size]], 100.0 * [self progress]];
else
[fProgressString appendFormat: @"%@, uploaded %@ (ratio: %@)", [NSString stringForFileSize:
[self size]], [NSString stringForFileSize: [self uploadedTotal]],
[NSString stringForRatioWithDownload: [self downloadedTotal] upload: [self uploadedTotal]]];
2006-03-23 12:39:39 +00:00
switch (fStat->status)
2006-03-23 12:39:39 +00:00
{
case TR_STATUS_PAUSE:
[fStatusString setString: fFinishedSeeding ? @"Seeding Complete" : @"Paused"];
2006-03-23 12:39:39 +00:00
break;
case TR_STATUS_CHECK:
[fStatusString setString: [@"Checking existing files" stringByAppendingEllipsis]];
2006-03-23 12:39:39 +00:00
break;
case TR_STATUS_DOWNLOAD:
[fStatusString setString: @""];
[fStatusString appendFormat:
@"Downloading from %d of %d peer%s", [self peersUploading], [self totalPeers],
([self totalPeers] == 1) ? "" : "s"];
int eta = [self eta];
if (eta < 0)
[fProgressString appendString: @" - remaining time unknown"];
2006-03-23 12:39:39 +00:00
else
{
if (eta < 60)
[fProgressString appendFormat: @" - %d sec remaining", eta];
else if (eta < 3600)
[fProgressString appendFormat: @" - %d min %02d sec remaining",
eta / 60, eta % 60];
else
[fProgressString appendFormat: @" - %d hr %02d min remaining",
eta / 3600, (eta / 60) % 60];
2006-03-23 12:39:39 +00:00
}
break;
case TR_STATUS_SEED:
[fStatusString setString: @""];
2006-03-23 12:39:39 +00:00
[fStatusString appendFormat:
@"Seeding to %d of %d peer%s",
[self peersDownloading], [self totalPeers],
([self totalPeers] == 1) ? "" : "s"];
2006-03-23 12:39:39 +00:00
break;
case TR_STATUS_STOPPING:
[fStatusString setString: [@"Stopping" stringByAppendingEllipsis]];
2006-03-23 12:39:39 +00:00
break;
}
2006-04-08 15:56:26 +00:00
if( fStat->error & TR_ETRACKER )
2006-03-23 12:39:39 +00:00
{
[fStatusString setString: [@"Error: " stringByAppendingString:
[NSString stringWithUTF8String: fStat->trackerError]]];
2006-03-23 12:39:39 +00:00
}
if ([self isActive])
2006-03-23 12:39:39 +00:00
{
2006-06-07 14:42:54 +00:00
[fStatusString appendString: @" - "];
if ([self progress] < 1.0)
2006-06-07 14:42:54 +00:00
[fStatusString appendFormat: @"DL: %@, ", [NSString stringForSpeed: [self downloadRate]]];
[fStatusString appendString: [@"UL: " stringByAppendingString:
[NSString stringForSpeed: [self uploadRate]]]];
2006-03-23 12:39:39 +00:00
}
}
- (void) startTransfer
2006-03-23 12:39:39 +00:00
{
if (![self isActive])
2006-03-23 12:39:39 +00:00
{
tr_torrentStart(fHandle);
fFinishedSeeding = NO;
2006-03-23 12:39:39 +00:00
}
}
- (void) stopTransfer
2006-03-23 12:39:39 +00:00
{
if ([self isActive])
tr_torrentStop(fHandle);
2006-03-23 12:39:39 +00:00
}
- (void) removeForever
{
if (fPrivateTorrent)
tr_torrentRemoveSaved(fHandle);
}
2006-03-23 12:39:39 +00:00
- (void) sleep
{
if ((fResumeOnWake = [self isActive]))
[self stopTransfer];
2006-03-23 12:39:39 +00:00
}
- (void) wakeUp
{
if (fResumeOnWake)
[self startTransfer];
2006-03-23 12:39:39 +00:00
}
- (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) revealData
2006-03-23 12:39:39 +00:00
{
[[NSWorkspace sharedWorkspace] selectFile: [self dataLocation] inFileViewerRootedAtPath: nil];
2006-03-23 12:39:39 +00:00
}
- (void) trashData
{
[self trashFile: [self dataLocation]];
2006-03-23 12:39:39 +00:00
}
- (void) trashTorrent
{
if (fPublicTorrent)
[self trashFile: [self publicTorrentLocation]];
}
2006-03-23 12:39:39 +00:00
- (NSImage *) icon
{
return fIcon;
}
- (NSImage *) iconFlipped
2006-03-23 12:39:39 +00:00
{
return fIconFlipped;
2006-03-23 12:39:39 +00:00
}
- (NSString *) name
{
return [NSString stringWithUTF8String: fInfo->name];
}
- (uint64_t) size
{
return fInfo->totalSize;
}
- (NSString *) tracker
{
return [NSString stringWithFormat: @"%s:%d", fInfo->trackerAddress, fInfo->trackerPort];
}
- (NSString *) announce
{
return [NSString stringWithUTF8String: fInfo->trackerAnnounce];
}
- (int) pieceSize
{
return fInfo->pieceSize;
}
- (int) pieceCount
{
return fInfo->pieceCount;
}
- (NSString *) hashString
{
return [NSString stringWithUTF8String: fInfo->hashString];
}
- (NSString *) torrentLocation
{
return [NSString stringWithUTF8String: fInfo->torrent];
}
- (NSString *) publicTorrentLocation
{
return fPublicTorrentLocation;
}
- (NSString *) torrentLocationString
{
2006-06-20 03:00:39 +00:00
return fPrivateTorrent ? @"Transmission Support Folder"
: [fPublicTorrentLocation stringByAbbreviatingWithTildeInPath];
}
- (NSString *) dataLocation
{
return [[self downloadFolder] stringByAppendingPathComponent: [self name]];
}
- (BOOL) publicTorrent
{
return fPublicTorrent;
}
- (BOOL) privateTorrent
{
return fPrivateTorrent;
}
/*- (NSString *) state
{
switch( fStat->status )
{
case TR_STATUS_PAUSE:
return @"Paused";
break;
case TR_STATUS_CHECK:
return [@"Checking existing files" stringByAppendingEllipsis];
break;
case TR_STATUS_DOWNLOAD:
return @"Downloading";
break;
case TR_STATUS_SEED:
return @"Seeding";
break;
case TR_STATUS_STOPPING:
return [@"Stopping" stringByAppendingEllipsis];
break;
default:
return @"N/A";
}
}*/
2006-03-23 12:39:39 +00:00
- (float) progress
{
return fStat->progress;
}
- (int) eta
{
return fStat->eta;
}
2006-03-23 12:39:39 +00:00
- (BOOL) isActive
{
return fStat->status & TR_STATUS_ACTIVE;
2006-03-23 12:39:39 +00:00
}
- (BOOL) isSeeding
{
return fStat->status == TR_STATUS_SEED;
2006-03-23 12:39:39 +00:00
}
- (BOOL) isPaused
{
return fStat->status == TR_STATUS_PAUSE;
2006-03-23 12:39:39 +00:00
}
- (BOOL) justFinished
{
return tr_getFinished(fHandle);
2006-03-23 12:39:39 +00:00
}
- (NSString *) progressString
{
return fProgressString;
}
2006-03-23 12:39:39 +00:00
- (NSString *) statusString
{
return fStatusString;
}
- (int) seeders
2006-03-23 12:39:39 +00:00
{
return fStat->seeders;
2006-03-23 12:39:39 +00:00
}
- (int) leechers
2006-03-23 12:39:39 +00:00
{
return fStat->leechers;
2006-03-23 12:39:39 +00:00
}
- (int) totalPeers
2006-03-23 12:39:39 +00:00
{
return fStat->peersTotal;
2006-03-23 12:39:39 +00:00
}
//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;
}
- (NSArray *) fileList
{
int count = fInfo->fileCount, i;
NSMutableArray * files = [NSMutableArray arrayWithCapacity: count];
for (i = 0; i < count; i++)
[files addObject: [[self downloadFolder] stringByAppendingPathComponent:
[NSString stringWithUTF8String: fInfo->files[i].name]]];
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
{
return [NSNumber numberWithFloat: [self progress]];
}
2006-03-23 12:39:39 +00:00
@end
@implementation Torrent (Private)
2006-06-15 05:30:08 +00:00
//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
2006-06-15 05:30:08 +00:00
privateTorrent: (NSNumber *) privateTorrent publicTorrent: (NSNumber *) publicTorrent
date: (NSDate *) date stopRatioSetting: (NSNumber *) stopRatioSetting
ratioLimit: (NSNumber *) ratioLimit
{
if (!(self = [super init]))
return nil;
fLib = lib;
2006-06-15 05:30:08 +00:00
fDefaults = [NSUserDefaults standardUserDefaults];
fPrivateTorrent = privateTorrent ? [privateTorrent boolValue] : [fDefaults boolForKey: @"SavePrivateTorrent"];
fPublicTorrent = !fPrivateTorrent || (publicTorrent ? [publicTorrent boolValue]
: ![fDefaults boolForKey: @"DeleteOriginalTorrent"]);
int error;
2006-06-15 05:30:08 +00:00
if (fPrivateTorrent && hashString)
2006-06-15 05:03:32 +00:00
fHandle = tr_torrentInitSaved(fLib, [hashString UTF8String], TR_FSAVEPRIVATE, & error);
2006-06-15 22:07:05 +00:00
if (!fHandle && path)
2006-06-15 05:03:32 +00:00
fHandle = tr_torrentInit(fLib, [path UTF8String], fPrivateTorrent ? TR_FSAVEPRIVATE : 0, & error);
2006-06-15 05:30:08 +00:00
2006-06-15 05:03:32 +00:00
if (!fHandle)
{
[self release];
return nil;
}
fInfo = tr_torrentInfo( fHandle );
2006-06-15 05:30:08 +00:00
if (fPublicTorrent)
fPublicTorrentLocation = [path retain];
fDate = date ? [date retain] : [[NSDate alloc] init];
fStopRatioSetting = stopRatioSetting ? [stopRatioSetting intValue] : -1;
fRatioLimit = ratioLimit ? [ratioLimit floatValue] : [fDefaults floatForKey: @"RatioLimit"];
fFinishedSeeding = NO;
NSString * fileType = fInfo->multifile ? NSFileTypeForHFSTypeCode('fldr') : [[self name] pathExtension];
fIcon = [[NSWorkspace sharedWorkspace] iconForFileType: fileType];
[fIcon retain];
fIconFlipped = [fIcon copy];
[fIconFlipped setFlipped: YES];
fProgressString = [[NSMutableString alloc] initWithCapacity: 50];
fStatusString = [[NSMutableString alloc] initWithCapacity: 75];
[self update];
return self;
}
- (void) trashFile: (NSString *) path
2006-03-23 12:39:39 +00:00
{
if (![[NSWorkspace sharedWorkspace] performFileOperation: NSWorkspaceRecycleOperation
source: [path stringByDeletingLastPathComponent] destination: @""
files: [NSArray arrayWithObject: [path lastPathComponent]] tag: nil])
2006-03-23 12:39:39 +00:00
{
2006-03-28 06:26:13 +00:00
/* We can't move it to the trash, let's try just to delete it
(will work if it is on a remote volume) */
if (![[NSFileManager defaultManager] removeFileAtPath: path handler: nil])
NSLog([@"Could not trash " stringByAppendingString: path]);
2006-03-23 12:39:39 +00:00
}
}
@end