MMAppController.m

   1 /* vi:set ts=8 sts=4 sw=4 ft=objc:
   2  *
   3  * VIM - Vi IMproved            by Bram Moolenaar
   4  *                              MacVim GUI port by Bjorn Winckler
   5  *
   6  * Do ":help uganda"  in Vim to read copying and usage conditions.
   7  * Do ":help credits" in Vim to see a list of people who contributed.
   8  * See README.txt for an overview of the Vim source code.
   9  */
  10
  11 #import "MMAppController.h"
  12 #import "MMVimController.h"
  13 #import "MMWindowController.h"
  14
  15
  16
  17 // Default timeout intervals on all connections.
  18 static NSTimeInterval MMRequestTimeout = 5;
  19 static NSTimeInterval MMReplyTimeout = 5;
  20
  21 // Timeout used when the app should terminate.
  22 static NSTimeInterval MMTerminateTimeout = 3;
  23
  24
  25
  26 @interface MMAppController (MMServices)
  27 - (void)openSelection:(NSPasteboard *)pboard userData:(NSString *)userData
  28                 error:(NSString **)error;
  29 - (void)openFile:(NSPasteboard *)pboard userData:(NSString *)userData
  30            error:(NSString **)error;
  31 @end
  32
  33
  34 @interface MMAppController (Private)
  35 - (MMVimController *)keyVimController;
  36 - (MMVimController *)topmostVimController;
  37 @end
  38
  39 @interface NSMenu (MMExtras)
  40 - (void)recurseSetAutoenablesItems:(BOOL)on;
  41 @end
  42
  43
  44
  45 @implementation MMAppController
  46
  47 + (void)initialize
  48 {
  49     NSDictionary *dict = [NSDictionary dictionaryWithObjectsAndKeys:
  50         [NSNumber numberWithBool:NO],   MMNoWindowKey,
  51         [NSNumber numberWithInt:64],    MMTabMinWidthKey,
  52         [NSNumber numberWithInt:6*64],  MMTabMaxWidthKey,
  53         [NSNumber numberWithInt:132],   MMTabOptimumWidthKey,
  54         [NSNumber numberWithInt:2],     MMTextInsetLeftKey,
  55         [NSNumber numberWithInt:1],     MMTextInsetRightKey,
  56         [NSNumber numberWithInt:1],     MMTextInsetTopKey,
  57         [NSNumber numberWithInt:1],     MMTextInsetBottomKey,
  58         [NSNumber numberWithBool:NO],   MMTerminateAfterLastWindowClosedKey,
  59         @"MMTypesetter",                MMTypesetterKey,
  60         [NSNumber numberWithFloat:1],   MMCellWidthMultiplierKey,
  61         [NSNumber numberWithFloat:-1],  MMBaselineOffsetKey,
  62         [NSNumber numberWithBool:YES],  MMTranslateCtrlClickKey,
  63         [NSNumber numberWithBool:NO],   MMOpenFilesInTabsKey,
  64         nil];
  65
  66     [[NSUserDefaults standardUserDefaults] registerDefaults:dict];
  67
  68     NSArray *types = [NSArray arrayWithObject:NSStringPboardType];
  69     [NSApp registerServicesMenuSendTypes:types returnTypes:types];
  70 }
  71
  72 - (id)init
  73 {
  74     if ((self = [super init])) {
  75         vimControllers = [NSMutableArray new];
  76
  77         // NOTE!  If the name of the connection changes here it must also be
  78         // updated in MMBackend.m.
  79         NSConnection *connection = [NSConnection defaultConnection];
  80         NSString *name = [NSString stringWithFormat:@"%@-connection",
  81                  [[NSBundle mainBundle] bundleIdentifier]];
  82         //NSLog(@"Registering connection with name '%@'", name);
  83         if ([connection registerName:name]) {
  84             [connection setRequestTimeout:MMRequestTimeout];
  85             [connection setReplyTimeout:MMReplyTimeout];
  86             [connection setRootObject:self];
  87
  88             // NOTE: When the user is resizing the window the AppKit puts the
  89             // run loop in event tracking mode.  Unless the connection listens
  90             // to request in this mode, live resizing won't work.
  91             [connection addRequestMode:NSEventTrackingRunLoopMode];
  92         } else {
  93             NSLog(@"WARNING: Failed to register connection with name '%@'",
  94                     name);
  95         }
  96     }
  97
  98     return self;
  99 }
 100
 101 - (void)dealloc
 102 {
 103     //NSLog(@"MMAppController dealloc");
 104
 105     [vimControllers release];
 106     [openSelectionString release];
 107
 108     [super dealloc];
 109 }
 110
 111 - (void)applicationDidFinishLaunching:(NSNotification *)notification
 112 {
 113     [NSApp setServicesProvider:self];
 114 }
 115
 116 - (BOOL)applicationShouldOpenUntitledFile:(NSApplication *)sender
 117 {
 118     // NOTE!  This way it possible to start the app with the command-line
 119     // argument '-nowindow yes' and no window will be opened by default.
 120     return ![[NSUserDefaults standardUserDefaults] boolForKey:MMNoWindowKey];
 121 }
 122
 123 - (BOOL)applicationOpenUntitledFile:(NSApplication *)sender
 124 {
 125     //NSLog(@"%s NSapp=%@ theApp=%@", _cmd, NSApp, sender);
 126
 127     [self newVimWindow:self];
 128     return YES;
 129 }
 130
 131 - (void)application:(NSApplication *)sender openFiles:(NSArray *)filenames
 132 {
 133     NSString *firstMissingFile = nil;
 134     NSMutableArray *files = [NSMutableArray array];
 135     int i, count = [filenames count];
 136     for (i = 0; i < count; ++i) {
 137         NSString *name = [filenames objectAtIndex:i];
 138         if ([NSFileHandle fileHandleForReadingAtPath:name]) {
 139             [files addObject:name];
 140         } else if (!firstMissingFile) {
 141             firstMissingFile = name;
 142         }
 143     }
 144
 145     if (firstMissingFile) {
 146         NSAlert *alert = [[NSAlert alloc] init];
 147         [alert addButtonWithTitle:@"OK"];
 148
 149         NSString *text;
 150         if ([files count] >= count-1) {
 151             [alert setMessageText:@"File not found"];
 152             text = [NSString stringWithFormat:@"Could not open file with "
 153                 "name %@.", firstMissingFile];
 154         } else {
 155             [alert setMessageText:@"Multiple files not found"];
 156             text = [NSString stringWithFormat:@"Could not open file with "
 157                 "name %@, and %d other files.", firstMissingFile,
 158                 count-[files count]-1];
 159         }
 160
 161         [alert setInformativeText:text];
 162         [alert setAlertStyle:NSWarningAlertStyle];
 163
 164         [alert runModal];
 165         [alert release];
 166
 167         [NSApp replyToOpenOrPrint:NSApplicationDelegateReplyFailure];
 168         return;
 169     }
 170
 171     MMVimController *vc;
 172     BOOL openInTabs = [[NSUserDefaults standardUserDefaults]
 173         boolForKey:MMOpenFilesInTabsKey];
 174
 175     if (openInTabs && (vc = [self topmostVimController])) {
 176         [vc dropFiles:files];
 177     } else {
 178         NSMutableArray *args = [NSMutableArray arrayWithObjects:
 179             @"-g", @"-p", nil];
 180         [args addObjectsFromArray:files];
 181
 182         NSString *path = [[NSBundle mainBundle]
 183                 pathForAuxiliaryExecutable:@"Vim"];
 184         if (!path) {
 185             NSLog(@"ERROR: Vim executable could not be found inside app "
 186                    "bundle!");
 187             [NSApp replyToOpenOrPrint:NSApplicationDelegateReplyFailure];
 188             return;
 189         }
 190
 191         [NSTask launchedTaskWithLaunchPath:path arguments:args];
 192     }
 193
 194     [NSApp replyToOpenOrPrint:NSApplicationDelegateReplySuccess];
 195     // NSApplicationDelegateReplySuccess = 0,
 196     // NSApplicationDelegateReplyCancel = 1,
 197     // NSApplicationDelegateReplyFailure = 2
 198 }
 199
 200 - (BOOL)applicationShouldTerminateAfterLastWindowClosed:(NSApplication *)sender
 201 {
 202     return [[NSUserDefaults standardUserDefaults]
 203             boolForKey:MMTerminateAfterLastWindowClosedKey];
 204 }
 205
 206 - (NSApplicationTerminateReply)applicationShouldTerminate:
 207     (NSApplication *)sender
 208 {
 209     int reply = NSTerminateNow;
 210     BOOL modifiedBuffers = NO;
 211     BOOL notResponding = NO;
 212
 213     // Go through vim controllers, checking for modified buffers.  If a process
 214     // is not responding then note this as well.
 215     unsigned i, count = [vimControllers count];
 216     for (i = 0; i < count; ++i) {
 217         MMVimController *controller = [vimControllers objectAtIndex:i];
 218         id proxy = [controller backendProxy];
 219         NSConnection *connection = [proxy connectionForProxy];
 220         if (connection) {
 221             NSTimeInterval req = [connection requestTimeout];
 222             NSTimeInterval rep = [connection replyTimeout];
 223             [connection setRequestTimeout:MMTerminateTimeout];
 224             [connection setReplyTimeout:MMTerminateTimeout];
 225
 226             @try {
 227                 if ([proxy checkForModifiedBuffers])
 228                     modifiedBuffers = YES;
 229             }
 230             @catch (NSException *e) {
 231                 NSLog(@"WARNING: Got exception while waiting for "
 232                         "checkForModifiedBuffers: \"%@\"", e);
 233                 notResponding = YES;
 234             }
 235             @finally {
 236                 [connection setRequestTimeout:req];
 237                 [connection setReplyTimeout:rep];
 238                 if (modifiedBuffers || notResponding)
 239                     break;
 240             }
 241         }
 242     }
 243
 244     if (modifiedBuffers || notResponding) {
 245         NSAlert *alert = [[NSAlert alloc] init];
 246         [alert addButtonWithTitle:@"Quit"];
 247         [alert addButtonWithTitle:@"Cancel"];
 248         if (modifiedBuffers) {
 249             [alert setMessageText:@"Quit without saving?"];
 250             [alert setInformativeText:@"There are modified buffers, "
 251                 "if you quit now all changes will be lost.  Quit anyway?"];
 252         } else {
 253             [alert setMessageText:@"Force Quit?"];
 254             [alert setInformativeText:@"At least one Vim process is not "
 255                 "responding, if you quit now any changes you have made "
 256                 "will be lost. Quit anyway?"];
 257         }
 258         [alert setAlertStyle:NSWarningAlertStyle];
 259
 260         if ([alert runModal] != NSAlertFirstButtonReturn) {
 261             reply = NSTerminateCancel;
 262         }
 263
 264         [alert release];
 265     }
 266
 267     return reply;
 268 }
 269
 270 - (void)applicationWillTerminate:(NSNotification *)aNotification
 271 {
 272     // Send a SIGINT to all running Vim processes, so that they are sure to
 273     // receive the connectionDidDie: notification (a process has to checking
 274     // the run-loop for this to happen).
 275     unsigned i, count = [vimControllers count];
 276     for (i = 0; i < count; ++i) {
 277         MMVimController *controller = [vimControllers objectAtIndex:i];
 278         int pid = [controller pid];
 279         if (pid > 0)
 280             kill(pid, SIGINT);
 281
 282         id proxy = [controller backendProxy];
 283         NSConnection *connection = [proxy connectionForProxy];
 284         if (connection) {
 285             [connection invalidate];
 286         }
 287     }
 288
 289     // NOTE! Is this a correct way of releasing the MMAppController?
 290     [NSApp setDelegate:nil];
 291     [self autorelease];
 292 }
 293
 294 - (void)removeVimController:(id)controller
 295 {
 296     //NSLog(@"%s%@", _cmd, controller);
 297
 298     [[controller windowController] close];
 299
 300     [vimControllers removeObject:controller];
 301
 302     if (![vimControllers count]) {
 303         // Turn on autoenabling of menus (because no Vim is open to handle it),
 304         // but do not touch the MacVim menu.
 305         NSMenu *mainMenu = [NSApp mainMenu];
 306         int i, count = [mainMenu numberOfItems];
 307         for (i = 1; i < count; ++i) {
 308             NSMenu *submenu = [[mainMenu itemAtIndex:i] submenu];
 309             [submenu recurseSetAutoenablesItems:YES];
 310         }
 311     }
 312 }
 313
 314 - (void)windowControllerWillOpen:(MMWindowController *)windowController
 315 {
 316     NSPoint topLeft = NSZeroPoint;
 317     NSWindow *keyWin = [NSApp keyWindow];
 318     NSWindow *win = [windowController window];
 319
 320     if (!win) return;
 321
 322     // If there is a key window, cascade from it, otherwise use the autosaved
 323     // window position (if any).
 324     if (keyWin) {
 325         NSRect frame = [keyWin frame];
 326         topLeft = NSMakePoint(frame.origin.x, NSMaxY(frame));
 327     } else {
 328         NSString *topLeftString = [[NSUserDefaults standardUserDefaults]
 329             stringForKey:MMTopLeftPointKey];
 330         if (topLeftString)
 331             topLeft = NSPointFromString(topLeftString);
 332     }
 333
 334     if (!NSEqualPoints(topLeft, NSZeroPoint)) {
 335         if (keyWin)
 336             topLeft = [win cascadeTopLeftFromPoint:topLeft];
 337
 338         [win setFrameTopLeftPoint:topLeft];
 339     }
 340
 341     if (openSelectionString) {
 342         // There is some text to paste into this window as a result of the
 343         // services menu "Open selection ..." being used.
 344         [[windowController vimController] dropString:openSelectionString];
 345         [openSelectionString release];
 346         openSelectionString = nil;
 347     }
 348 }
 349
 350 - (IBAction)newVimWindow:(id)sender
 351 {
 352     NSMutableArray *args = [NSMutableArray arrayWithObject:@"-g"];
 353     NSString *path = [[NSBundle mainBundle]
 354             pathForAuxiliaryExecutable:@"Vim"];
 355     if (!path) {
 356         NSLog(@"ERROR: Vim executable could not be found inside app bundle!");
 357         return;
 358     }
 359
 360
 361     //NSLog(@"Launching a new VimTask...");
 362     [NSTask launchedTaskWithLaunchPath:path arguments:args];
 363 }
 364
 365 - (IBAction)selectNextWindow:(id)sender
 366 {
 367     unsigned i, count = [vimControllers count];
 368     if (!count) return;
 369
 370     NSWindow *keyWindow = [NSApp keyWindow];
 371     for (i = 0; i < count; ++i) {
 372         MMVimController *vc = [vimControllers objectAtIndex:i];
 373         if ([[[vc windowController] window] isEqual:keyWindow])
 374             break;
 375     }
 376
 377     if (i < count) {
 378         if (++i >= count)
 379             i = 0;
 380         MMVimController *vc = [vimControllers objectAtIndex:i];
 381         [[vc windowController] showWindow:self];
 382     }
 383 }
 384
 385 - (IBAction)selectPreviousWindow:(id)sender
 386 {
 387     unsigned i, count = [vimControllers count];
 388     if (!count) return;
 389
 390     NSWindow *keyWindow = [NSApp keyWindow];
 391     for (i = 0; i < count; ++i) {
 392         MMVimController *vc = [vimControllers objectAtIndex:i];
 393         if ([[[vc windowController] window] isEqual:keyWindow])
 394             break;
 395     }
 396
 397     if (i < count) {
 398         if (i > 0) {
 399             --i;
 400         } else {
 401             i = count - 1;
 402         }
 403         MMVimController *vc = [vimControllers objectAtIndex:i];
 404         [[vc windowController] showWindow:self];
 405     }
 406 }
 407
 408 - (byref id <MMFrontendProtocol>)connectBackend:
 409     (byref in id <MMBackendProtocol>)backend pid:(int)pid
 410 {
 411     //NSLog(@"Frontend got connection request from backend...adding new "
 412     //        "MMVimController");
 413
 414     [(NSDistantObject*)backend
 415             setProtocolForProxy:@protocol(MMBackendProtocol)];
 416
 417     MMVimController *vc = [[[MMVimController alloc]
 418             initWithBackend:backend pid:pid] autorelease];
 419
 420     if (![vimControllers count]) {
 421         // The first window autosaves its position.  (The autosaving features
 422         // of Cocoa are not used because we need more control over what is
 423         // autosaved and when it is restored.)
 424         [[vc windowController] setWindowAutosaveKey:MMTopLeftPointKey];
 425     }
 426
 427     [vimControllers addObject:vc];
 428
 429     // HACK!  MacVim does not get activated if it is launched from the
 430     // terminal, so we forcibly activate here.
 431     [NSApp activateIgnoringOtherApps:YES];
 432
 433     return vc;
 434 }
 435
 436 - (NSArray *)serverList
 437 {
 438     NSMutableArray *array = [NSMutableArray array];
 439
 440     unsigned i, count = [vimControllers count];
 441     for (i = 0; i < count; ++i) {
 442         MMVimController *controller = [vimControllers objectAtIndex:i];
 443         id proxy = [controller backendProxy];
 444         NSConnection *connection = [proxy connectionForProxy];
 445         if (!connection)
 446             continue;
 447
 448         // Set low timeouts since we don't want this call to potentially lock
 449         // up MacVim for a while.
 450         NSTimeInterval req = [connection requestTimeout];
 451         NSTimeInterval rep = [connection replyTimeout];
 452         [connection setRequestTimeout:0];
 453         [connection setReplyTimeout:.1];
 454
 455         @try {
 456             NSString *eval = [proxy evaluateExpression:@"v:servername"];
 457             if (eval) {
 458                 [array addObject:eval];
 459             }
 460         }
 461         @catch (NSException *e) {
 462             NSLog(@"WARNING: Got exception while listing servers: \"%@\"", e);
 463         }
 464         @finally {
 465             [connection setRequestTimeout:req];
 466             [connection setReplyTimeout:rep];
 467         }
 468     }
 469
 470     return array;
 471 }
 472
 473 @end // MMAppController
 474
 475
 476
 477
 478 @implementation MMAppController (MMServices)
 479
 480 - (void)openSelection:(NSPasteboard *)pboard userData:(NSString *)userData
 481                 error:(NSString **)error
 482 {
 483     if (![[pboard types] containsObject:NSStringPboardType]) {
 484         NSLog(@"WARNING: Pasteboard contains no object of type "
 485                 "NSStringPboardType");
 486         return;
 487     }
 488
 489     MMVimController *vc = [self topmostVimController];
 490     if (vc) {
 491         // Open a new tab first, since dropString: does not do this.
 492         [vc sendMessage:AddNewTabMsgID data:nil wait:NO];
 493         [vc dropString:[pboard stringForType:NSStringPboardType]];
 494     } else {
 495         // NOTE: There is no window to paste the selection into, so save the
 496         // text, open a new window, and paste the text when the next window
 497         // opens.  (If this is called several times in a row, then all but the
 498         // last call might be ignored.)
 499         if (openSelectionString) [openSelectionString release];
 500         openSelectionString = [[pboard stringForType:NSStringPboardType] copy];
 501
 502         [self newVimWindow:self];
 503     }
 504 }
 505
 506 - (void)openFile:(NSPasteboard *)pboard userData:(NSString *)userData
 507            error:(NSString **)error
 508 {
 509     if (![[pboard types] containsObject:NSStringPboardType]) {
 510         NSLog(@"WARNING: Pasteboard contains no object of type "
 511                 "NSStringPboardType");
 512         return;
 513     }
 514
 515     // TODO: Parse multiple filenames and create array with names.
 516     NSString *string = [pboard stringForType:NSStringPboardType];
 517     string = [string stringByTrimmingCharactersInSet:
 518             [NSCharacterSet whitespaceAndNewlineCharacterSet]];
 519     string = [string stringByStandardizingPath];
 520
 521     MMVimController *vc = nil;
 522     if (userData && [userData isEqual:@"Tab"])
 523         vc = [self topmostVimController];
 524
 525     if (vc) {
 526         [vc dropFiles:[NSArray arrayWithObject:string]];
 527     } else {
 528         [self application:NSApp openFiles:[NSArray arrayWithObject:string]];
 529     }
 530 }
 531
 532 @end // MMAppController (MMServices)
 533
 534
 535
 536
 537 @implementation MMAppController (Private)
 538
 539 - (MMVimController *)keyVimController
 540 {
 541     NSWindow *keyWindow = [NSApp keyWindow];
 542     if (keyWindow) {
 543         unsigned i, count = [vimControllers count];
 544         for (i = 0; i < count; ++i) {
 545             MMVimController *vc = [vimControllers objectAtIndex:i];
 546             if ([[[vc windowController] window] isEqual:keyWindow])
 547                 return vc;
 548         }
 549     }
 550
 551     return nil;
 552 }
 553
 554 - (MMVimController *)topmostVimController
 555 {
 556     NSArray *windows = [NSApp orderedWindows];
 557     if ([windows count] > 0) {
 558         NSWindow *window = [windows objectAtIndex:0];
 559         unsigned i, count = [vimControllers count];
 560         for (i = 0; i < count; ++i) {
 561             MMVimController *vc = [vimControllers objectAtIndex:i];
 562             if ([[[vc windowController] window] isEqual:window])
 563                 return vc;
 564         }
 565     }
 566
 567     return nil;
 568 }
 569
 570 @end
 571
 572
 573
 574
 575 @implementation NSMenu (MMExtras)
 576
 577 - (void)recurseSetAutoenablesItems:(BOOL)on
 578 {
 579     [self setAutoenablesItems:on];
 580
 581     int i, count = [self numberOfItems];
 582     for (i = 0; i < count; ++i) {
 583         NSMenu *submenu = [[self itemAtIndex:i] submenu];
 584         if (submenu) {
 585             [submenu recurseSetAutoenablesItems:on];
 586         }
 587     }
 588 }
 589
 590 @end  // NSMenu (MMExtras)