357 lines
12 KiB
Mathematica
357 lines
12 KiB
Mathematica
![]() |
/******************************************************************************
|
||
|
* $Id$
|
||
|
*
|
||
|
* Copyright (c) 2010 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 "InfoTrackersViewController.h"
|
||
|
#import "NSApplicationAdditions.h"
|
||
|
#import "Torrent.h"
|
||
|
#import "TrackerCell.h"
|
||
|
#import "TrackerNode.h"
|
||
|
#import "TrackerTableView.h"
|
||
|
|
||
|
#define TRACKER_GROUP_SEPARATOR_HEIGHT 14.0
|
||
|
|
||
|
#define TRACKER_ADD_TAG 0
|
||
|
#define TRACKER_REMOVE_TAG 1
|
||
|
|
||
|
@interface InfoTrackersViewController (Private)
|
||
|
|
||
|
- (void) resetInfo;
|
||
|
|
||
|
- (void) addTrackers;
|
||
|
- (void) removeTrackers;
|
||
|
|
||
|
@end
|
||
|
|
||
|
@implementation InfoTrackersViewController
|
||
|
|
||
|
- (id) init
|
||
|
{
|
||
|
if ((self = [super initWithNibName: @"InfoTrackersView" bundle: nil]))
|
||
|
{
|
||
|
fTrackerCell = [[TrackerCell alloc] init];
|
||
|
}
|
||
|
|
||
|
return self;
|
||
|
}
|
||
|
|
||
|
- (void) dealloc
|
||
|
{
|
||
|
[fTorrents release];
|
||
|
[fTrackers release];
|
||
|
[fTrackerCell release];
|
||
|
|
||
|
[super dealloc];
|
||
|
}
|
||
|
|
||
|
- (void) setInfoForTorrents: (NSArray *) torrents
|
||
|
{
|
||
|
if (fTorrents && [fTorrents isEqualToArray: torrents])
|
||
|
return;
|
||
|
|
||
|
[fTorrents release];
|
||
|
fTorrents = [torrents retain];
|
||
|
|
||
|
[self resetInfo];
|
||
|
}
|
||
|
|
||
|
- (void) updateInfo
|
||
|
{
|
||
|
if ([fTorrents count] == 0)
|
||
|
return;
|
||
|
|
||
|
//get updated tracker stats
|
||
|
if ([fTrackerTable editedRow] == -1)
|
||
|
{
|
||
|
[fTrackers release];
|
||
|
|
||
|
if ([fTorrents count] == 1)
|
||
|
fTrackers = [[[fTorrents objectAtIndex: 0] allTrackerStats] retain];
|
||
|
else
|
||
|
{
|
||
|
fTrackers = [[NSMutableArray alloc] init];
|
||
|
for (Torrent * torrent in fTorrents)
|
||
|
[fTrackers addObjectsFromArray: [torrent allTrackerStats]];
|
||
|
}
|
||
|
|
||
|
[fTrackerTable setTrackers: fTrackers];
|
||
|
[fTrackerTable reloadData];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
NSAssert1([fTorrents count] == 1, @"Attempting to add tracker with %d transfers selected", [fTorrents count]);
|
||
|
|
||
|
if ([NSApp isOnSnowLeopardOrBetter])
|
||
|
{
|
||
|
NSIndexSet * addedIndexes = [NSIndexSet indexSetWithIndexesInRange: NSMakeRange([fTrackers count]-2, 2)];
|
||
|
NSArray * tierAndTrackerBeingAdded = [fTrackers objectsAtIndexes: addedIndexes];
|
||
|
|
||
|
[fTrackers release];
|
||
|
fTrackers = [[[fTorrents objectAtIndex: 0] allTrackerStats] retain];
|
||
|
[fTrackers addObjectsFromArray: tierAndTrackerBeingAdded];
|
||
|
|
||
|
[fTrackerTable setTrackers: fTrackers];
|
||
|
|
||
|
NSIndexSet * updateIndexes = [NSIndexSet indexSetWithIndexesInRange: NSMakeRange(0, [fTrackers count]-2)],
|
||
|
* columnIndexes = [NSIndexSet indexSetWithIndexesInRange: NSMakeRange(0, [[fTrackerTable tableColumns] count])];
|
||
|
[fTrackerTable reloadDataForRowIndexes: updateIndexes columnIndexes: columnIndexes];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
- (void) clearTrackers
|
||
|
{
|
||
|
[fTrackers release];
|
||
|
fTrackers = nil;
|
||
|
}
|
||
|
|
||
|
- (NSInteger) numberOfRowsInTableView: (NSTableView *) tableView
|
||
|
{
|
||
|
return fTrackers ? [fTrackers count] : 0;
|
||
|
}
|
||
|
|
||
|
- (id) tableView: (NSTableView *) tableView objectValueForTableColumn: (NSTableColumn *) column row: (NSInteger) row
|
||
|
{
|
||
|
id item = [fTrackers objectAtIndex: row];
|
||
|
|
||
|
if ([item isKindOfClass: [NSDictionary class]])
|
||
|
{
|
||
|
const NSInteger tier = [[item objectForKey: @"Tier"] integerValue];
|
||
|
NSString * tierString = tier == -1 ? NSLocalizedString(@"New Tier", "Inspector -> tracker table")
|
||
|
: [NSString stringWithFormat: NSLocalizedString(@"Tier %d", "Inspector -> tracker table"), tier];
|
||
|
|
||
|
if ([fTorrents count] > 1)
|
||
|
tierString = [tierString stringByAppendingFormat: @" - %@", [item objectForKey: @"Name"]];
|
||
|
return tierString;
|
||
|
}
|
||
|
else
|
||
|
return item; //TrackerNode or NSString
|
||
|
}
|
||
|
|
||
|
- (NSCell *) tableView: (NSTableView *) tableView dataCellForTableColumn: (NSTableColumn *) tableColumn row: (NSInteger) row
|
||
|
{
|
||
|
const BOOL tracker = [[fTrackers objectAtIndex: row] isKindOfClass: [TrackerNode class]];
|
||
|
return tracker ? fTrackerCell : [tableColumn dataCellForRow: row];
|
||
|
}
|
||
|
|
||
|
- (CGFloat) tableView: (NSTableView *) tableView heightOfRow: (NSInteger) row
|
||
|
{
|
||
|
//check for NSDictionay instead of TrackerNode because of display issue when adding a row
|
||
|
if ([[fTrackers objectAtIndex: row] isKindOfClass: [NSDictionary class]])
|
||
|
return TRACKER_GROUP_SEPARATOR_HEIGHT;
|
||
|
else
|
||
|
return [tableView rowHeight];
|
||
|
}
|
||
|
|
||
|
- (void) tableViewSelectionDidChange: (NSNotification *) notification
|
||
|
{
|
||
|
[fTrackerAddRemoveControl setEnabled: [fTrackerTable numberOfSelectedRows] > 0 forSegment: TRACKER_REMOVE_TAG];
|
||
|
}
|
||
|
|
||
|
- (BOOL) tableView: (NSTableView *) tableView isGroupRow: (NSInteger) row
|
||
|
{
|
||
|
return ![[fTrackers objectAtIndex: row] isKindOfClass: [TrackerNode class]] && [tableView editedRow] != row;
|
||
|
}
|
||
|
|
||
|
- (NSString *) tableView: (NSTableView *) tableView toolTipForCell: (NSCell *) cell rect: (NSRectPointer) rect
|
||
|
tableColumn: (NSTableColumn *) column row: (NSInteger) row mouseLocation: (NSPoint) mouseLocation
|
||
|
{
|
||
|
id node = [fTrackers objectAtIndex: row];
|
||
|
if ([node isKindOfClass: [TrackerNode class]])
|
||
|
return [(TrackerNode *)node fullAnnounceAddress];
|
||
|
else
|
||
|
return nil;
|
||
|
}
|
||
|
|
||
|
- (void) tableView: (NSTableView *) tableView setObjectValue: (id) object forTableColumn: (NSTableColumn *) tableColumn
|
||
|
row: (NSInteger) row
|
||
|
{
|
||
|
Torrent * torrent= [fTorrents objectAtIndex: 0];
|
||
|
|
||
|
BOOL added = NO;
|
||
|
for (NSString * tracker in [object componentsSeparatedByString: @"\n"])
|
||
|
if ([torrent addTrackerToNewTier: tracker])
|
||
|
added = YES;
|
||
|
|
||
|
if (!added)
|
||
|
NSBeep();
|
||
|
|
||
|
//reset table with either new or old value
|
||
|
[fTrackers release];
|
||
|
fTrackers = [[torrent allTrackerStats] retain];
|
||
|
|
||
|
[fTrackerTable setTrackers: fTrackers];
|
||
|
[fTrackerTable reloadData];
|
||
|
[fTrackerTable deselectAll: self];
|
||
|
|
||
|
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateUI" object: nil]; //incase sort by tracker
|
||
|
}
|
||
|
|
||
|
- (void) addRemoveTracker: (id) sender
|
||
|
{
|
||
|
//don't allow add/remove when currently adding - it leads to weird results
|
||
|
if ([fTrackerTable editedRow] != -1)
|
||
|
return;
|
||
|
|
||
|
if ([[sender cell] tagForSegment: [sender selectedSegment]] == TRACKER_REMOVE_TAG)
|
||
|
[self removeTrackers];
|
||
|
else
|
||
|
[self addTrackers];
|
||
|
}
|
||
|
|
||
|
@end
|
||
|
|
||
|
@implementation InfoTrackersViewController (Private)
|
||
|
|
||
|
- (void) resetInfo
|
||
|
{
|
||
|
const NSUInteger numberSelected = [fTorrents count];
|
||
|
if (numberSelected != 1)
|
||
|
{
|
||
|
if (numberSelected == 0)
|
||
|
{
|
||
|
[fTrackers release];
|
||
|
fTrackers = nil;
|
||
|
|
||
|
[fTrackerTable setTrackers: nil];
|
||
|
[fTrackerTable reloadData];
|
||
|
}
|
||
|
|
||
|
[fTrackerTable setTorrent: nil];
|
||
|
|
||
|
[fTrackerAddRemoveControl setEnabled: NO forSegment: TRACKER_ADD_TAG];
|
||
|
[fTrackerAddRemoveControl setEnabled: NO forSegment: TRACKER_REMOVE_TAG];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
[fTrackerTable setTorrent: [fTorrents objectAtIndex: 0]];
|
||
|
|
||
|
[fTrackerAddRemoveControl setEnabled: YES forSegment: TRACKER_ADD_TAG];
|
||
|
[fTrackerAddRemoveControl setEnabled: NO forSegment: TRACKER_REMOVE_TAG];
|
||
|
}
|
||
|
|
||
|
[fTrackerTable deselectAll: self];
|
||
|
}
|
||
|
|
||
|
#warning doesn't like blank addresses
|
||
|
- (void) addTrackers
|
||
|
{
|
||
|
[[[self view] window] makeKeyWindow];
|
||
|
|
||
|
NSAssert1([fTorrents count] == 1, @"Attempting to add tracker with %d transfers selected", [fTorrents count]);
|
||
|
|
||
|
[fTrackers addObject: [NSDictionary dictionaryWithObject: [NSNumber numberWithInteger: -1] forKey: @"Tier"]];
|
||
|
[fTrackers addObject: @""];
|
||
|
|
||
|
[fTrackerTable setTrackers: fTrackers];
|
||
|
[fTrackerTable reloadData];
|
||
|
[fTrackerTable selectRowIndexes: [NSIndexSet indexSetWithIndex: [fTrackers count]-1] byExtendingSelection: NO];
|
||
|
[fTrackerTable editColumn: [fTrackerTable columnWithIdentifier: @"Tracker"] row: [fTrackers count]-1 withEvent: nil select: YES];
|
||
|
}
|
||
|
|
||
|
- (void) removeTrackers
|
||
|
{
|
||
|
NSMutableDictionary * removeIdentifiers = [NSMutableDictionary dictionaryWithCapacity: [fTorrents count]];
|
||
|
NSUInteger removeCount = 0;
|
||
|
|
||
|
NSIndexSet * selectedIndexes = [fTrackerTable selectedRowIndexes];
|
||
|
BOOL groupSelected = NO;
|
||
|
for (NSUInteger i = 0; i < [fTrackers count]; ++i)
|
||
|
{
|
||
|
id object = [fTrackers objectAtIndex: i];
|
||
|
if ([object isKindOfClass: [TrackerNode class]])
|
||
|
{
|
||
|
if (groupSelected || [selectedIndexes containsIndex: i])
|
||
|
{
|
||
|
Torrent * torrent = [(TrackerNode *)object torrent];
|
||
|
NSMutableIndexSet * removeIndexSet;
|
||
|
if (!(removeIndexSet = [removeIdentifiers objectForKey: torrent]))
|
||
|
{
|
||
|
removeIndexSet = [NSMutableIndexSet indexSet];
|
||
|
[removeIdentifiers setObject: removeIndexSet forKey: torrent];
|
||
|
}
|
||
|
|
||
|
[removeIndexSet addIndex: [(TrackerNode *)object identifier]];
|
||
|
++removeCount;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
groupSelected = [selectedIndexes containsIndex: i];
|
||
|
if (!groupSelected && i > [selectedIndexes lastIndex])
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
NSAssert(removeCount > 0, @"Trying to remove no trackers.");
|
||
|
|
||
|
if ([[NSUserDefaults standardUserDefaults] boolForKey: @"WarningRemoveTrackers"])
|
||
|
{
|
||
|
NSAlert * alert = [[NSAlert alloc] init];
|
||
|
|
||
|
if (removeCount > 1)
|
||
|
{
|
||
|
[alert setMessageText: [NSString stringWithFormat: NSLocalizedString(@"Are you sure you want to remove %d trackers?",
|
||
|
"Remove trackers alert -> title"), removeCount]];
|
||
|
[alert setInformativeText: NSLocalizedString(@"Once removed, Transmission will no longer attempt to contact them."
|
||
|
" This cannot be undone.", "Remove trackers alert -> message")];
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
[alert setMessageText: NSLocalizedString(@"Are you sure you want to remove this tracker?", "Remove trackers alert -> title")];
|
||
|
[alert setInformativeText: NSLocalizedString(@"Once removed, Transmission will no longer attempt to contact it."
|
||
|
" This cannot be undone.", "Remove trackers alert -> message")];
|
||
|
}
|
||
|
|
||
|
[alert addButtonWithTitle: NSLocalizedString(@"Remove", "Remove trackers alert -> button")];
|
||
|
[alert addButtonWithTitle: NSLocalizedString(@"Cancel", "Remove trackers alert -> button")];
|
||
|
|
||
|
[alert setShowsSuppressionButton: YES];
|
||
|
|
||
|
NSInteger result = [alert runModal];
|
||
|
if ([[alert suppressionButton] state] == NSOnState)
|
||
|
[[NSUserDefaults standardUserDefaults] setBool: NO forKey: @"WarningRemoveTrackers"];
|
||
|
[alert release];
|
||
|
|
||
|
if (result != NSAlertFirstButtonReturn)
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
for (Torrent * torrent in removeIdentifiers)
|
||
|
[torrent removeTrackersWithIdentifiers: [removeIdentifiers objectForKey: torrent]];
|
||
|
|
||
|
//reset table with either new or old value
|
||
|
[fTrackers release];
|
||
|
fTrackers = [[NSMutableArray alloc] init];
|
||
|
for (Torrent * torrent in fTorrents)
|
||
|
[fTrackers addObjectsFromArray: [torrent allTrackerStats]];
|
||
|
|
||
|
[fTrackerTable setTrackers: fTrackers];
|
||
|
[fTrackerTable reloadData];
|
||
|
[fTrackerTable deselectAll: self];
|
||
|
|
||
|
[[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateUI" object: nil]; //incase sort by tracker
|
||
|
}
|
||
|
|
||
|
@end
|