ITCategory-NSMenu.m

   1 #import "ITCategory-NSMenu.h"
   2
   3 extern void _NSGetMenuItemForCommandKeyEvent(NSMenu *menu, void *chicken, void *taco, void *food);
   4 extern MenuRef _NSGetCarbonMenu(NSMenu *menu);
   5
   6 @interface NSMenu (ITKitCategoryHacks)
   7 - (id)_menuImpl;
   8 @end
   9
  10 @protocol _ITKitCategoryNSMenuItemPantherCompatability <NSMenuItem>
  11 - (int)indentationLevel;
  12 - (void)setIndentationLevel:(int)indentationLevel;
  13 @end
  14
  15 @implementation NSMenu (ITKitCategory)
  16
  17 - (void)indentItem:(id <NSMenuItem>)item {
  18     [self indentItem:item toLevel:1];
  19 }
  20
  21 - (void)indentItemAtIndex:(int)index {
  22     [self indentItemAtIndex:index toLevel:1];
  23 }
  24
  25 - (void)indentItem:(id <NSMenuItem>)item toLevel:(int)indentLevel {
  26     [self indentItemAtIndex:[self indexOfItem:item] toLevel:indentLevel];
  27 }
  28
  29 - (void)indentItemAtIndex:(int)index toLevel:(int)indentLevel {
  30         id <NSMenuItem> theItem = [self itemAtIndex:index];
  31     if ([theItem respondsToSelector:@selector(setIndentationLevel:)]) {
  32         [(id <_ITKitCategoryNSMenuItemPantherCompatability>)theItem setIndentationLevel:indentLevel];
  33     } else {
  34         MenuRef carbonMenu = [self menuRef];
  35         if (carbonMenu) {
  36             SetMenuItemIndent(carbonMenu, index + 1, indentLevel);
  37         }
  38     }
  39 }
  40
  41 - (MenuRef)menuRef {
  42     MenuRef carbonMenu;
  43     int w00t, m00f;
  44
  45     if ([self respondsToSelector:@selector(_menuImpl)]) {
  46         (void)[self _menuImpl];
  47     } else {
  48         return nil;
  49     }
  50
  51     _NSGetMenuItemForCommandKeyEvent(self, NULL, &w00t, &m00f);
  52     carbonMenu = _NSGetCarbonMenu(self);
  53     return carbonMenu;
  54 }
  55
  56 - (void)removeAllItems {
  57         int numOfItems = [self numberOfItems];
  58         int i = numOfItems;
  59
  60         while (i != 0) {
  61                 [self removeItemAtIndex:(i-1)];
  62                 i--;
  63         }
  64 }
  65
  66 @end