undmail/guiproto/ToolbarSource.m

#import "ToolbarSource.h"
#import "MainController.h"

@implementation ToolbarSource
- (id)init{
    if(self = [super init]){
        toolbarItemList = [[NSMutableArray alloc] init];
        defaultToolbarItemList = [[NSMutableArray alloc] init];
        itemTable = [[NSMutableDictionary alloc] init];
    }
    return self;
}
-(void)dealloc{
    [toolbarItemList release];
    [defaultToolbarItemList release];
    [itemTable release];
    [super dealloc];
}
- (NSToolbarItem *)addToolbarItemWithIdentifier:(NSString *)identifier label:(NSString *)label imageName:(NSString *)imageName  isDefault:(BOOL)isDefault{

    NSToolbarItem *item;
    NSBundle *mainBundle;
    mainBundle = [NSBundle mainBundle];
    item = [[[NSToolbarItem alloc] initWithItemIdentifier:identifier] autorelease];
    [item setLabel:[mainBundle localizedStringForKey:label value:nil table:@"Toolbar"]];
    [item setImage:[[[NSImage alloc] initByReferencingFile:
                    [mainBundle pathForImageResource:imageName]] autorelease]];
    [item setEnabled:YES];
    [itemTable setObject:item forKey:identifier];
    [toolbarItemList addObject:identifier];
    if(isDefault){
        [defaultToolbarItemList addObject:identifier];
    }
    return item;
}
- (NSToolbarItem *)toolbar:(NSToolbar *)toolbar 
                itemForItemIdentifier:(NSString *)itemIdentifier 
                willBeInsertedIntoToolbar:(BOOL)flag{
    return [itemTable objectForKey:itemIdentifier];
}
- (NSArray *)toolbarDefaultItemIdentifiers:(NSToolbar*)toolbar{
    return toolbarItemList;
}
- (NSArray *)toolbarAllowedItemIdentifiers:(NSToolbar*)toolbar{
    return defaultToolbarItemList;
}

@end