From f36685ae1bee7b9274de27414ab8b64dbc57770f Mon Sep 17 00:00:00 2001
From: drx777 <david.rekowski@gmx.de>
Date: Wed, 28 Apr 2010 17:11:41 +0000
Subject: This changeset implements renaming and duplicating databases on a
 server. Details as follows: * altered MCPConnection listTablesFromDB:like: to
 return NSArray * altered MCPConnection listFieldsFromTable:like: to use
 backtick quoted   strings for table and fieldnames * added MCPConnection
 listTablesFromDB for a complete table list * added SPStringAdditions.h to
 various files to prevent warnings * added sheets for duplicate/rename DB in
 DBView.xib * added duplicate/rename menu items to MainMenu.xib * added
 outlets in TableDocument: databaseNewSheet databaseRenameSheet  
 databaseCopyNameField databaseRenameNameField copyOnlyStructureButton  
 copyDatabaseButton renameDatabaseButton * added methods in TableDocument:
 getConnection, copyDatabase, renameDatabase,   _copyDatabase, _renameDatabase
 * added OCMock Framework for object mocking in tests * added group
 Others/DatabaseActions

---
 Source/SPDatabaseCopy.m | 113 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 Source/SPDatabaseCopy.m

(limited to 'Source/SPDatabaseCopy.m')

diff --git a/Source/SPDatabaseCopy.m b/Source/SPDatabaseCopy.m
new file mode 100644
index 00000000..98c97587
--- /dev/null
+++ b/Source/SPDatabaseCopy.m
@@ -0,0 +1,113 @@
+//
+//  $Id: $
+//
+//  SPDatabaseCopy.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 <MCPKit/MCPResult.h>
+#import "SPAlertSheets.h"
+#import "SPStringAdditions.h"
+#import "SPDatabaseCopy.h"
+#import "SPTableCopy.h"
+#import "SPDatabaseInfo.h"
+#import "Sequel-Pro.pch"
+
+@implementation SPDatabaseCopy
+
+@synthesize connection;
+@synthesize dbInfo;
+@synthesize parent;
+
+- (SPDatabaseInfo *)getDBInfoObject {
+	if (dbInfo != nil) {
+		return dbInfo;
+	} else {
+		dbInfo = [[SPDatabaseInfo alloc] init];
+		[dbInfo setConnection:[self connection]];
+		[dbInfo setParent:[self parent]];
+	}
+	return dbInfo;
+}
+
+- (NSObject *)getTableWindow {
+	return [NSApp mainWindow];
+}
+
+- (BOOL)copyDatabaseFrom: (NSString *)sourceDatabaseName to: (NSString *)targetDatabaseName withContent:(BOOL)copyWithContent {
+
+	SPDatabaseInfo *databaseInfo = [self getDBInfoObject];
+	// check, whether the source database exists and the target database doesn't.
+	NSArray *tables = [NSArray array]; 
+	BOOL sourceExists = [databaseInfo databaseExists:sourceDatabaseName];
+	BOOL targetExists = [databaseInfo databaseExists:targetDatabaseName];
+	if (sourceExists && !targetExists) {
+		// retrieve the list of tables/views/funcs/triggers from the source database
+
+		tables = [connection listTablesFromDB:sourceDatabaseName];
+	} else {
+		SPBeginAlertSheet(NSLocalizedString(@"Cannot create existing database", @"create database exists error message"), 
+						  NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, 
+						  [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to create the target database.\n\nDatabase %@ already exists.", 
+																	   @"create database error informative message"), 
+						   targetDatabaseName]);
+		
+		return NO;
+	}
+	DLog(@"list of found tables of source db: %@", tables);
+	
+	[self createDatabase:targetDatabaseName];
+	SPTableCopy *dbActionTableCopy = [[SPTableCopy alloc] init];
+	[dbActionTableCopy setConnection:connection];
+	
+	for (NSString *currentTable in tables) {
+		if ([dbActionTableCopy copyTable:currentTable 						
+									from:sourceDatabaseName
+									  to:targetDatabaseName
+							 withContent:copyWithContent]) {
+		}
+	}
+}
+
+- (BOOL) createDatabase: (NSString *)newDatabaseName {
+	NSString *createStatement = [NSString stringWithFormat:@"CREATE DATABASE %@", 
+								 [newDatabaseName backtickQuotedString]];
+	[connection queryString:createStatement];	
+
+	if ([connection queryErrored]) {
+		SPBeginAlertSheet(NSLocalizedString(@"Failed to create database", @"create database error message"), 
+						  NSLocalizedString(@"OK", @"OK button"), nil, nil, [self getTableWindow], self, nil, nil, nil, 
+						  [NSString stringWithFormat:NSLocalizedString(@"An error occured while trying to create the target database.\n\nMySQL said: %@", 
+																	   @"create database error informative message"), 
+						   [connection getLastErrorMessage]]);
+		return NO;
+	}
+	return YES;
+	
+	
+}
+
+- (void)dealloc {
+	[dbInfo dealloc];
+}
+
+
+@end
\ No newline at end of file
-- 
cgit v1.2.3