1 files changed, 72 insertions, 0 deletions
diff --git a/Source/SPUserMO.m b/Source/SPUserMO.m
new file mode 100644
index 00000000..05d699ae
--- /dev/null
+++ b/Source/SPUserMO.m
@@ -0,0 +1,72 @@
+//
+//  $Id: SPUserMO.m 856 2009-06-12 05:31:39Z mltownsend $
+//
+//  SPUserMO.m
+//  sequel-pro
+//
+//  Created by Mark Townsend on Jan 01, 2009
+//
+//  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 "SPUserMO.h"
+
+@implementation NSManagedObject (CoreDataGeneratedAccessors)
+
+@dynamic user, host;
+@dynamic parent;
+@dynamic children;
+
+- (NSString *)displayName
+{
+	if ([self valueForKey:@"parent"] == nil) {
+		return self.user;
+	} else {
+		return self.host;
+	}
+}
+
+- (void)setDisplayName:(NSString *)value
+{
+	[self setValue:value forKey:@"host"];
+}
+
+- (void)addChildrenObject:(NSManagedObject *)value 
+{    
+    NSSet *changedObjects = [[NSSet alloc] initWithObjects:&value count:1];
+
+    [self willChangeValueForKey:@"children" withSetMutation:NSKeyValueUnionSetMutation usingObjects:changedObjects];
+    [[self primitiveValueForKey:@"children"] addObject:value];
+    [self didChangeValueForKey:@"children" withSetMutation:NSKeyValueUnionSetMutation usingObjects:changedObjects];
+    
+    [changedObjects release];
+	value.user = self.user;
+}
+
+- (void)removeChildrenObject:(NSManagedObject *)value 
+{
+    NSSet *changedObjects = [[NSSet alloc] initWithObjects:&value count:1];
+    
+    [self willChangeValueForKey:@"children" withSetMutation:NSKeyValueMinusSetMutation usingObjects:changedObjects];
+    [[self primitiveValueForKey:@"children"] removeObject:value];
+    [self didChangeValueForKey:@"children" withSetMutation:NSKeyValueMinusSetMutation usingObjects:changedObjects];
+    
+    [changedObjects release];
+}
+@end
+
+
+