NSTableView + Contextual Menu

#import <Cocoa/Cocoa.h>
 
@protocol ContextMenuDelegate <NSObject>
- (NSMenu*)tableView:(NSTableView*)aTableView menuForRows:(NSIndexSet*)rows;
@end
 
@interface NSTableView (ContextMenu)
	
@end
 
----------
 
@implementation NSTableView (ContextMenu)
 
- (NSMenu*)menuForEvent:(NSEvent*)event
{
	NSPoint location = [self convertPoint:[event locationInWindow] fromView:nil];
	NSInteger row = [self rowAtPoint:location];
	if (!(row >= 0) || ([event type] != NSRightMouseDown)) { 
		return [super menuForEvent:event]; 
	}
	NSIndexSet *selected = [self selectedRowIndexes];
	if (![selected containsIndex:row]) {
		selected = [NSIndexSet indexSetWithIndex:row];
		[self selectRowIndexes:selected byExtendingSelection:NO];
	}
	if ([[self delegate] respondsToSelector:@selector(tableView:menuForRows:)]) {
		return [(id)[self delegate] tableView:self menuForRows:selected];
	}
	return [super menuForEvent:event];
}
@end

This category adds functionality to NSTableView that should have been there in the first place: contextual (right click) menus. Usage is simple:

1. Add the category to your project
2. Make sure that the NSTableView's delegate outlet is connected
3. Implement the ContextMenuDelegate protocol in your controller and return the NSMenu that you want to display