/****************************************************************************** * $Id$ * * Copyright (c) 2005-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 "PrefsController.h" #import "StringAdditions.h" #import "UKKQueue.h" #define DOWNLOAD_FOLDER 0 #define DOWNLOAD_TORRENT 2 #define DOWNLOAD_ASK 3 #define UPDATE_DAILY 0 #define UPDATE_WEEKLY 1 #define UPDATE_NEVER 2 #define TOOLBAR_GENERAL @"TOOLBAR_GENERAL" #define TOOLBAR_TRANSFERS @"TOOLBAR_TRANSFERS" #define TOOLBAR_BANDWIDTH @"TOOLBAR_BANDWIDTH" #define TOOLBAR_NETWORK @"TOOLBAR_NETWORK" @interface PrefsController (Private) - (void) setPrefView: (id) sender; - (void) folderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info; - (void) incompleteFolderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info; - (void) importFolderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info; @end @implementation PrefsController - (id) initWithWindowNibName: (NSString *) name handle: (tr_handle_t *) handle { if ((self = [self initWithWindowNibName: name])) { fDefaults = [NSUserDefaults standardUserDefaults]; fHandle = handle; //checks for old version upload speed of -1 if ([fDefaults integerForKey: @"UploadLimit"] < 0) { [fDefaults setInteger: 20 forKey: @"UploadLimit"]; [fDefaults setBool: NO forKey: @"CheckUpload"]; } //set auto import if ([fDefaults boolForKey: @"AutoImport"]) [[UKKQueue sharedFileWatcher] addPath: [[fDefaults stringForKey: @"AutoImportDirectory"] stringByExpandingTildeInPath]]; //set bind port int bindPort = [fDefaults integerForKey: @"BindPort"]; tr_setBindPort(fHandle, bindPort); //set NAT tr_natTraversalEnable(fHandle, [fDefaults boolForKey: @"NatTraversal"]); //actually set bandwidth limits [self applySpeedSettings: nil]; //set play sound NSMutableArray * sounds = [NSMutableArray array]; NSEnumerator * soundEnumerator, * soundDirectoriesEnumerator = [[NSArray arrayWithObjects: @"System/Library/Sounds", [NSHomeDirectory() stringByAppendingPathComponent: @"Library/Sounds"], nil] objectEnumerator]; NSString * soundPath, * sound; //get list of all sounds and sort alphabetically while ((soundPath = [soundDirectoriesEnumerator nextObject])) if (soundEnumerator = [[NSFileManager defaultManager] enumeratorAtPath: soundPath]) while ((sound = [soundEnumerator nextObject])) { sound = [sound stringByDeletingPathExtension]; if ([NSSound soundNamed: sound]) [sounds addObject: sound]; } fSounds = [[sounds sortedArrayUsingSelector: @selector(caseInsensitiveCompare:)] retain]; } return self; } - (void) dealloc { if (fNatStatusTimer) [fNatStatusTimer invalidate]; [fSounds release]; [super dealloc]; } - (void) awakeFromNib { hasLoaded = YES; fToolbar = [[NSToolbar alloc] initWithIdentifier: @"Preferences Toolbar"]; [fToolbar setDelegate: self]; [fToolbar setAllowsUserCustomization: NO]; [[self window] setToolbar: fToolbar]; [fToolbar setDisplayMode: NSToolbarDisplayModeIconAndLabel]; [fToolbar setSizeMode: NSToolbarSizeModeRegular]; [fToolbar setSelectedItemIdentifier: TOOLBAR_GENERAL]; [self setPrefView: nil]; //set download folder NSString * downloadChoice = [fDefaults stringForKey: @"DownloadChoice"]; if ([downloadChoice isEqualToString: @"Constant"]) [fFolderPopUp selectItemAtIndex: DOWNLOAD_FOLDER]; else if ([downloadChoice isEqualToString: @"Torrent"]) [fFolderPopUp selectItemAtIndex: DOWNLOAD_TORRENT]; else [fFolderPopUp selectItemAtIndex: DOWNLOAD_ASK]; //set stop ratio [self updateRatioStopField]; //set limits [self updateLimitFields]; //set speed limit [fSpeedLimitUploadField setIntValue: [fDefaults integerForKey: @"SpeedLimitUploadLimit"]]; [fSpeedLimitDownloadField setIntValue: [fDefaults integerForKey: @"SpeedLimitDownloadLimit"]]; //set port [fPortField setIntValue: [fDefaults integerForKey: @"BindPort"]]; [self updatePortStatus]; fNatStatus = -1; [self updateNatStatus]; fNatStatusTimer = [NSTimer scheduledTimerWithTimeInterval: 5.0 target: self selector: @selector(updateNatStatus) userInfo: nil repeats: YES]; //set queue values [fQueueDownloadField setIntValue: [fDefaults integerForKey: @"QueueDownloadNumber"]]; [fQueueSeedField setIntValue: [fDefaults integerForKey: @"QueueSeedNumber"]]; //set update check NSString * updateCheck = [fDefaults stringForKey: @"UpdateCheck"]; if ([updateCheck isEqualToString: @"Weekly"]) [fUpdatePopUp selectItemAtIndex: UPDATE_WEEKLY]; else if ([updateCheck isEqualToString: @"Never"]) [fUpdatePopUp selectItemAtIndex: UPDATE_NEVER]; else [fUpdatePopUp selectItemAtIndex: UPDATE_DAILY]; } - (void) setUpdater: (SUUpdater *) updater { fUpdater = updater; } - (NSToolbarItem *) toolbar: (NSToolbar *) toolbar itemForItemIdentifier: (NSString *) ident willBeInsertedIntoToolbar: (BOOL) flag { NSToolbarItem * item; item = [[NSToolbarItem alloc] initWithItemIdentifier: ident]; if ([ident isEqualToString: TOOLBAR_GENERAL]) { [item setLabel: NSLocalizedString(@"General", "Preferences -> General toolbar item title")]; [item setImage: [NSImage imageNamed: @"Preferences.png"]]; [item setTarget: self]; [item setAction: @selector(setPrefView:)]; [item setAutovalidates: NO]; } else if ([ident isEqualToString: TOOLBAR_TRANSFERS]) { [item setLabel: NSLocalizedString(@"Transfers", "Preferences -> Transfers toolbar item title")]; [item setImage: [NSImage imageNamed: @"Transfers.png"]]; [item setTarget: self]; [item setAction: @selector(setPrefView:)]; [item setAutovalidates: NO]; } else if ([ident isEqualToString: TOOLBAR_BANDWIDTH]) { [item setLabel: NSLocalizedString(@"Bandwidth", "Preferences -> Bandwidth toolbar item title")]; [item setImage: [NSImage imageNamed: @"Bandwidth.png"]]; [item setTarget: self]; [item setAction: @selector(setPrefView:)]; [item setAutovalidates: NO]; } else if ([ident isEqualToString: TOOLBAR_NETWORK]) { [item setLabel: NSLocalizedString(@"Network", "Preferences -> Network toolbar item title")]; [item setImage: [NSImage imageNamed: @"Network.png"]]; [item setTarget: self]; [item setAction: @selector(setPrefView:)]; [item setAutovalidates: NO]; } else { [item release]; return nil; } return item; } - (NSArray *) toolbarSelectableItemIdentifiers: (NSToolbar *) toolbar { return [self toolbarDefaultItemIdentifiers: toolbar]; } - (NSArray *) toolbarDefaultItemIdentifiers: (NSToolbar *) toolbar { return [self toolbarAllowedItemIdentifiers: toolbar]; } - (NSArray *) toolbarAllowedItemIdentifiers: (NSToolbar *) toolbar { return [NSArray arrayWithObjects: TOOLBAR_GENERAL, TOOLBAR_TRANSFERS, TOOLBAR_BANDWIDTH, TOOLBAR_NETWORK, nil]; } - (void) setPort: (id) sender { int port = [sender intValue]; if (![[sender stringValue] isEqualToString: [NSString stringWithFormat: @"%d", port]]) { NSBeep(); [sender setIntValue: [fDefaults integerForKey: @"BindPort"]]; return; } [fDefaults setInteger: port forKey: @"BindPort"]; tr_setBindPort(fHandle, [fDefaults integerForKey: @"BindPort"]); [self updateNatStatus]; [self updatePortStatus]; } - (void) updatePortStatus { PortChecker * portChecker = [[PortChecker alloc] initWithDelegate: self]; [fPortStatusField setStringValue: [NSLocalizedString(@"Checking port status", "Preferences -> Network -> port status") stringByAppendingEllipsis]]; [fPortStatusImage setImage: nil]; [fPortStatusProgress startAnimation: self]; tr_handle_status_t * stat = tr_handleStatus(fHandle); [portChecker probePort: stat->publicPort]; } - (void) portCheckerDidFinishProbing: (PortChecker *) portChecker { [fPortStatusProgress stopAnimation: self]; switch ([portChecker status]) { case PORT_STATUS_OPEN: [fPortStatusField setStringValue: NSLocalizedString(@"Port is open", "Preferences -> Network -> port status")]; [fPortStatusImage setImage: [NSImage imageNamed: @"GreenDot.tiff"]]; break; case PORT_STATUS_STEALTH: [fPortStatusField setStringValue: NSLocalizedString(@"Port is stealth", "Preferences -> Network -> port status")]; [fPortStatusImage setImage: [NSImage imageNamed: @"RedDot.tiff"]]; break; case PORT_STATUS_CLOSED: [fPortStatusField setStringValue: NSLocalizedString(@"Port is closed", "Preferences -> Network -> port status")]; [fPortStatusImage setImage: [NSImage imageNamed: @"RedDot.tiff"]]; break; case PORT_STATUS_ERROR: [fPortStatusField setStringValue: NSLocalizedString(@"Unable to check port status", "Preferences -> Network -> port status")]; [fPortStatusImage setImage: [NSImage imageNamed: @"YellowDot.tiff"]]; break; } [portChecker release]; } - (void) setNat: (id) sender { tr_natTraversalEnable(fHandle, [fDefaults boolForKey: @"NatTraversal"]); [self updateNatStatus]; } - (void) updateNatStatus { tr_handle_status_t * stat = tr_handleStatus(fHandle); if (fNatStatus == stat->natTraversalStatus) return; fNatStatus = stat->natTraversalStatus; if (fNatStatus == TR_NAT_TRAVERSAL_MAPPED) { [fNatStatusField setStringValue: NSLocalizedString(@"Port successfully mapped", "Preferences -> Network -> port map status")]; [fNatStatusImage setImage: [NSImage imageNamed: @"GreenDot.tiff"]]; } else if (fNatStatus == TR_NAT_TRAVERSAL_NOTFOUND || fNatStatus == TR_NAT_TRAVERSAL_ERROR) { [fNatStatusField setStringValue: NSLocalizedString(@"Error mapping port", "Preferences -> Network -> port map status")]; [fNatStatusImage setImage: [NSImage imageNamed: @"RedDot.tiff"]]; } else { [fNatStatusField setStringValue: @""]; [fNatStatusImage setImage: nil]; } [self updatePortStatus]; } - (void) applySpeedSettings: (id) sender { if ([fDefaults boolForKey: @"SpeedLimit"]) { tr_setGlobalUploadLimit(fHandle, [fDefaults integerForKey: @"SpeedLimitUploadLimit"]); tr_setGlobalDownloadLimit(fHandle, [fDefaults integerForKey: @"SpeedLimitDownloadLimit"]); } else { tr_setGlobalUploadLimit(fHandle, [fDefaults boolForKey: @"CheckUpload"] ? [fDefaults integerForKey: @"UploadLimit"] : -1); tr_setGlobalDownloadLimit(fHandle, [fDefaults boolForKey: @"CheckDownload"] ? [fDefaults integerForKey: @"DownloadLimit"] : -1); } } - (void) updateRatioStopField { if (!hasLoaded) return; [fRatioStopField setFloatValue: [fDefaults floatForKey: @"RatioLimit"]]; } - (void) setRatioStop: (id) sender { float ratio = [sender floatValue]; if (![[sender stringValue] isEqualToString: [NSString stringWithFormat: @"%.2f", ratio]] || ratio < 0) { NSBeep(); [sender setFloatValue: [fDefaults floatForKey: @"RatioLimit"]]; return; } [fDefaults setFloat: ratio forKey: @"RatioLimit"]; } - (void) updateLimitFields { if (!hasLoaded) return; [fUploadField setIntValue: [fDefaults integerForKey: @"UploadLimit"]]; [fDownloadField setIntValue: [fDefaults integerForKey: @"DownloadLimit"]]; } - (void) setGlobalLimit: (id) sender { BOOL upload = sender == fUploadField; int limit = [sender intValue]; if (![[sender stringValue] isEqualToString: [NSString stringWithFormat: @"%d", limit]] || limit < 0) { NSBeep(); [sender setIntValue: [fDefaults integerForKey: upload ? @"UploadLimit" : @"DownloadLimit"]]; return; } [fDefaults setInteger: limit forKey: upload ? @"UploadLimit" : @"DownloadLimit"]; [self applySpeedSettings: self]; } - (void) setSpeedLimit: (id) sender { BOOL upload = sender == fSpeedLimitUploadField; int limit = [sender intValue]; if (![[sender stringValue] isEqualToString: [NSString stringWithFormat: @"%d", limit]]) { NSBeep(); [sender setIntValue: [fDefaults integerForKey: upload ? @"SpeedLimitUploadLimit" : @"SpeedLimitDownloadLimit"]]; return; } [fDefaults setInteger: limit forKey: upload ? @"SpeedLimitUploadLimit" : @"SpeedLimitDownloadLimit"]; [self applySpeedSettings: self]; } - (void) setAutoSpeedLimit: (id) sender { [[NSNotificationCenter defaultCenter] postNotificationName: @"AutoSpeedLimitChange" object: self]; } - (void) setBadge: (id) sender { [[NSNotificationCenter defaultCenter] postNotificationName: @"DockBadgeChange" object: self]; } - (void) setSound: (id) sender { //play sound when selecting NSString * soundName = [sender titleOfSelectedItem]; NSSound * sound; if ((sound = [NSSound soundNamed: soundName])) [sound play]; } - (void) setUpdate: (id) sender { int index = [fUpdatePopUp indexOfSelectedItem]; NSTimeInterval seconds; if (index == UPDATE_DAILY) { [fDefaults setObject: @"Daily" forKey: @"UpdateCheck"]; seconds = 86400; } else if (index == UPDATE_WEEKLY) { [fDefaults setObject: @"Weekly" forKey: @"UpdateCheck"]; seconds = 604800; } else { [fDefaults setObject: @"Never" forKey: @"UpdateCheck"]; seconds = 0; } [fDefaults setInteger: seconds forKey: @"SUScheduledCheckInterval"]; if (fUpdater) [fUpdater scheduleCheckWithInterval: seconds]; } - (void) setQueue: (id) sender { [[NSNotificationCenter defaultCenter] postNotificationName: @"UpdateQueue" object: self]; } - (void) setQueueNumber: (id) sender { BOOL download = sender == fQueueDownloadField; int limit = [sender intValue]; if (![[sender stringValue] isEqualToString: [NSString stringWithFormat: @"%d", limit]] || limit < 0) { NSBeep(); [sender setIntValue: [fDefaults integerForKey: download ? @"QueueDownloadNumber" : @"QueueSeedNumber"]]; return; } [fDefaults setInteger: limit forKey: download ? @"QueueDownloadNumber" : @"QueueSeedNumber"]; [self setQueue: nil]; } - (void) setDownloadLocation: (id) sender { switch ([fFolderPopUp indexOfSelectedItem]) { case DOWNLOAD_FOLDER: [fDefaults setObject: @"Constant" forKey: @"DownloadChoice"]; break; case DOWNLOAD_TORRENT: [fDefaults setObject: @"Torrent" forKey: @"DownloadChoice"]; break; case DOWNLOAD_ASK: [fDefaults setObject: @"Ask" forKey: @"DownloadChoice"]; break; } } - (void) folderSheetShow: (id) sender { NSOpenPanel * panel = [NSOpenPanel openPanel]; [panel setPrompt: @"Select"]; [panel setAllowsMultipleSelection: NO]; [panel setCanChooseFiles: NO]; [panel setCanChooseDirectories: YES]; [panel setCanCreateDirectories: YES]; [panel beginSheetForDirectory: nil file: nil types: nil modalForWindow: [self window] modalDelegate: self didEndSelector: @selector(folderSheetClosed:returnCode:contextInfo:) contextInfo: nil]; } - (void) incompleteFolderSheetShow: (id) sender { NSOpenPanel * panel = [NSOpenPanel openPanel]; [panel setPrompt: @"Select"]; [panel setAllowsMultipleSelection: NO]; [panel setCanChooseFiles: NO]; [panel setCanChooseDirectories: YES]; [panel setCanCreateDirectories: YES]; [panel beginSheetForDirectory: nil file: nil types: nil modalForWindow: [self window] modalDelegate: self didEndSelector: @selector(incompleteFolderSheetClosed:returnCode:contextInfo:) contextInfo: nil]; } - (void) setAutoImport: (id) sender { NSString * path = [[fDefaults stringForKey: @"AutoImportDirectory"] stringByExpandingTildeInPath]; if ([fDefaults boolForKey: @"AutoImport"]) [[UKKQueue sharedFileWatcher] addPath: path]; else [[UKKQueue sharedFileWatcher] removePathFromQueue: path]; [[NSNotificationCenter defaultCenter] postNotificationName: @"AutoImportSettingChange" object: self]; } - (void) importFolderSheetShow: (id) sender { NSOpenPanel * panel = [NSOpenPanel openPanel]; [panel setPrompt: @"Select"]; [panel setAllowsMultipleSelection: NO]; [panel setCanChooseFiles: NO]; [panel setCanChooseDirectories: YES]; [panel setCanCreateDirectories: YES]; [panel beginSheetForDirectory: nil file: nil types: nil modalForWindow: [self window] modalDelegate: self didEndSelector: @selector(importFolderSheetClosed:returnCode:contextInfo:) contextInfo: nil]; } - (void) setAutoSize: (id) sender { [[NSNotificationCenter defaultCenter] postNotificationName: @"AutoSizeSettingChange" object: self]; } - (void) helpForNetwork: (id) sender { [[NSHelpManager sharedHelpManager] openHelpAnchor: @"PortForwarding" inBook: [[NSBundle mainBundle] objectForInfoDictionaryKey: @"CFBundleHelpBookName"]]; } @end @implementation PrefsController (Private) - (void) setPrefView: (id) sender { NSView * view = fGeneralView; if (sender) { NSString * identifier = [sender itemIdentifier]; if ([identifier isEqualToString: TOOLBAR_TRANSFERS]) view = fTransfersView; else if ([identifier isEqualToString: TOOLBAR_BANDWIDTH]) view = fBandwidthView; else if ([identifier isEqualToString: TOOLBAR_NETWORK]) view = fNetworkView; else; } NSWindow * window = [self window]; if ([window contentView] == view) return; NSRect windowRect = [window frame]; int difference = [view frame].size.height - [[window contentView] frame].size.height; windowRect.origin.y -= difference; windowRect.size.height += difference; [view setHidden: YES]; [window setContentView: view]; [window setFrame: windowRect display: YES animate: YES]; [view setHidden: NO]; //set title label if (sender) [window setTitle: [sender label]]; else { NSToolbarItem * item; NSEnumerator * enumerator = [[fToolbar items] objectEnumerator]; while ((item = [enumerator nextObject])) if ([[item itemIdentifier] isEqualToString: [fToolbar selectedItemIdentifier]]) { [window setTitle: [item label]]; break; } } //for network view make sure progress indicator hides itself if (view == fNetworkView && [fPortStatusImage image]) [fPortStatusProgress setDisplayedWhenStopped: NO]; } - (void) folderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info { if (code == NSOKButton) { [fFolderPopUp selectItemAtIndex: DOWNLOAD_FOLDER]; [fDefaults setObject: [[openPanel filenames] objectAtIndex: 0] forKey: @"DownloadFolder"]; [fDefaults setObject: @"Constant" forKey: @"DownloadChoice"]; } else { //reset if cancelled NSString * downloadChoice = [fDefaults stringForKey: @"DownloadChoice"]; if ([downloadChoice isEqualToString: @"Constant"]) [fFolderPopUp selectItemAtIndex: DOWNLOAD_FOLDER]; else if ([downloadChoice isEqualToString: @"Torrent"]) [fFolderPopUp selectItemAtIndex: DOWNLOAD_TORRENT]; else [fFolderPopUp selectItemAtIndex: DOWNLOAD_ASK]; } } - (void) incompleteFolderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info { if (code == NSOKButton) [fDefaults setObject: [[openPanel filenames] objectAtIndex: 0] forKey: @"IncompleteDownloadFolder"]; [fIncompleteFolderPopUp selectItemAtIndex: 0]; } - (void) importFolderSheetClosed: (NSOpenPanel *) openPanel returnCode: (int) code contextInfo: (void *) info { if (code == NSOKButton) { UKKQueue * sharedQueue = [UKKQueue sharedFileWatcher]; [sharedQueue removePathFromQueue: [[fDefaults stringForKey: @"AutoImportDirectory"] stringByExpandingTildeInPath]]; [fDefaults setObject: [[openPanel filenames] objectAtIndex: 0] forKey: @"AutoImportDirectory"]; [sharedQueue addPath: [[fDefaults stringForKey: @"AutoImportDirectory"] stringByExpandingTildeInPath]]; [[NSNotificationCenter defaultCenter] postNotificationName: @"AutoImportSettingChange" object: self]; } [fImportFolderPopUp selectItemAtIndex: 0]; } @end