From 48f80fe6342b14913ffbebf5449895ff9d5e9af8 Mon Sep 17 00:00:00 2001
From: rowanbeentje <rowan@beent.je>
Date: Thu, 18 Mar 2010 01:29:17 +0000
Subject:  - Fix a couple of calls which crashed if the window had already been
 closed, fixing window close on query startup or after queries; this fixes
 http://log.sequelpro.com/view/9  - Fix an exception caused in the
 NavigatorController if the dbstructure has not been fetched yet

---
 Source/CustomQuery.m            |  4 ++++
 Source/SPConnectionController.h |  2 ++
 Source/SPConnectionController.m | 16 ++++++++++++++++
 Source/SPNavigatorController.m  |  6 ++++--
 Source/TableDocument.m          |  1 +
 5 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/Source/CustomQuery.m b/Source/CustomQuery.m
index 35338463..d0553f27 100644
--- a/Source/CustomQuery.m
+++ b/Source/CustomQuery.m
@@ -2806,6 +2806,10 @@
  */
 - (void)historyItemsHaveBeenUpdated:(id)manager
 {
+
+	// Abort if the connection has been closed already - sign of a closed window
+	if (![mySQLConnection isConnected]) return;
+
 	// Refresh history popup menu
 	NSMenu* historyMenu = [queryHistoryButton menu];
 	while([queryHistoryButton numberOfItems] > 7)
diff --git a/Source/SPConnectionController.h b/Source/SPConnectionController.h
index f2c6ba9a..b1cc541e 100644
--- a/Source/SPConnectionController.h
+++ b/Source/SPConnectionController.h
@@ -58,6 +58,7 @@
 	SPSSHTunnel *sshTunnel;
 	MCPConnection *mySQLConnection;
 	BOOL automaticFavoriteSelection;
+	BOOL cancellingConnection;
 
 	NSInteger previousType;
 	NSInteger type;
@@ -131,6 +132,7 @@
 - (void)initiateSSHTunnelConnection;
 - (void)sshTunnelCallback:(SPSSHTunnel *)theTunnel;
 - (void)initiateMySQLConnection;
+- (void)cancelConnection;
 - (void)failConnectionWithTitle:(NSString *)theTitle errorMessage:(NSString *)theErrorMessage detail:(NSString *)errorDetail;
 - (void)errorSheetDidEnd:(NSWindow *)sheet returnCode:(NSInteger)returnCode contextInfo:(NSString *)contextInfo;
 - (void)addConnectionToDocument;
diff --git a/Source/SPConnectionController.m b/Source/SPConnectionController.m
index 2e48106c..869bd35c 100644
--- a/Source/SPConnectionController.m
+++ b/Source/SPConnectionController.m
@@ -68,6 +68,7 @@
 		connectionSSHKeychainItemAccount = nil;
 		mySQLConnection = nil;
 		sshTunnel = nil;
+		cancellingConnection = NO;
 
 		// Load the connection nib
 		[NSBundle loadNibNamed:@"ConnectionView" owner:self];
@@ -165,6 +166,7 @@
 	if (![self checkHost]) return;
 	
 	// Basic details have validated - start the connection process animating
+	cancellingConnection = NO;
 	[addToFavoritesButton setHidden:YES];
 	[addToFavoritesButton display];
 	[helpButton setHidden:YES];
@@ -250,6 +252,19 @@
 	[sshTunnel connect];
 }
 
+/*
+ * Cancel connection.
+ * Currently only cleans up the SSH connection (MySQL connection isn't threaded)
+ */
+- (void)cancelConnection
+{
+	if (!sshTunnel) return;
+	cancellingConnection = YES;
+	[sshTunnel disconnect];
+	[sshTunnel release];
+	sshTunnel = nil;
+}
+
 /*
  * A callback function for the SSH Tunnel setup process - will be called on a connection
  * state change, allowing connection to fail or proceed as appropriate.  If successful,
@@ -257,6 +272,7 @@
  */
 - (void)sshTunnelCallback:(SPSSHTunnel *)theTunnel
 {
+	if (cancellingConnection) return;
 	NSInteger newState = [theTunnel state];
 
 	if (newState == PROXY_STATE_IDLE) {
diff --git a/Source/SPNavigatorController.m b/Source/SPNavigatorController.m
index ec3458ad..107019af 100644
--- a/Source/SPNavigatorController.m
+++ b/Source/SPNavigatorController.m
@@ -90,8 +90,10 @@ static SPNavigatorController *sharedNavigatorController = nil;
 	if ([[[NSDocumentController sharedDocumentController] documents] count]) {
 		for(id doc in [[NSDocumentController sharedDocumentController] documents]) {
 			NSString *connectionName = [NSString stringWithFormat:@"%@@%@", [doc user], [doc host]];
-			if(![schemaData objectForKey:connectionName])
-				[schemaData setObject:[[doc valueForKeyPath:@"mySQLConnection"] getDbStructure] forKey:connectionName];
+			if(![schemaData objectForKey:connectionName]) {
+				NSDictionary *dbStructure = [[doc valueForKeyPath:@"mySQLConnection"] getDbStructure];
+				if (dbStructure) [schemaData setObject:dbStructure forKey:connectionName];
+			}
 		}
 	}
 
diff --git a/Source/TableDocument.m b/Source/TableDocument.m
index a262f09d..3a7fd59b 100644
--- a/Source/TableDocument.m
+++ b/Source/TableDocument.m
@@ -3591,6 +3591,7 @@
 {
 	[mySQLConnection setDelegate:nil];
 	if (_isConnected) [self closeConnection];
+	else [connectionController cancelConnection];
 	if ([[[SPQueryController sharedQueryController] window] isVisible]) [self toggleConsole:self];
 	if ([[[SPNavigatorController sharedNavigatorController] window] isVisible]) [self toggleNavigator:self];
 	[createTableSyntaxWindow orderOut:nil];
-- 
cgit v1.2.3