MVPreferencesController.m

//
//  NAPreferencesController.m
//  FileShuttle
//
//  Created by Michael Villar on 8/15/11.
//

#import "MVPreferencesController.h"
#import "EMKeychainItem.h"
#import "MVDictionaryKeyCombo.h"

#define kMVGeneralIdentifier @"general"
#define kMVAdvancedIdentifier @"advanced"

#define kMVTopMargin 78

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
@interface MVPreferencesController ()

@property (strong, readwrite) NSString *selectedIdentifier;
@property (strong, readwrite) NSView *selectedView;
@property (assign, nonatomic) BOOL showDockIcon;
@property (assign, nonatomic) BOOL use_filename;
@property (assign, nonatomic) BOOL use_hash;
@property (retain) NSTimer *passwordTimer;

- (void)updateLaunchAtLoginFromValue;
- (void)savePassword;
- (void)setShowDockIconValue:(BOOL)show;
- (void)setFront;
- (void)layoutView:(BOOL)animated;

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark Launch At Login helper Methods
- (void)enableLoginItemWithLoginItemsReference:(LSSharedFileListRef )theLoginItemsRefs
                                       ForPath:(NSString *)appPath;
- (void)disableLoginItemWithLoginItemsReference:(LSSharedFileListRef )theLoginItemsRefs
                                        ForPath:(NSString *)appPath;
- (BOOL)loginItemExistsWithLoginItemReference:(LSSharedFileListRef)theLoginItemsRefs
                                      ForPath:(NSString *)appPath;

@end

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
@implementation MVPreferencesController

@synthesize passwordTextField           = passwordTextField_,
            window                      = window_,
            toolbar                     = toolbar_,
            generalView                 = generalView_,
            advancedView                = advancedView_,
            showInPopUpButton           = showInPopUpButton_,
            setFilenamePopUpButton      = setFilenamePopUpButton_,
            clipboardRecorderControl	= clipboardRecorderControl_,
            selectedIdentifier          = selectedIdentifier_,
            selectedView                = selectedView_,
            showDockIcon                = showDockIcon_,
            passwordTimer               = passwordTimer_;

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)awakeFromNib
{
	[[NSNotificationCenter defaultCenter] addObserver:self
                                           selector:@selector(controlTextDidChange:)
                                               name:NSControlTextDidChangeNotification
                                             object:self.passwordTextField];
	
	NSString *password = @"";
	EMGenericKeychainItem *keychainItem = [EMGenericKeychainItem
                                         genericKeychainItemForService:@"FileShuttle"
                                         withUsername:@""];
	if(keychainItem != nil) {
		password = [keychainItem password];
	}
	[self.passwordTextField setStringValue:password];
	
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	[defaults addObserver:self
             forKeyPath:@"dock_icon"
                options:NSKeyValueObservingOptionNew
                context:nil];
	[defaults addObserver:self
             forKeyPath:@"launch_at_login"
                options:NSKeyValueObservingOptionNew
                context:nil];
	self.showDockIcon = ![[[[NSBundle mainBundle] infoDictionary] objectForKey:@"LSUIElement"]
                        boolValue];
	[self setShowDockIconValue:[defaults boolForKey:@"dock_icon"]];
	[self updateLaunchAtLoginFromValue];
	
	if([defaults boolForKey:@"dock_icon"]) {
		if([defaults boolForKey:@"menubar_icon"]) {
			[self.showInPopUpButton selectItemWithTag:2];
		}
		else {
			[self.showInPopUpButton selectItemWithTag:1];
		}
	}
	else
		[self.showInPopUpButton selectItemWithTag:0];
    
    if([defaults boolForKey:@"use_hash"]) {
		if([defaults boolForKey:@"use_filename"]) {
			[self.setFilenamePopUpButton selectItemWithTag:2];
		}
		else {
			[self.setFilenamePopUpButton selectItemWithTag:1];
		}
	}
	else
		[self.setFilenamePopUpButton selectItemWithTag:0];

	NSString *protocol = [defaults valueForKey:@"protocol"];
	if([protocol compare:@"SFTP"] == NSOrderedSame) {
		BOOL changePermissions = [defaults boolForKey:@"change_permissions"];
		self.changePermissionsCheckbox.enabled = changePermissions;
		self.changePermissionsCheckbox.state = changePermissions ? NSOnState : NSOffState;
		self.permissionsTextField.enabled = YES;
	} else {
		self.changePermissionsCheckbox.enabled = NO;
		self.changePermissionsCheckbox.state = NSOffState;
		self.permissionsTextField.enabled = NO;
	}
	
	self.clipboardRecorderControl.style = SRGreyStyle;
	self.clipboardRecorderControl.animates = YES;
	self.clipboardRecorderControl.canCaptureGlobalHotKeys = YES;
	self.clipboardRecorderControl.delegate = self;
	NSDictionary *dic = [defaults dictionaryForKey:@"clipboard_upload_shortcut"];
	if(dic) {
		self.clipboardRecorderControl.keyCombo = MVKeyComboFromDictionary(dic);
	}
    
    [self.toolbar setSelectedItemIdentifier:kMVGeneralIdentifier];
    self.selectedIdentifier = kMVGeneralIdentifier;
    self.selectedView = nil;
    [self layoutView:NO];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)observeValueForKeyPath:(NSString *)keyPath
                      ofObject:(id)object
                        change:(NSDictionary *)change
                       context:(void *)context
{
	if([keyPath isEqualToString:@"dock_icon"])
	{
		NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
		[self setShowDockIconValue:[defaults boolForKey:@"dock_icon"]];
	}
	else if([keyPath isEqualToString:@"launch_at_login"])
	{
		[self updateLaunchAtLoginFromValue];
	}
}

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark IBActions methods

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)toolbarItemAction:(id)sender {
  NSString *selectedIdentifier = self.toolbar.selectedItemIdentifier;
  if([self.selectedIdentifier isEqualToString:selectedIdentifier])
    return;
  self.selectedIdentifier = selectedIdentifier;
  [self layoutView:YES];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)protocolChanged:(id)sender {
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	NSString *protocol = [defaults valueForKey:@"protocol"];
	
	NSString *port = @"21";
  if([protocol compare:@"SFTP"] == NSOrderedSame) {
		port = @"22";
		self.changePermissionsCheckbox.state = [defaults boolForKey:@"change_permissions"];
		self.permissionsTextField.enabled = [defaults boolForKey:@"change_permissions"];
    self.changePermissionsCheckbox.enabled = YES;
	} else {
		self.changePermissionsCheckbox.enabled = NO;
		self.changePermissionsCheckbox.state = NSOffState;
		self.permissionsTextField.enabled = NO;
	}
	[[NSUserDefaults standardUserDefaults] setValue:port forKey:@"port"];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)passwordChanged:(id)sender {
	[self savePassword];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)showInPopUpButtonChanged:(id)sender {
	NSInteger tag = [self.showInPopUpButton selectedTag];
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	if(tag == 0) {
		[defaults setBool:YES forKey:@"menubar_icon"];
		[defaults setBool:NO forKey:@"dock_icon"];
	}
	else if(tag == 1) {
		[defaults setBool:NO forKey:@"menubar_icon"];
		[defaults setBool:YES forKey:@"dock_icon"];
	}
	else if(tag == 2) {
		[defaults setBool:YES forKey:@"menubar_icon"];
		[defaults setBool:YES forKey:@"dock_icon"];
	}
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)setFilenamePopUpButtonChanged:(id)sender {
	NSInteger tag = [self.setFilenamePopUpButton selectedTag];
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	if(tag == 0) {
		[defaults setBool:YES forKey:@"use_filename"];
		[defaults setBool:NO forKey:@"use_hash"];
	}
	else if(tag == 1) {
		[defaults setBool:NO forKey:@"use_filename"];
		[defaults setBool:YES forKey:@"use_hash"];
	}
	else if(tag == 2) {
		[defaults setBool:YES forKey:@"use_filename"];
		[defaults setBool:YES forKey:@"use_hash"];
	}
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (IBAction)changePermissionsCheckboxChanged:(id)sender
{
    NSButton *checkbox = sender;
    BOOL changePermissions = checkbox.state == NSOnState;
    
    NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
    [defaults setBool:changePermissions forKey:@"change_permissions"];
    
    self.permissionsTextField.enabled = changePermissions;
    if (changePermissions) {
			[self.permissionsTextField becomeFirstResponder];
			if (![defaults stringForKey:@"permission_string"]) {
				[defaults setObject:@"644" forKey:@"permission_string"];
			}
    }
}

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark Private Methods

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)updateLaunchAtLoginFromValue
{
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	BOOL launch = [defaults boolForKey:@"launch_at_login"];
	LSSharedFileListRef loginItems = LSSharedFileListCreate(NULL,
                                                          kLSSharedFileListSessionLoginItems,
                                                          NULL);
	NSString *path = [[NSBundle mainBundle] bundlePath];
	BOOL alreadyLaunch = [self loginItemExistsWithLoginItemReference:loginItems
                                                           ForPath:path];
	if(launch != alreadyLaunch) {
		if(launch)
			[self enableLoginItemWithLoginItemsReference:loginItems ForPath:path];
		else
			[self disableLoginItemWithLoginItemsReference:loginItems ForPath:path];
	}
	CFRelease(loginItems);
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)savePassword
{
	NSString *password = [self.passwordTextField stringValue];
	EMGenericKeychainItem *item = [EMGenericKeychainItem genericKeychainItemForService:@"FileShuttle"
                                                                        withUsername:@""];
	if(item)
		item.password = password;
	else
		[EMGenericKeychainItem addGenericKeychainItemForService:@"FileShuttle"
                                               withUsername:@""
                                                   password:password];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)setShowDockIconValue:(BOOL)show
{
	if (show) {
		if (!self.showDockIcon) {
			ProcessSerialNumber psn = { 0, kCurrentProcess };
			OSStatus returnCode = TransformProcessType(& psn,
                                                 kProcessTransformToForegroundApplication);
			if( returnCode == 0) {
				ProcessSerialNumber psnx = { 0, kNoProcess };
				GetNextProcess(&psnx);
				SetFrontProcess(&psnx);
				self.showDockIcon = YES;
				[self performSelector:@selector(setFront) withObject:nil afterDelay:0.5];
			} else {
				NSLog(@"Could not bring the application to front. Error %d", (int)returnCode);
			}
		}
	}
	else {
		if(self.showDockIcon) {
			NSAlert *alert = [[[NSAlert alloc] init] autorelease];
			[alert addButtonWithTitle:@"Restart"];
			[alert setMessageText:@"You must now restart"];
			[alert setInformativeText:@"This change requires to restart the application"];
			[alert setAlertStyle:NSWarningAlertStyle];
			[alert beginSheetModalForWindow:self.window
                        modalDelegate:self
                       didEndSelector:@selector(restartDialogDidEnd:returnCode:contextInfo:)
                          contextInfo:nil];
		}
	}
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)setFront
{
	ProcessSerialNumber psn = { 0, kCurrentProcess };
	SetFrontProcess(&psn);
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)layoutView:(BOOL)animated
{
  NSView *nextView = nil;
  if([self.selectedIdentifier isEqualToString:kMVGeneralIdentifier])
    nextView = self.generalView;
  else if([self.selectedIdentifier isEqualToString:kMVAdvancedIdentifier])
    nextView = self.advancedView;
  
  if(nextView != self.selectedView)
  {
    if(self.selectedView)
      [self.selectedView removeFromSuperview];
    
    CGRect windowFrame = self.window.frame;
    windowFrame.size.height = nextView.frame.size.height + kMVTopMargin;
    windowFrame.origin.y += self.window.frame.size.height - windowFrame.size.height;
    
    [self.window setFrame:windowFrame display:YES animate:animated];
    
    [self.window.contentView addSubview:nextView];
    
    self.selectedView = nextView;
  }
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)restartDialogDidEnd:(NSAlert *)alert
                 returnCode:(NSInteger)returnCode
                contextInfo:(void *)contextInfo
{
	// Not quite sure why we can't directly execute outselves, but
	// we seem to require the open command to make it work
	[NSTask launchedTaskWithLaunchPath:@"/bin/sh"
                           arguments:[NSArray arrayWithObjects:@"-c",
                                      [NSString stringWithFormat:
                                       @"sleep 1 ; /usr/bin/open '%@'",
                                       [[NSBundle mainBundle] bundlePath]],
                                      nil]];
	[NSApp terminate:self];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark Launch At Login helper Methods

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)enableLoginItemWithLoginItemsReference:(LSSharedFileListRef )theLoginItemsRefs
                                       ForPath:(NSString *)appPath
{
	// We call LSSharedFileListInsertItemURL to insert the item at the bottom of Login Items list.
	CFURLRef url = (CFURLRef)[NSURL fileURLWithPath:appPath];
	LSSharedFileListItemRef item = LSSharedFileListInsertItemURL(theLoginItemsRefs,
                                                               kLSSharedFileListItemLast,
                                                               NULL, NULL, url, NULL, NULL);
	if (item)
		CFRelease(item);
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)disableLoginItemWithLoginItemsReference:(LSSharedFileListRef )theLoginItemsRefs
                                        ForPath:(NSString *)appPath
{
	UInt32 seedValue;
	CFURLRef thePath = NULL;
	// We're going to grab the contents of the shared file list (LSSharedFileListItemRef objects)
	// and pop it in an array so we can iterate through it to find our item.
	CFArrayRef loginItemsArray = LSSharedFileListCopySnapshot(theLoginItemsRefs, &seedValue);
	for (id item in (NSArray *)loginItemsArray) {
		LSSharedFileListItemRef itemRef = (LSSharedFileListItemRef)item;
		if (LSSharedFileListItemResolve(itemRef, 0, (CFURLRef*) &thePath, NULL) == noErr) {
			if ([[(NSURL *)thePath path] hasPrefix:appPath]) {
				LSSharedFileListItemRemove(theLoginItemsRefs, itemRef); // Deleting the item
			}
			// Docs for LSSharedFileListItemResolve say we're responsible
			// for releasing the CFURLRef that is returned
			if (thePath != NULL) CFRelease(thePath);
		}
	}
	if (loginItemsArray != NULL) CFRelease(loginItemsArray);
}

///////////////////////////////////////////////////////////////////////////////////////////////////
- (BOOL)loginItemExistsWithLoginItemReference:(LSSharedFileListRef)theLoginItemsRefs
                                      ForPath:(NSString *)appPath
{
	BOOL found = NO;
	UInt32 seedValue;
	CFURLRef thePath = NULL;
	
	// We're going to grab the contents of the shared file list (LSSharedFileListItemRef objects)
	// and pop it in an array so we can iterate through it to find our item.
	CFArrayRef loginItemsArray = LSSharedFileListCopySnapshot(theLoginItemsRefs, &seedValue);
	for (id item in (NSArray *)loginItemsArray) {
		LSSharedFileListItemRef itemRef = (LSSharedFileListItemRef)item;
		if (LSSharedFileListItemResolve(itemRef, 0, (CFURLRef*) &thePath, NULL) == noErr) {
			if ([[(NSURL *)thePath path] hasPrefix:appPath]) {
				found = YES;
				break;
			}
      // Docs for LSSharedFileListItemResolve say we're responsible
      // for releasing the CFURLRef that is returned
      if (thePath != NULL) CFRelease(thePath);
		}
	}
	if (loginItemsArray != NULL) CFRelease(loginItemsArray);
	
	return found;
}

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark NSTextFieldDelegate Methods

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)controlTextDidChange:(NSNotification *)aNotification
{
	if(self.passwordTimer)
		[self.passwordTimer invalidate];
	self.passwordTimer = [NSTimer scheduledTimerWithTimeInterval:2 
                                                        target:self 
                                                      selector:@selector(savePassword) 
                                                      userInfo:nil 
                                                       repeats:NO];
}

///////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////
#pragma mark -
#pragma mark SRRecorderDelegate Methods

///////////////////////////////////////////////////////////////////////////////////////////////////
- (void)shortcutRecorder:(SRRecorderControl *)aRecorder 
       keyComboDidChange:(KeyCombo)newKeyCombo
{
	NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
	[defaults setValue:MVDictionaryFromKeyCombo(newKeyCombo)
              forKey:@"clipboard_upload_shortcut"];
}

@end