preliminary support for constraint editing

1 files changed, 219 insertions, 0 deletions

diff --git a/Source/TableRelations.m b/Source/TableRelations.m
new file mode 100644
index 00000000..60f19aca
--- /dev/null
+++ b/Source/TableRelations.m
@@ -0,0 +1,219 @@
+//
+//  TableRelations.m
+//  sequel-pro
+//
+//  Created by J Knight on 13/05/09.
+//  Copyright 2009 TalonEdge Ltd.. All rights reserved.
+//
+
+#import "TableRelations.h"
+#import "TableDocument.h"
+#import "TablesList.h"
+#import "CMMCPConnection.h"
+#import "CMMCPResult.h"
+#import "SPTableData.h"
+
+@implementation TableRelations
+
+- (id)init
+{
+	if (![super init])
+		return nil;
+	
+	relData = [[NSMutableArray alloc] init];
+
+	return self;
+}
+
+- (void)dealloc
+{	
+	[relData release], relData = nil;
+	
+	[super dealloc];
+}
+
+- (void)awakeFromNib
+{
+	[[NSNotificationCenter defaultCenter] addObserver:self
+											 selector:@selector(tableChanged:) 
+												 name:NSTableViewSelectionDidChangeNotification 
+											   object:tableList];
+}
+
+- (void)setConnection:(CMMCPConnection *)theConnection
+{
+	mySQLConnection = theConnection;			
+}
+
+- (IBAction)closeRelationSheet:(id)sender
+{
+	[NSApp stopModalWithCode:1];
+}
+
+- (IBAction)addRelation:(id)sender
+{
+	[NSApp stopModalWithCode:0];	
+}
+
+// user choose a reference table
+- (IBAction)chooseRefTable:(id)sender
+{
+	NSString *table = [refTableSelect titleOfSelectedItem];
+	
+	[refColumnSelect removeAllItems];
+
+	NSDictionary *info = [tableDataInstance informationForTable:table];
+	NSArray *cols = [info objectForKey:@"columns"];
+	NSMutableArray *colNames = [[NSMutableArray alloc] init];
+	for( int i = 0; i < [cols count]; i++ ) {
+		[colNames addObject:[[cols objectAtIndex:i] objectForKey:@"name"]];
+	}
+	[refColumnSelect addItemsWithTitles:colNames];
+	[colNames release];
+}
+
+- (IBAction)addRow:(id)sender
+{
+	// set up the controls
+	[tableBox setTitle:[NSString stringWithFormat:@"Table: %@",[tablesListInstance tableName] ]];
+	[columnSelect removeAllItems];
+	[columnSelect addItemsWithTitles:[tableDataInstance columnNames]];
+	[refTableSelect removeAllItems];
+	// grab only real tables
+	NSArray *tables = [tablesListInstance tables];
+	NSArray *types = [tablesListInstance tableTypes];
+	NSMutableArray *validTables = [[NSMutableArray alloc] init];
+	for( int i = 0; i < [tables count]; i++ ) {
+		NSLog( @"%@ %@", [tables objectAtIndex:i], [types objectAtIndex:i] );
+		if( [[types objectAtIndex:i] intValue] == SP_TABLETYPE_TABLE ) {
+			[validTables addObject:[tables objectAtIndex:i]];
+		}
+	}
+	[refTableSelect addItemsWithTitles:validTables];
+	[validTables release];
+	[self chooseRefTable:nil];
+	
+	[NSApp beginSheet:relationSheet
+			modalForWindow:tableWindow
+			modalDelegate:self
+			didEndSelector:nil 
+		  contextInfo:nil];
+	
+	
+	[NSApp runModalForWindow:relationSheet];
+	
+	[NSApp endSheet:relationSheet];
+	[relationSheet orderOut:nil];
+}
+
+- (IBAction)removeRow:(id)sender
+{
+	if ( [relationsView numberOfSelectedRows] ) {
+		int resp = NSRunAlertPanel(@"Remove Relations",
+		@"Are you sure you want to remove the selected relations?",
+								   @"OK", @"Cancel", nil );
+		if( resp == NSAlertDefaultReturn ) {
+			
+		}
+	}
+}
+
+- (IBAction)refresh:(id)sender
+{
+
+	[relData removeAllObjects];
+	
+	if( [tablesListInstance tableType] == SP_TABLETYPE_TABLE ) {
+		[labelText setStringValue:[NSString stringWithFormat:@"Relations for table: %@",[tablesListInstance tableName]]];			
+		[tableDataInstance updateInformationForCurrentTable];
+		NSArray *constraints = [tableDataInstance getConstraints];
+		for( int i = 0; i < [constraints count]; i++ ) {
+			[relData addObject:[NSDictionary dictionaryWithObjectsAndKeys:
+								[tablesListInstance tableName], @"table",
+								[[constraints objectAtIndex:i] objectForKey:@"name"], @"name",
+								[[constraints objectAtIndex:i] objectForKey:@"columns"], @"columns",
+								[[constraints objectAtIndex:i] objectForKey:@"ref_table"], @"fk_table",
+								[[constraints objectAtIndex:i] objectForKey:@"ref_columns"], @"fk_columns",
+								[[constraints objectAtIndex:i] objectForKey:@"update"], @"on_update",
+								[[constraints objectAtIndex:i] objectForKey:@"delete"], @"on_delete",
+								nil]];
+			
+		}
+	} else {
+		[labelText setStringValue:@""];		
+	}
+	
+	[relationsView reloadData];
+
+}
+
+- (void)tableChanged:(NSNotification *)notification
+{
+	if( [tablesListInstance tableType] == SP_TABLETYPE_TABLE ) {
+		[addButton setEnabled:YES];
+	} else {
+		[addButton setEnabled:NO];		
+	}	
+	
+	[self refresh:nil];
+}
+
+
+//tableView datasource methods
+- (int)numberOfRowsInTableView:(NSTableView *)aTableView
+{
+	return [relData count];
+}
+
+- (id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)aTableColumn
+			row:(int)rowIndex
+{
+	//NSNumber *theIdentifier = [aTableColumn identifier];
+	NSDictionary *theRow = [relData objectAtIndex:rowIndex];
+	return [theRow objectForKey:[aTableColumn identifier]];
+}
+
+- (void)tableView:(NSTableView *)aTableView
+   setObjectValue:(id)anObject
+   forTableColumn:(NSTableColumn *)aTableColumn
+			  row:(int)rowIndex
+{
+	
+}
+
+//tableView delegate methods
+- (void)tableView:(NSTableView*)tableView didClickTableColumn:(NSTableColumn *)tableColumn
+{
+	
+}
+- (void)tableViewSelectionDidChange:(NSNotification *)aNotification
+{
+	if ( [relationsView numberOfSelectedRows] ) {
+		[removeButton setEnabled:YES];
+	} else {
+		[removeButton setEnabled:NO];		
+	}
+}
+- (void)tableViewSelectionIsChanging:(NSNotification *)aNotification
+{
+	
+}
+- (void)tableViewColumnDidResize:(NSNotification *)aNotification
+{
+	
+}
+- (BOOL)tableView:(NSTableView *)aTableView shouldEditTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex
+{
+	return NO;
+}
+- (BOOL)tableView:(NSTableView *)tableView writeRows:(NSArray*)rows toPasteboard:(NSPasteboard*)pboard
+{
+	return FALSE;
+}
+- (BOOL)control:(NSControl *)control textView:(NSTextView *)textView doCommandBySelector:(SEL)command
+{
+	return FALSE;
+}
+
+
+@end