aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorstuconnolly <stuart02@gmail.com>2009-11-17 01:29:01 +0000
committerstuconnolly <stuart02@gmail.com>2009-11-17 01:29:01 +0000
commit79a8715c6860dd51bf43daf0210d0c84684632fa (patch)
tree1f85487ca73ca0b20198d1919981f56374606b42
parent4e3acd5942e775908be9222650eb0fc8cc7b1e10 (diff)
downloadsequelpro-79a8715c6860dd51bf43daf0210d0c84684632fa.tar.gz
sequelpro-79a8715c6860dd51bf43daf0210d0c84684632fa.tar.bz2
sequelpro-79a8715c6860dd51bf43daf0210d0c84684632fa.zip
Various user manager changes. Fixes issue #437.
-rw-r--r--Source/SPUserMO.m3
-rw-r--r--Source/SPUserManager.h8
-rw-r--r--Source/SPUserManager.m86
3 files changed, 52 insertions, 45 deletions
diff --git a/Source/SPUserMO.m b/Source/SPUserMO.m
index 8b6af9b2..386e4831 100644
--- a/Source/SPUserMO.m
+++ b/Source/SPUserMO.m
@@ -26,7 +26,8 @@
@implementation NSManagedObject (CoreDataGeneratedAccessors)
-@dynamic user, host;
+@dynamic user;
+@dynamic host;
@dynamic parent;
@dynamic children;
diff --git a/Source/SPUserManager.h b/Source/SPUserManager.h
index 7eb0b8e0..48c8ad3f 100644
--- a/Source/SPUserManager.h
+++ b/Source/SPUserManager.h
@@ -35,9 +35,9 @@
BOOL isInitializing;
- MCPConnection* mySqlConnection;
+ MCPConnection *mySqlConnection;
- IBOutlet NSOutlineView* outlineView;
+ IBOutlet NSOutlineView *outlineView;
IBOutlet NSTabView *tabView;
IBOutlet NSTreeController *treeController;
IBOutlet NSMutableDictionary *privsSupportedByServer;
@@ -49,7 +49,7 @@
@property (nonatomic, retain, readonly) NSManagedObjectContext *managedObjectContext;
@property (nonatomic, retain) NSMutableDictionary *privsSupportedByServer;
-// Add/Remove Users
+// Add/Remove users
- (IBAction)addUser:(id)sender;
- (IBAction)removeUser:(id)sender;
- (IBAction)addHost:(id)sender;
@@ -62,7 +62,7 @@
- (IBAction)checkAllPrivileges:(id)sender;
- (IBAction)uncheckAllPrivileges:(id)sender;
-// Core Data Notifications
+// Core Data notifications
- (void)contextDidSave:(NSNotification *)notification;
- (BOOL)insertUsers:(NSArray *)insertedUsers;
- (BOOL)deleteUsers:(NSArray *)deletedUsers;
diff --git a/Source/SPUserManager.m b/Source/SPUserManager.m
index ebca5ff9..40ec31e4 100644
--- a/Source/SPUserManager.m
+++ b/Source/SPUserManager.m
@@ -33,7 +33,7 @@
#define COLUMNIDNAME @"NameColumn"
-@interface SPUserManager (PrivateMethods)
+@interface SPUserManager (PrivateAPI)
- (void)_initializeTree:(NSArray *)items;
- (void)_initializeUsers;
@@ -54,11 +54,10 @@
@synthesize managedObjectModel;
@synthesize persistentStoreCoordinator;
-
-(id)init
{
- if ((self = [super initWithWindowNibName:@"UserManagerView"]))
- {
+ if ((self = [super initWithWindowNibName:@"UserManagerView"])) {
+
// When reading privileges from the database, they are converted automatically to a
// lowercase key used in the user privileges stores, from which a GRANT syntax
// is derived automatically. While most keys can be automatically converted without
@@ -78,6 +77,7 @@
object:nil];
}
+
return self;
}
@@ -99,8 +99,10 @@
[super dealloc];
}
-// UI specific items to set up when the window loads. This is different
-// than awakeFromNib as it's only called once.
+/**
+ * UI specific items to set up when the window loads. This is different than awakeFromNib
+ * as it's only called once.
+ */
-(void)windowDidLoad
{
[tabView selectTabViewItemAtIndex:0];
@@ -115,12 +117,12 @@
[super windowDidLoad];
}
+/**
+ * This method reads in the users from the mysql.user table of the current
+ * connection. Then uses this information to initialize the NSOutlineView.
+ */
- (void)_initializeUsers
{
- /**
- * This method reads in the users from the mysql.user table of the current
- * connection. Then uses this information to initialize the NSOutlineView.
- */
isInitializing = TRUE; // Don't want to do some of the notifications if initializing
NSMutableString *privKey;
NSArray *privRow;
@@ -132,16 +134,17 @@
// Select users from the mysql.user table
MCPResult *result = [self.mySqlConnection queryString:@"SELECT * FROM `mysql`.`user` ORDER BY `user`"];
int rows = [result numOfRows];
- if (rows > 0)
- {
+
+ if (rows > 0) {
// Go to the beginning
[result dataSeek:0];
}
- for(int i = 0; i < rows; i++)
+ for (int i = 0; i < rows; i++)
{
[resultAsArray addObject:[result fetchRowAsDictionary]];
}
+
[usersResultArray addObjectsFromArray:resultAsArray];
[self _initializeTree:usersResultArray];
@@ -225,7 +228,9 @@
[outlineView reloadData];
}
-// Set NSManagedObject with values from the passed in dictionary
+/**
+ * Set NSManagedObject with values from the passed in dictionary.
+ */
- (void)initializeChild:(NSManagedObject *)child withItem:(NSDictionary *)item
{
for (NSString *key in item)
@@ -261,7 +266,6 @@
DLog(@"%@ not implemented yet.", key);
NS_ENDHANDLER
}
-
}
/**
@@ -332,8 +336,7 @@
[image1 release];
}
- else
- {
+ else {
NSImage *image1 = [[NSImage imageNamed:NSImageNameUser] retain];
[image1 setScalesWhenResized:YES];
[image1 setSize:(NSSize){16,16}];
@@ -361,14 +364,12 @@
- (void)outlineViewSelectionDidChange:(NSNotification *)notification
{
id selectedObject = [[treeController selectedObjects] objectAtIndex:0];
- if ([selectedObject parent] == nil && !([[[tabView selectedTabViewItem] identifier] isEqualToString:@"General"]))
- {
+
+ if ([selectedObject parent] == nil && !([[[tabView selectedTabViewItem] identifier] isEqualToString:@"General"])) {
[tabView selectTabViewItemWithIdentifier:@"General"];
}
- else
- {
- if ([[[tabView selectedTabViewItem] identifier] isEqualToString:@"General"])
- {
+ else {
+ if ([[[tabView selectedTabViewItem] identifier] isEqualToString:@"General"]) {
[tabView selectTabViewItemWithIdentifier:@"Global Privileges"];
}
}
@@ -427,6 +428,9 @@
}
}
+/**
+ * Disables all privileges.
+ */
- (IBAction)uncheckAllPrivileges:(id)sender
{
id selectedUser = [[treeController selectedObjects] objectAtIndex:0];
@@ -448,13 +452,12 @@
- (IBAction)addUser:(id)sender
{
// Adds a new SPUser objects to the managedObjectContext and sets default values
- if ([[treeController selectedObjects] count] > 0)
- {
- if ([[[treeController selectedObjects] objectAtIndex:0] parent] != nil)
- {
+ if ([[treeController selectedObjects] count] > 0) {
+ if ([[[treeController selectedObjects] objectAtIndex:0] parent] != nil) {
[self _selectParentFromSelection];
}
}
+
NSManagedObject *newItem = [self _createNewSPUser];
NSManagedObject *newChild = [self _createNewSPUser];
[newChild setValue:@"localhost" forKey:@"host"];
@@ -496,7 +499,6 @@
[treeController remove:sender];
}
-
- (void)_clearData
{
[managedObjectContext reset];
@@ -541,9 +543,11 @@
#pragma mark -
#pragma mark Notifications
-// This notification is called when the managedObjectContext save happens.
-// This takes the inserted, updated, and deleted arrays and applys them to
-// the database.
+/**
+ * This notification is called when the managedObjectContext save happens.
+ * This takes the inserted, updated, and deleted arrays and applys them to
+ * the database.
+ */
- (void)contextDidSave:(NSNotification *)notification
{
if (!isInitializing)
@@ -552,6 +556,8 @@
NSArray *inserted = [[notification userInfo] valueForKey:NSInsertedObjectsKey];
NSArray *deleted = [[notification userInfo] valueForKey:NSDeletedObjectsKey];
+ NSLog(@"%d", [inserted count]);
+
if ([inserted count] > 0) {
[self insertUsers:inserted];
}
@@ -567,9 +573,7 @@
}
- (void)contextDidChange:(NSNotification *)notification
-{
- DLog(@"contextDidChange:");
-
+{
if (!isInitializing) [outlineView reloadData];
}
@@ -578,7 +582,8 @@
for (NSManagedObject *user in updatedUsers) {
[self grantPrivilegesToUser:user];
}
- return TRUE;
+
+ return YES;
}
- (BOOL)deleteUsers:(NSArray *)deletedUsers
@@ -607,7 +612,7 @@
{
for (NSManagedObject *user in insertedUsers)
{
- if ([user parent] != nil) {
+ if ([user parent] && [[user parent] valueForKey:@"user"] && [[user parent] valueForKey:@"password"]) {
NSString *createStatement = [NSString stringWithFormat:@"CREATE USER %@@%@ IDENTIFIED BY %@;",
[[[user parent] valueForKey:@"user"] tickQuotedString],
@@ -731,17 +736,18 @@
- (BOOL)tabView:(NSTabView *)tabView shouldSelectTabViewItem:(NSTabViewItem *)tabViewItem
{
- if ([[treeController selectedObjects] count] == 0)
- return FALSE;
+ if ([[treeController selectedObjects] count] == 0) return NO;
id selectedObject = [[treeController selectedObjects] objectAtIndex:0];
+
if ([[tabViewItem identifier] isEqualToString:@"General"]) {
return ([selectedObject parent] == nil);
- } else if ([[tabViewItem identifier] isEqualToString:@"Global Privileges"] || [[tabViewItem identifier] isEqualToString:@"Resources"]) {
+ }
+ else if ([[tabViewItem identifier] isEqualToString:@"Global Privileges"] || [[tabViewItem identifier] isEqualToString:@"Resources"]) {
return ([selectedObject parent] != nil);
}
- return TRUE;
+ return NO;
}
#pragma mark -