Add support for saving the state of the connection favorites outline view.

1 files changed, 22 insertions, 0 deletions

diff --git a/Source/SPConnectionControllerDelegate.m b/Source/SPConnectionControllerDelegate.m
index 389bd46c..bc6800c3 100644
--- a/Source/SPConnectionControllerDelegate.m
+++ b/Source/SPConnectionControllerDelegate.m
@@ -47,6 +47,8 @@ static NSString *SPDatabaseImage = @"database-small";
 
 - (NSString *)_stripInvalidCharactersFromString:(NSString *)subject;
 
+- (void)_setNodeIsExpanded:(BOOL)expanded fromNotification:(NSNotification *)notification;
+
 @end
 
 @implementation SPConnectionController (SPConnectionControllerDelegate)
@@ -154,6 +156,16 @@ static NSString *SPDatabaseImage = @"database-small";
 	return ([[item parentNode] parentNode] != nil);
 }
 
+- (void)outlineViewItemDidCollapse:(NSNotification *)notification
+{	
+	[self _setNodeIsExpanded:NO fromNotification:notification];
+}
+
+- (void)outlineViewItemDidExpand:(NSNotification *)notification
+{
+	[self _setNodeIsExpanded:YES fromNotification:notification];
+}
+
 #pragma mark -
 #pragma mark Outline view drag & drop
 
@@ -557,4 +569,14 @@ static NSString *SPDatabaseImage = @"database-small";
 	}
 }
 
+#pragma mark -
+#pragma mark Private API
+
+- (void)_setNodeIsExpanded:(BOOL)expanded fromNotification:(NSNotification *)notification
+{
+	SPGroupNode *node = [[[notification userInfo] valueForKey:@"NSObject"] representedObject];
+	
+	[node setNodeIsExpanded:expanded];	
+}
+
 @end