release/src/router/transmission/macosx/AddWindowController.m

   1 /******************************************************************************
   2  * $Id: AddWindowController.m 13251 2012-03-13 02:52:11Z livings124 $
   3  *
   4  * Copyright (c) 2008-2012 Transmission authors and contributors
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22  * DEALINGS IN THE SOFTWARE.
  23  *****************************************************************************/
  24
  25 #import "AddWindowController.h"
  26 #import "Controller.h"
  27 #import "ExpandedPathToIconTransformer.h"
  28 #import "FileOutlineController.h"
  29 #import "GroupsController.h"
  30 #import "NSStringAdditions.h"
  31 #import "Torrent.h"
  32
  33 #define UPDATE_SECONDS 1.0
  34
  35 #define POPUP_PRIORITY_HIGH 0
  36 #define POPUP_PRIORITY_NORMAL 1
  37 #define POPUP_PRIORITY_LOW 2
  38
  39 @interface AddWindowController (Private)
  40
  41 - (void) updateFiles;
  42
  43 - (void) confirmAdd;
  44
  45 - (void) setDestinationPath: (NSString *) destination;
  46
  47 - (void) setGroupsMenu;
  48 - (void) changeGroupValue: (id) sender;
  49
  50 - (void) sameNameAlertDidEnd: (NSAlert *) alert returnCode: (NSInteger) returnCode contextInfo: (void *) contextInfo;
  51
  52 @end
  53
  54 @implementation AddWindowController
  55
  56 - (id) initWithTorrent: (Torrent *) torrent destination: (NSString *) path lockDestination: (BOOL) lockDestination
  57     controller: (Controller *) controller torrentFile: (NSString *) torrentFile
  58     deleteTorrent: (BOOL) deleteTorrent canToggleDelete: (BOOL) canToggleDelete
  59 {
  60     if ((self = [super initWithWindowNibName: @"AddWindow"]))
  61     {
  62         fTorrent = torrent;
  63         fDestination = [[path stringByExpandingTildeInPath] retain];
  64         fLockDestination = lockDestination;
  65
  66         fController = controller;
  67
  68         fTorrentFile = [[torrentFile stringByExpandingTildeInPath] retain];
  69
  70         fDeleteTorrentInitial = deleteTorrent;
  71         fDeleteEnableInitial = canToggleDelete;
  72
  73         fGroupValue = [torrent groupValue];
  74
  75         [fVerifyIndicator setUsesThreadedAnimation: YES];
  76     }
  77     return self;
  78 }
  79
  80 - (void) awakeFromNib
  81 {
  82     [[NSNotificationCenter defaultCenter] addObserver: self selector: @selector(updateStatusField:)
  83         name: @"TorrentFileCheckChange" object: fTorrent];
  84
  85     [[NSNotificationCenter defaultCenter] addObserver: self selector: @selector(updateGroupMenu:)
  86         name: @"UpdateGroups" object: nil];
  87
  88     [fFileController setTorrent: fTorrent];
  89
  90     NSString * name = [fTorrent name];
  91     [[self window] setTitle: name];
  92     [fNameField setStringValue: name];
  93     [fNameField setToolTip: name];
  94
  95     [fIconView setImage: [fTorrent icon]];
  96
  97     [self updateStatusField: nil];
  98
  99     [self setGroupsMenu];
 100     [fGroupPopUp selectItemWithTag: fGroupValue];
 101
 102     NSInteger priorityIndex;
 103     switch ([fTorrent priority])
 104     {
 105         case TR_PRI_HIGH: priorityIndex = POPUP_PRIORITY_HIGH; break;
 106         case TR_PRI_NORMAL: priorityIndex = POPUP_PRIORITY_NORMAL; break;
 107         case TR_PRI_LOW: priorityIndex = POPUP_PRIORITY_LOW; break;
 108         default: NSAssert1(NO, @"Unknown priority for adding torrent: %d", [fTorrent priority]);
 109     }
 110     [fPriorityPopUp selectItemAtIndex: priorityIndex];
 111
 112     [fStartCheck setState: [[NSUserDefaults standardUserDefaults] boolForKey: @"AutoStartDownload"] ? NSOnState : NSOffState];
 113
 114     [fDeleteCheck setState: fDeleteTorrentInitial ? NSOnState : NSOffState];
 115     [fDeleteCheck setEnabled: fDeleteEnableInitial];
 116
 117     if (fDestination)
 118         [self setDestinationPath: fDestination];
 119     else
 120     {
 121         [fLocationField setStringValue: @""];
 122         [fLocationImageView setImage: nil];
 123     }
 124
 125     fTimer = [NSTimer scheduledTimerWithTimeInterval: UPDATE_SECONDS target: self
 126                 selector: @selector(updateFiles) userInfo: nil repeats: YES];
 127     [self updateFiles];
 128 }
 129
 130 - (void) windowDidLoad
 131 {
 132     //if there is no destination, prompt for one right away
 133     if (!fDestination)
 134         [self setDestination: nil];
 135 }
 136
 137 - (void) dealloc
 138 {
 139     [[NSNotificationCenter defaultCenter] removeObserver: self];
 140
 141     [fTimer invalidate];
 142
 143     [fDestination release];
 144     [fTorrentFile release];
 145
 146     [super dealloc];
 147 }
 148
 149 - (Torrent *) torrent
 150 {
 151     return fTorrent;
 152 }
 153
 154 - (void) setDestination: (id) sender
 155 {
 156     NSOpenPanel * panel = [NSOpenPanel openPanel];
 157
 158     [panel setPrompt: NSLocalizedString(@"Select", "Open torrent -> prompt")];
 159     [panel setAllowsMultipleSelection: NO];
 160     [panel setCanChooseFiles: NO];
 161     [panel setCanChooseDirectories: YES];
 162     [panel setCanCreateDirectories: YES];
 163
 164     [panel setMessage: [NSString stringWithFormat: NSLocalizedString(@"Select the download folder for \"%@\"",
 165                         "Add -> select destination folder"), [fTorrent name]]];
 166
 167     [panel beginSheetModalForWindow: [self window] completionHandler: ^(NSInteger result) {
 168         if (result == NSFileHandlingPanelOKButton)
 169         {
 170             fLockDestination = NO;
 171             [self setDestinationPath: [[[panel URLs] objectAtIndex: 0] path]];
 172         }
 173         else
 174         {
 175             if (!fDestination)
 176                 [self performSelectorOnMainThread: @selector(cancelAdd:) withObject: nil waitUntilDone: NO];
 177         }
 178     }];
 179 }
 180
 181 - (void) add: (id) sender
 182 {
 183     if ([[fDestination lastPathComponent] isEqualToString: [fTorrent name]]
 184         && [[NSUserDefaults standardUserDefaults] boolForKey: @"WarningFolderDataSameName"])
 185     {
 186         NSAlert * alert = [[NSAlert alloc] init];
 187         [alert setMessageText: NSLocalizedString(@"The destination directory and root data directory have the same name.",
 188                                 "Add torrent -> same name -> title")];
 189         [alert setInformativeText: NSLocalizedString(@"If you are attempting to use already existing data,"
 190             " the root data directory should be inside the destination directory.", "Add torrent -> same name -> message")];
 191         [alert setAlertStyle: NSWarningAlertStyle];
 192         [alert addButtonWithTitle: NSLocalizedString(@"Cancel", "Add torrent -> same name -> button")];
 193         [alert addButtonWithTitle: NSLocalizedString(@"Add", "Add torrent -> same name -> button")];
 194         [alert setShowsSuppressionButton: YES];
 195
 196         [alert beginSheetModalForWindow: [self window] modalDelegate: self
 197             didEndSelector: @selector(sameNameAlertDidEnd:returnCode:contextInfo:) contextInfo: nil];
 198     }
 199     else
 200         [self confirmAdd];
 201 }
 202
 203 - (void) cancelAdd: (id) sender
 204 {
 205     [[self window] performClose: sender];
 206 }
 207
 208 //only called on cancel
 209 - (BOOL) windowShouldClose: (id) window
 210 {
 211     [fTimer invalidate];
 212     fTimer = nil;
 213
 214     [fFileController setTorrent: nil]; //avoid a crash when window tries to update
 215
 216     [fController askOpenConfirmed: self add: NO];
 217     return YES;
 218 }
 219
 220 - (void) verifyLocalData: (id) sender
 221 {
 222     [fTorrent resetCache];
 223     [self updateFiles];
 224 }
 225
 226 - (void) changePriority: (id) sender
 227 {
 228     tr_priority_t priority;
 229     switch ([sender indexOfSelectedItem])
 230     {
 231         case POPUP_PRIORITY_HIGH: priority = TR_PRI_HIGH; break;
 232         case POPUP_PRIORITY_NORMAL: priority = TR_PRI_NORMAL; break;
 233         case POPUP_PRIORITY_LOW: priority = TR_PRI_LOW; break;
 234         default: NSAssert1(NO, @"Unknown priority tag for adding torrent: %d", [sender tag]);
 235     }
 236     [fTorrent setPriority: priority];
 237 }
 238
 239 - (void) updateStatusField: (NSNotification *) notification
 240 {
 241     NSString * statusString = [NSString stringForFileSize: [fTorrent size]];
 242     if ([fTorrent isFolder])
 243     {
 244         NSString * fileString;
 245         NSInteger count = [fTorrent fileCount];
 246         if (count != 1)
 247             fileString = [NSString stringWithFormat: NSLocalizedString(@"%@ files", "Add torrent -> info"),
 248                             [NSString formattedUInteger: count]];
 249         else
 250             fileString = NSLocalizedString(@"1 file", "Add torrent -> info");
 251
 252         NSString * selectedString = [NSString stringWithFormat: NSLocalizedString(@"%@ selected", "Add torrent -> info"),
 253                                         [NSString stringForFileSize: [fTorrent totalSizeSelected]]];
 254
 255         statusString = [NSString stringWithFormat: @"%@, %@ (%@)", fileString, statusString, selectedString];
 256     }
 257
 258     [fStatusField setStringValue: statusString];
 259 }
 260
 261 - (void) updateGroupMenu: (NSNotification *) notification
 262 {
 263     [self setGroupsMenu];
 264     if (![fGroupPopUp selectItemWithTag: fGroupValue])
 265     {
 266         fGroupValue = -1;
 267         [fGroupPopUp selectItemWithTag: fGroupValue];
 268     }
 269 }
 270
 271 @end
 272
 273 @implementation AddWindowController (Private)
 274
 275 - (void) updateFiles
 276 {
 277     [fTorrent update];
 278
 279     [fFileController refresh];
 280
 281     if ([fTorrent isChecking])
 282     {
 283         const BOOL waiting = [fTorrent isCheckingWaiting];
 284         [fVerifyIndicator setIndeterminate: waiting];
 285         if (waiting)
 286             [fVerifyIndicator startAnimation: self];
 287         else
 288             [fVerifyIndicator setDoubleValue: [fTorrent checkingProgress]];
 289     }
 290     else {
 291         [fVerifyIndicator setIndeterminate: YES]; //we want to hide when stopped, which only applies when indeterminate
 292         [fVerifyIndicator stopAnimation: self];
 293     }
 294 }
 295
 296 - (void) confirmAdd
 297 {
 298     [fTimer invalidate];
 299     fTimer = nil;
 300     [fTorrent setGroupValue: fGroupValue];
 301
 302     if (fTorrentFile && [fDeleteCheck state] == NSOnState)
 303         [Torrent trashFile: fTorrentFile];
 304
 305     if ([fStartCheck state] == NSOnState)
 306         [fTorrent startTransfer];
 307
 308     [fFileController setTorrent: nil]; //avoid a crash when window tries to update
 309
 310     [self close];
 311     [fController askOpenConfirmed: self add: YES]; //ensure last, since it releases this controller
 312 }
 313
 314 - (void) setDestinationPath: (NSString *) destination
 315 {
 316     destination = [destination stringByExpandingTildeInPath];
 317     if (!fDestination || ![fDestination isEqualToString: destination])
 318     {
 319         [fDestination release];
 320         fDestination = [destination retain];
 321
 322         [fTorrent changeDownloadFolderBeforeUsing: fDestination];
 323     }
 324
 325     [fLocationField setStringValue: [fDestination stringByAbbreviatingWithTildeInPath]];
 326     [fLocationField setToolTip: fDestination];
 327
 328     ExpandedPathToIconTransformer * iconTransformer = [[ExpandedPathToIconTransformer alloc] init];
 329     [fLocationImageView setImage: [iconTransformer transformedValue: fDestination]];
 330     [iconTransformer release];
 331 }
 332
 333 - (void) setGroupsMenu
 334 {
 335     NSMenu * groupMenu = [[GroupsController groups] groupMenuWithTarget: self action: @selector(changeGroupValue:) isSmall: NO];
 336     [fGroupPopUp setMenu: groupMenu];
 337 }
 338
 339 - (void) changeGroupValue: (id) sender
 340 {
 341     NSInteger previousGroup = fGroupValue;
 342     fGroupValue = [sender tag];
 343
 344     if (!fLockDestination)
 345     {
 346         if ([[GroupsController groups] usesCustomDownloadLocationForIndex: fGroupValue])
 347             [self setDestinationPath: [[GroupsController groups] customDownloadLocationForIndex: fGroupValue]];
 348         else if ([fDestination isEqualToString: [[GroupsController groups] customDownloadLocationForIndex: previousGroup]])
 349             [self setDestinationPath: [[NSUserDefaults standardUserDefaults] stringForKey: @"DownloadFolder"]];
 350         else;
 351     }
 352 }
 353
 354 - (void) sameNameAlertDidEnd: (NSAlert *) alert returnCode: (NSInteger) returnCode contextInfo: (void *) contextInfo
 355 {
 356     if ([[alert suppressionButton] state] == NSOnState)
 357         [[NSUserDefaults standardUserDefaults] setBool: NO forKey: @"WarningFolderDataSameName"];
 358
 359     [alert release];
 360
 361     if (returnCode == NSAlertSecondButtonReturn)
 362         [self performSelectorOnMainThread: @selector(confirmAdd) withObject: nil waitUntilDone: NO];
 363 }
 364
 365 @end