transmission/macosx/Torrent.m

481 lines
12 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"
#import "Utils.h"
2006-03-23 12:39:39 +00:00
@interface Torrent (Private)
- (void) trashPath: (NSString *) path;
- (id) initWithPath: (NSString *) path lib: (tr_handle_t *) lib date: (NSDate *) date
stopRatioSetting: (NSNumber *) stopRatioSetting ratioLimit: (NSNumber *) ratioLimit;
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
{
return [self initWithPath: path lib: lib
date: nil stopRatioSetting: nil
ratioLimit: nil];
}
2006-03-23 12:39:39 +00:00
- (id) initWithHistory: (NSDictionary *) history lib: (tr_handle_t *) lib
{
self = [self initWithPath: [history objectForKey: @"TorrentPath"]
lib: lib date: [history objectForKey: @"Date"]
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 setFolder: downloadFolder];
NSString * paused;
if (!(paused = [history objectForKey: @"Paused"]) || [paused isEqualToString: @"NO"])
[self start];
2006-03-23 12:39:39 +00:00
}
2006-03-23 12:39:39 +00:00
return self;
}
- (NSDictionary *) history
{
return [NSDictionary dictionaryWithObjectsAndKeys:
[self path], @"TorrentPath",
[self getFolder], @"DownloadFolder",
[self isActive] ? @"NO" : @"YES", @"Paused",
[self date], @"Date",
[NSNumber numberWithInt: fStopRatioSetting], @"StopRatioSetting",
[NSNumber numberWithFloat: fRatioLimit], @"RatioLimit", nil];
}
2006-03-23 12:39:39 +00:00
- (void) dealloc
{
if( fHandle )
{
tr_torrentClose( fLib, fHandle );
[fDate release];
2006-03-23 12:39:39 +00:00
[fIcon release];
[fIconNonFlipped release];
2006-03-23 12:39:39 +00:00
[fStatusString release];
[fInfoString release];
[fDownloadString release];
[fUploadString release];
}
[super dealloc];
}
- (void) setFolder: (NSString *) path
{
tr_torrentSetFolder( fHandle, [path UTF8String] );
}
- (NSString *) getFolder
{
return [NSString stringWithUTF8String: tr_torrentGetFolder( fHandle )];
}
- (void) getAvailability: (int8_t *) tab size: (int) size
{
tr_torrentAvailability( fHandle, tab, size );
}
- (void) update
{
fStat = tr_torrentStat( fHandle );
if ([self isSeeding])
if ((fStopRatioSetting == 1 && [self ratio] >= fRatioLimit)
|| (fStopRatioSetting == -1 && [fDefaults boolForKey: @"RatioCheck"]
&& [self ratio] >= [fDefaults floatForKey: @"RatioLimit"]))
{
[self stop];
[self setStopRatioSetting: 0];
fStat = tr_torrentStat( fHandle );
}
2006-03-23 12:39:39 +00:00
[fStatusString setString: @""];
[fInfoString setString: @""];
2006-03-23 12:39:39 +00:00
switch( fStat->status )
{
case TR_STATUS_PAUSE:
[fStatusString appendFormat: @"Paused (%.2f %%)",
100 * fStat->progress];
break;
case TR_STATUS_CHECK:
[fStatusString appendFormat:
@"Checking existing files (%.2f %%)",
100 * fStat->progress];
break;
case TR_STATUS_DOWNLOAD:
if( fStat->eta < 0 )
{
[fStatusString appendFormat:
@"Finishing in --:--:-- (%.2f %%)",
100 * fStat->progress];
}
else
{
[fStatusString appendFormat:
@"Finishing in %02d:%02d:%02d (%.2f %%)",
fStat->eta / 3600, ( fStat->eta / 60 ) % 60,
fStat->eta % 60, 100 * fStat->progress];
}
[fInfoString appendFormat:
@"Downloading from %d of %d peer%s",
fStat->peersUploading, fStat->peersTotal,
( fStat->peersTotal == 1 ) ? "" : "s"];
break;
case TR_STATUS_SEED:
[fStatusString appendFormat:
@"Seeding, uploading to %d of %d peer%s",
fStat->peersDownloading, fStat->peersTotal,
( fStat->peersTotal == 1 ) ? "" : "s"];
break;
case TR_STATUS_STOPPING:
[fStatusString setString: [@"Stopping"
stringByAppendingString: NS_ELLIPSIS]];
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
{
[fInfoString setString: [@"Error: " stringByAppendingString:
[NSString stringWithUTF8String: fStat->trackerError]]];
2006-03-23 12:39:39 +00:00
}
if( fStat->progress == 1.0 )
{
[fDownloadString setString: [@"Ratio: " stringByAppendingString:
[NSString stringForRatio: fStat->downloaded
upload: fStat->uploaded]]];
2006-03-23 12:39:39 +00:00
}
else
{
[fDownloadString setString: [@"DL: " stringByAppendingString:
[NSString stringForSpeed: fStat->rateDownload]]];
2006-03-23 12:39:39 +00:00
}
[fUploadString setString: [@"UL: " stringByAppendingString:
[NSString stringForSpeed: fStat->rateUpload]]];
2006-03-23 12:39:39 +00:00
}
- (void) start
{
if( fStat->status & TR_STATUS_INACTIVE )
{
tr_torrentStart( fHandle );
}
}
- (void) stop
{
if( fStat->status & TR_STATUS_ACTIVE )
{
tr_torrentStop( fHandle );
}
}
- (void) sleep
{
if( ( fResumeOnWake = ( fStat->status & TR_STATUS_ACTIVE ) ) )
2006-03-23 12:39:39 +00:00
{
[self stop];
}
}
- (void) wakeUp
{
if( fResumeOnWake )
{
[self start];
}
}
- (float) ratio
{
uint64_t downloaded = [self downloaded];
return downloaded > 0 ? [self uploaded] / downloaded : -1;
}
/* 1: Check ratio
0: Don't check ratio
-1: Use defaults */
- (int) stopRatioSetting
{
return fStopRatioSetting;
}
- (void) setStopRatioSetting: (int) setting
{
fStopRatioSetting = setting;
}
- (float) ratioLimit
{
return fRatioLimit;
}
- (void) setRatioLimit: (float) limit
{
if (limit >= 0)
fRatioLimit = limit;
}
2006-03-23 12:39:39 +00:00
- (void) reveal
{
2006-04-22 16:54:20 +00:00
[[NSWorkspace sharedWorkspace] selectFile: [[self getFolder]
stringByAppendingPathComponent: [self name]]
2006-03-23 12:39:39 +00:00
inFileViewerRootedAtPath: nil];
}
- (void) trashTorrent
{
[self trashPath: [self path]];
}
- (void) trashData
{
2006-04-22 16:54:20 +00:00
[self trashPath: [[self getFolder]
stringByAppendingPathComponent: [self name]]];
2006-03-23 12:39:39 +00:00
}
- (NSImage *) icon
{
return fIcon;
}
- (NSImage *) iconNonFlipped
{
return fIconNonFlipped;
}
2006-03-23 12:39:39 +00:00
- (NSString *) path
{
return [NSString stringWithUTF8String: fInfo->torrent];
}
- (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 *) hash
{
NSMutableString * string = [NSMutableString
stringWithCapacity: SHA_DIGEST_LENGTH];
int i;
for( i = 0; i < SHA_DIGEST_LENGTH; i++ )
{
[string appendFormat: @"%02x", fInfo->hash[i]];
}
return string;
}
2006-03-23 12:39:39 +00:00
- (float) progress
{
return fStat->progress;
}
- (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) justFinished
{
return tr_getFinished( fHandle );
}
- (NSString *) statusString
{
return fStatusString;
}
- (NSString *) infoString
{
return fInfoString;
}
- (NSString *) downloadString
{
return fDownloadString;
}
- (NSString *) uploadString
{
return fUploadString;
}
- (int) seeders
{
return fStat->seeders;
}
- (int) leechers
{
return fStat->leechers;
}
- (uint64_t) downloaded
{
return fStat->downloaded;
}
- (uint64_t) uploaded
{
return fStat->uploaded;
}
- (NSDate *) date
{
return fDate;
}
- (NSNumber *) stateSortKey
{
if (fStat->status & TR_STATUS_INACTIVE)
return [NSNumber numberWithInt: 0];
else if (fStat->status == TR_STATUS_SEED)
return [NSNumber numberWithInt: 1];
else
return [NSNumber numberWithInt: 2];
}
2006-03-23 12:39:39 +00:00
@end
@implementation Torrent (Private)
- (id) initWithPath: (NSString *) path lib: (tr_handle_t *) lib date: (NSDate *) date
stopRatioSetting: (NSNumber *) stopRatioSetting ratioLimit: (NSNumber *) ratioLimit
{
if (!(self = [super init]))
return nil;
fLib = lib;
int error;
if (!path || !(fHandle = tr_torrentInit(fLib, [path UTF8String], &error)))
{
[self release];
return nil;
}
fInfo = tr_torrentInfo( fHandle );
fDefaults = [NSUserDefaults standardUserDefaults];
fDate = date ? [date retain] : [[NSDate alloc] init];
fStopRatioSetting = stopRatioSetting ? [stopRatioSetting intValue] : -1;
fRatioLimit = ratioLimit ? [ratioLimit floatValue] : [fDefaults floatForKey: @"RatioLimit"];
NSString * fileType = ( fInfo->fileCount > 1 ) ?
NSFileTypeForHFSTypeCode('fldr') : [[self name] pathExtension];
fIcon = [[NSWorkspace sharedWorkspace] iconForFileType: fileType];
[fIcon setFlipped: YES];
[fIcon retain];
fIconNonFlipped = [[NSWorkspace sharedWorkspace] iconForFileType: fileType];
[fIconNonFlipped retain];
fStatusString = [[NSMutableString alloc] initWithCapacity: 50];
fInfoString = [[NSMutableString alloc] initWithCapacity: 50];
fDownloadString = [[NSMutableString alloc] initWithCapacity: 10];
fUploadString = [[NSMutableString alloc] initWithCapacity: 10];
[self update];
return self;
}
2006-03-23 12:39:39 +00:00
- (void) trashPath: (NSString *) path
{
2006-03-28 06:26:13 +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-28 06:26:13 +00:00
}
2006-03-23 12:39:39 +00:00
}
}
@end