MMVimController.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 "MMVimController.h"
  12 #import "MMWindowController.h"
  13 #import "MMTextView.h"
  14 #import "MMAppController.h"
  15 #import "MMTextStorage.h"
  16
  17
  18 // This is taken from gui.h
  19 #define DRAW_CURSOR 0x20
  20
  21 static NSString *MMDefaultToolbarImageName = @"Attention";
  22 static int MMAlertTextFieldHeight = 22;
  23
  24 // NOTE: By default a message sent to the backend will be dropped if it cannot
  25 // be delivered instantly; otherwise there is a possibility that MacVim will
  26 // 'beachball' while waiting to deliver DO messages to an unresponsive Vim
  27 // process.  This means that you cannot rely on any message sent with
  28 // sendMessage: to actually reach Vim.
  29 static NSTimeInterval MMBackendProxyRequestTimeout = 0;
  30
  31 #if MM_RESEND_LAST_FAILURE
  32 // If a message send fails, the message will be resent after this many seconds
  33 // have passed.  (No queue is kept, only the very last message is resent.)
  34 static NSTimeInterval MMResendInterval = 0.5;
  35 #endif
  36
  37
  38 @interface MMAlert : NSAlert {
  39     NSTextField *textField;
  40 }
  41 - (void)setTextFieldString:(NSString *)textFieldString;
  42 - (NSTextField *)textField;
  43 @end
  44
  45
  46 @interface MMVimController (Private)
  47 - (void)handleMessage:(int)msgid data:(NSData *)data;
  48 - (void)performBatchDrawWithData:(NSData *)data;
  49 - (void)savePanelDidEnd:(NSSavePanel *)panel code:(int)code
  50                 context:(void *)context;
  51 - (void)alertDidEnd:(MMAlert *)alert code:(int)code context:(void *)context;
  52 - (NSMenuItem *)menuItemForTag:(int)tag;
  53 - (NSMenu *)menuForTag:(int)tag;
  54 - (NSMenu *)topLevelMenuForTitle:(NSString *)title;
  55 - (void)addMenuWithTag:(int)tag parent:(int)parentTag title:(NSString *)title
  56                atIndex:(int)idx;
  57 - (void)addMenuItemWithTag:(int)tag parent:(NSMenu *)parent
  58                      title:(NSString *)title tip:(NSString *)tip
  59              keyEquivalent:(int)key modifiers:(int)mask
  60                     action:(NSString *)action atIndex:(int)idx;
  61 - (void)updateMainMenu;
  62 - (NSToolbarItem *)toolbarItemForTag:(int)tag index:(int *)index;
  63 - (void)addToolbarItemToDictionaryWithTag:(int)tag label:(NSString *)title
  64         toolTip:(NSString *)tip icon:(NSString *)icon;
  65 - (void)addToolbarItemWithTag:(int)tag label:(NSString *)label
  66                           tip:(NSString *)tip icon:(NSString *)icon
  67                       atIndex:(int)idx;
  68 - (void)connectionDidDie:(NSNotification *)notification;
  69 #if MM_RESEND_LAST_FAILURE
  70 - (void)resendTimerFired:(NSTimer *)timer;
  71 #endif
  72 @end
  73
  74
  75
  76 // TODO: Move to separate file
  77 @interface NSColor (MMProtocol)
  78 + (NSColor *)colorWithRgbInt:(unsigned)rgb;
  79 + (NSColor *)colorWithArgbInt:(unsigned)argb;
  80 @end
  81
  82
  83
  84 static NSMenuItem *findMenuItemWithTagInMenu(NSMenu *root, int tag)
  85 {
  86     if (root) {
  87         NSMenuItem *item = [root itemWithTag:tag];
  88         if (item) return item;
  89
  90         NSArray *items = [root itemArray];
  91         unsigned i, count = [items count];
  92         for (i = 0; i < count; ++i) {
  93             item = [items objectAtIndex:i];
  94             if ([item hasSubmenu]) {
  95                 item = findMenuItemWithTagInMenu([item submenu], tag);
  96                 if (item) return item;
  97             }
  98         }
  99     }
 100
 101     return nil;
 102 }
 103
 104
 105
 106 @implementation MMVimController
 107
 108 - (id)initWithBackend:(id)backend pid:(int)processIdentifier
 109 {
 110     if ((self = [super init])) {
 111         windowController =
 112             [[MMWindowController alloc] initWithVimController:self];
 113         backendProxy = [backend retain];
 114         sendQueue = [NSMutableArray new];
 115         mainMenuItems = [[NSMutableArray alloc] init];
 116         popupMenuItems = [[NSMutableArray alloc] init];
 117         toolbarItemDict = [[NSMutableDictionary alloc] init];
 118         pid = processIdentifier;
 119
 120         NSConnection *connection = [backendProxy connectionForProxy];
 121
 122         // TODO: Check that this will not set the timeout for the root proxy
 123         // (in MMAppController).
 124         [connection setRequestTimeout:MMBackendProxyRequestTimeout];
 125
 126         [[NSNotificationCenter defaultCenter] addObserver:self
 127                 selector:@selector(connectionDidDie:)
 128                     name:NSConnectionDidDieNotification object:connection];
 129
 130
 131         NSWindow *win = [windowController window];
 132
 133         [[NSNotificationCenter defaultCenter]
 134                 addObserver:self
 135                    selector:@selector(windowDidBecomeMain:)
 136                        name:NSWindowDidBecomeMainNotification
 137                      object:win];
 138
 139         isInitialized = YES;
 140     }
 141
 142     return self;
 143 }
 144
 145 - (void)dealloc
 146 {
 147     //NSLog(@"%@ %s", [self className], _cmd);
 148     isInitialized = NO;
 149
 150 #if MM_RESEND_LAST_FAILURE
 151     [resendData release];  resendData = nil;
 152 #endif
 153
 154     [serverName release];  serverName = nil;
 155     [backendProxy release];  backendProxy = nil;
 156     [sendQueue release];  sendQueue = nil;
 157
 158     [toolbarItemDict release];  toolbarItemDict = nil;
 159     [toolbar release];  toolbar = nil;
 160     [popupMenuItems release];  popupMenuItems = nil;
 161     [mainMenuItems release];  mainMenuItems = nil;
 162     [windowController release];  windowController = nil;
 163
 164     [super dealloc];
 165 }
 166
 167 - (MMWindowController *)windowController
 168 {
 169     return windowController;
 170 }
 171
 172 - (void)setServerName:(NSString *)name
 173 {
 174     if (name != serverName) {
 175         [serverName release];
 176         serverName = [name copy];
 177     }
 178 }
 179
 180 - (NSString *)serverName
 181 {
 182     return serverName;
 183 }
 184
 185 - (int)pid
 186 {
 187     return pid;
 188 }
 189
 190 - (void)dropFiles:(NSArray *)filenames
 191 {
 192     int i, numberOfFiles = [filenames count];
 193     NSMutableData *data = [NSMutableData data];
 194
 195     [data appendBytes:&numberOfFiles length:sizeof(int)];
 196
 197     for (i = 0; i < numberOfFiles; ++i) {
 198         NSString *file = [filenames objectAtIndex:i];
 199         int len = [file lengthOfBytesUsingEncoding:NSUTF8StringEncoding];
 200
 201         if (len > 0) {
 202             ++len;  // include NUL as well
 203             [data appendBytes:&len length:sizeof(int)];
 204             [data appendBytes:[file UTF8String] length:len];
 205         }
 206     }
 207
 208     [self sendMessage:DropFilesMsgID data:data];
 209 }
 210
 211 - (void)dropString:(NSString *)string
 212 {
 213     int len = [string lengthOfBytesUsingEncoding:NSUTF8StringEncoding] + 1;
 214     if (len > 0) {
 215         NSMutableData *data = [NSMutableData data];
 216
 217         [data appendBytes:&len length:sizeof(int)];
 218         [data appendBytes:[string UTF8String] length:len];
 219
 220         [self sendMessage:DropStringMsgID data:data];
 221     }
 222 }
 223
 224 - (void)sendMessage:(int)msgid data:(NSData *)data
 225 {
 226     //NSLog(@"sendMessage:%s (isInitialized=%d inProcessCommandQueue=%d)",
 227     //        MessageStrings[msgid], isInitialized, inProcessCommandQueue);
 228
 229     if (!isInitialized) return;
 230
 231     if (inProcessCommandQueue) {
 232         //NSLog(@"In process command queue; delaying message send.");
 233         [sendQueue addObject:[NSNumber numberWithInt:msgid]];
 234         if (data)
 235             [sendQueue addObject:data];
 236         else
 237             [sendQueue addObject:[NSNull null]];
 238         return;
 239     }
 240
 241 #if MM_RESEND_LAST_FAILURE
 242     if (resendTimer) {
 243         //NSLog(@"cancelling scheduled resend of %s",
 244         //        MessageStrings[resendMsgid]);
 245
 246         [resendTimer invalidate];
 247         [resendTimer release];
 248         resendTimer = nil;
 249     }
 250
 251     if (resendData) {
 252         [resendData release];
 253         resendData = nil;
 254     }
 255 #endif
 256
 257     @try {
 258         [backendProxy processInput:msgid data:data];
 259     }
 260     @catch (NSException *e) {
 261         //NSLog(@"%@ %s Exception caught during DO call: %@",
 262         //        [self className], _cmd, e);
 263 #if MM_RESEND_LAST_FAILURE
 264         //NSLog(@"%s failed, scheduling message %s for resend", _cmd,
 265         //        MessageStrings[msgid]);
 266
 267         resendMsgid = msgid;
 268         resendData = [data retain];
 269         resendTimer = [NSTimer
 270             scheduledTimerWithTimeInterval:MMResendInterval
 271                                     target:self
 272                                   selector:@selector(resendTimerFired:)
 273                                   userInfo:nil
 274                                    repeats:NO];
 275         [resendTimer retain];
 276 #endif
 277     }
 278 }
 279
 280 - (BOOL)sendMessageNow:(int)msgid data:(NSData *)data
 281                timeout:(NSTimeInterval)timeout
 282 {
 283     if (!isInitialized || inProcessCommandQueue)
 284         return NO;
 285
 286     if (timeout < 0) timeout = 0;
 287
 288     BOOL sendOk = YES;
 289     NSConnection *conn = [backendProxy connectionForProxy];
 290     NSTimeInterval oldTimeout = [conn requestTimeout];
 291
 292     [conn setRequestTimeout:timeout];
 293
 294     @try {
 295         [backendProxy processInput:msgid data:data];
 296     }
 297     @catch (NSException *e) {
 298         sendOk = NO;
 299     }
 300     @finally {
 301         [conn setRequestTimeout:oldTimeout];
 302     }
 303
 304     return sendOk;
 305 }
 306
 307 - (id)backendProxy
 308 {
 309     return backendProxy;
 310 }
 311
 312 - (void)cleanup
 313 {
 314     //NSLog(@"%@ %s", [self className], _cmd);
 315     if (!isInitialized) return;
 316
 317     isInitialized = NO;
 318     [toolbar setDelegate:nil];
 319     [[NSNotificationCenter defaultCenter] removeObserver:self];
 320     [windowController cleanup];
 321 }
 322
 323 - (oneway void)showSavePanelForDirectory:(in bycopy NSString *)dir
 324                                    title:(in bycopy NSString *)title
 325                                   saving:(int)saving
 326 {
 327     if (!isInitialized) return;
 328
 329     if (saving) {
 330         [[NSSavePanel savePanel] beginSheetForDirectory:dir file:nil
 331                 modalForWindow:[windowController window]
 332                  modalDelegate:self
 333                 didEndSelector:@selector(savePanelDidEnd:code:context:)
 334                    contextInfo:NULL];
 335     } else {
 336         NSOpenPanel *panel = [NSOpenPanel openPanel];
 337         [panel setAllowsMultipleSelection:NO];
 338         [panel beginSheetForDirectory:dir file:nil types:nil
 339                 modalForWindow:[windowController window]
 340                  modalDelegate:self
 341                 didEndSelector:@selector(savePanelDidEnd:code:context:)
 342                    contextInfo:NULL];
 343     }
 344 }
 345
 346 - (oneway void)presentDialogWithStyle:(int)style
 347                               message:(in bycopy NSString *)message
 348                       informativeText:(in bycopy NSString *)text
 349                          buttonTitles:(in bycopy NSArray *)buttonTitles
 350                       textFieldString:(in bycopy NSString *)textFieldString
 351 {
 352     if (!(windowController && buttonTitles && [buttonTitles count])) return;
 353
 354     MMAlert *alert = [[MMAlert alloc] init];
 355
 356     // NOTE! This has to be done before setting the informative text.
 357     if (textFieldString)
 358         [alert setTextFieldString:textFieldString];
 359
 360     [alert setAlertStyle:style];
 361
 362     if (message) {
 363         [alert setMessageText:message];
 364     } else {
 365         // If no message text is specified 'Alert' is used, which we don't
 366         // want, so set an empty string as message text.
 367         [alert setMessageText:@""];
 368     }
 369
 370     if (text) {
 371         [alert setInformativeText:text];
 372     } else if (textFieldString) {
 373         // Make sure there is always room for the input text field.
 374         [alert setInformativeText:@""];
 375     }
 376
 377     unsigned i, count = [buttonTitles count];
 378     for (i = 0; i < count; ++i) {
 379         NSString *title = [buttonTitles objectAtIndex:i];
 380         // NOTE: The title of the button may contain the character '&' to
 381         // indicate that the following letter should be the key equivalent
 382         // associated with the button.  Extract this letter and lowercase it.
 383         NSString *keyEquivalent = nil;
 384         NSRange hotkeyRange = [title rangeOfString:@"&"];
 385         if (NSNotFound != hotkeyRange.location) {
 386             if ([title length] > NSMaxRange(hotkeyRange)) {
 387                 NSRange keyEquivRange = NSMakeRange(hotkeyRange.location+1, 1);
 388                 keyEquivalent = [[title substringWithRange:keyEquivRange]
 389                     lowercaseString];
 390             }
 391
 392             NSMutableString *string = [NSMutableString stringWithString:title];
 393             [string deleteCharactersInRange:hotkeyRange];
 394             title = string;
 395         }
 396
 397         [alert addButtonWithTitle:title];
 398
 399         // Set key equivalent for the button, but only if NSAlert hasn't
 400         // already done so.  (Check the documentation for
 401         // - [NSAlert addButtonWithTitle:] to see what key equivalents are
 402         // automatically assigned.)
 403         NSButton *btn = [[alert buttons] lastObject];
 404         if ([[btn keyEquivalent] length] == 0 && keyEquivalent) {
 405             [btn setKeyEquivalent:keyEquivalent];
 406         }
 407     }
 408
 409     [alert beginSheetModalForWindow:[windowController window]
 410                       modalDelegate:self
 411                      didEndSelector:@selector(alertDidEnd:code:context:)
 412                         contextInfo:NULL];
 413
 414     [alert release];
 415 }
 416
 417 - (oneway void)processCommandQueue:(in NSArray *)queue
 418 {
 419     if (!isInitialized) return;
 420
 421     unsigned i, count = [queue count];
 422     if (count % 2) {
 423         NSLog(@"WARNING: Uneven number of components (%d) in flush queue "
 424                 "message; ignoring this message.", count);
 425         return;
 426     }
 427
 428     inProcessCommandQueue = YES;
 429
 430     //NSLog(@"======== %s BEGIN ========", _cmd);
 431     for (i = 0; i < count; i += 2) {
 432         NSData *value = [queue objectAtIndex:i];
 433         NSData *data = [queue objectAtIndex:i+1];
 434
 435         int msgid = *((int*)[value bytes]);
 436 #if 0
 437         if (msgid != EnableMenuItemMsgID && msgid != AddMenuItemMsgID
 438                 && msgid != AddMenuMsgID) {
 439             NSLog(@"%s%s", _cmd, MessageStrings[msgid]);
 440         }
 441 #endif
 442
 443         [self handleMessage:msgid data:data];
 444     }
 445     //NSLog(@"======== %s  END  ========", _cmd);
 446
 447     if (shouldUpdateMainMenu) {
 448         [self updateMainMenu];
 449     }
 450
 451     [windowController processCommandQueueDidFinish];
 452
 453     inProcessCommandQueue = NO;
 454
 455     if ([sendQueue count] > 0) {
 456         @try {
 457             [backendProxy processInputAndData:sendQueue];
 458         }
 459         @catch (NSException *e) {
 460             // Connection timed out, just ignore this.
 461             //NSLog(@"WARNING! Connection timed out in %s", _cmd);
 462         }
 463
 464         [sendQueue removeAllObjects];
 465     }
 466 }
 467
 468 - (void)windowDidBecomeMain:(NSNotification *)notification
 469 {
 470     if (isInitialized)
 471         [self updateMainMenu];
 472 }
 473
 474 - (NSToolbarItem *)toolbar:(NSToolbar *)theToolbar
 475     itemForItemIdentifier:(NSString *)itemId
 476     willBeInsertedIntoToolbar:(BOOL)flag
 477 {
 478     NSToolbarItem *item = [toolbarItemDict objectForKey:itemId];
 479     if (!item) {
 480         NSLog(@"WARNING:  No toolbar item with id '%@'", itemId);
 481     }
 482
 483     return item;
 484 }
 485
 486 - (NSArray *)toolbarAllowedItemIdentifiers:(NSToolbar *)theToolbar
 487 {
 488     return nil;
 489 }
 490
 491 - (NSArray *)toolbarDefaultItemIdentifiers:(NSToolbar *)theToolbar
 492 {
 493     return nil;
 494 }
 495
 496 @end // MMVimController
 497
 498
 499
 500 @implementation MMVimController (Private)
 501
 502 - (void)handleMessage:(int)msgid data:(NSData *)data
 503 {
 504     //NSLog(@"%@ %s", [self className], _cmd);
 505
 506     if (OpenVimWindowMsgID == msgid) {
 507         [windowController openWindow];
 508     } else if (BatchDrawMsgID == msgid) {
 509         [self performBatchDrawWithData:data];
 510     } else if (SelectTabMsgID == msgid) {
 511 #if 0   // NOTE: Tab selection is done inside updateTabsWithData:.
 512         const void *bytes = [data bytes];
 513         int idx = *((int*)bytes);
 514         //NSLog(@"Selecting tab with index %d", idx);
 515         [windowController selectTabWithIndex:idx];
 516 #endif
 517     } else if (UpdateTabBarMsgID == msgid) {
 518         [windowController updateTabsWithData:data];
 519     } else if (ShowTabBarMsgID == msgid) {
 520         [windowController showTabBar:YES];
 521     } else if (HideTabBarMsgID == msgid) {
 522         [windowController showTabBar:NO];
 523     } else if (SetTextDimensionsMsgID == msgid) {
 524         const void *bytes = [data bytes];
 525         int rows = *((int*)bytes);  bytes += sizeof(int);
 526         int cols = *((int*)bytes);  bytes += sizeof(int);
 527
 528         [windowController setTextDimensionsWithRows:rows columns:cols];
 529     } else if (SetWindowTitleMsgID == msgid) {
 530         const void *bytes = [data bytes];
 531         int len = *((int*)bytes);  bytes += sizeof(int);
 532
 533         NSString *string = [[NSString alloc] initWithBytes:(void*)bytes
 534                 length:len encoding:NSUTF8StringEncoding];
 535
 536         [[windowController window] setTitle:string];
 537
 538         [string release];
 539     } else if (AddMenuMsgID == msgid) {
 540         NSString *title = nil;
 541         const void *bytes = [data bytes];
 542         int tag = *((int*)bytes);  bytes += sizeof(int);
 543         int parentTag = *((int*)bytes);  bytes += sizeof(int);
 544         int len = *((int*)bytes);  bytes += sizeof(int);
 545         if (len > 0) {
 546             title = [[NSString alloc] initWithBytes:(void*)bytes length:len
 547                                            encoding:NSUTF8StringEncoding];
 548             bytes += len;
 549         }
 550         int idx = *((int*)bytes);  bytes += sizeof(int);
 551
 552         if (MenuToolbarType == parentTag) {
 553             if (!toolbar) {
 554                 // NOTE! Each toolbar must have a unique identifier, else each
 555                 // window will have the same toolbar.
 556                 NSString *ident = [NSString stringWithFormat:@"%d.%d",
 557                          (int)self, tag];
 558                 toolbar = [[NSToolbar alloc] initWithIdentifier:ident];
 559
 560                 [toolbar setShowsBaselineSeparator:NO];
 561                 [toolbar setDelegate:self];
 562                 [toolbar setDisplayMode:NSToolbarDisplayModeIconOnly];
 563                 [toolbar setSizeMode:NSToolbarSizeModeSmall];
 564
 565                 NSWindow *win = [windowController window];
 566                 [win setToolbar:toolbar];
 567
 568                 // HACK! Redirect the pill button so that we can ask Vim to
 569                 // hide the toolbar.
 570                 NSButton *pillButton = [win
 571                     standardWindowButton:NSWindowToolbarButton];
 572                 if (pillButton) {
 573                     [pillButton setAction:@selector(toggleToolbar:)];
 574                     [pillButton setTarget:windowController];
 575                 }
 576             }
 577         } else if (title) {
 578             [self addMenuWithTag:tag parent:parentTag title:title atIndex:idx];
 579         }
 580
 581         [title release];
 582     } else if (AddMenuItemMsgID == msgid) {
 583         NSString *title = nil, *tip = nil, *icon = nil, *action = nil;
 584         const void *bytes = [data bytes];
 585         int tag = *((int*)bytes);  bytes += sizeof(int);
 586         int parentTag = *((int*)bytes);  bytes += sizeof(int);
 587         int namelen = *((int*)bytes);  bytes += sizeof(int);
 588         if (namelen > 0) {
 589             title = [[NSString alloc] initWithBytes:(void*)bytes length:namelen
 590                                            encoding:NSUTF8StringEncoding];
 591             bytes += namelen;
 592         }
 593         int tiplen = *((int*)bytes);  bytes += sizeof(int);
 594         if (tiplen > 0) {
 595             tip = [[NSString alloc] initWithBytes:(void*)bytes length:tiplen
 596                                            encoding:NSUTF8StringEncoding];
 597             bytes += tiplen;
 598         }
 599         int iconlen = *((int*)bytes);  bytes += sizeof(int);
 600         if (iconlen > 0) {
 601             icon = [[NSString alloc] initWithBytes:(void*)bytes length:iconlen
 602                                            encoding:NSUTF8StringEncoding];
 603             bytes += iconlen;
 604         }
 605         int actionlen = *((int*)bytes);  bytes += sizeof(int);
 606         if (actionlen > 0) {
 607             action = [[NSString alloc] initWithBytes:(void*)bytes
 608                                               length:actionlen
 609                                             encoding:NSUTF8StringEncoding];
 610             bytes += actionlen;
 611         }
 612         int idx = *((int*)bytes);  bytes += sizeof(int);
 613         if (idx < 0) idx = 0;
 614         int key = *((int*)bytes);  bytes += sizeof(int);
 615         int mask = *((int*)bytes);  bytes += sizeof(int);
 616
 617         NSString *ident = [NSString stringWithFormat:@"%d.%d",
 618                 (int)self, parentTag];
 619         if (toolbar && [[toolbar identifier] isEqual:ident]) {
 620             [self addToolbarItemWithTag:tag label:title tip:tip icon:icon
 621                                 atIndex:idx];
 622         } else {
 623             NSMenu *parent = [self menuForTag:parentTag];
 624             [self addMenuItemWithTag:tag parent:parent title:title tip:tip
 625                        keyEquivalent:key modifiers:mask action:action
 626                              atIndex:idx];
 627         }
 628
 629         [title release];
 630         [tip release];
 631         [icon release];
 632         [action release];
 633     } else if (RemoveMenuItemMsgID == msgid) {
 634         const void *bytes = [data bytes];
 635         int tag = *((int*)bytes);  bytes += sizeof(int);
 636
 637         id item;
 638         int idx;
 639         if ((item = [self toolbarItemForTag:tag index:&idx])) {
 640             [toolbar removeItemAtIndex:idx];
 641         } else if ((item = [self menuItemForTag:tag])) {
 642             [item retain];
 643
 644             if ([item menu] == [NSApp mainMenu] || ![item menu]) {
 645                 // NOTE: To be on the safe side we try to remove the item from
 646                 // both arrays (it is ok to call removeObject: even if an array
 647                 // does not contain the object to remove).
 648                 [mainMenuItems removeObject:item];
 649                 [popupMenuItems removeObject:item];
 650             }
 651
 652             if ([item menu])
 653                 [[item menu] removeItem:item];
 654
 655             [item release];
 656         }
 657     } else if (EnableMenuItemMsgID == msgid) {
 658         const void *bytes = [data bytes];
 659         int tag = *((int*)bytes);  bytes += sizeof(int);
 660         int state = *((int*)bytes);  bytes += sizeof(int);
 661
 662         id item = [self toolbarItemForTag:tag index:NULL];
 663         if (!item)
 664             item = [self menuItemForTag:tag];
 665
 666         [item setEnabled:state];
 667     } else if (ShowToolbarMsgID == msgid) {
 668         const void *bytes = [data bytes];
 669         int enable = *((int*)bytes);  bytes += sizeof(int);
 670         int flags = *((int*)bytes);  bytes += sizeof(int);
 671
 672         int mode = NSToolbarDisplayModeDefault;
 673         if (flags & ToolbarLabelFlag) {
 674             mode = flags & ToolbarIconFlag ? NSToolbarDisplayModeIconAndLabel
 675                     : NSToolbarDisplayModeLabelOnly;
 676         } else if (flags & ToolbarIconFlag) {
 677             mode = NSToolbarDisplayModeIconOnly;
 678         }
 679
 680         int size = flags & ToolbarSizeRegularFlag ? NSToolbarSizeModeRegular
 681                 : NSToolbarSizeModeSmall;
 682
 683         [windowController showToolbar:enable size:size mode:mode];
 684     } else if (CreateScrollbarMsgID == msgid) {
 685         const void *bytes = [data bytes];
 686         long ident = *((long*)bytes);  bytes += sizeof(long);
 687         int type = *((int*)bytes);  bytes += sizeof(int);
 688
 689         [windowController createScrollbarWithIdentifier:ident type:type];
 690     } else if (DestroyScrollbarMsgID == msgid) {
 691         const void *bytes = [data bytes];
 692         long ident = *((long*)bytes);  bytes += sizeof(long);
 693
 694         [windowController destroyScrollbarWithIdentifier:ident];
 695     } else if (ShowScrollbarMsgID == msgid) {
 696         const void *bytes = [data bytes];
 697         long ident = *((long*)bytes);  bytes += sizeof(long);
 698         int visible = *((int*)bytes);  bytes += sizeof(int);
 699
 700         [windowController showScrollbarWithIdentifier:ident state:visible];
 701     } else if (SetScrollbarPositionMsgID == msgid) {
 702         const void *bytes = [data bytes];
 703         long ident = *((long*)bytes);  bytes += sizeof(long);
 704         int pos = *((int*)bytes);  bytes += sizeof(int);
 705         int len = *((int*)bytes);  bytes += sizeof(int);
 706
 707         [windowController setScrollbarPosition:pos length:len
 708                                     identifier:ident];
 709     } else if (SetScrollbarThumbMsgID == msgid) {
 710         const void *bytes = [data bytes];
 711         long ident = *((long*)bytes);  bytes += sizeof(long);
 712         float val = *((float*)bytes);  bytes += sizeof(float);
 713         float prop = *((float*)bytes);  bytes += sizeof(float);
 714
 715         [windowController setScrollbarThumbValue:val proportion:prop
 716                                       identifier:ident];
 717     } else if (SetFontMsgID == msgid) {
 718         const void *bytes = [data bytes];
 719         float size = *((float*)bytes);  bytes += sizeof(float);
 720         int len = *((int*)bytes);  bytes += sizeof(int);
 721         NSString *name = [[NSString alloc]
 722                 initWithBytes:(void*)bytes length:len
 723                      encoding:NSUTF8StringEncoding];
 724         NSFont *font = [NSFont fontWithName:name size:size];
 725
 726         if (font)
 727             [windowController setFont:font];
 728
 729         [name release];
 730     } else if (SetDefaultColorsMsgID == msgid) {
 731         const void *bytes = [data bytes];
 732         unsigned bg = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 733         unsigned fg = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 734         NSColor *back = [NSColor colorWithArgbInt:bg];
 735         NSColor *fore = [NSColor colorWithRgbInt:fg];
 736
 737         [windowController setDefaultColorsBackground:back foreground:fore];
 738     } else if (ExecuteActionMsgID == msgid) {
 739         const void *bytes = [data bytes];
 740         int len = *((int*)bytes);  bytes += sizeof(int);
 741         NSString *actionName = [[NSString alloc]
 742                 initWithBytesNoCopy:(void*)bytes
 743                              length:len
 744                            encoding:NSUTF8StringEncoding
 745                        freeWhenDone:NO];
 746
 747         SEL sel = NSSelectorFromString(actionName);
 748         [NSApp sendAction:sel to:nil from:self];
 749
 750         [actionName release];
 751     } else if (ShowPopupMenuMsgID == msgid) {
 752         const void *bytes = [data bytes];
 753         int row = *((int*)bytes);  bytes += sizeof(int);
 754         int col = *((int*)bytes);  bytes += sizeof(int);
 755         int len = *((int*)bytes);  bytes += sizeof(int);
 756         NSString *title = [[NSString alloc]
 757                 initWithBytesNoCopy:(void*)bytes
 758                              length:len
 759                            encoding:NSUTF8StringEncoding
 760                        freeWhenDone:NO];
 761
 762         NSMenu *menu = [self topLevelMenuForTitle:title];
 763         if (menu) {
 764             [windowController popupMenu:menu atRow:row column:col];
 765         } else {
 766             NSLog(@"WARNING: Cannot popup menu with title %@; no such menu.",
 767                     title);
 768         }
 769
 770         [title release];
 771     } else if (SetMouseShapeMsgID == msgid) {
 772         const void *bytes = [data bytes];
 773         int shape = *((int*)bytes);  bytes += sizeof(int);
 774
 775         [windowController setMouseShape:shape];
 776     } else if (AdjustLinespaceMsgID == msgid) {
 777         const void *bytes = [data bytes];
 778         int linespace = *((int*)bytes);  bytes += sizeof(int);
 779
 780         [windowController adjustLinespace:linespace];
 781     } else if (ActivateMsgID == msgid) {
 782         [NSApp activateIgnoringOtherApps:YES];
 783         [[windowController window] makeKeyAndOrderFront:self];
 784     } else if (SetServerNameMsgID == msgid) {
 785         NSString *name = [[NSString alloc] initWithData:data
 786                                                encoding:NSUTF8StringEncoding];
 787         [self setServerName:name];
 788         [name release];
 789     } else if (EnterFullscreenMsgID == msgid) {
 790         [windowController enterFullscreen];
 791     } else if (LeaveFullscreenMsgID == msgid) {
 792         [windowController leaveFullscreen];
 793     } else {
 794         NSLog(@"WARNING: Unknown message received (msgid=%d)", msgid);
 795     }
 796 }
 797
 798
 799 #define MM_DEBUG_DRAWING 0
 800
 801 - (void)performBatchDrawWithData:(NSData *)data
 802 {
 803     // TODO!  Move to window controller.
 804     MMTextStorage *textStorage = [windowController textStorage];
 805     MMTextView *textView = [windowController textView];
 806     if (!(textStorage && textView))
 807         return;
 808
 809     const void *bytes = [data bytes];
 810     const void *end = bytes + [data length];
 811
 812 #if MM_DEBUG_DRAWING
 813     NSLog(@"====> BEGIN %s", _cmd);
 814 #endif
 815     [textStorage beginEditing];
 816
 817     // TODO: Sanity check input
 818
 819     while (bytes < end) {
 820         int type = *((int*)bytes);  bytes += sizeof(int);
 821
 822         if (ClearAllDrawType == type) {
 823 #if MM_DEBUG_DRAWING
 824             NSLog(@"   Clear all");
 825 #endif
 826             [textStorage clearAll];
 827         } else if (ClearBlockDrawType == type) {
 828             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 829             int row1 = *((int*)bytes);  bytes += sizeof(int);
 830             int col1 = *((int*)bytes);  bytes += sizeof(int);
 831             int row2 = *((int*)bytes);  bytes += sizeof(int);
 832             int col2 = *((int*)bytes);  bytes += sizeof(int);
 833
 834 #if MM_DEBUG_DRAWING
 835             NSLog(@"   Clear block (%d,%d) -> (%d,%d)", row1, col1,
 836                     row2,col2);
 837 #endif
 838             [textStorage clearBlockFromRow:row1 column:col1
 839                     toRow:row2 column:col2
 840                     color:[NSColor colorWithArgbInt:color]];
 841         } else if (DeleteLinesDrawType == type) {
 842             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 843             int row = *((int*)bytes);  bytes += sizeof(int);
 844             int count = *((int*)bytes);  bytes += sizeof(int);
 845             int bot = *((int*)bytes);  bytes += sizeof(int);
 846             int left = *((int*)bytes);  bytes += sizeof(int);
 847             int right = *((int*)bytes);  bytes += sizeof(int);
 848
 849 #if MM_DEBUG_DRAWING
 850             NSLog(@"   Delete %d line(s) from %d", count, row);
 851 #endif
 852             [textStorage deleteLinesFromRow:row lineCount:count
 853                     scrollBottom:bot left:left right:right
 854                            color:[NSColor colorWithArgbInt:color]];
 855         } else if (ReplaceStringDrawType == type) {
 856             int bg = *((int*)bytes);  bytes += sizeof(int);
 857             int fg = *((int*)bytes);  bytes += sizeof(int);
 858             int sp = *((int*)bytes);  bytes += sizeof(int);
 859             int row = *((int*)bytes);  bytes += sizeof(int);
 860             int col = *((int*)bytes);  bytes += sizeof(int);
 861             int flags = *((int*)bytes);  bytes += sizeof(int);
 862             int len = *((int*)bytes);  bytes += sizeof(int);
 863             NSString *string = [[NSString alloc]
 864                     initWithBytesNoCopy:(void*)bytes
 865                                  length:len
 866                                encoding:NSUTF8StringEncoding
 867                            freeWhenDone:NO];
 868             bytes += len;
 869
 870 #if MM_DEBUG_DRAWING
 871             NSLog(@"   Draw string at (%d,%d) length=%d flags=%d fg=0x%x "
 872                     "bg=0x%x sp=0x%x (%@)", row, col, len, flags, fg, bg, sp,
 873                     len > 0 ? [string substringToIndex:1] : @"");
 874 #endif
 875             // NOTE: If this is a call to draw the (block) cursor, then cancel
 876             // any previous request to draw the insertion point, or it might
 877             // get drawn as well.
 878             if (flags & DRAW_CURSOR) {
 879                 [textView setShouldDrawInsertionPoint:NO];
 880                 //NSColor *color = [NSColor colorWithRgbInt:bg];
 881                 //[textView drawInsertionPointAtRow:row column:col
 882                 //                            shape:MMInsertionPointBlock
 883                 //                            color:color];
 884             }
 885             [textStorage replaceString:string
 886                                  atRow:row column:col
 887                              withFlags:flags
 888                        foregroundColor:[NSColor colorWithRgbInt:fg]
 889                        backgroundColor:[NSColor colorWithArgbInt:bg]
 890                           specialColor:[NSColor colorWithRgbInt:sp]];
 891
 892             [string release];
 893         } else if (InsertLinesDrawType == type) {
 894             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 895             int row = *((int*)bytes);  bytes += sizeof(int);
 896             int count = *((int*)bytes);  bytes += sizeof(int);
 897             int bot = *((int*)bytes);  bytes += sizeof(int);
 898             int left = *((int*)bytes);  bytes += sizeof(int);
 899             int right = *((int*)bytes);  bytes += sizeof(int);
 900
 901 #if MM_DEBUG_DRAWING
 902             NSLog(@"   Insert %d line(s) at row %d", count, row);
 903 #endif
 904             [textStorage insertLinesAtRow:row lineCount:count
 905                              scrollBottom:bot left:left right:right
 906                                     color:[NSColor colorWithArgbInt:color]];
 907         } else if (DrawCursorDrawType == type) {
 908             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 909             int row = *((int*)bytes);  bytes += sizeof(int);
 910             int col = *((int*)bytes);  bytes += sizeof(int);
 911             int shape = *((int*)bytes);  bytes += sizeof(int);
 912             int percent = *((int*)bytes);  bytes += sizeof(int);
 913
 914 #if MM_DEBUG_DRAWING
 915             NSLog(@"   Draw cursor at (%d,%d)", row, col);
 916 #endif
 917             [textView drawInsertionPointAtRow:row column:col shape:shape
 918                                      fraction:percent
 919                                         color:[NSColor colorWithRgbInt:color]];
 920         } else {
 921             NSLog(@"WARNING: Unknown draw type (type=%d)", type);
 922         }
 923     }
 924
 925     [textStorage endEditing];
 926 #if MM_DEBUG_DRAWING
 927     NSLog(@"<==== END   %s", _cmd);
 928 #endif
 929 }
 930
 931 - (void)savePanelDidEnd:(NSSavePanel *)panel code:(int)code
 932                 context:(void *)context
 933 {
 934     NSString *string = (code == NSOKButton) ? [panel filename] : nil;
 935     @try {
 936         [backendProxy setDialogReturn:string];
 937     }
 938     @catch (NSException *e) {
 939         NSLog(@"Exception caught in %s %@", _cmd, e);
 940     }
 941 }
 942
 943 - (void)alertDidEnd:(MMAlert *)alert code:(int)code context:(void *)context
 944 {
 945     NSArray *ret = nil;
 946
 947     code = code - NSAlertFirstButtonReturn + 1;
 948
 949     if ([alert isKindOfClass:[MMAlert class]] && [alert textField]) {
 950         ret = [NSArray arrayWithObjects:[NSNumber numberWithInt:code],
 951             [[alert textField] stringValue], nil];
 952     } else {
 953         ret = [NSArray arrayWithObject:[NSNumber numberWithInt:code]];
 954     }
 955
 956     @try {
 957         [backendProxy setDialogReturn:ret];
 958     }
 959     @catch (NSException *e) {
 960         NSLog(@"Exception caught in %s %@", _cmd, e);
 961     }
 962 }
 963
 964 - (NSMenuItem *)menuItemForTag:(int)tag
 965 {
 966     // Search the main menu.
 967     int i, count = [mainMenuItems count];
 968     for (i = 0; i < count; ++i) {
 969         NSMenuItem *item = [mainMenuItems objectAtIndex:i];
 970         if ([item tag] == tag) return item;
 971         item = findMenuItemWithTagInMenu([item submenu], tag);
 972         if (item) return item;
 973     }
 974
 975     // Search the popup menus.
 976     count = [popupMenuItems count];
 977     for (i = 0; i < count; ++i) {
 978         NSMenuItem *item = [popupMenuItems objectAtIndex:i];
 979         if ([item tag] == tag) return item;
 980         item = findMenuItemWithTagInMenu([item submenu], tag);
 981         if (item) return item;
 982     }
 983
 984     return nil;
 985 }
 986
 987 - (NSMenu *)menuForTag:(int)tag
 988 {
 989     return [[self menuItemForTag:tag] submenu];
 990 }
 991
 992 - (NSMenu *)topLevelMenuForTitle:(NSString *)title
 993 {
 994     // Search only the top-level menus.
 995
 996     unsigned i, count = [popupMenuItems count];
 997     for (i = 0; i < count; ++i) {
 998         NSMenuItem *item = [popupMenuItems objectAtIndex:i];
 999         if ([title isEqual:[item title]])
1000             return [item submenu];
1001     }
1002
1003     count = [mainMenuItems count];
1004     for (i = 0; i < count; ++i) {
1005         NSMenuItem *item = [mainMenuItems objectAtIndex:i];
1006         if ([title isEqual:[item title]])
1007             return [item submenu];
1008     }
1009
1010     return nil;
1011 }
1012
1013 - (void)addMenuWithTag:(int)tag parent:(int)parentTag title:(NSString *)title
1014                atIndex:(int)idx
1015 {
1016     NSMenu *parent = [self menuForTag:parentTag];
1017     NSMenuItem *item = [[NSMenuItem alloc] init];
1018     NSMenu *menu = [[NSMenu alloc] initWithTitle:title];
1019
1020     [menu setAutoenablesItems:NO];
1021     [item setTag:tag];
1022     [item setTitle:title];
1023     [item setSubmenu:menu];
1024
1025     if (parent) {
1026         if ([parent numberOfItems] <= idx) {
1027             [parent addItem:item];
1028         } else {
1029             [parent insertItem:item atIndex:idx];
1030         }
1031     } else {
1032         NSMutableArray *items = (MenuPopupType == parentTag)
1033             ? popupMenuItems : mainMenuItems;
1034         if ([items count] <= idx) {
1035             [items addObject:item];
1036         } else {
1037             [items insertObject:item atIndex:idx];
1038         }
1039
1040         shouldUpdateMainMenu = (MenuPopupType != parentTag);
1041     }
1042
1043     [item release];
1044     [menu release];
1045 }
1046
1047 - (void)addMenuItemWithTag:(int)tag parent:(NSMenu *)parent
1048                      title:(NSString *)title tip:(NSString *)tip
1049              keyEquivalent:(int)key modifiers:(int)mask
1050                     action:(NSString *)action atIndex:(int)idx
1051 {
1052     if (parent) {
1053         NSMenuItem *item = nil;
1054         if (!title || ([title hasPrefix:@"-"] && [title hasSuffix:@"-"])) {
1055             item = [NSMenuItem separatorItem];
1056         } else {
1057             item = [[[NSMenuItem alloc] init] autorelease];
1058             [item setTitle:title];
1059             // TODO: Check that 'action' is a valid action (nothing will happen
1060             // if it isn't, but it would be nice with a warning).
1061             if (action) [item setAction:NSSelectorFromString(action)];
1062             else        [item setAction:@selector(vimMenuItemAction:)];
1063             if (tip) [item setToolTip:tip];
1064
1065             if (key != 0) {
1066                 NSString *keyString =
1067                     [NSString stringWithFormat:@"%C", key];
1068                 [item setKeyEquivalent:keyString];
1069                 [item setKeyEquivalentModifierMask:mask];
1070             }
1071         }
1072
1073         // NOTE!  The tag is used to idenfity which menu items were
1074         // added by Vim (tag != 0) and which were added by the AppKit
1075         // (tag == 0).
1076         [item setTag:tag];
1077
1078         if ([parent numberOfItems] <= idx) {
1079             [parent addItem:item];
1080         } else {
1081             [parent insertItem:item atIndex:idx];
1082         }
1083     } else {
1084         NSLog(@"WARNING: Menu item '%@' (tag=%d) has no parent.", title, tag);
1085     }
1086 }
1087
1088 - (void)updateMainMenu
1089 {
1090     NSMenu *mainMenu = [NSApp mainMenu];
1091
1092     // Stop NSApp from updating the Window menu.
1093     [NSApp setWindowsMenu:nil];
1094
1095     // Remove all menus from main menu (except the MacVim menu).
1096     int i, count = [mainMenu numberOfItems];
1097     for (i = count-1; i > 0; --i) {
1098         [mainMenu removeItemAtIndex:i];
1099     }
1100
1101     // Add menus from 'mainMenuItems' to main menu.
1102     count = [mainMenuItems count];
1103     for (i = 0; i < count; ++i) {
1104         [mainMenu addItem:[mainMenuItems objectAtIndex:i]];
1105     }
1106
1107     // Set the new Window menu.
1108     // TODO!  Need to look for 'Window' in all localized languages.
1109     NSMenu *windowMenu = [[mainMenu itemWithTitle:@"Window"] submenu];
1110     if (windowMenu) {
1111         // Remove all AppKit owned menu items (tag == 0); they will be added
1112         // again when setWindowsMenu: is called.
1113         count = [windowMenu numberOfItems];
1114         for (i = count-1; i >= 0; --i) {
1115             NSMenuItem *item = [windowMenu itemAtIndex:i];
1116             if (![item tag]) {
1117                 [windowMenu removeItem:item];
1118             }
1119         }
1120
1121         [NSApp setWindowsMenu:windowMenu];
1122     }
1123
1124     shouldUpdateMainMenu = NO;
1125 }
1126
1127 - (NSToolbarItem *)toolbarItemForTag:(int)tag index:(int *)index
1128 {
1129     if (!toolbar) return nil;
1130
1131     NSArray *items = [toolbar items];
1132     int i, count = [items count];
1133     for (i = 0; i < count; ++i) {
1134         NSToolbarItem *item = [items objectAtIndex:i];
1135         if ([item tag] == tag) {
1136             if (index) *index = i;
1137             return item;
1138         }
1139     }
1140
1141     return nil;
1142 }
1143
1144 - (void)addToolbarItemToDictionaryWithTag:(int)tag label:(NSString *)title
1145         toolTip:(NSString *)tip icon:(NSString *)icon
1146 {
1147     // If the item corresponds to a separator then do nothing, since it is
1148     // already defined by Cocoa.
1149     if (!title || [title isEqual:NSToolbarSeparatorItemIdentifier]
1150                || [title isEqual:NSToolbarSpaceItemIdentifier]
1151                || [title isEqual:NSToolbarFlexibleSpaceItemIdentifier])
1152         return;
1153
1154     NSToolbarItem *item = [[NSToolbarItem alloc] initWithItemIdentifier:title];
1155     [item setTag:tag];
1156     [item setLabel:title];
1157     [item setToolTip:tip];
1158     [item setAction:@selector(vimMenuItemAction:)];
1159     [item setAutovalidates:NO];
1160
1161     NSImage *img = [NSImage imageNamed:icon];
1162     if (!img) {
1163         NSLog(@"WARNING: Could not find image with name '%@' to use as toolbar"
1164                " image for identifier '%@';"
1165                " using default toolbar icon '%@' instead.",
1166                icon, title, MMDefaultToolbarImageName);
1167
1168         img = [NSImage imageNamed:MMDefaultToolbarImageName];
1169     }
1170
1171     [item setImage:img];
1172
1173     [toolbarItemDict setObject:item forKey:title];
1174
1175     [item release];
1176 }
1177
1178 - (void)addToolbarItemWithTag:(int)tag label:(NSString *)label tip:(NSString
1179                    *)tip icon:(NSString *)icon atIndex:(int)idx
1180 {
1181     if (!toolbar) return;
1182
1183     // Check for separator items.
1184     if (!label) {
1185         label = NSToolbarSeparatorItemIdentifier;
1186     } else if ([label length] >= 2 && [label hasPrefix:@"-"]
1187                                    && [label hasSuffix:@"-"]) {
1188         // The label begins and ends with '-'; decided which kind of separator
1189         // item it is by looking at the prefix.
1190         if ([label hasPrefix:@"-space"]) {
1191             label = NSToolbarSpaceItemIdentifier;
1192         } else if ([label hasPrefix:@"-flexspace"]) {
1193             label = NSToolbarFlexibleSpaceItemIdentifier;
1194         } else {
1195             label = NSToolbarSeparatorItemIdentifier;
1196         }
1197     }
1198
1199     [self addToolbarItemToDictionaryWithTag:tag label:label toolTip:tip
1200                                        icon:icon];
1201
1202     int maxIdx = [[toolbar items] count];
1203     if (maxIdx < idx) idx = maxIdx;
1204
1205     [toolbar insertItemWithItemIdentifier:label atIndex:idx];
1206 }
1207
1208 - (void)connectionDidDie:(NSNotification *)notification
1209 {
1210     //NSLog(@"%@ %s%@", [self className], _cmd, notification);
1211
1212     [self cleanup];
1213
1214     // NOTE!  This causes the call to removeVimController: to be delayed.
1215     [[NSApp delegate]
1216             performSelectorOnMainThread:@selector(removeVimController:)
1217                              withObject:self waitUntilDone:NO];
1218 }
1219
1220 - (NSString *)description
1221 {
1222     return [NSString stringWithFormat:@"%@ : isInitialized=%d inProcessCommandQueue=%d mainMenuItems=%@ popupMenuItems=%@ toolbar=%@", [self className], isInitialized, inProcessCommandQueue, mainMenuItems, popupMenuItems, toolbar];
1223 }
1224
1225 #if MM_RESEND_LAST_FAILURE
1226 - (void)resendTimerFired:(NSTimer *)timer
1227 {
1228     int msgid = resendMsgid;
1229     NSData *data = nil;
1230
1231     [resendTimer release];
1232     resendTimer = nil;
1233
1234     if (!isInitialized)
1235         return;
1236
1237     if (resendData)
1238         data = [resendData copy];
1239
1240     //NSLog(@"Resending message: %s", MessageStrings[msgid]);
1241     [self sendMessage:msgid data:data];
1242 }
1243 #endif
1244
1245 @end // MMVimController (Private)
1246
1247
1248
1249 @implementation NSColor (MMProtocol)
1250
1251 + (NSColor *)colorWithRgbInt:(unsigned)rgb
1252 {
1253     float r = ((rgb>>16) & 0xff)/255.0f;
1254     float g = ((rgb>>8) & 0xff)/255.0f;
1255     float b = (rgb & 0xff)/255.0f;
1256
1257     return [NSColor colorWithCalibratedRed:r green:g blue:b alpha:1.0f];
1258 }
1259
1260 + (NSColor *)colorWithArgbInt:(unsigned)argb
1261 {
1262     float a = ((argb>>24) & 0xff)/255.0f;
1263     float r = ((argb>>16) & 0xff)/255.0f;
1264     float g = ((argb>>8) & 0xff)/255.0f;
1265     float b = (argb & 0xff)/255.0f;
1266
1267     return [NSColor colorWithCalibratedRed:r green:g blue:b alpha:a];
1268 }
1269
1270 @end // NSColor (MMProtocol)
1271
1272
1273
1274 @implementation MMAlert
1275 - (void)dealloc
1276 {
1277     [textField release];
1278     [super dealloc];
1279 }
1280
1281 - (void)setTextFieldString:(NSString *)textFieldString
1282 {
1283     [textField release];
1284     textField = [[NSTextField alloc] init];
1285     [textField setStringValue:textFieldString];
1286 }
1287
1288 - (NSTextField *)textField
1289 {
1290     return textField;
1291 }
1292
1293 - (void)setInformativeText:(NSString *)text
1294 {
1295     if (textField) {
1296         // HACK! Add some space for the text field.
1297         [super setInformativeText:[text stringByAppendingString:@"\n\n\n"]];
1298     } else {
1299         [super setInformativeText:text];
1300     }
1301 }
1302
1303 - (void)beginSheetModalForWindow:(NSWindow *)window
1304                    modalDelegate:(id)delegate
1305                   didEndSelector:(SEL)didEndSelector
1306                      contextInfo:(void *)contextInfo
1307 {
1308     [super beginSheetModalForWindow:window
1309                       modalDelegate:delegate
1310                      didEndSelector:didEndSelector
1311                         contextInfo:contextInfo];
1312
1313     // HACK! Place the input text field at the bottom of the informative text
1314     // (which has been made a bit larger by adding newline characters).
1315     NSView *contentView = [[self window] contentView];
1316     NSRect rect = [contentView frame];
1317     rect.origin.y = rect.size.height;
1318
1319     NSArray *subviews = [contentView subviews];
1320     unsigned i, count = [subviews count];
1321     for (i = 0; i < count; ++i) {
1322         NSView *view = [subviews objectAtIndex:i];
1323         if ([view isKindOfClass:[NSTextField class]]
1324                 && [view frame].origin.y < rect.origin.y) {
1325             // NOTE: The informative text field is the lowest NSTextField in
1326             // the alert dialog.
1327             rect = [view frame];
1328         }
1329     }
1330
1331     rect.size.height = MMAlertTextFieldHeight;
1332     [textField setFrame:rect];
1333     [contentView addSubview:textField];
1334     [textField becomeFirstResponder];
1335 }
1336
1337 @end // MMAlert