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
  13 #define FileChangesTableViewType @"GitFileChangedType"
  14
  15 @interface PBGitIndexController (PrivateMethods)
  16 - (void)stopTrackingIndex;
  17 - (void)resumeTrackingIndex;
  18 @end
  19
  20 @implementation PBGitIndexController
  21
  22 @synthesize contextSize;
  23
  24 - (void)awakeFromNib
  25 {
  26         contextSize = 3;
  27
  28         [unstagedTable setDoubleAction:@selector(tableClicked:)];
  29         [stagedTable setDoubleAction:@selector(tableClicked:)];
  30
  31         [unstagedTable setTarget:self];
  32         [stagedTable setTarget:self];
  33
  34         [unstagedTable registerForDraggedTypes: [NSArray arrayWithObject:FileChangesTableViewType]];
  35         [stagedTable registerForDraggedTypes: [NSArray arrayWithObject:FileChangesTableViewType]];
  36
  37 }
  38
  39 - (void) stageFiles:(NSArray *)files
  40 {
  41         NSMutableString *input = [NSMutableString string];
  42
  43         for (PBChangedFile *file in files) {
  44                 [input appendFormat:@"%@\0", file.path];
  45         }
  46
  47         int ret = 1;
  48         [commitController.repository outputForArguments:[NSArray arrayWithObjects:@"update-index", @"--add", @"--remove", @"-z", @"--stdin", nil]
  49          inputString:input retValue:&ret];
  50
  51         if (ret)
  52         {
  53                 NSLog(@"Error when updating index. Retvalue: %i", ret);
  54                 return;
  55         }
  56
  57         [self stopTrackingIndex];
  58         for (PBChangedFile *file in files)
  59         {
  60                 file.hasUnstagedChanges = NO;
  61                 file.hasCachedChanges = YES;
  62         }
  63         [self resumeTrackingIndex];
  64 }
  65
  66 - (void) unstageFiles:(NSArray *)files
  67 {
  68         NSMutableString *input = [NSMutableString string];
  69
  70         for (PBChangedFile *file in files) {
  71                 [input appendString:[file indexInfo]];
  72         }
  73
  74         int ret = 1;
  75         [commitController.repository outputForArguments:[NSArray arrayWithObjects:@"update-index", @"-z", @"--index-info", nil]
  76          inputString:input retValue:&ret];
  77
  78         if (ret)
  79         {
  80                 NSLog(@"Error when updating index. Retvalue: %i", ret);
  81                 return;
  82         }
  83
  84         [self stopTrackingIndex];
  85         for (PBChangedFile *file in files)
  86         {
  87                 file.hasUnstagedChanges = YES;
  88                 file.hasCachedChanges = NO;
  89         }
  90         [self resumeTrackingIndex];
  91 }
  92
  93 - (void) ignoreFiles:(NSArray *)files
  94 {
  95         // Build output string
  96         NSMutableArray *fileList = [NSMutableArray array];
  97         for (PBChangedFile *file in files) {
  98                 NSString *name = file.path;
  99                 if ([name length] > 0)
 100                         [fileList addObject:name];
 101         }
 102         NSString *filesAsString = [fileList componentsJoinedByString:@"\n"];
 103
 104         // Write to the file
 105         NSString *gitIgnoreName = [commitController.repository gitIgnoreFilename];
 106
 107         NSStringEncoding enc = NSUTF8StringEncoding;
 108         NSError *error = nil;
 109         NSMutableString *ignoreFile;
 110
 111         if (![[NSFileManager defaultManager] fileExistsAtPath:gitIgnoreName]) {
 112                 ignoreFile = [filesAsString mutableCopy];
 113         } else {
 114                 ignoreFile = [NSMutableString stringWithContentsOfFile:gitIgnoreName usedEncoding:&enc error:&error];
 115                 if (error) {
 116                         [[NSAlert alertWithError:error] runModal];
 117                         return;
 118                 }
 119                 // Add a newline if not yet present
 120                 if ([ignoreFile characterAtIndex:([ignoreFile length] - 1)] != '\n')
 121                         [ignoreFile appendString:@"\n"];
 122                 [ignoreFile appendString:filesAsString];
 123         }
 124
 125         [ignoreFile writeToFile:gitIgnoreName atomically:YES encoding:enc error:&error];
 126         if (error)
 127                 [[NSAlert alertWithError:error] runModal];
 128 }
 129
 130 # pragma mark Displaying diffs
 131
 132 - (NSString *) stagedChangesForFile:(PBChangedFile *)file
 133 {
 134         NSString *indexPath = [@":0:" stringByAppendingString:file.path];
 135
 136         if (file.status == NEW)
 137                 return [commitController.repository outputForArguments:[NSArray arrayWithObjects:@"show", indexPath, nil]];
 138
 139         return [commitController.repository outputInWorkdirForArguments:[NSArray arrayWithObjects:@"diff-index", [self contextParameter], @"--cached", [commitController parentTree], @"--", file.path, nil]];
 140 }
 141
 142 - (NSString *)unstagedChangesForFile:(PBChangedFile *)file
 143 {
 144         if (file.status == NEW) {
 145                 NSStringEncoding encoding;
 146                 NSError *error = nil;
 147                 NSString *path = [[commitController.repository workingDirectory] stringByAppendingPathComponent:file.path];
 148                 NSString *contents = [NSString stringWithContentsOfFile:path
 149                                                                                                    usedEncoding:&encoding
 150                                                                                                                   error:&error];
 151                 if (error)
 152                         return nil;
 153
 154                 return contents;
 155         }
 156
 157         return [commitController.repository outputInWorkdirForArguments:[NSArray arrayWithObjects:@"diff-files", [self contextParameter], @"--", file.path, nil]];
 158 }
 159
 160 - (void) forceRevertChangesForFiles:(NSArray *)files
 161 {
 162         NSArray *paths = [files valueForKey:@"path"];
 163         NSString *input = [paths componentsJoinedByString:@"\0"];
 164
 165         NSArray *arguments = [NSArray arrayWithObjects:@"checkout-index", @"--index", @"--quiet", @"--force", @"-z", @"--stdin", nil];
 166         int ret = 1;
 167         [commitController.repository outputForArguments:arguments inputString:input retValue:&ret];
 168         if (ret) {
 169                 [[NSAlert alertWithMessageText:@"Reverting changes failed"
 170                                                  defaultButton:nil
 171                                            alternateButton:nil
 172                                                    otherButton:nil
 173                          informativeTextWithFormat:@"Reverting changes failed with error code %i", ret] runModal];
 174                 return;
 175         }
 176
 177         for (PBChangedFile *file in files)
 178                 file.hasUnstagedChanges = NO;
 179 }
 180
 181 - (void) revertChangesForFiles:(NSArray *)files
 182 {
 183         int ret = [[NSAlert alertWithMessageText:@"Revert changes"
 184                                          defaultButton:nil
 185                                    alternateButton:@"Cancel"
 186                                            otherButton:nil
 187                  informativeTextWithFormat:@"Are you sure you wish to revert changes?\n\n You cannot undo this operation."] runModal];
 188
 189         if (ret == NSAlertDefaultReturn)
 190                 [self forceRevertChangesForFiles:files];
 191 }
 192
 193
 194 # pragma mark Context Menu methods
 195 - (BOOL) allSelectedCanBeIgnored:(NSArray *)selectedFiles
 196 {
 197         for (PBChangedFile *selectedItem in selectedFiles) {
 198                 if (selectedItem.status != NEW) {
 199                         return NO;
 200                 }
 201         }
 202         return YES;
 203 }
 204
 205 - (NSMenu *) menuForTable:(NSTableView *)table
 206 {
 207         NSMenu *menu = [[NSMenu alloc] init];
 208         id controller = [table tag] == 0 ? unstagedFilesController : stagedFilesController;
 209         NSArray *selectedFiles = [controller selectedObjects];
 210
 211         // Unstaged changes
 212         if ([table tag] == 0) {
 213                 NSMenuItem *stageItem = [[NSMenuItem alloc] initWithTitle:@"Stage Changes" action:@selector(stageFilesAction:) keyEquivalent:@""];
 214                 [stageItem setTarget:self];
 215                 [stageItem setRepresentedObject:selectedFiles];
 216                 [menu addItem:stageItem];
 217         }
 218         else if ([table tag] == 1) {
 219                 NSMenuItem *unstageItem = [[NSMenuItem alloc] initWithTitle:@"Unstage Changes" action:@selector(unstageFilesAction:) keyEquivalent:@""];
 220                 [unstageItem setTarget:self];
 221                 [unstageItem setRepresentedObject:selectedFiles];
 222                 [menu addItem:unstageItem];
 223         }
 224
 225         NSString *title = [selectedFiles count] == 1 ? @"Open file" : @"Open files";
 226         NSMenuItem *openItem = [[NSMenuItem alloc] initWithTitle:title action:@selector(openFilesAction:) keyEquivalent:@""];
 227         [openItem setTarget:self];
 228         [openItem setRepresentedObject:selectedFiles];
 229         [menu addItem:openItem];
 230
 231         // Attempt to ignore
 232         if ([self allSelectedCanBeIgnored:selectedFiles]) {
 233                 NSString *ignoreText = [selectedFiles count] == 1 ? @"Ignore File": @"Ignore Files";
 234                 NSMenuItem *ignoreItem = [[NSMenuItem alloc] initWithTitle:ignoreText action:@selector(ignoreFilesAction:) keyEquivalent:@""];
 235                 [ignoreItem setTarget:self];
 236                 [ignoreItem setRepresentedObject:selectedFiles];
 237                 [menu addItem:ignoreItem];
 238         }
 239
 240         for (PBChangedFile *file in selectedFiles)
 241                 if (!file.hasUnstagedChanges)
 242                         return menu;
 243
 244         NSMenuItem *revertItem = [[NSMenuItem alloc] initWithTitle:@"Revert Changes…" action:@selector(revertFilesAction:) keyEquivalent:@""];
 245         [revertItem setTarget:self];
 246         [revertItem setAlternate:NO];
 247         [revertItem setRepresentedObject:selectedFiles];
 248
 249         [menu addItem:revertItem];
 250
 251         NSMenuItem *revertForceItem = [[NSMenuItem alloc] initWithTitle:@"Revert Changes" action:@selector(forceRevertFilesAction:) keyEquivalent:@""];
 252         [revertForceItem setTarget:self];
 253         [revertForceItem setAlternate:YES];
 254         [revertForceItem setRepresentedObject:selectedFiles];
 255         [revertForceItem setKeyEquivalentModifierMask:NSAlternateKeyMask];
 256         [menu addItem:revertForceItem];
 257
 258         return menu;
 259 }
 260
 261 - (void) stageFilesAction:(id) sender
 262 {
 263         [self stageFiles:[sender representedObject]];
 264 }
 265
 266 - (void) unstageFilesAction:(id) sender
 267 {
 268         [self unstageFiles:[sender representedObject]];
 269 }
 270
 271 - (void) openFilesAction:(id) sender
 272 {
 273         NSArray *files = [sender representedObject];
 274         NSString *workingDirectory = [commitController.repository workingDirectory];
 275         for (PBChangedFile *file in files)
 276                 [[NSWorkspace sharedWorkspace] openFile:[workingDirectory stringByAppendingPathComponent:[file path]]];
 277 }
 278
 279 - (void) ignoreFilesAction:(id) sender
 280 {
 281         NSArray *selectedFiles = [sender representedObject];
 282         if ([selectedFiles count] > 0) {
 283                 [self ignoreFiles:selectedFiles];
 284         }
 285         [commitController refresh:NULL];
 286 }
 287
 288 - (void) revertFilesAction:(id) sender
 289 {
 290         NSArray *selectedFiles = [sender representedObject];
 291         if ([selectedFiles count] > 0)
 292                 [self revertChangesForFiles:selectedFiles];
 293 }
 294
 295 - (void) forceRevertFilesAction:(id) sender
 296 {
 297         NSArray *selectedFiles = [sender representedObject];
 298         if ([selectedFiles count] > 0)
 299                 [self forceRevertChangesForFiles:selectedFiles];
 300 }
 301
 302
 303 # pragma mark TableView icon delegate
 304 - (void)tableView:(NSTableView*)tableView willDisplayCell:(id)cell forTableColumn:(NSTableColumn*)tableColumn row:(NSInteger)rowIndex
 305 {
 306         id controller = [tableView tag] == 0 ? unstagedFilesController : stagedFilesController;
 307         [[tableColumn dataCell] setImage:[[[controller arrangedObjects] objectAtIndex:rowIndex] icon]];
 308 }
 309
 310 - (void) tableClicked:(NSTableView *) tableView
 311 {
 312         NSArrayController *controller = [tableView tag] == 0 ? unstagedFilesController : stagedFilesController;
 313
 314         NSIndexSet *selectionIndexes = [tableView selectedRowIndexes];
 315         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:selectionIndexes];
 316         if ([tableView tag] == 0)
 317                 [self stageFiles:files];
 318         else
 319                 [self unstageFiles:files];
 320 }
 321
 322 - (void) rowClicked:(NSCell *)sender
 323 {
 324         NSTableView *tableView = (NSTableView *)[sender controlView];
 325         if([tableView numberOfSelectedRows] != 1)
 326                 return;
 327         [self tableClicked: tableView];
 328 }
 329
 330 - (BOOL)tableView:(NSTableView *)tv
 331 writeRowsWithIndexes:(NSIndexSet *)rowIndexes
 332          toPasteboard:(NSPasteboard*)pboard
 333 {
 334     // Copy the row numbers to the pasteboard.
 335     [pboard declareTypes:[NSArray arrayWithObjects:FileChangesTableViewType, NSFilenamesPboardType, nil] owner:self];
 336
 337         // Internal, for dragging from one tableview to the other
 338         NSData *data = [NSKeyedArchiver archivedDataWithRootObject:rowIndexes];
 339     [pboard setData:data forType:FileChangesTableViewType];
 340
 341         // External, to drag them to for example XCode or Textmate
 342         NSArrayController *controller = [tv tag] == 0 ? unstagedFilesController : stagedFilesController;
 343         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:rowIndexes];
 344         NSString *workingDirectory = [commitController.repository workingDirectory];
 345
 346         NSMutableArray *filenames = [NSMutableArray arrayWithCapacity:[rowIndexes count]];
 347         for (PBChangedFile *file in files)
 348                 [filenames addObject:[workingDirectory stringByAppendingPathComponent:[file path]]];
 349
 350         [pboard setPropertyList:filenames forType:NSFilenamesPboardType];
 351     return YES;
 352 }
 353
 354 - (NSDragOperation)tableView:(NSTableView*)tableView
 355                                 validateDrop:(id <NSDraggingInfo>)info
 356                                  proposedRow:(NSInteger)row
 357            proposedDropOperation:(NSTableViewDropOperation)operation
 358 {
 359         if ([info draggingSource] == tableView)
 360                 return NSDragOperationNone;
 361
 362         [tableView setDropRow:-1 dropOperation:NSTableViewDropOn];
 363     return NSDragOperationCopy;
 364 }
 365
 366 - (BOOL)tableView:(NSTableView *)aTableView
 367            acceptDrop:(id <NSDraggingInfo>)info
 368                           row:(NSInteger)row
 369         dropOperation:(NSTableViewDropOperation)operation
 370 {
 371     NSPasteboard* pboard = [info draggingPasteboard];
 372     NSData* rowData = [pboard dataForType:FileChangesTableViewType];
 373     NSIndexSet* rowIndexes = [NSKeyedUnarchiver unarchiveObjectWithData:rowData];
 374
 375         NSArrayController *controller = [aTableView tag] == 0 ? stagedFilesController : unstagedFilesController;
 376         NSArray *files = [[controller arrangedObjects] objectsAtIndexes:rowIndexes];
 377
 378         if ([aTableView tag] == 0)
 379                 [self unstageFiles:files];
 380         else
 381                 [self stageFiles:files];
 382
 383         return YES;
 384 }
 385
 386 - (NSString *) contextParameter
 387 {
 388         return [[NSString alloc] initWithFormat:@"-U%i", contextSize];
 389 }
 390
 391 # pragma mark WebKit Accessibility
 392
 393 + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector
 394 {
 395         return NO;
 396 }
 397
 398 #pragma mark Private Methods
 399 - (void)stopTrackingIndex
 400 {
 401         [stagedFilesController setAutomaticallyRearrangesObjects:NO];
 402         [unstagedFilesController setAutomaticallyRearrangesObjects:NO];
 403 }
 404 - (void)resumeTrackingIndex
 405 {
 406         [stagedFilesController setAutomaticallyRearrangesObjects:YES];
 407         [unstagedFilesController setAutomaticallyRearrangesObjects:YES];
 408         [stagedFilesController rearrangeObjects];
 409         [unstagedFilesController rearrangeObjects];
 410 }
 411 @end