diff options
author | abhibeckert <abhi@abhibeckert.com> | 2008-04-04 11:37:36 +0000 |
---|---|---|
committer | abhibeckert <abhi@abhibeckert.com> | 2008-04-04 11:37:36 +0000 |
commit | 0f586da34dac2c764a12055158e4929cf69340a1 (patch) | |
tree | bca3868c9715a10c08017a0a7b6d5148f8bd3556 /TableDocument.m | |
parent | 5fb284b7c12887b8eea6c3a02fa83abf50793992 (diff) | |
download | sequelpro-0f586da34dac2c764a12055158e4929cf69340a1.tar.gz sequelpro-0f586da34dac2c764a12055158e4929cf69340a1.tar.bz2 sequelpro-0f586da34dac2c764a12055158e4929cf69340a1.zip |
Initial import from (slightly modified) CocoaMySQL v0.7b4
Diffstat (limited to 'TableDocument.m')
-rw-r--r-- | TableDocument.m | 1625 |
1 files changed, 1625 insertions, 0 deletions
diff --git a/TableDocument.m b/TableDocument.m new file mode 100644 index 00000000..5274c688 --- /dev/null +++ b/TableDocument.m @@ -0,0 +1,1625 @@ +// +// TableDocument.m +// CocoaMySQL +// +// Created by lorenz textor (lorenz@textor.ch) on Wed May 01 2002. +// Copyright (c) 2002-2003 Lorenz Textor. All rights reserved. +// +// This program is free software; you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation; either version 2 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// More info at <http://cocoamysql.sourceforge.net/> +// Or mail to <lorenz@textor.ch> + +#import "TableDocument.h" +#import "KeyChain.h" +#import "TablesList.h" +#import "TableSource.h" +#import "TableContent.h" +#import "CustomQuery.h" +#import "TableDump.h" +#import "TableStatus.h" + +NSString *TableDocumentFavoritesControllerSelectionIndexDidChange = @"TableDocumentFavoritesControllerSelectionIndexDidChange"; + +@implementation TableDocument + +- (void)awakeFromNib +{ + [favoritesController addObserver:self forKeyPath:@"selectionIndex" options:NSKeyValueChangeInsertion context:TableDocumentFavoritesControllerSelectionIndexDidChange]; +} + +- (void) observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context +{ + if (context == TableDocumentFavoritesControllerSelectionIndexDidChange) { + [self chooseFavorite:self]; + } + else { + [super observeValueForKeyPath:keyPath ofObject:object change:change context:context]; + } +} + + + +//start sheet +- (IBAction)toggleUseSSH:(id)sender +/* +enables/disables ssh tunneling +*/ +{ + if ([sshCheckbox state] == NSOnState) { + [sshUserField setEnabled:YES]; + [sshPasswordField setEnabled:YES]; + [sshHostField setEnabled:YES]; + [sshPortField setEnabled:YES]; + } else { + [sshUserField setEnabled:NO]; + [sshPasswordField setEnabled:NO]; + [sshHostField setEnabled:NO]; + [sshPortField setEnabled:NO]; + } +} + +- (IBAction)connectToDB:(id)sender +/* +tries to connect to the db +alert-sheets when no success +*/ +{ + CMMCPResult *theResult; + NSString *encoding; + int code; + id version; + + [self setFavorites]; + + [NSApp beginSheet:connectSheet + modalForWindow:tableWindow modalDelegate:self + didEndSelector:nil contextInfo:nil]; + code = [NSApp runModalForWindow:connectSheet]; + + [NSApp endSheet:connectSheet]; + [connectSheet orderOut:nil]; + + if ( code == 1) { +//connected with success + //register as delegate + [mySQLConnection setDelegate:self]; + // set encoding + encoding = [prefs objectForKey:@"encoding"]; + if ( [encoding isEqualToString:@"Autodetect"] ) { + [self detectEncoding]; + } else { + [chooseEncodingButton selectItemWithTitle:encoding]; + [self setEncoding:[self getSelectedEncoding]]; + } + // get selected db + if ( ![[databaseField stringValue] isEqualToString:@""] ) + selectedDatabase = [[databaseField stringValue] retain]; + //get mysql version +// theResult = [mySQLConnection queryString:@"SHOW VARIABLES LIKE \"version\""]; + theResult = [mySQLConnection queryString:@"SHOW VARIABLES LIKE 'version'"]; + version = [[theResult fetchRowAsArray] objectAtIndex:1]; + if ( [version isKindOfClass:[NSData class]] ) { + // starting with MySQL 4.1.14 the mysql variables are returned as nsdata + mySQLVersion = [[NSString alloc] initWithData:version encoding:[mySQLConnection encoding]]; + } else { + mySQLVersion = [[NSString stringWithString:version] retain]; + } + [self setDatabases:self]; + [tablesListInstance setConnection:mySQLConnection]; + [tableSourceInstance setConnection:mySQLConnection]; + [tableContentInstance setConnection:mySQLConnection]; + [customQueryInstance setConnection:mySQLConnection]; + [tableDumpInstance setConnection:mySQLConnection]; + [tableStatusInstance setConnection:mySQLConnection]; + [self setFileName:[NSString stringWithFormat:@"(MySQL %@) %@@%@ %@", mySQLVersion, [userField stringValue], + [hostField stringValue], [databaseField stringValue]]]; + [tableWindow setTitle:[NSString stringWithFormat:@"(MySQL %@) %@@%@/%@", mySQLVersion, [userField stringValue], + [hostField stringValue], [databaseField stringValue]]]; + } else if (code == 2) { +//can't connect to host + NSBeginAlertSheet(NSLocalizedString(@"Connection failed!", @"connection failed"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, + @selector(sheetDidEnd:returnCode:contextInfo:), @"connect", + [NSString stringWithFormat:NSLocalizedString(@"Unable to connect to host %@.\nBe sure that the address is correct and that you have the necessary privileges.\nMySQL said: %@", @"message of panel when connection to host failed"), [hostField stringValue], [mySQLConnection getLastErrorMessage]]); + } else if (code == 3) { +//can't connect to db + NSBeginAlertSheet(NSLocalizedString(@"Connection failed!", @"connection failed"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, + @selector(sheetDidEnd:returnCode:contextInfo:), @"connect", + [NSString stringWithFormat:NSLocalizedString(@"Unable to connect to database %@.\nBe sure that the database exists and that you have the necessary privileges.\nMySQL said: %@", @"message of panel when connection to db failed"), [databaseField stringValue], [mySQLConnection getLastErrorMessage]]); + } else if (code == 4) { +//no host is given + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, + @selector(sheetDidEnd:returnCode:contextInfo:), @"connect", NSLocalizedString(@"Please enter at least a host or socket.", @"message of panel when host/socket are missing")); + } else { +//cancel button was pressed + //since the window is getting ready to be toast ignore events for awhile + //so as not to crash, this happens to me when hitten esc key instead of + //cancel button, but with this code it does not crash + [[NSApplication sharedApplication] discardEventsMatchingMask:NSAnyEventMask + beforeEvent:[[NSApplication sharedApplication] nextEventMatchingMask:NSLeftMouseDownMask | NSLeftMouseUpMask |NSRightMouseDownMask | NSRightMouseUpMask | NSFlagsChangedMask | NSKeyDownMask | NSKeyUpMask untilDate:[NSDate distantPast] inMode:NSEventTrackingRunLoopMode dequeue:YES]]; + [tableWindow close]; + } +} + + +- (IBAction)connect:(id)sender +/* +invoked when user hits the connect-button of the connectSheet +stops modal session with code: +1 when connected with success +2 when no connection to host +3 when no connection to db +4 when hostField and socketField are empty +*/ +{ + int code; + + [connectProgressBar startAnimation:self]; + + code = 0; + if ( [[hostField stringValue] isEqualToString:@""] && [[socketField stringValue] isEqualToString:@""] ) { + code = 4; + } else { + if ( ![[socketField stringValue] isEqualToString:@""] ) { + //connect to socket + mySQLConnection = [[CMMCPConnection alloc] initToSocket:[socketField stringValue] + withLogin:[userField stringValue] + password:[passwordField stringValue]]; + [hostField setStringValue:@"localhost"]; + } else { + //connect to host + mySQLConnection = [[CMMCPConnection alloc] initToHost:[hostField stringValue] + withLogin:[userField stringValue] + password:[passwordField stringValue] + usingPort:[portField intValue]]; + } + if ( ![mySQLConnection isConnected] ) + code = 2; + if ( !code && ![[databaseField stringValue] isEqualToString:@""] ) + if ( ![mySQLConnection selectDB:[databaseField stringValue]] ) + code = 3; + if ( !code ) + code = 1; + } + [NSApp stopModalWithCode:code]; + + [connectProgressBar stopAnimation:self]; +} + +- (IBAction)closeSheet:(id)sender +/* +invoked when user hits the cancel button of the connectSheet +stops modal session with code 0 +reused when user hits the close button of the variablseSheet or of the createTableSyntaxSheet +*/ +{ + [NSApp stopModalWithCode:0]; +} + +- (IBAction)chooseFavorite:(id)sender +/* + sets fields for the choosen favorite + */ +{ + /* + BOOL useSSH = NO; + + if ( [favoritesButton indexOfSelectedItem] == 0 ) { + [hostField setStringValue:@""]; + [socketField setStringValue:@""]; + [userField setStringValue:@""]; + [portField setStringValue:@""]; + [databaseField setStringValue:@""]; + [passwordField setStringValue:@""]; + [sshCheckbox setState:NSOffState]; + [sshUserField setEnabled:NO]; + [sshPasswordField setEnabled:NO]; + [sshHostField setEnabled:NO]; + [sshPortField setEnabled:NO]; + [sshHostField setStringValue:@""]; + [sshUserField setStringValue:@""]; + [sshPasswordField setStringValue:@""]; + [sshPortField setStringValue:@"8888"]; + [hostField selectText:self]; + [selectedFavorite release]; + selectedFavorite = [[favoritesButton titleOfSelectedItem] retain]; + } else if ( [favoritesButton indexOfSelectedItem] == 1 ) { + if ( ![[socketField stringValue] isEqualToString:@""] ) { + [hostField setStringValue:@"localhost"]; + } + if ( [sshCheckbox state] == NSOnState ) { + useSSH = YES; + } else { + useSSH = NO; + } + [self addToFavoritesHost:[hostField stringValue] + socket:[socketField stringValue] + user:[userField stringValue] + password:[passwordField stringValue] + port:[portField stringValue] + database:[databaseField stringValue] + useSSH:useSSH + sshHost:[sshHostField stringValue] + sshUser:[sshUserField stringValue] + sshPassword:[sshPasswordField stringValue] + sshPort:[sshPortField stringValue]]; + // } else if ( [favoritesButton indexOfSelectedItem] == 2 ) { + // [favoritesButton selectItemWithTitle:selectedFavorite]; + } else {*/ + NSDictionary *favorite = [[prefs objectForKey:@"favorites"] objectAtIndex:[favoritesController selectionIndex]]; + NSString *name = [favorite objectForKey:@"name"]; + NSString *host = [favorite objectForKey:@"host"]; + NSString *socket = [favorite objectForKey:@"socket"]; + NSString *user = [favorite objectForKey:@"user"]; + NSString *port = [favorite objectForKey:@"port"]; + NSString *database = [favorite objectForKey:@"database"]; + int useSSH = [[favorite objectForKey:@"useSSH"] intValue]; + NSString *sshHost = [favorite objectForKey:@"sshHost"]; + NSString *sshUser = [favorite objectForKey:@"sshUser"]; + NSString *sshPort = [favorite objectForKey:@"sshPort"]; + + [hostField setStringValue:host]; + [socketField setStringValue:socket]; + [userField setStringValue:user]; + [portField setStringValue:port]; + [databaseField setStringValue:database]; + [passwordField setStringValue:[keyChainInstance + getPasswordForName:[NSString stringWithFormat:@"CocoaMySQL : %@", name] + account:[NSString stringWithFormat:@"%@@%@/%@", user, host, database]]]; + if ( useSSH ) { + [sshCheckbox setState:NSOnState]; + [sshHostField setStringValue:sshHost]; + [sshUserField setStringValue:sshUser]; + [sshPortField setStringValue:sshPort]; + [sshPasswordField setStringValue:[keyChainInstance + getPasswordForName:[NSString stringWithFormat:@"CocoaMySQL SSHTunnel : %@", name] + account:[NSString stringWithFormat:@"%@@%@/%@", user, host, database]]]; + [sshUserField setEnabled:YES]; + [sshPasswordField setEnabled:YES]; + [sshHostField setEnabled:YES]; + [sshPortField setEnabled:YES]; + } else { + [sshCheckbox setState:NSOffState]; + [sshHostField setStringValue:@""]; + [sshUserField setStringValue:@""]; + [sshPortField setStringValue:@""]; + [sshPasswordField setStringValue:@""]; + [sshUserField setEnabled:NO]; + [sshPasswordField setEnabled:NO]; + [sshHostField setEnabled:NO]; + [sshPortField setEnabled:NO]; + } + + [selectedFavorite release]; + selectedFavorite = [[favoritesButton titleOfSelectedItem] retain]; + /*}*/ +} + +- (NSArray *)favorites +{ + return favorites; +} + +- (void)setFavorites +/* + set up the favorites popUpButton and notifiy bindings that it's changed + */ +{ + [self willChangeValueForKey:@"favorites"]; + [self didChangeValueForKey:@"favorites"]; + + NSEnumerator *enumerator = [favorites objectEnumerator]; + id favorite; + + [favoritesButton removeAllItems]; + [favoritesButton addItemWithTitle:NSLocalizedString(@"Custom", @"menu item for custom connection")]; + [favoritesButton addItemWithTitle:NSLocalizedString(@"Save to favorites...", @"menu item for saving connection to favorites")]; + // [favoritesButton addItemWithTitle:@""]; + [[favoritesButton menu] addItem:[NSMenuItem separatorItem]]; + while ( (favorite = [enumerator nextObject]) ) { + [favoritesButton addItemWithTitle:[favorite objectForKey:@"name"]]; + } +} + +- (void)addToFavoritesHost:(NSString *)host socket:(NSString *)socket + user:(NSString *)user password:(NSString *)password + port:(NSString *)port database:(NSString *)database + useSSH:(BOOL)useSSH sshHost:(NSString *)sshHost + sshUser:(NSString *)sshUser sshPassword:(NSString *)sshPassword + sshPort:(NSString *)sshPort +/* +add actual connection to favorites +*/ +{ + NSEnumerator *enumerator = [favorites objectEnumerator]; + id favorite; + NSString *favoriteName = [NSString stringWithFormat:@"%@@%@/%@", user, host, database]; + NSNumber *ssh; + +//test if host and socket are not nil + if ( [host isEqualToString:@""] && [socket isEqualToString:@""] ) + { + NSRunAlertPanel(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"Please enter at least a host or socket.", @"message of panel when host/socket are missing"), NSLocalizedString(@"OK", @"OK button"), nil, nil); + [favoritesButton selectItemWithTitle:selectedFavorite]; + return; + } +//test if all fields are specified for ssh tunnel + if ( useSSH ) { + if ( [sshHost isEqualToString:@""] ) { + sshHost = host; + } + if ( [sshUser isEqualToString:@""] ) { + sshUser = user; + } + if ( [sshPassword isEqualToString:@""] ) { + sshPassword = password; + } + if ( [sshPort isEqualToString:@""] ) { + sshPort = port; + } + ssh = [NSNumber numberWithInt:1]; + } else { + sshHost = @""; + sshUser = @""; + sshPassword = @""; + sshPort = @""; + ssh = [NSNumber numberWithInt:0]; + } + +//test if favorite name isn't used by another favorite and if no favorite with the same host, user and db exists + while ( (favorite = [enumerator nextObject]) ) { + if ( [[favorite objectForKey:@"name"] isEqualToString:favoriteName] ) + { + NSRunAlertPanel(NSLocalizedString(@"Error", @"error"), [NSString stringWithFormat:NSLocalizedString(@"Favorite %@ has already been saved!\nOpen Preferences to change the names of the favorites.", @"message of panel when favorite name has already been used"), favoriteName], NSLocalizedString(@"OK", @"OK button"), nil, nil); + [favoritesButton selectItemWithTitle:selectedFavorite]; + return; + } +/* + if ( [[favorite objectForKey:@"host"] isEqualToString:host] && + [[favorite objectForKey:@"user"] isEqualToString:user] && + [[favorite objectForKey:@"database"] isEqualToString:database] ) { + NSRunAlertPanel(@"Error", @"There is already a favorite with the same host, user and database!", @"OK", nil, nil); + [favoritesButton selectItemWithTitle:selectedFavorite]; + return; + } +*/ + } + +//write favorites and password + NSDictionary *newFavorite = [NSDictionary + dictionaryWithObjects:[NSArray arrayWithObjects:favoriteName, host, socket, user, port, database, ssh, sshHost, sshUser, sshPort, nil] + forKeys:[NSArray arrayWithObjects:@"name", @"host", @"socket", @"user", @"port", @"database", @"useSSH", @"sshHost", @"sshUser", @"sshPort", nil]]; + favorites = [[favorites arrayByAddingObject:newFavorite] retain]; + if ( ![password isEqualToString:@""] ) + [keyChainInstance addPassword:password forName:[NSString stringWithFormat:@"CocoaMySQL : %@", favoriteName] + account:[NSString stringWithFormat:@"%@@%@/%@", user, host, database]]; + if ( ![sshPassword isEqualToString:@""] ) + [keyChainInstance addPassword:sshPassword forName:[NSString stringWithFormat:@"CocoaMySQL SSHTunnel : %@", favoriteName] + account:[NSString stringWithFormat:@"%@@%@/%@", user, host, database]]; + [prefs setObject:favorites forKey:@"favorites"]; + +//reload favorites and select new favorite + [self setFavorites]; + [favoritesButton selectItemWithTitle:favoriteName]; + selectedFavorite = [favoriteName retain]; +} + + +//alert sheets method +- (void)sheetDidEnd:(NSWindow *)sheet returnCode:(int)returnCode contextInfo:(NSString *)contextInfo +/* +invoked when alertSheet get closed +if contextInfo == connect -> reopens the connectSheet +if contextInfo == removedatabase -> tries to remove the selected database +*/ +{ + [sheet orderOut:self]; + + if ( [contextInfo isEqualToString:@"connect"] ) { + [self connectToDB:nil]; + } else if ( [contextInfo isEqualToString:@"removedatabase"] ) { + if ( returnCode == NSAlertDefaultReturn ) { + [mySQLConnection queryString:[NSString stringWithFormat:@"DROP DATABASE `%@`", [self database]]]; + if ( [[mySQLConnection getLastErrorMessage] isEqualToString:@""] ) { + //db deleted with success + selectedDatabase = nil; + [self setDatabases:self]; + [tablesListInstance setConnection:mySQLConnection]; + [tableDumpInstance setConnection:mySQLConnection]; + [tableWindow setTitle:[NSString stringWithFormat:@"(MySQL %@) %@@%@/", + mySQLVersion, [userField stringValue], [hostField stringValue]]]; + } else { + //error while deleting db + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Couldn't remove database.\nMySQL said: %@", @"message of panel when removing db failed"), [mySQLConnection getLastErrorMessage]]); + } + } + } +} + + +//database methods +- (IBAction)setDatabases:(id)sender; +/* +sets up the chooseDatabaseButton (adds all databases) +*/ +{ + CMMCPResult *queryResult; + int i; + + [chooseDatabaseButton removeAllItems]; + [chooseDatabaseButton addItemWithTitle:NSLocalizedString(@"Choose database...", @"menu item for choose db")]; + queryResult = [mySQLConnection listDBs]; + for ( i = 0 ; i < [queryResult numOfRows] ; i++ ) { + [queryResult dataSeek:i]; + [chooseDatabaseButton addItemWithTitle:[[queryResult fetchRowAsArray] objectAtIndex:0]]; + } + if ( ![self database] ) { + [chooseDatabaseButton selectItemWithTitle:NSLocalizedString(@"Choose database...", @"menu item for choose db")]; + } else { + [chooseDatabaseButton selectItemWithTitle:[self database]]; + } +} + +- (IBAction)chooseDatabase:(id)sender +/* +selects the database choosen by the user +errorsheet if connection failed +*/ +{ + if ( ![tablesListInstance selectionShouldChangeInTableView:nil] ) { + [chooseDatabaseButton selectItemWithTitle:[self database]]; + return; + } + + if ( [chooseDatabaseButton indexOfSelectedItem] == 0 ) { + if ( ![self database] ) { + [chooseDatabaseButton selectItemWithTitle:NSLocalizedString(@"Choose database...", @"menu item for choose db")]; + } else { + [chooseDatabaseButton selectItemWithTitle:[self database]]; + } + return; + } + + if ( ![mySQLConnection selectDB:[chooseDatabaseButton titleOfSelectedItem]] ) { +//connection failed + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Unable to connect to database %@.\nBe sure that you have the necessary privileges.", @"message of panel when connection to db failed after selecting from popupbutton"), + [chooseDatabaseButton titleOfSelectedItem]]); + [self setDatabases:self]; + } else { +//changed database with success +//setConnection of TablesList and TablesDump to reload tables in db + [selectedDatabase release]; + selectedDatabase = nil; + selectedDatabase = [[chooseDatabaseButton titleOfSelectedItem] retain]; + [tablesListInstance setConnection:mySQLConnection]; + [tableDumpInstance setConnection:mySQLConnection]; + [tableWindow setTitle:[NSString stringWithFormat:@"(MySQL %@) %@@%@/%@", mySQLVersion, [userField stringValue], + [hostField stringValue], [self database]]]; + } +} + +- (IBAction)addDatabase:(id)sender +/* +opens the add-db sheet and creates the new db +*/ +{ + int code = 0; + + if ( ![tablesListInstance selectionShouldChangeInTableView:nil] ) + return; + + [databaseNameField setStringValue:@""]; + [NSApp beginSheet:databaseSheet + modalForWindow:tableWindow modalDelegate:self + didEndSelector:nil contextInfo:nil]; + code = [NSApp runModalForWindow:databaseSheet]; + + [NSApp endSheet:databaseSheet]; + [databaseSheet orderOut:nil]; + + if ( code ) { + if ( [[databaseNameField stringValue] isEqualToString:@""] ) { + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, NSLocalizedString(@"Database must have a name.", @"message of panel when no db name is given")); + } else { + [mySQLConnection queryString:[NSString stringWithFormat:@"CREATE DATABASE `%@`", [databaseNameField stringValue]]]; + if ( [[mySQLConnection getLastErrorMessage] isEqualToString:@""] ) { + //db created with success + if ( ![mySQLConnection selectDB:[databaseNameField stringValue]] ) { + //error while selecting new db (is this possible?!) + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Unable to connect to database %@.\nBe sure that you have the necessary privileges.", @"message of panel when connection to db failed after selecting from popupbutton"), + [databaseNameField stringValue]]); + [self setDatabases:self]; + } else { + //select new db + [selectedDatabase release]; + selectedDatabase = nil; + selectedDatabase = [[databaseNameField stringValue] retain]; + [self setDatabases:self]; + [tablesListInstance setConnection:mySQLConnection]; + [tableDumpInstance setConnection:mySQLConnection]; + [tableWindow setTitle:[NSString stringWithFormat:@"(MySQL %@) %@@%@/%@", + mySQLVersion, [userField stringValue], [hostField stringValue], + selectedDatabase]]; + } + } else { + //error while creating db + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Couldn't create database.\nMySQL said: %@", @"message of panel when creation of db failed"), [mySQLConnection getLastErrorMessage]]); + } + } + } +} + +- (IBAction)closeDatabaseSheet:(id)sender +/* +closes the add-db sheet and stops modal session +*/ +{ + [NSApp stopModalWithCode:[sender tag]]; +} + +- (IBAction)removeDatabase:(id)sender +/* +opens sheet to ask user if he really wants to delete the db +*/ +{ + if ( [chooseDatabaseButton indexOfSelectedItem] == 0 ) + return; + if ( ![tablesListInstance selectionShouldChangeInTableView:nil] ) + return; + + NSBeginAlertSheet(NSLocalizedString(@"Warning", @"warning"), NSLocalizedString(@"Delete", @"delete button"), NSLocalizedString(@"Cancel", @"cancel button"), nil, tableWindow, self, nil, + @selector(sheetDidEnd:returnCode:contextInfo:), @"removedatabase", + [NSString stringWithFormat:NSLocalizedString(@"Do you really want to delete the database %@?", @"message of panel asking for confirmation for deleting db"), [self database]] ); +} + + +//console methods +- (void)toggleConsole +/* +shows or hides the console +*/ +{ + NSDrawerState state = [consoleDrawer state]; + if (NSDrawerOpeningState == state || NSDrawerOpenState == state) { + [consoleDrawer close]; + } else { + [consoleTextView scrollRangeToVisible:[consoleTextView selectedRange]]; + [consoleDrawer openOnEdge:NSMinYEdge]; + } +} + +- (void)clearConsole +/* +clears the console +*/ +{ + [consoleTextView setString:@""]; +} + +- (BOOL)consoleIsOpened +/* +returns YES if the console is visible +*/ +{ + if ( [consoleDrawer state] == NSDrawerOpeningState || [consoleDrawer state] == NSDrawerOpenState ) + { + return YES; + } else { + return NO; + } +} + +- (void)showMessageInConsole:(NSString *)message +/* +shows a message in the console +*/ +{ + int begin, end; + + [consoleTextView setSelectedRange:NSMakeRange([[consoleTextView string] length],0)]; + begin = [[consoleTextView string] length]; + [consoleTextView replaceCharactersInRange:NSMakeRange(begin,0) + withString:message]; + end = [[consoleTextView string] length]; + [consoleTextView setTextColor:[NSColor blackColor] range:NSMakeRange(begin,end-begin)]; + if ( [self consoleIsOpened] ) { +/* + NSClipView *clipView = [consoleTextView superview]; + if (![clipView isKindOfClass:[NSClipView class]]) return; + [clipView scrollToPoint:[clipView constrainScrollPoint:NSMakePoint(0,[consoleTextView frame].size.height)]]; + [[clipView superview] reflectScrolledClipView:clipView]; +*/ + [consoleTextView displayIfNeeded]; + [consoleTextView scrollRangeToVisible:[consoleTextView selectedRange]]; + } +} + +- (void)showErrorInConsole:(NSString *)error +/* +shows an error in the console (red) +*/ +{ + int begin, end; + + [consoleTextView setSelectedRange:NSMakeRange([[consoleTextView string] length],0)]; + begin = [[consoleTextView string] length]; + [consoleTextView replaceCharactersInRange:NSMakeRange(begin,0) + withString:error]; + end = [[consoleTextView string] length]; + [consoleTextView setTextColor:[NSColor redColor] range:NSMakeRange(begin,end-begin)]; + if ( [self consoleIsOpened] ) { +/* + NSClipView *clipView = [consoleTextView superview]; + if (![clipView isKindOfClass:[NSClipView class]]) return; + [clipView scrollToPoint:[clipView constrainScrollPoint:NSMakePoint(0,[consoleTextView frame].size.height)]]; + [[clipView superview] reflectScrolledClipView:clipView]; +*/ + [consoleTextView displayIfNeeded]; + [consoleTextView scrollRangeToVisible:[consoleTextView selectedRange]]; + } +} + + +//encoding methods +- (void)setEncoding:(NSString *)encoding +/* +set the encoding for the database +*/ +{ +// set encoding of connection and client + [mySQLConnection queryString:[NSString stringWithFormat:@"SET NAMES '%@'", encoding]]; + if ( [[mySQLConnection getLastErrorMessage] isEqualToString:@""] ) { + [mySQLConnection setEncoding:[CMMCPConnection encodingForMySQLEncoding:[encoding cString]]]; + } else { + [self detectEncoding]; + } +//NSLog(@"set encoding to %@", encoding); + + [tableSourceInstance reloadTable:self]; + [tableContentInstance reloadTable:self]; + [tableStatusInstance reloadTable:self]; + +// int encodingCode; + +/* if( [encoding isEqualToString:@"ISO Latin 1"] ) { + encodingCode = NSISOLatin1StringEncoding; + } else if( [encoding isEqualToString:@"ISO Latin 2"] ) { + encodingCode = NSISOLatin2StringEncoding; + } else if( [encoding isEqualToString:@"Win Latin 1"] ) { + encodingCode = NSWindowsCP1252StringEncoding; + } else if( [encoding isEqualToString:@"Win Latin 2"] ) { + encodingCode = NSWindowsCP1250StringEncoding; + } else if( [encoding isEqualToString:@"Cyrillic"] ) { + encodingCode = NSWindowsCP1251StringEncoding; + } else if( [encoding isEqualToString:@"Greek"] ) { + encodingCode = NSWindowsCP1253StringEncoding; + } else if( [encoding isEqualToString:@"Turkish"] ) { + encodingCode = NSWindowsCP1254StringEncoding; + } else if ( [encoding isEqualToString:@"Shift-JIS"] ) { + encodingCode = NSShiftJISStringEncoding; + } else if ( [encoding isEqualToString:@"EUC-JP"] ) { + encodingCode = NSJapaneseEUCStringEncoding; + } else if ( [encoding isEqualToString:@"ISO 2022-JP"] ) { + encodingCode = NSISO2022JPStringEncoding; + } else if ( [encoding isEqualToString:@"UTF-8"] ) { + encodingCode = NSUTF8StringEncoding; +*/ +/* + if( [encoding isEqualToString:@"ISO Latin 1"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatin1); + } else if( [encoding isEqualToString:@"ISO Latin 2"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatin2); + } else if ( [encoding isEqualToString:@"ISO Cyrillic"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatinCyrillic); + } else if ( [encoding isEqualToString:@"ISO Greek"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatinGreek); + } else if ( [encoding isEqualToString:@"ISO Turkish"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatin5); + } else if ( [encoding isEqualToString:@"ISO Arabic"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatinArabic); + } else if ( [encoding isEqualToString:@"ISO Hebrew"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatinHebrew); + } else if ( [encoding isEqualToString:@"ISO Thai"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISOLatinThai); + + } else if( [encoding isEqualToString:@"Win Latin 1"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsLatin1); + } else if( [encoding isEqualToString:@"Win Latin 2"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsLatin2); + } else if( [encoding isEqualToString:@"Win Cyrillic"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsCyrillic); + } else if( [encoding isEqualToString:@"Win Greek"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsGreek); + } else if( [encoding isEqualToString:@"Win Turkish"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsLatin5); + } else if( [encoding isEqualToString:@"Win Arabic"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsArabic); + } else if( [encoding isEqualToString:@"Win Baltic Rim"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsBalticRim); + } else if( [encoding isEqualToString:@"Win Korean"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsKoreanJohab); + } else if( [encoding isEqualToString:@"Win Vietnamese"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingWindowsVietnamese); + + } else if ( [encoding isEqualToString:@"Shift-JIS"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingShiftJIS); + } else if ( [encoding isEqualToString:@"EUC-JP"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingEUC_JP); + } else if ( [encoding isEqualToString:@"ISO 2022-JP"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISO_2022_JP); + + } else if ( [encoding isEqualToString:@"EUC-CN"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingEUC_CN); + } else if ( [encoding isEqualToString:@"EUC-TW"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingEUC_TW); + } else if ( [encoding isEqualToString:@"EUC-KR"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingEUC_KR); + + } else if ( [encoding isEqualToString:@"ISO 2022-KR"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISO_2022_KR); + } else if ( [encoding isEqualToString:@"ISO 2022-CN"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingISO_2022_CN); + + } else if ( [encoding isEqualToString:@"KOI8-R"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingKOI8_R); + } else if ( [encoding isEqualToString:@"HZ"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingHZ_GB_2312); + + } else if ( [encoding isEqualToString:@"UTF-8"] ) { + encodingCode = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingUTF8); + + } else { + encodingCode = NSISOLatin1StringEncoding; // default is ISO Latin 1 + } + + if(encodingCode == kCFStringEncodingInvalidId) + encodingCode = NSISOLatin1StringEncoding; +*/ +} + +- (void)detectEncoding +/* +autodetects the connection encoding and sets the encoding dropdown +*/ +{ + id mysqlEncoding; + + // mysql > 4.0 + mysqlEncoding = [[[mySQLConnection queryString:@"SHOW VARIABLES LIKE 'character_set_connection'"] fetchRowAsDictionary] objectForKey:@"Value"]; + if ( [mysqlEncoding isKindOfClass:[NSData class]] ) { + // MySQL 4.1.14 returns the mysql variables as nsdata + mysqlEncoding = [mySQLConnection stringWithText:mysqlEncoding]; + } + if ( !mysqlEncoding ) { + // mysql 4.0 or older -> only default character set possible, cannot choose others using "set names xy" + mysqlEncoding = [[[mySQLConnection queryString:@"SHOW VARIABLES LIKE 'character_set'"] fetchRowAsDictionary] objectForKey:@"Value"]; + [chooseEncodingButton setEnabled:NO]; + } + if ( !mysqlEncoding ) { + // older version? -> set encoding to mysql default encoding latin1, chooseEncodingButton is already disabled + NSLog(@"error: no character encoding found, mysql version is %@", [self mySQLVersion]); + mysqlEncoding = @"latin1"; + } + [mySQLConnection setEncoding:[CMMCPConnection encodingForMySQLEncoding:[mysqlEncoding cString]]]; + +//NSLog(@"autodetected %@", mysqlEncoding); + + if ( [mysqlEncoding isEqualToString:@"ucs2"] ) { + [chooseEncodingButton selectItemWithTitle:@"UCS-2 Unicode (ucs2)"]; + } else if ( [mysqlEncoding isEqualToString:@"utf8"] ) { + [chooseEncodingButton selectItemWithTitle:@"UTF-8 Unicode (utf8)"]; + } else if ( [mysqlEncoding isEqualToString:@"ascii"] ) { + [chooseEncodingButton selectItemWithTitle:@"US ASCII (ascii)"]; + } else if ( [mysqlEncoding isEqualToString:@"latin1"] ) { + [chooseEncodingButton selectItemWithTitle:@"ISO Latin 1 (latin1)"]; + } else if ( [mysqlEncoding isEqualToString:@"macroman"] ) { + [chooseEncodingButton selectItemWithTitle:@"Mac Roman (macroman)"]; + } else if ( [mysqlEncoding isEqualToString:@"cp1250"] ) { + [chooseEncodingButton selectItemWithTitle:@"Windows Latin 2 (cp1250)"]; + } else if ( [mysqlEncoding isEqualToString:@"latin2"] ) { + [chooseEncodingButton selectItemWithTitle:@"ISO Latin 2 (latin2)"]; + } else if ( [mysqlEncoding isEqualToString:@"cp1256"] ) { + [chooseEncodingButton selectItemWithTitle:@"Windows Arabic (cp1256)"]; + } else if ( [mysqlEncoding isEqualToString:@"greek"] ) { + [chooseEncodingButton selectItemWithTitle:@"ISO Greek (greek)"]; + } else if ( [mysqlEncoding isEqualToString:@"hebrew"] ) { + [chooseEncodingButton selectItemWithTitle:@"ISO Hebrew (hebrew)"]; + } else if ( [mysqlEncoding isEqualToString:@"latin5"] ) { + [chooseEncodingButton selectItemWithTitle:@"ISO Turkish (latin5)"]; + } else if ( [mysqlEncoding isEqualToString:@"cp1257"] ) { + [chooseEncodingButton selectItemWithTitle:@"Windows Baltic (cp1257)"]; + } else if ( [mysqlEncoding isEqualToString:@"cp1251"] ) { + [chooseEncodingButton selectItemWithTitle:@"Windows Cyrillic (cp1251)"]; + } else if ( [mysqlEncoding isEqualToString:@"big5"] ) { + [chooseEncodingButton selectItemWithTitle:@"Big5 Traditional Chinese (big5)"]; + } else if ( [mysqlEncoding isEqualToString:@"sjis"] ) { + [chooseEncodingButton selectItemWithTitle:@"Shift-JIS Japanese (sjis)"]; + } else if ( [mysqlEncoding isEqualToString:@"ujis"] ) { + [chooseEncodingButton selectItemWithTitle:@"EUC-JP Japanese (ujis)"]; + } else { + NSLog(@"unsupported encoding %@! falling back to utf8.", mysqlEncoding); + [chooseEncodingButton selectItemWithTitle:@"UTF-8 Unicode (utf8)"]; + [self setEncoding:[self getSelectedEncoding]]; + } +} + +- (NSString *)getSelectedEncoding +/* +gets the selected mysql encoding +*/ +{ + NSString *mysqlEncoding; + NSString *encoding = [chooseEncodingButton titleOfSelectedItem]; + +// unicode + if ( [encoding isEqualToString:@"UCS-2 Unicode (ucs2)"] ) { + mysqlEncoding = @"ucs2"; + } else if ( [encoding isEqualToString:@"UTF-8 Unicode (utf8)"] ) { + mysqlEncoding = @"utf8"; +// west european + } else if( [encoding isEqualToString:@"US ASCII (ascii)"] ) { + mysqlEncoding = @"ascii"; + } else if ( [encoding isEqualToString:@"ISO Latin 1 (latin1)"] ) { + mysqlEncoding = @"latin1"; + } else if ( [encoding isEqualToString:@"Mac Roman (macroman)"] ) { + mysqlEncoding = @"macroman"; +// central european + } else if ( [encoding isEqualToString:@"Windows Latin 2 (cp1250)"] ) { + mysqlEncoding = @"cp1250"; + } else if ( [encoding isEqualToString:@"ISO Latin 2 (latin2)"] ) { + mysqlEncoding = @"latin2"; +// south european and middle east + } else if ( [encoding isEqualToString:@"Windows Arabic (cp1256)"] ) { + mysqlEncoding = @"cp1256"; + } else if ( [encoding isEqualToString:@"ISO Greek (greek)"] ) { + mysqlEncoding = @"greek"; + } else if ( [encoding isEqualToString:@"ISO Hebrew (hebrew)"] ) { + mysqlEncoding = @"hebrew"; + } else if ( [encoding isEqualToString:@"ISO Turkish (latin5)"] ) { + mysqlEncoding = @"latin5"; +// baltic + } else if ( [encoding isEqualToString:@"Windows Baltic (cp1257)"] ) { + mysqlEncoding = @"cp1257"; +// cyrillic + } else if ( [encoding isEqualToString:@"Windows Cyrillic (cp1251)"] ) { + mysqlEncoding = @"cp1251"; +// asian + } else if ( [encoding isEqualToString:@"Big5 Traditional Chinese (big5)"] ) { + mysqlEncoding = @"big5"; + } else if ( [encoding isEqualToString:@"Shift-JIS Japanese (sjis)"] ) { + mysqlEncoding = @"sjis"; + } else if ( [encoding isEqualToString:@"EUC-JP Japanese (ujis)"] ) { + mysqlEncoding = @"ujis"; + } else { +// unknown encoding + NSLog(@"error: unknown encoding %@", encoding); + mysqlEncoding = @"utf8"; + } + + return [mysqlEncoding autorelease]; +} + +- (IBAction)chooseEncoding:(id)sender +/* +choose encoding +*/ +{ + // Set encoding + [self setEncoding:[self getSelectedEncoding]]; +} + +- (BOOL)supportsEncoding +/* +returny YES if MySQL server supports choosing connection and table encodings (MySQL 4.1 and newer) +*/ +{ + return [chooseEncodingButton isEnabled]; +} + + +//other methods +- (NSString *)host +/* +returns the host +*/ +{ + return [hostField stringValue]; +} + +- (void)doPerformQueryService:(NSString *)query +/* +passes query to tablesListInstance +*/ +{ + [tableWindow makeKeyAndOrderFront:self]; + [tablesListInstance doPerformQueryService:query]; +} + +- (void)flushPrivileges +/* +flushes the mysql privileges +*/ +{ + [mySQLConnection queryString:@"FLUSH PRIVILEGES"]; + + if ( [[mySQLConnection getLastErrorMessage] isEqualToString:@""] ) { + //flushed privileges without errors + NSBeginAlertSheet(NSLocalizedString(@"Flushed Privileges", @"title of panel when successfully flushed privs"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + NSLocalizedString(@"Succesfully flushed privileges.", @"message of panel when successfully flushed privs")); + } else { + //error while flushing privileges + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Couldn't flush privileges.\nMySQL said: %@", @"message of panel when flushing privs failed"), + [mySQLConnection getLastErrorMessage]]); + } +} + +- (void)openTableOperationsSheet +/* +opens the sheet for table operations (check/analyze/optimize/repair/flush) and performs desired operation +*/ +{ + int code, operation; + CMMCPResult *theResult; + NSDictionary *theRow; + NSString *query; + NSString *operationText; + NSString *messageType; + NSString *messageText; + + [NSApp beginSheet:tableOperationsSheet + modalForWindow:tableWindow modalDelegate:self + didEndSelector:nil contextInfo:nil]; + code = [NSApp runModalForWindow:tableOperationsSheet]; + + [NSApp endSheet:tableOperationsSheet]; + [tableOperationsSheet orderOut:nil]; +NSLog(@"%d",code); + if ( !code ) + return; + + // get operation + operation = [[chooseTableOperationButton selectedItem] tag]; + switch ( operation ) { + case 0: + // check table + query = [NSString stringWithFormat:@"CHECK TABLE `%@`", [self table]]; + break; + case 1: + // analyze table + query = [NSString stringWithFormat:@"ANALYZE TABLE `%@`", [self table]]; + break; + case 2: + // optimize table + query = [NSString stringWithFormat:@"OPTIMIZE TABLE `%@`", [self table]]; + break; + case 3: + // repair table + query = [NSString stringWithFormat:@"REPAIR TABLE `%@`", [self table]]; + break; + case 4: + // flush table + query = [NSString stringWithFormat:@"FLUSH TABLE `%@`", [self table]]; + break; + } + + // perform operation + theResult = [mySQLConnection queryString:query]; + + if ( [[mySQLConnection getLastErrorMessage] isEqualToString:@""] ) { + // no errors + if ( operation == 4 ) { + // flushed -> no return values + operationText = [NSString stringWithString:@"flush"]; + messageType = [NSString stringWithString:@"-"]; + messageText = [NSString stringWithString:@"-"]; + } else { + // other operations -> get return values + theRow = [[theResult fetch2DResultAsType:MCPTypeDictionary] lastObject]; + operationText = [NSString stringWithString:[theRow objectForKey:@"Op"]]; + messageType = [NSString stringWithString:[theRow objectForKey:@"Msg_type"]]; + messageText = [NSString stringWithString:[theRow objectForKey:@"Msg_text"]]; + } + NSBeginAlertSheet(NSLocalizedString(@"Successfully performed table operation", @"title of panel when successfully performed table operation"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Operation: %@\nMsg_type: %@\nMsg_text: %@", @"message of panel when successfully performed table operation"), + operationText, messageType, messageText]); + } else { + // error + NSBeginAlertSheet(NSLocalizedString(@"Error", @"error"), NSLocalizedString(@"OK", @"OK button"), nil, nil, tableWindow, self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"Couldn't perform table operation.\nMySQL said: %@", @"message of panel when table operation failed"), + [mySQLConnection getLastErrorMessage]]); + } +} + +- (IBAction)doTableOperation:(id)sender +/* +closes the sheet and ends modal with 0 if cancel and 1 if ok +*/ +{ + [NSApp stopModalWithCode:[sender tag]]; +} + +- (void)showVariables +/* +shows the mysql variables +*/ +{ + CMMCPResult *theResult; + NSMutableArray *tempResult = [NSMutableArray array]; + int i; + + if ( variables ) { + [variables release]; + variables = nil; + } + //get variables + theResult = [mySQLConnection queryString:@"SHOW VARIABLES"]; + for ( i = 0 ; i < [theResult numOfRows] ; i++ ) { + [theResult dataSeek:i]; + [tempResult addObject:[theResult fetchRowAsDictionary]]; + } + variables = [[NSArray arrayWithArray:tempResult] retain]; + [variablesTableView reloadData]; + //show variables sheet + [NSApp beginSheet:variablesSheet + modalForWindow:tableWindow modalDelegate:self + didEndSelector:nil contextInfo:nil]; + [NSApp runModalForWindow:variablesSheet]; + + [NSApp endSheet:variablesSheet]; + [variablesSheet orderOut:nil]; +} + +- (void)showCreateTable +/* +shows the mysql command used to create the selected table +*/ +{ + id createTableSyntax; + + CMMCPResult *result = [mySQLConnection queryString:[NSString stringWithFormat:@"SHOW CREATE TABLE `%@`", + [self table]]]; + createTableSyntax = [[result fetchRowAsArray] objectAtIndex:1]; + if ( [createTableSyntax isKindOfClass:[NSData class]] ) { + createTableSyntax = [[NSString alloc] initWithData:createTableSyntax encoding:[mySQLConnection encoding]]; + } + + [createTableSyntaxView setString:createTableSyntax]; + [createTableSyntaxView selectAll:self]; + + //show createTableSyntaxSheet + [NSApp beginSheet:createTableSyntaxSheet + modalForWindow:tableWindow modalDelegate:self + didEndSelector:nil contextInfo:nil]; + [NSApp runModalForWindow:createTableSyntaxSheet]; + + [NSApp endSheet:createTableSyntaxSheet]; + [createTableSyntaxSheet orderOut:nil]; +} + +- (void)closeConnection +{ + [mySQLConnection disconnect]; +} + + +//getter methods +- (NSString *)database +/* +returns the currently selected database +*/ +{ + return selectedDatabase; +} + +- (NSString *)table +/* +returns the currently selected table (passing the request to TablesList) +*/ +{ + return [tablesListInstance table]; +} + +- (NSString *)mySQLVersion +/* +returns the mysql version +*/ +{ + return mySQLVersion; +} + +- (NSString *)user +/* +returns the mysql version +*/ +{ + return [userField stringValue]; +} + + +//notification center methods +- (void)willPerformQuery:(NSNotification *)notification +/* +invoked before a query is performed +*/ +{ + [queryProgressBar startAnimation:self]; +} + +- (void)hasPerformedQuery:(NSNotification *)notification +/* +invoked after a query has been performed +*/ +{ + [queryProgressBar stopAnimation:self]; +} + +- (void)applicationWillTerminate:(NSNotification *)notification +/* +invoked when the application will terminate +*/ +{ + [tablesListInstance selectionShouldChangeInTableView:nil]; +} + +- (void)tunnelStatusChanged:(NSNotification *)notification +/* +the status of the tunnel has changed +*/ +{ + NSLog([tunnel status]); +} + +//menu methods +- (IBAction)import:(id)sender +/* +passes the request to the tableDump object +*/ +{ + [tableDumpInstance importFile:[sender tag]]; +} + +- (IBAction)export:(id)sender +/* +passes the request to the tableDump object +*/ +{ + [tableDumpInstance exportFile:[sender tag]]; +} + +- (BOOL)validateMenuItem:(NSMenuItem *)anItem +/* +do menu validation +*/ +{ + switch ( [anItem tag] ) { + case 1: + //import dump + if ( ![self database] ) { + return NO; + } + break; + case 2: + //import CSV + if ( ![self database] || ![self table] ) { + return NO; + } + break; + case 5: + //export dump + if ( ![self database] ) { + return NO; + } + break; + case 6: + //export table content as CSV + if ( ![self database] || ![self table] ) { + return NO; + } + break; + case 7: + //export table content as XML + if ( ![self database] || ![self table] ) { + return NO; + } + break; + case 8: + //export custom result as CSV + return YES; + break; + case 9: + //export custom result as XML + return YES; + break; + case 10: + //export multiple tables as CSV + if ( ![self database] ) { + return NO; + } + break; + case 11: + //export multiple tables as XML + if ( ![self database] ) { + return NO; + } + break; + } + return YES; +} + +- (IBAction)viewStructure:(id)sender +{ + [tableTabView selectTabViewItemAtIndex:0]; +} + +- (IBAction)viewContent:(id)sender +{ + [tableTabView selectTabViewItemAtIndex:1]; +} + +- (IBAction)viewQuery:(id)sender +{ + [tableTabView selectTabViewItemAtIndex:2]; +} + +- (IBAction)viewStatus:(id)sender +{ + [tableTabView selectTabViewItemAtIndex:3]; +} + + +//toolbar methods +- (void)setupToolbar +/* +set up the standard toolbar +*/ +{ + //create a new toolbar instance, and attach it to our document window + NSToolbar *toolbar = [[[NSToolbar alloc] initWithIdentifier:@"TableWindowToolbar"] autorelease]; + + //set up toolbar properties + [toolbar setAllowsUserCustomization: YES]; + [toolbar setAutosavesConfiguration: YES]; + [toolbar setDisplayMode:NSToolbarDisplayModeIconAndLabel]; + + //set ourself as the delegate + [toolbar setDelegate:self]; + + //attach the toolbar to the document window + [tableWindow setToolbar:toolbar]; +} + +- (NSToolbarItem *)toolbar:(NSToolbar *)toolbar itemForItemIdentifier:(NSString *)itemIdentifier willBeInsertedIntoToolbar:(BOOL)flag +/* +toolbar delegate method +*/ +{ + NSToolbarItem *toolbarItem = [[[NSToolbarItem alloc] initWithItemIdentifier:itemIdentifier] autorelease]; + + if ([itemIdentifier isEqualToString:@"ToggleConsoleIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setPaletteLabel:NSLocalizedString(@"Show/Hide Console", @"toolbar item for show/hide console")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Show or hide the console which shows all MySQL commands performed by CocoaMySQL", @"tooltip for toolbar item for show/hide console")]; + if ( [self consoleIsOpened] ) { + [toolbarItem setLabel:NSLocalizedString(@"Hide Console", @"toolbar item for hide console")]; + [toolbarItem setImage:[NSImage imageNamed:@"hideconsole"]]; + } else { + [toolbarItem setLabel:NSLocalizedString(@"Show Console", @"toolbar item for showconsole")]; + [toolbarItem setImage:[NSImage imageNamed:@"showconsole"]]; + } + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(toggleConsole)]; + } else if ([itemIdentifier isEqualToString:@"ClearConsoleIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setLabel:NSLocalizedString(@"Clear Console", @"toolbar item for clear console")]; + [toolbarItem setPaletteLabel:NSLocalizedString(@"Clear Console", @"toolbar item for clear console")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Clear the console which shows all MySQL commands performed by CocoaMySQL", @"tooltip for toolbar item for clear console")]; + [toolbarItem setImage:[NSImage imageNamed:@"clearconsole"]]; + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(clearConsole)]; + } else if ([itemIdentifier isEqualToString:@"FlushPrivilegesIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setLabel:NSLocalizedString(@"Flush Privileges", @"toolbar item for flush privileges")]; + [toolbarItem setPaletteLabel:NSLocalizedString(@"Flush Privileges", @"toolbar item for flush privileges")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Reload the MySQL privileges saved in the mysql database", @"tooltip for toolbar item for flush privileges")]; + [toolbarItem setImage:[NSImage imageNamed:@"flushprivileges"]]; + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(flushPrivileges)]; + } else if ([itemIdentifier isEqualToString:@"OptimizeTableIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setLabel:NSLocalizedString(@"Table Operations", @"toolbar item for perform table operations")]; + [toolbarItem setPaletteLabel:NSLocalizedString(@"Table Operations", @"toolbar item for perform table operations")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Perform table operations for the selected table", @"tooltip for toolbar item for perform table operations")]; + [toolbarItem setImage:[NSImage imageNamed:@"optimizetable"]]; + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(openTableOperationsSheet)]; + } else if ([itemIdentifier isEqualToString:@"ShowVariablesIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setLabel:NSLocalizedString(@"Show Variables", @"toolbar item for show variables")]; + [toolbarItem setPaletteLabel:NSLocalizedString(@"Show Variables", @"toolbar item for show variables")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Show the MySQL Variables", @"tooltip for toolbar item for show variables")]; + [toolbarItem setImage:[NSImage imageNamed:@"showvariables"]]; + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(showVariables)]; + } else if ([itemIdentifier isEqualToString:@"ShowCreateTableIdentifier"]) { + //set the text label to be displayed in the toolbar and customization palette + [toolbarItem setLabel:NSLocalizedString(@"Create Table Syntax", @"toolbar item for create table syntax")]; + [toolbarItem setPaletteLabel:NSLocalizedString(@"Create Table Syntax", @"toolbar item for create table syntax")]; + //set up tooltip and image + [toolbarItem setToolTip:NSLocalizedString(@"Show the MySQL command used to create the selected table", @"tooltip for toolbar item for create table syntax")]; + [toolbarItem setImage:[NSImage imageNamed:@"createtablesyntax"]]; + //set up the target action + [toolbarItem setTarget:self]; + [toolbarItem setAction:@selector(showCreateTable)]; + } else { + //itemIdentifier refered to a toolbar item that is not provided or supported by us or cocoa + toolbarItem = nil; + } + + return toolbarItem; +} + +- (NSArray *)toolbarAllowedItemIdentifiers:(NSToolbar*)toolbar +/* +toolbar delegate method +*/ +{ + return [NSArray arrayWithObjects:@"ToggleConsoleIdentifier", @"ClearConsoleIdentifier", @"ShowVariablesIdentifier", @"FlushPrivilegesIdentifier", @"OptimizeTableIdentifier", @"ShowCreateTableIdentifier", NSToolbarCustomizeToolbarItemIdentifier, NSToolbarFlexibleSpaceItemIdentifier, NSToolbarSpaceItemIdentifier, NSToolbarSeparatorItemIdentifier, nil]; +} + +- (NSArray *)toolbarDefaultItemIdentifiers:(NSToolbar*)toolbar +/* +toolbar delegate method +*/ +{ + return [NSArray arrayWithObjects:@"ToggleConsoleIdentifier", @"ClearConsoleIdentifier", NSToolbarSeparatorItemIdentifier, @"ShowVariablesIdentifier", @"FlushPrivilegesIdentifier", NSToolbarSeparatorItemIdentifier, @"OptimizeTableIdentifier", @"ShowCreateTableIdentifier", nil]; +} + +- (BOOL)validateToolbarItem:(NSToolbarItem *)toolbarItem; +/* +validates the toolbar items +*/ +{ + if ( [[toolbarItem itemIdentifier] isEqualToString:@"OptimizeTableIdentifier"] ) { + if ( ![self table] ) + return NO; + } else if ( [[toolbarItem itemIdentifier] isEqualToString:@"ShowCreateTableIdentifier"] ) { + if ( ![self table] ) + return NO; + } else if ( [[toolbarItem itemIdentifier] isEqualToString:@"ToggleConsoleIdentifier"] ) { + if ( [self consoleIsOpened] ) { + [toolbarItem setLabel:@"Hide Console"]; + [toolbarItem setImage:[NSImage imageNamed:@"hideconsole"]]; + } else { + [toolbarItem setLabel:@"Show Console"]; + [toolbarItem setImage:[NSImage imageNamed:@"showconsole"]]; + } + } + + return YES; +} + + +//NSDocument methods +- (NSString *)windowNibName +/* +returns the name of the nib file +*/ +{ + return @"DBView"; +} + +- (void)windowControllerDidLoadNib:(NSWindowController *) aController +/* +code that need to be executed once the windowController has loaded the document's window +sets upt the interface (small fonts) +*/ +{ + [aController setShouldCascadeWindows:NO]; + [super windowControllerDidLoadNib:aController]; + + NSEnumerator *theCols = [[variablesTableView tableColumns] objectEnumerator]; + NSTableColumn *theCol; + +// [tableWindow makeKeyAndOrderFront:self]; + + prefs = [[NSUserDefaults standardUserDefaults] retain]; + if ( [prefs objectForKey:@"favorites"] != nil ) { + favorites = [[NSArray alloc] initWithArray:[prefs objectForKey:@"favorites"]]; + } else { + favorites = [[NSArray array] retain]; + } + selectedFavorite = [[NSString alloc] initWithString:NSLocalizedString(@"Custom", @"menu item for custom connection")]; + + //register for notifications + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(willPerformQuery:) + name:@"SMySQLQueryWillBePerformed" object:nil]; + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(hasPerformedQuery:) + name:@"SMySQLQueryHasBeenPerformed" object:nil]; + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(applicationWillTerminate:) + name:@"NSApplicationWillTerminateNotification" object:nil]; + [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(tunnelStatusChanged:) + name: @"STMStatusChanged" object: nil]; + + //set up interface + if ( [prefs boolForKey:@"useMonospacedFonts"] ) { + [consoleTextView setFont:[NSFont fontWithName:@"Monaco" size:[NSFont smallSystemFontSize]]]; + [createTableSyntaxView setFont:[NSFont fontWithName:@"Monaco" size:[NSFont smallSystemFontSize]]]; + while ( (theCol = [theCols nextObject]) ) { + [[theCol dataCell] setFont:[NSFont fontWithName:@"Monaco" size:10]]; + } + } else { + [consoleTextView setFont:[NSFont systemFontOfSize:[NSFont smallSystemFontSize]]]; + [createTableSyntaxView setFont:[NSFont systemFontOfSize:[NSFont smallSystemFontSize]]]; + while ( (theCol = [theCols nextObject]) ) { + [[theCol dataCell] setFont:[NSFont systemFontOfSize:[NSFont smallSystemFontSize]]]; + } + } + [consoleDrawer setContentSize:NSMakeSize(110,110)]; + + //set up toolbar + [self setupToolbar]; + + + +//tunnel test +/* +NSDictionary *args = [NSDictionary dictionaryWithObjectsAndKeys:@"CocoaMySQL Tunnel",@"connName", + @"xy",@"connUser", + @"textor.ch",@"connHost", + [NSNumber numberWithBool:YES],@"connAuth", + [NSArray arrayWithObject:[NSDictionary dictionaryWithObjectsAndKeys:@"8888",@"port", + @"textor.ch",@"host", + @"3306",@"hostport", + nil]],@"tunnelsLocal", + nil]; +tunnel = [[SSHTunnel alloc] initWithDictionary:args]; +[tunnel startTunnel]; +*/ +//sleep(3); +//[tunnel startTunnelWithArguments:args]; +//end tunnel test + + + [self connectToDB:nil]; +} + +- (void)windowWillClose:(NSNotification *)aNotification +{ + [self closeConnection]; + + [[NSNotificationCenter defaultCenter] removeObserver:self]; +} + + +//NSWindow delegate methods +- (BOOL)windowShouldClose:(id)sender +/* +invoked when the document window should close +*/ +{ + if ( ![tablesListInstance selectionShouldChangeInTableView:nil] ) { + return NO; + } else { + return YES; + } + +} + + +//SMySQL delegate methods +- (void)willQueryString:(NSString *)query +/* +invoked when framework will perform a query +*/ +{ + NSString *currentTime = [[NSDate date] descriptionWithCalendarFormat:@"%H:%M:%S" timeZone:nil locale:nil]; + + [self showMessageInConsole:[NSString stringWithFormat:@"/* MySQL %@ */ %@;\n", currentTime, query]]; +} + +- (void)queryGaveError:(NSString *)error +/* +invoked when query gave an error +*/ +{ + NSString *currentTime = [[NSDate date] descriptionWithCalendarFormat:@"%H:%M:%S" timeZone:nil locale:nil]; + + [self showErrorInConsole:[NSString stringWithFormat:@"/* ERROR %@ */ %@;\n", currentTime, error]]; +} + + +//splitView delegate methods +- (BOOL)splitView:(NSSplitView *)sender canCollapseSubview:(NSView *)subview +/* +tells the splitView that it can collapse views +*/ +{ + return YES; +} + +- (float)splitView:(NSSplitView *)sender constrainMaxCoordinate:(float)proposedMax ofSubviewAt:(int)offset +/* +defines max position of splitView +*/ +{ + return proposedMax - 600; +} + +- (float)splitView:(NSSplitView *)sender constrainMinCoordinate:(float)proposedMin ofSubviewAt:(int)offset +/* +defines min position of splitView +*/ +{ + return proposedMin + 160; +} + + +//tableView datasource methods +- (int)numberOfRowsInTableView:(NSTableView *)aTableView +{ + return [variables count]; +} + +- (id)tableView:(NSTableView *)aTableView + objectValueForTableColumn:(NSTableColumn *)aTableColumn + row:(int)rowIndex +{ + id theValue; + + theValue = [[variables objectAtIndex:rowIndex] objectForKey:[aTableColumn identifier]]; + + if ( [theValue isKindOfClass:[NSData class]] ) { + theValue = [[NSString alloc] initWithData:theValue encoding:[mySQLConnection encoding]]; + } + + return theValue; +} + + +//for freeing up memory +- (void)dealloc +{ +// NSLog(@"TableDocument dealloc"); + + [mySQLConnection release]; + [favorites release]; + if (nil != variables ) + { + [variables release]; + } + [selectedDatabase release]; + [selectedFavorite release]; + [mySQLVersion release]; + [prefs release]; + + [super dealloc]; +} + +@end |