diff options
Diffstat (limited to 'Source/SPTableCopy.m')
| -rw-r--r-- | Source/SPTableCopy.m | 151 |
1 files changed, 151 insertions, 0 deletions
diff --git a/Source/SPTableCopy.m b/Source/SPTableCopy.m new file mode 100644 index 00000000..2e8c2461 --- /dev/null +++ b/Source/SPTableCopy.m @@ -0,0 +1,151 @@ +// +// $Id: $ +// +// SPTableCopy.m +// sequel-pro +// +// Created by David Rekowski on Apr 13, 2010 +// +// 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://code.google.com/p/sequel-pro/> + +#import <MCPKit/MCPConnection.h> +#import "SPAlertSheets.h" +#import "SPStringAdditions.h" +#import "SPTableCopy.h" +#import "Sequel-Pro.pch" + +@implementation SPTableCopy + +@synthesize connection; +@synthesize parent; + +- (NSObject *)getTableWindow { + return [NSApp mainWindow]; +} + +- (NSString *)getCreateTableStatementFor: (NSString *)tableName inDB: (NSString *)sourceDB { + NSString *showCreateTableStatment = [NSString stringWithFormat:@"SHOW CREATE TABLE %@.%@", + [sourceDB backtickQuotedString], + [tableName backtickQuotedString] + ]; + NSLog(showCreateTableStatment); + MCPResult *theResult = [connection queryString:showCreateTableStatment]; + + if ([connection queryErrored]) { + SPBeginAlertSheet(NSLocalizedString(@"Failed to show create table statement", @"show create table error message"), + NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to retrieve the create table statement for a table.\n\nMySQL said: %@", + @"show create table error informative message"), + [connection getLastErrorMessage]]); + } + NSLog(@"%i", [theResult numOfRows]); + if ([theResult numOfRows] != 0) { + NSString *createTableStatment = [[theResult fetchRowAsArray] objectAtIndex:1]; + return createTableStatment; + } +} + +- (BOOL)copyTable:(NSString *)tableName from: (NSString *)sourceDB to: (NSString *)targetDB { + + NSMutableString *createTableStatement = [[NSMutableString alloc] initWithString:[self getCreateTableStatementFor:tableName inDB:sourceDB]]; + + // adding the target DB name and the separator dot after "CREATE TABLE ". + [createTableStatement insertString:@"." atIndex:13]; + [createTableStatement insertString:[targetDB backtickQuotedString] atIndex:13]; + /* + // this only works with MySQL >= 4.1 + NSString *copyStatement = [NSString stringWithFormat:@"CREATE TABLE %@.%@ LIKE %@.%@", + [targetDB backtickQuotedString], + [tableName backtickQuotedString], + [sourceDB backtickQuotedString], + [tableName backtickQuotedString] + ]; + DLog(@"Copying table %@ from %@ to %@", tableName, sourceDB, targetDB); + DLog(@"Copying table: %@", copyStatement); + [connection queryString:copyStatement]; + */ + + [connection queryString:createTableStatement]; + [createTableStatement release]; + + + if ([connection queryErrored]) { + SPBeginAlertSheet(NSLocalizedString(@"Failed to copy table", @"copy table error message"), + NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to copy a table.\n\nMySQL said: %@", + @"copy table error informative message"), + [connection getLastErrorMessage]]); + return NO; + } + return YES; +} + +- (BOOL)copyTable:(NSString *)tableName from: (NSString *)sourceDB to: (NSString *)targetDB withContent:(BOOL)copyWithContent{ + // copy the table structure + BOOL structureCopyResult = [self copyTable:tableName from:sourceDB to:targetDB]; + + // optionally copy the table data using an insert select + if (copyWithContent == YES) { + NSString *copyDataStatement = [NSString stringWithFormat:@"INSERT INTO %@.%@ SELECT * FROM %@.%@", + [targetDB backtickQuotedString], + [tableName backtickQuotedString], + [sourceDB backtickQuotedString], + [tableName backtickQuotedString] + ]; + DLog(@"Copying table data: %@", copyDataStatement); + [connection queryString:copyDataStatement]; + + if ([connection queryErrored]) { + SPBeginAlertSheet(NSLocalizedString(@"Failed to copy table data", @"copy table data error message"), + NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to copy a table's data.\n\nMySQL said: %@", + @"copy table data error informative message"), + [connection getLastErrorMessage]]); + return NO; + } + return YES; + + } + return structureCopyResult; +} + +- (BOOL)moveTable:(NSString *)tableName from: (NSString *)sourceDB to: (NSString *)targetDB { + + NSString *moveStatement = [NSString stringWithFormat:@"RENAME TABLE %@.%@ TO %@.%@", + [sourceDB backtickQuotedString], + [tableName backtickQuotedString], + [targetDB backtickQuotedString], + [tableName backtickQuotedString] + ]; + // moving the table + DLog(@"Moving table %@ from %@ to %@", tableName, sourceDB, targetDB); + DLog(@"Moving table: %@", moveStatement); + [connection queryString:moveStatement]; + + if ([connection queryErrored]) { + SPBeginAlertSheet(NSLocalizedString(@"Failed to move table", @"move table error message"), + NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, + [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to move a table.\n\nMySQL said: %@", + @"move table error informative message"), + [connection getLastErrorMessage]]); + return NO; + } + return YES; +} + + +@end |