PBGitIndexController.m

   1 //
   2 //  PBGitIndexController.m
   3 //  GitX
   4 //
   5 //  Created by Pieter de Bie on 18-11-08.
   6 //  Copyright 2008 Pieter de Bie. All rights reserved.
   7 //
   8
   9 #import "PBGitIndexController.h"
  10 #import "PBChangedFile.h"
  11 #import "PBGitRepository.h"
  12 #import "PBGitIndex.h"
  13
  14 #define FileChangesTableViewType @"GitFileChangedType"
  15
  16 @interface PBGitIndexController ()
  17 - (void)discardChangesForFiles:(NSArray *)files force:(BOOL)force;
  18 @end
  19
  20 @implementation PBGitIndexController
  21
  22 - (void)awakeFromNib
  23 {
  24         [unstagedTable setDoubleAction:@selector(tableClicked:)];
  25         [stagedTable setDoubleAction:@selector(tableClicked:)];
  26
  27         [unstagedTable setTarget:self];
  28         [stagedTable setTarget:self];
  29
  30         [unstagedTable registerForDraggedTypes: [NSArray arrayWithObject:FileChangesTableViewType]];
  31         [stagedTable registerForDraggedTypes: [NSArray arrayWithObject:FileChangesTableViewType]];
  32 }
  33
  34 // FIXME: Find a proper place for this method -- this is not it.
  35 - (void)ignoreFiles:(NSArray *)files
  36 {
  37         // Build output string
  38         NSMutableArray *fileList = [NSMutableArray array];
  39         for (PBChangedFile *file in files) {
  40                 NSString *name = file.path;
  41                 if ([name length] > 0)
  42                         [fileList addObject:name];
  43         }
  44         NSString *filesAsString = [fileList componentsJoinedByString:@"\n"];
  45
  46         // Write to the file
  47         NSString *gitIgnoreName = [commitController.repository gitIgnoreFilename];
  48
  49         NSStringEncoding enc = NSUTF8StringEncoding;
  50         NSError *error = nil;
  51         NSMutableString *ignoreFile;
  52
  53         if (![[NSFileManager defaultManager] fileExistsAtPath:gitIgnoreName]) {
  54                 ignoreFile = [filesAsString mutableCopy];
  55         } else {
  56                 ignoreFile = [NSMutableString stringWithContentsOfFile:gitIgnoreName usedEncoding:&enc error:&error];
  57                 if (error) {
  58                         [[commitController.repository windowController] showErrorSheet:error];
  59                         return;
  60                 }
  61                 // Add a newline if not yet present
  62                 if ([ignoreFile characterAtIndex:([ignoreFile length] - 1)] != '\n')
  63                         [ignoreFile appendString:@"\n"];
  64                 [ignoreFile appendString:filesAsString];
  65         }
  66
  67         [ignoreFile writeToFile:gitIgnoreName atomically:YES encoding:enc error:&error];
  68         if (error)
  69                 [[commitController.repository windowController] showErrorSheet:error];
  70 }
  71
  72 # pragma mark Context Menu methods
  73 - (BOOL) allSelectedCanBeIgnored:(NSArray *)selectedFiles
  74 {
  75         for (PBChangedFile *selectedItem in selectedFiles) {
  76                 if (selectedItem.status != NEW) {
  77                         return NO;
  78                 }
  79         }
  80         return YES;
  81 }
  82
  83 - (NSMenu *) menuForTable:(NSTableView *)table
  84 {
  85         NSMenu *menu = [[NSMenu alloc] init];
  86         id controller = [table tag] == 0 ? unstagedFilesController : stagedFilesController;
  87         NSArray *selectedFiles = [controller selectedObjects];
  88
  89         // Unstaged changes
  90         if ([table tag] == 0) {
  91                 NSMenuItem *stageItem = [[NSMenuItem alloc] initWithTitle:@"Stage Changes" action:@selector(stageFilesAction:) keyEquivalent:@""];
  92                 [stageItem setTarget:self];
  93                 [stageItem setRepresentedObject:selectedFiles];
  94                 [menu addItem:stageItem];
  95         }
  96         else if ([table tag] == 1) {
  97                 NSMenuItem *unstageItem = [[NSMenuItem alloc] initWithTitle:@"Unstage Changes" action:@selector(unstageFilesAction:) keyEquivalent:@""];
  98                 [unstageItem setTarget:self];
  99                 [unstageItem setRepresentedObject:selectedFiles];
 100                 [menu addItem:unstageItem];
 101         }
 102
 103         NSString *title = [selectedFiles count] == 1 ? @"Open file" : @"Open files";
 104         NSMenuItem *openItem = [[NSMenuItem alloc] initWithTitle:title action:@selector(openFilesAction:) keyEquivalent:@""];
 105         [openItem setTarget:self];
 106         [openItem setRepresentedObject:selectedFiles];
 107         [menu addItem:openItem];
 108
 109         // Attempt to ignore
 110         if ([self allSelectedCanBeIgnored:selectedFiles]) {
 111                 NSString *ignoreText = [selectedFiles count] == 1 ? @"Ignore File": @"Ignore Files";
 112                 NSMenuItem *ignoreItem = [[NSMenuItem alloc] initWithTitle:ignoreText action:@selector(ignoreFilesAction:) keyEquivalent:@""];
 113                 [ignoreItem setTarget:self];
 114                 [ignoreItem setRepresentedObject:selectedFiles];
 115                 [menu addItem:ignoreItem];
 116         }
 117
 118         if ([selectedFiles count] == 1) {
 119                 NSMenuItem *showInFinderItem = [[NSMenuItem alloc] initWithTitle:@"Show in Finder" action:@selector(showInFinderAction:) keyEquivalent:@""];
 120                 [showInFinderItem setTarget:self];
 121                 [showInFinderItem setRepresentedObject:selectedFiles];
 122                 [menu addItem:showInFinderItem];
 123     }
 124
 125         for (PBChangedFile *file in selectedFiles)
 126                 if (!file.hasUnstagedChanges)
 127                         return menu;
 128
 129         NSMenuItem *discardItem = [[NSMenuItem alloc] initWithTitle:@"Discard changes…" action:@selector(discardFilesAction:) keyEquivalent:@""];
 130         [discardItem setTarget:self];
 131         [discardItem setAlternate:NO];
 132         [discardItem setRepresentedObject:selectedFiles];
 133
 134         [menu addItem:discardItem];
 135
 136         NSMenuItem *discardForceItem = [[NSMenuItem alloc] initWithTitle:@"Discard changes" action:@selector(forceDiscardFilesAction:) keyEquivalent:@""];
 137         [discardForceItem setTarget:self];
 138         [discardForceItem setAlternate:YES];
 139         [discardForceItem setRepresentedObject:selectedFiles];
 140         [discardForceItem setKeyEquivalentModifierMask:NSAlternateKeyMask];
 141         [menu addItem:discardForceItem];
 142
 143         return menu;
 144 }
 145
 146 - (void) stageFilesAction:(id) sender
 147 {
 148         [commitController.index stageFiles:[sender representedObject]];
 149 }
 150
 151 - (void) unstageFilesAction:(id) sender
 152 {
 153         [commitController.index unstageFiles:[sender representedObject]];
 154 }
 155
 156 - (void) openFilesAction:(id) sender
 157 {
 158         NSArray *files = [sender representedObject];
 159         NSString *workingDirectory = [commitController.repository workingDirectory];
 160         for (PBChangedFile *file in files)
 161                 [[NSWorkspace sharedWorkspace] openFile:[workingDirectory stringByAppendingPathComponent:[file path]]];
 162 }
 163
 164 - (void) ignoreFilesAction:(id) sender
 165 {
 166         NSArray *selectedFiles = [sender representedObject];
 167         if ([selectedFiles count] == 0)
 168                 return;
 169
 170         [self ignoreFiles:selectedFiles];
 171         [commitController.index refresh];
 172 }
 173
 174 - (void)discardFilesAction:(id) sender
 175 {
 176         NSArray *selectedFiles = [sender representedObject];
 177         if ([selectedFiles count] > 0)
 178                 [self discardChangesForFiles:selectedFiles force:FALSE];
 179 }
 180
 181 - (void)forceDiscardFilesAction:(id) sender
 182 {
 183         NSArray *selectedFiles = [sender representedObject];
 184         if ([selectedFiles count] > 0)
 185                 [self discardChangesForFiles:selectedFiles force:TRUE];
 186 }
 187
 188 - (void) showInFinderAction:(id) sender
 189 {
 190         NSArray *selectedFiles = [sender representedObject];
 191         if ([selectedFiles count] == 0)
 192                 return;
 193         NSString *workingDirectory = [[commitController.repository workingDirectory] stringByAppendingString:@"/"];
 194         NSString *path = [workingDirectory stringByAppendingPathComponent:[[selectedFiles objectAtIndex:0] path]];
 195         NSWorkspace *ws = [NSWorkspace sharedWorkspace];
 196         [ws selectFile: path inFileViewerRootedAtPath:nil];
 197 }
 198
 199 - (void)discardChangesForFiles:(NSArray *)files force:(BOOL)force
 200 {
 201         if (!force) {
 202                 int ret = [[NSAlert alertWithMessageText:@"Discard changes"
 203                                                                    defaultButton:nil
 204                                                                  alternateButton:@"Cancel"
 205                                                                          otherButton:nil
 206                                            informativeTextWithFormat:@"Are you sure you wish to discard the changes to this file?\n\nYou cannot undo this operation."] runModal];
 207                 if (ret != NSAlertDefaultReturn)
 208                         return;
 209         }
 210
 211         [commitController.index discardChangesForFiles:files];
 212 }
 213
 214 # pragma mark TableView icon delegate
 215 - (void)tableView:(NSTableView*)tableView willDisplayCell:(id)cell forTableColumn:(NSTableColumn*)tableColumn row:(NSInteger)rowIndex
 216 {
 217         id controller = [tableView tag] == 0 ? unstagedFilesController : stagedFilesController;
 218         [[tableColumn dataCell] setImage:[[[controller arrangedObjects] objectAtIndex:rowIndex] icon]];
 219 }
 220
 221 - (void) tableClicked:(NSTableView *) tableView
 222 {
 223         NSArrayController *controller = [tableView tag] == 0 ? unstagedFilesController : stagedFilesController;
 224
 225         NSIndexSet *selectionIndexes = [tableView selectedRowIndexes];
 226         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:selectionIndexes];
 227         if ([tableView tag] == 0)
 228                 [commitController.index stageFiles:files];
 229         else
 230                 [commitController.index unstageFiles:files];
 231 }
 232
 233 - (void) rowClicked:(NSCell *)sender
 234 {
 235         NSTableView *tableView = (NSTableView *)[sender controlView];
 236         if([tableView numberOfSelectedRows] != 1)
 237                 return;
 238         [self tableClicked: tableView];
 239 }
 240
 241 - (BOOL)tableView:(NSTableView *)tv
 242 writeRowsWithIndexes:(NSIndexSet *)rowIndexes
 243          toPasteboard:(NSPasteboard*)pboard
 244 {
 245     // Copy the row numbers to the pasteboard.
 246     [pboard declareTypes:[NSArray arrayWithObjects:FileChangesTableViewType, NSFilenamesPboardType, nil] owner:self];
 247
 248         // Internal, for dragging from one tableview to the other
 249         NSData *data = [NSKeyedArchiver archivedDataWithRootObject:rowIndexes];
 250     [pboard setData:data forType:FileChangesTableViewType];
 251
 252         // External, to drag them to for example XCode or Textmate
 253         NSArrayController *controller = [tv tag] == 0 ? unstagedFilesController : stagedFilesController;
 254         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:rowIndexes];
 255         NSString *workingDirectory = [commitController.repository workingDirectory];
 256
 257         NSMutableArray *filenames = [NSMutableArray arrayWithCapacity:[rowIndexes count]];
 258         for (PBChangedFile *file in files)
 259                 [filenames addObject:[workingDirectory stringByAppendingPathComponent:[file path]]];
 260
 261         [pboard setPropertyList:filenames forType:NSFilenamesPboardType];
 262     return YES;
 263 }
 264
 265 - (NSDragOperation)tableView:(NSTableView*)tableView
 266                                 validateDrop:(id <NSDraggingInfo>)info
 267                                  proposedRow:(NSInteger)row
 268            proposedDropOperation:(NSTableViewDropOperation)operation
 269 {
 270         if ([info draggingSource] == tableView)
 271                 return NSDragOperationNone;
 272
 273         [tableView setDropRow:-1 dropOperation:NSTableViewDropOn];
 274     return NSDragOperationCopy;
 275 }
 276
 277 - (BOOL)tableView:(NSTableView *)aTableView
 278            acceptDrop:(id <NSDraggingInfo>)info
 279                           row:(NSInteger)row
 280         dropOperation:(NSTableViewDropOperation)operation
 281 {
 282     NSPasteboard* pboard = [info draggingPasteboard];
 283     NSData* rowData = [pboard dataForType:FileChangesTableViewType];
 284     NSIndexSet* rowIndexes = [NSKeyedUnarchiver unarchiveObjectWithData:rowData];
 285
 286         NSArrayController *controller = [aTableView tag] == 0 ? stagedFilesController : unstagedFilesController;
 287         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:rowIndexes];
 288
 289         if ([aTableView tag] == 0)
 290                 [commitController.index unstageFiles:files];
 291         else
 292                 [commitController.index stageFiles:files];
 293
 294         return YES;
 295 }
 296
 297 @end