Adding API to ITStatusItem for callbacks to get dynamic menus.

diff --git a/ITStatusItem.h b/ITStatusItem.h
index f2f6070..7085c78 100644 (file)
--- a/ITStatusItem.h
+++ b/ITStatusItem.h
@@ -14,8 +14,14 @@
 
 #import <Cocoa/Cocoa.h>
 
-@interface ITStatusItem : NSStatusItem {
+@class ITStatusItem;
+
+@protocol ITStatusItemMenuProvider
+- (NSMenu *)menuForStatusItem:(ITStatusItem *)statusItem;
+@end
 
+@interface ITStatusItem : NSStatusItem {
+       id <ITStatusItemMenuProvider> _menuProvider;
 }
 
 - (id)initWithStatusBar:(NSStatusBar *)statusBar withLength:(float)length;
@@ -23,4 +29,7 @@
 - (NSImage *)alternateImage;
 - (void)setAlternateImage:(NSImage *)image;
 
+- (id <ITStatusItemMenuProvider>)menuProvider;
+- (void)setMenuProvider:(id <ITStatusItemMenuProvider>)provider;
+
 @end
\ No newline at end of file

diff --git a/ITStatusItem.m b/ITStatusItem.m
index ea31d3a..561ecf2 100644 (file)
--- a/ITStatusItem.m
+++ b/ITStatusItem.m
@@ -51,4 +51,21 @@ static BOOL _ITStatusItemShouldKillShadow = NO;
        [(NSButton *)[self _button] setAlternateImage:image];
 }
 
+- (id <ITStatusItemMenuProvider>)menuProvider {
+       return _menuProvider;
+}
+
+- (void)setMenuProvider:(id <ITStatusItemMenuProvider>)provider {
+       [_menuProvider autorelease];
+       _menuProvider = [provider retain];
+}
+
+- (NSMenu *)menu {
+       if (_menuProvider) {
+               return [_menuProvider menuForStatusItem:self];
+       } else {
+               return [super menu];
+       }
+}
+
 @end
\ No newline at end of file