AddConnectionController.m

//
//  AddConnectionController.m
//  MongoHub
//
//  Created by Syd on 10-4-24.
//  Copyright 2010 MusicPeace.ORG. All rights reserved.
//

#import "Configure.h"
#import "AddConnectionController.h"
#import "ConnectionsArrayController.h"

@implementation AddConnectionController

@synthesize hostTextField;
@synthesize hostportTextField;
@synthesize aliasTextField;
@synthesize adminuserTextField;
@synthesize adminpassTextField;
@synthesize defaultdbTextField;
@synthesize usesshCheckBox;
@synthesize bindaddressTextField;
@synthesize bindportTextField;
@synthesize sshhostTextField;
@synthesize sshportTextField;
@synthesize sshuserTextField;
@synthesize sshpasswordTextField;
@synthesize connectionInfo;
@synthesize connectionsArrayController;
@synthesize managedObjectContext;

- (id)init {
    if (![super initWithWindowNibName:@"NewConnection"]) return nil;
    return self;
}

- (void)dealloc {
    [hostTextField release];
    [hostportTextField release];
    [aliasTextField release];
    [adminuserTextField release];
    [adminpassTextField release];
    [defaultdbTextField release];
    [usesshCheckBox release];
    [bindaddressTextField release];
    [bindportTextField release];
    [sshhostTextField release];
    [sshportTextField release];
    [sshuserTextField release];
    [sshpasswordTextField release];
    [connectionInfo release];
    [connectionsArrayController release];
    [managedObjectContext release];
    [super dealloc];
}

- (void)windowDidLoad {
    //NSLog(@"New Connection Window Loaded");
    [super windowDidLoad];
}

- (void)windowWillClose:(NSNotification *)notification {
    [[NSNotificationCenter defaultCenter] postNotificationName:kNewConnectionWindowWillClose object:connectionInfo];
}

- (IBAction)cancel:(id)sender {
    connectionInfo = nil;
    [self close];
}

- (IBAction)add:(id)sender {
    NSString *host;
    NSUInteger hostport;
    NSString *alias;
    NSString *adminuser = [adminuserTextField stringValue];
    NSString *adminpass = [adminpassTextField stringValue];
    NSString *defaultdb = [defaultdbTextField stringValue];
    NSUInteger usessh = 0;
    NSString *bindaddress;
    NSUInteger bindport;
    NSString *sshhost;
    NSUInteger sshport;
    NSString *sshuser;
    NSString *sshpassword;
    if ([ [hostTextField stringValue] length] == 0) {
        host = [[NSString alloc] initWithString:@"localhost"];
    }else{
        host = [[NSString alloc] initWithString:[hostTextField stringValue]];
    }
    if ([hostportTextField intValue] == 0) {
        hostport = 27017;
    }else{
        hostport = [hostportTextField intValue];
    }
    if ([ [aliasTextField stringValue] length] == 0) {
        alias = [[NSString alloc] initWithString:@"localhost"];
    }else{
        alias = [[NSString alloc] initWithString:[aliasTextField stringValue]];
    }
    if ([usesshCheckBox state])
    {
        usessh = 1;
        if ([ [bindaddressTextField stringValue] length] == 0) {
            bindaddress = [[NSString alloc] initWithString:@"127.0.0.1"];
        }else{
            bindaddress = [[NSString alloc] initWithString:[bindaddressTextField stringValue]];
        }
        if ([ [bindportTextField stringValue] length] == 0) {
            bindport = 8888;
        }else{
            bindport = [bindportTextField intValue];
        }
        sshhost = [[NSString alloc] initWithString:[sshhostTextField stringValue]];
        if ([[sshportTextField stringValue] length] == 0) {
            sshport = 22;
        }else {
            sshport = [sshportTextField intValue];
        }

        sshuser = [[NSString alloc] initWithString:[sshuserTextField stringValue]];
        sshpassword = [[NSString alloc] initWithString:[sshpasswordTextField stringValue]];
    }else{
        bindaddress = @"127.0.0.1";
        bindport = 8888;
        sshhost = @"";
        sshport = 22;
        sshuser = @"";
        sshpassword = @"";
    }
    NSArray *keys = [[NSArray alloc] initWithObjects:@"host", @"hostport", @"alias", @"adminuser", @"adminpass", @"defaultdb", @"usessh", @"bindaddress", @"bindport", @"sshhost", @"sshport", @"sshuser", @"sshpassword", nil];
    NSArray *objs = [[NSArray alloc] initWithObjects:host, [NSNumber numberWithInt:hostport], alias, adminuser, adminpass, defaultdb, [NSNumber numberWithInt:usessh], bindaddress, [NSNumber numberWithInt:bindport], sshhost, [NSNumber numberWithInt:sshport], sshuser, sshpassword, nil];
    if (!connectionInfo) {
        connectionInfo = [[NSMutableDictionary alloc] initWithCapacity:13]; 
    }
    connectionInfo = [NSMutableDictionary dictionaryWithObjects:objs forKeys:keys];
    [keys release];
    [objs release];
    [host release];
    [alias release];
    if (usessh == 1)
    {
        [sshhost release];
        [sshuser release];
        [sshpassword release];
        [bindaddress release];
    }
    if ([self validateConnection]) {
        [self close];
    }
}

- (BOOL)validateConnection
{
    if ([[connectionInfo objectForKey:@"host"] length] == 0) {
        NSRunAlertPanel(@"Error", @"Connection host should not be empty", @"OK", nil, nil);
        return NO;
    }
    if ([[connectionInfo objectForKey:@"host"] isEqualToString:@"flame.mongohq.com"] && [[connectionInfo objectForKey:@"defaultdb"] length] == 0) {
        NSRunAlertPanel(@"Error", @"DB should not be empty if you are using mongohq", @"OK", nil, nil);
        return NO;
    }
    if ([[connectionInfo objectForKey:@"alias"] length]<3) {
        NSRunAlertPanel(@"Error", @"Connection name should not be less than 3 charaters", @"OK", nil, nil);
        return NO;
    }
    if ([connectionsArrayController checkDuplicate:[connectionInfo objectForKey:@"alias"]]) {
        NSRunAlertPanel(@"Error", @"Connection alias name has been existed!", @"OK", nil, nil);
        return NO;
    }
    if ([usesshCheckBox state] == 1 && ([[connectionInfo objectForKey:@"bindaddress"] length] == 0 || [[connectionInfo objectForKey:@"sshhost"] length] == 0)) {
        NSRunAlertPanel(@"Error", @"Please full fill ssh information!", @"OK", nil, nil);
        return NO;
    }
    return YES;
}

- (IBAction)enableSSH:(id)sender
{
    if ([usesshCheckBox state] == 1)
    {
        [bindaddressTextField setEnabled:YES];
        [bindportTextField setEnabled:YES];
        [sshhostTextField setEnabled:YES];
        [sshuserTextField setEnabled:YES];
        [sshpasswordTextField setEnabled:YES];
        [sshportTextField setEnabled:YES];
    }else {
        [bindaddressTextField setEnabled:NO];
        [bindportTextField setEnabled:NO];
        [sshhostTextField setEnabled:NO];
        [sshuserTextField setEnabled:NO];
        [sshpasswordTextField setEnabled:NO];
        [sshportTextField setEnabled:NO];
    }

}
@end