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;  // append 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 {
 790         NSLog(@"WARNING: Unknown message received (msgid=%d)", msgid);
 791     }
 792 }
 793
 794
 795 #define MM_DEBUG_DRAWING 0
 796
 797 - (void)performBatchDrawWithData:(NSData *)data
 798 {
 799     // TODO!  Move to window controller.
 800     MMTextStorage *textStorage = [windowController textStorage];
 801     MMTextView *textView = [windowController textView];
 802     if (!(textStorage && textView))
 803         return;
 804
 805     const void *bytes = [data bytes];
 806     const void *end = bytes + [data length];
 807
 808 #if MM_DEBUG_DRAWING
 809     NSLog(@"====> BEGIN %s", _cmd);
 810 #endif
 811     [textStorage beginEditing];
 812
 813     // TODO: Sanity check input
 814
 815     while (bytes < end) {
 816         int type = *((int*)bytes);  bytes += sizeof(int);
 817
 818         if (ClearAllDrawType == type) {
 819 #if MM_DEBUG_DRAWING
 820             NSLog(@"   Clear all");
 821 #endif
 822             [textStorage clearAll];
 823         } else if (ClearBlockDrawType == type) {
 824             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 825             int row1 = *((int*)bytes);  bytes += sizeof(int);
 826             int col1 = *((int*)bytes);  bytes += sizeof(int);
 827             int row2 = *((int*)bytes);  bytes += sizeof(int);
 828             int col2 = *((int*)bytes);  bytes += sizeof(int);
 829
 830 #if MM_DEBUG_DRAWING
 831             NSLog(@"   Clear block (%d,%d) -> (%d,%d)", row1, col1,
 832                     row2,col2);
 833 #endif
 834             [textStorage clearBlockFromRow:row1 column:col1
 835                     toRow:row2 column:col2
 836                     color:[NSColor colorWithArgbInt:color]];
 837         } else if (DeleteLinesDrawType == type) {
 838             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 839             int row = *((int*)bytes);  bytes += sizeof(int);
 840             int count = *((int*)bytes);  bytes += sizeof(int);
 841             int bot = *((int*)bytes);  bytes += sizeof(int);
 842             int left = *((int*)bytes);  bytes += sizeof(int);
 843             int right = *((int*)bytes);  bytes += sizeof(int);
 844
 845 #if MM_DEBUG_DRAWING
 846             NSLog(@"   Delete %d line(s) from %d", count, row);
 847 #endif
 848             [textStorage deleteLinesFromRow:row lineCount:count
 849                     scrollBottom:bot left:left right:right
 850                            color:[NSColor colorWithArgbInt:color]];
 851         } else if (ReplaceStringDrawType == type) {
 852             int bg = *((int*)bytes);  bytes += sizeof(int);
 853             int fg = *((int*)bytes);  bytes += sizeof(int);
 854             int sp = *((int*)bytes);  bytes += sizeof(int);
 855             int row = *((int*)bytes);  bytes += sizeof(int);
 856             int col = *((int*)bytes);  bytes += sizeof(int);
 857             int flags = *((int*)bytes);  bytes += sizeof(int);
 858             int len = *((int*)bytes);  bytes += sizeof(int);
 859             NSString *string = [[NSString alloc]
 860                     initWithBytesNoCopy:(void*)bytes
 861                                  length:len
 862                                encoding:NSUTF8StringEncoding
 863                            freeWhenDone:NO];
 864             bytes += len;
 865
 866 #if MM_DEBUG_DRAWING
 867             NSLog(@"   Draw string at (%d,%d) length=%d flags=%d fg=0x%x "
 868                     "bg=0x%x sp=0x%x (%@)", row, col, len, flags, fg, bg, sp,
 869                     len > 0 ? [string substringToIndex:1] : @"");
 870 #endif
 871             // NOTE: If this is a call to draw the (block) cursor, then cancel
 872             // any previous request to draw the insertion point, or it might
 873             // get drawn as well.
 874             if (flags & DRAW_CURSOR) {
 875                 [textView setShouldDrawInsertionPoint:NO];
 876                 //[textView drawInsertionPointAtRow:row column:col
 877                 //                            shape:MMInsertionPointBlock
 878                 //                            color:[NSColor colorWithRgbInt:bg]];
 879             }
 880             [textStorage replaceString:string
 881                                  atRow:row column:col
 882                              withFlags:flags
 883                        foregroundColor:[NSColor colorWithRgbInt:fg]
 884                        backgroundColor:[NSColor colorWithArgbInt:bg]
 885                           specialColor:[NSColor colorWithRgbInt:sp]];
 886
 887             [string release];
 888         } else if (InsertLinesDrawType == type) {
 889             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 890             int row = *((int*)bytes);  bytes += sizeof(int);
 891             int count = *((int*)bytes);  bytes += sizeof(int);
 892             int bot = *((int*)bytes);  bytes += sizeof(int);
 893             int left = *((int*)bytes);  bytes += sizeof(int);
 894             int right = *((int*)bytes);  bytes += sizeof(int);
 895
 896 #if MM_DEBUG_DRAWING
 897             NSLog(@"   Insert %d line(s) at row %d", count, row);
 898 #endif
 899             [textStorage insertLinesAtRow:row lineCount:count
 900                              scrollBottom:bot left:left right:right
 901                                     color:[NSColor colorWithArgbInt:color]];
 902         } else if (DrawCursorDrawType == type) {
 903             unsigned color = *((unsigned*)bytes);  bytes += sizeof(unsigned);
 904             int row = *((int*)bytes);  bytes += sizeof(int);
 905             int col = *((int*)bytes);  bytes += sizeof(int);
 906             int shape = *((int*)bytes);  bytes += sizeof(int);
 907             int percent = *((int*)bytes);  bytes += sizeof(int);
 908
 909 #if MM_DEBUG_DRAWING
 910             NSLog(@"   Draw cursor at (%d,%d)", row, col);
 911 #endif
 912             [textView drawInsertionPointAtRow:row column:col shape:shape
 913                                      fraction:percent
 914                                         color:[NSColor colorWithRgbInt:color]];
 915         } else {
 916             NSLog(@"WARNING: Unknown draw type (type=%d)", type);
 917         }
 918     }
 919
 920     [textStorage endEditing];
 921 #if MM_DEBUG_DRAWING
 922     NSLog(@"<==== END   %s", _cmd);
 923 #endif
 924 }
 925
 926 - (void)savePanelDidEnd:(NSSavePanel *)panel code:(int)code
 927                 context:(void *)context
 928 {
 929     NSString *string = (code == NSOKButton) ? [panel filename] : nil;
 930     @try {
 931         [backendProxy setDialogReturn:string];
 932     }
 933     @catch (NSException *e) {
 934         NSLog(@"Exception caught in %s %@", _cmd, e);
 935     }
 936 }
 937
 938 - (void)alertDidEnd:(MMAlert *)alert code:(int)code context:(void *)context
 939 {
 940     NSArray *ret = nil;
 941
 942     code = code - NSAlertFirstButtonReturn + 1;
 943
 944     if ([alert isKindOfClass:[MMAlert class]] && [alert textField]) {
 945         ret = [NSArray arrayWithObjects:[NSNumber numberWithInt:code],
 946             [[alert textField] stringValue], nil];
 947     } else {
 948         ret = [NSArray arrayWithObject:[NSNumber numberWithInt:code]];
 949     }
 950
 951     @try {
 952         [backendProxy setDialogReturn:ret];
 953     }
 954     @catch (NSException *e) {
 955         NSLog(@"Exception caught in %s %@", _cmd, e);
 956     }
 957 }
 958
 959 - (NSMenuItem *)menuItemForTag:(int)tag
 960 {
 961     // Search the main menu.
 962     int i, count = [mainMenuItems count];
 963     for (i = 0; i < count; ++i) {
 964         NSMenuItem *item = [mainMenuItems objectAtIndex:i];
 965         if ([item tag] == tag) return item;
 966         item = findMenuItemWithTagInMenu([item submenu], tag);
 967         if (item) return item;
 968     }
 969
 970     // Search the popup menus.
 971     count = [popupMenuItems count];
 972     for (i = 0; i < count; ++i) {
 973         NSMenuItem *item = [popupMenuItems objectAtIndex:i];
 974         if ([item tag] == tag) return item;
 975         item = findMenuItemWithTagInMenu([item submenu], tag);
 976         if (item) return item;
 977     }
 978
 979     return nil;
 980 }
 981
 982 - (NSMenu *)menuForTag:(int)tag
 983 {
 984     return [[self menuItemForTag:tag] submenu];
 985 }
 986
 987 - (NSMenu *)topLevelMenuForTitle:(NSString *)title
 988 {
 989     // Search only the top-level menus.
 990
 991     unsigned i, count = [popupMenuItems count];
 992     for (i = 0; i < count; ++i) {
 993         NSMenuItem *item = [popupMenuItems objectAtIndex:i];
 994         if ([title isEqual:[item title]])
 995             return [item submenu];
 996     }
 997
 998     count = [mainMenuItems count];
 999     for (i = 0; i < count; ++i) {
1000         NSMenuItem *item = [mainMenuItems objectAtIndex:i];
1001         if ([title isEqual:[item title]])
1002             return [item submenu];
1003     }
1004
1005     return nil;
1006 }
1007
1008 - (void)addMenuWithTag:(int)tag parent:(int)parentTag title:(NSString *)title
1009                atIndex:(int)idx
1010 {
1011     NSMenu *parent = [self menuForTag:parentTag];
1012     NSMenuItem *item = [[NSMenuItem alloc] init];
1013     NSMenu *menu = [[NSMenu alloc] initWithTitle:title];
1014
1015     [menu setAutoenablesItems:NO];
1016     [item setTag:tag];
1017     [item setTitle:title];
1018     [item setSubmenu:menu];
1019
1020     if (parent) {
1021         if ([parent numberOfItems] <= idx) {
1022             [parent addItem:item];
1023         } else {
1024             [parent insertItem:item atIndex:idx];
1025         }
1026     } else {
1027         NSMutableArray *items = (MenuPopupType == parentTag)
1028             ? popupMenuItems : mainMenuItems;
1029         if ([items count] <= idx) {
1030             [items addObject:item];
1031         } else {
1032             [items insertObject:item atIndex:idx];
1033         }
1034
1035         shouldUpdateMainMenu = (MenuPopupType != parentTag);
1036     }
1037
1038     [item release];
1039     [menu release];
1040 }
1041
1042 - (void)addMenuItemWithTag:(int)tag parent:(NSMenu *)parent
1043                      title:(NSString *)title tip:(NSString *)tip
1044              keyEquivalent:(int)key modifiers:(int)mask
1045                     action:(NSString *)action atIndex:(int)idx
1046 {
1047     if (parent) {
1048         NSMenuItem *item = nil;
1049         if (!title || ([title hasPrefix:@"-"] && [title hasSuffix:@"-"])) {
1050             item = [NSMenuItem separatorItem];
1051         } else {
1052             item = [[[NSMenuItem alloc] init] autorelease];
1053             [item setTitle:title];
1054             // TODO: Check that 'action' is a valid action (nothing will happen
1055             // if it isn't, but it would be nice with a warning).
1056             if (action) [item setAction:NSSelectorFromString(action)];
1057             else        [item setAction:@selector(vimMenuItemAction:)];
1058             if (tip) [item setToolTip:tip];
1059
1060             if (key != 0) {
1061                 NSString *keyString =
1062                     [NSString stringWithFormat:@"%C", key];
1063                 [item setKeyEquivalent:keyString];
1064                 [item setKeyEquivalentModifierMask:mask];
1065             }
1066         }
1067
1068         // NOTE!  The tag is used to idenfity which menu items were
1069         // added by Vim (tag != 0) and which were added by the AppKit
1070         // (tag == 0).
1071         [item setTag:tag];
1072
1073         if ([parent numberOfItems] <= idx) {
1074             [parent addItem:item];
1075         } else {
1076             [parent insertItem:item atIndex:idx];
1077         }
1078     } else {
1079         NSLog(@"WARNING: Menu item '%@' (tag=%d) has no parent.", title, tag);
1080     }
1081 }
1082
1083 - (void)updateMainMenu
1084 {
1085     NSMenu *mainMenu = [NSApp mainMenu];
1086
1087     // Stop NSApp from updating the Window menu.
1088     [NSApp setWindowsMenu:nil];
1089
1090     // Remove all menus from main menu (except the MacVim menu).
1091     int i, count = [mainMenu numberOfItems];
1092     for (i = count-1; i > 0; --i) {
1093         [mainMenu removeItemAtIndex:i];
1094     }
1095
1096     // Add menus from 'mainMenuItems' to main menu.
1097     count = [mainMenuItems count];
1098     for (i = 0; i < count; ++i) {
1099         [mainMenu addItem:[mainMenuItems objectAtIndex:i]];
1100     }
1101
1102     // Set the new Window menu.
1103     // TODO!  Need to look for 'Window' in all localized languages.
1104     NSMenu *windowMenu = [[mainMenu itemWithTitle:@"Window"] submenu];
1105     if (windowMenu) {
1106         // Remove all AppKit owned menu items (tag == 0); they will be added
1107         // again when setWindowsMenu: is called.
1108         count = [windowMenu numberOfItems];
1109         for (i = count-1; i >= 0; --i) {
1110             NSMenuItem *item = [windowMenu itemAtIndex:i];
1111             if (![item tag]) {
1112                 [windowMenu removeItem:item];
1113             }
1114         }
1115
1116         [NSApp setWindowsMenu:windowMenu];
1117     }
1118
1119     shouldUpdateMainMenu = NO;
1120 }
1121
1122 - (NSToolbarItem *)toolbarItemForTag:(int)tag index:(int *)index
1123 {
1124     if (!toolbar) return nil;
1125
1126     NSArray *items = [toolbar items];
1127     int i, count = [items count];
1128     for (i = 0; i < count; ++i) {
1129         NSToolbarItem *item = [items objectAtIndex:i];
1130         if ([item tag] == tag) {
1131             if (index) *index = i;
1132             return item;
1133         }
1134     }
1135
1136     return nil;
1137 }
1138
1139 - (void)addToolbarItemToDictionaryWithTag:(int)tag label:(NSString *)title
1140         toolTip:(NSString *)tip icon:(NSString *)icon
1141 {
1142     // If the item corresponds to a separator then do nothing, since it is
1143     // already defined by Cocoa.
1144     if (!title || [title isEqual:NSToolbarSeparatorItemIdentifier]
1145                || [title isEqual:NSToolbarSpaceItemIdentifier]
1146                || [title isEqual:NSToolbarFlexibleSpaceItemIdentifier])
1147         return;
1148
1149     NSToolbarItem *item = [[NSToolbarItem alloc] initWithItemIdentifier:title];
1150     [item setTag:tag];
1151     [item setLabel:title];
1152     [item setToolTip:tip];
1153     [item setAction:@selector(vimMenuItemAction:)];
1154     [item setAutovalidates:NO];
1155
1156     NSImage *img = [NSImage imageNamed:icon];
1157     if (!img) {
1158         NSLog(@"WARNING: Could not find image with name '%@' to use as toolbar"
1159                " image for identifier '%@';"
1160                " using default toolbar icon '%@' instead.",
1161                icon, title, MMDefaultToolbarImageName);
1162
1163         img = [NSImage imageNamed:MMDefaultToolbarImageName];
1164     }
1165
1166     [item setImage:img];
1167
1168     [toolbarItemDict setObject:item forKey:title];
1169
1170     [item release];
1171 }
1172
1173 - (void)addToolbarItemWithTag:(int)tag label:(NSString *)label tip:(NSString
1174                    *)tip icon:(NSString *)icon atIndex:(int)idx
1175 {
1176     if (!toolbar) return;
1177
1178     // Check for separator items.
1179     if (!label) {
1180         label = NSToolbarSeparatorItemIdentifier;
1181     } else if ([label length] >= 2 && [label hasPrefix:@"-"]
1182                                    && [label hasSuffix:@"-"]) {
1183         // The label begins and ends with '-'; decided which kind of separator
1184         // item it is by looking at the prefix.
1185         if ([label hasPrefix:@"-space"]) {
1186             label = NSToolbarSpaceItemIdentifier;
1187         } else if ([label hasPrefix:@"-flexspace"]) {
1188             label = NSToolbarFlexibleSpaceItemIdentifier;
1189         } else {
1190             label = NSToolbarSeparatorItemIdentifier;
1191         }
1192     }
1193
1194     [self addToolbarItemToDictionaryWithTag:tag label:label toolTip:tip
1195                                        icon:icon];
1196
1197     int maxIdx = [[toolbar items] count];
1198     if (maxIdx < idx) idx = maxIdx;
1199
1200     [toolbar insertItemWithItemIdentifier:label atIndex:idx];
1201 }
1202
1203 - (void)connectionDidDie:(NSNotification *)notification
1204 {
1205     //NSLog(@"%@ %s%@", [self className], _cmd, notification);
1206
1207     [self cleanup];
1208
1209     // NOTE!  This causes the call to removeVimController: to be delayed.
1210     [[NSApp delegate]
1211             performSelectorOnMainThread:@selector(removeVimController:)
1212                              withObject:self waitUntilDone:NO];
1213 }
1214
1215 - (NSString *)description
1216 {
1217     return [NSString stringWithFormat:@"%@ : isInitialized=%d inProcessCommandQueue=%d mainMenuItems=%@ popupMenuItems=%@ toolbar=%@", [self className], isInitialized, inProcessCommandQueue, mainMenuItems, popupMenuItems, toolbar];
1218 }
1219
1220 #if MM_RESEND_LAST_FAILURE
1221 - (void)resendTimerFired:(NSTimer *)timer
1222 {
1223     int msgid = resendMsgid;
1224     NSData *data = nil;
1225
1226     [resendTimer release];
1227     resendTimer = nil;
1228
1229     if (!isInitialized)
1230         return;
1231
1232     if (resendData)
1233         data = [resendData copy];
1234
1235     //NSLog(@"Resending message: %s", MessageStrings[msgid]);
1236     [self sendMessage:msgid data:data];
1237 }
1238 #endif
1239
1240 @end // MMVimController (Private)
1241
1242
1243
1244 @implementation NSColor (MMProtocol)
1245
1246 + (NSColor *)colorWithRgbInt:(unsigned)rgb
1247 {
1248     float r = ((rgb>>16) & 0xff)/255.0f;
1249     float g = ((rgb>>8) & 0xff)/255.0f;
1250     float b = (rgb & 0xff)/255.0f;
1251
1252     return [NSColor colorWithCalibratedRed:r green:g blue:b alpha:1.0f];
1253 }
1254
1255 + (NSColor *)colorWithArgbInt:(unsigned)argb
1256 {
1257     float a = ((argb>>24) & 0xff)/255.0f;
1258     float r = ((argb>>16) & 0xff)/255.0f;
1259     float g = ((argb>>8) & 0xff)/255.0f;
1260     float b = (argb & 0xff)/255.0f;
1261
1262     return [NSColor colorWithCalibratedRed:r green:g blue:b alpha:a];
1263 }
1264
1265 @end // NSColor (MMProtocol)
1266
1267
1268
1269 @implementation MMAlert
1270 - (void)dealloc
1271 {
1272     [textField release];
1273     [super dealloc];
1274 }
1275
1276 - (void)setTextFieldString:(NSString *)textFieldString
1277 {
1278     [textField release];
1279     textField = [[NSTextField alloc] init];
1280     [textField setStringValue:textFieldString];
1281 }
1282
1283 - (NSTextField *)textField
1284 {
1285     return textField;
1286 }
1287
1288 - (void)setInformativeText:(NSString *)text
1289 {
1290     if (textField) {
1291         // HACK! Add some space for the text field.
1292         [super setInformativeText:[text stringByAppendingString:@"\n\n\n"]];
1293     } else {
1294         [super setInformativeText:text];
1295     }
1296 }
1297
1298 - (void)beginSheetModalForWindow:(NSWindow *)window
1299                    modalDelegate:(id)delegate
1300                   didEndSelector:(SEL)didEndSelector
1301                      contextInfo:(void *)contextInfo
1302 {
1303     [super beginSheetModalForWindow:window
1304                       modalDelegate:delegate
1305                      didEndSelector:didEndSelector
1306                         contextInfo:contextInfo];
1307
1308     // HACK! Place the input text field at the bottom of the informative text
1309     // (which has been made a bit larger by adding newline characters).
1310     NSView *contentView = [[self window] contentView];
1311     NSRect rect = [contentView frame];
1312     rect.origin.y = rect.size.height;
1313
1314     NSArray *subviews = [contentView subviews];
1315     unsigned i, count = [subviews count];
1316     for (i = 0; i < count; ++i) {
1317         NSView *view = [subviews objectAtIndex:i];
1318         if ([view isKindOfClass:[NSTextField class]]
1319                 && [view frame].origin.y < rect.origin.y) {
1320             // NOTE: The informative text field is the lowest NSTextField in
1321             // the alert dialog.
1322             rect = [view frame];
1323         }
1324     }
1325
1326     rect.size.height = MMAlertTextFieldHeight;
1327     [textField setFrame:rect];
1328     [contentView addSubview:textField];
1329     [textField becomeFirstResponder];
1330 }
1331
1332 @end // MMAlert