root/lang/objective-c/TerminalColoreopard/TCLANSIColors.m

#import "TCLANSIColors.h"

@implementation TCLANSIColors
+(TCLANSIColors *)defaultANSIColors
{
        TCLANSIColors *colors = [[TCLANSIColors alloc] init];
        return [colors autorelease];
}

+(TCLANSIColors *)fromDictionary:(NSDictionary *)aDictionary
{
        TCLANSIColors *colors = [TCLANSIColors defaultANSIColors];
        for(NSString *name in [TCLANSIColors ansiColorNames]) {
                [colors setValue:[aDictionary objectForKey:name] forKey:[name stringByAppendingString:@"Color"]];
        }
        return colors;
}

+(NSArray *)ansiColorNames
{
        static NSArray *colorNames;
        if(colorNames == nil) {
                colorNames = [[NSArray arrayWithObjects:@"black", @"red", @"green", @"yellow", @"blue", @"magenta", @"cyan", @"white", nil] retain];
        }
        return colorNames;
}

-(void)dealloc
{
        [super dealloc];
}

-(id)init
{
        self = [super init];
        for(NSString *name in [TCLANSIColors ansiColorNames]) {
                NSString *key = [name stringByAppendingString:@"Color"];
                [self setValue:[NSColor valueForKey:key] forKey:key];
        }
        return self;
}

-(NSDictionary *)ansiColorsAsDictionary
{
        NSMutableDictionary *dict = [NSMutableDictionary dictionary];
        for(NSString *name in [TCLANSIColors ansiColorNames]) {
                [dict setValue:[self valueForKey:[name stringByAppendingString:@"Color"]] forKey:name];
        }
        return dict;
}

-(NSColor *)colorFor:(unsigned int)code
{
        NSColor *color = nil;
        if(code > 0) {
                NSString *name = [[TCLANSIColors ansiColorNames] objectAtIndex:code-1];
                if(name) {
                        color = [self valueForKey:[name stringByAppendingString:@"Color"]];
                }
        }
        return color;
}

@end