root/tags/2D14/JVChatEvent.m

#import "JVChatEvent.h"
#import "NSAttributedStringMoreAdditions.h"

#import <ChatCore/NSAttributedStringAdditions.h>
#import <ChatCore/NSStringAdditions.h>
#import <ChatCore/NSDataAdditions.h>
#import <libxml/xinclude.h>

@implementation JVChatEvent
- (id) init {
        if( ( self = [super init] ) ) {
                _loadedMessage = NO;
                _loadedAttributes = NO;
                _loadedSmall = NO;
                _objectSpecifier = nil;
                _transcript = nil;
                _eventIdentifier = nil;
                _message = nil;
                _date = nil;
                _name = nil;
                _attributes = nil;
        }

        return self;
}

- (id) initWithNode:(xmlNode *) node andTranscript:(JVChatTranscript *) transcript {
        if( ( self = [self init] ) ) {
                _node = node;
                _transcript = transcript; // weak reference

                if( ! _node || node -> type != XML_ELEMENT_NODE ) {
                        [self release];
                        return nil;
                }

                @synchronized( [self transcript] ) {
                        xmlChar *prop = xmlGetProp( (xmlNode *) _node, (xmlChar *) "id" );
                        _eventIdentifier = ( prop ? [[NSString allocWithZone:[self zone]] initWithUTF8String:(char *) prop] : nil );
                        xmlFree( prop );
                }
        }

        return self;
}

+ (id) eventWithNode:(xmlNode *) node andTranscript:(JVChatTranscript *) transcript {
        return [[[self alloc] initWithNode:node andTranscript:transcript] autorelease];
}

- (void) dealloc {
        [_eventIdentifier release];
        [_date release];
        [_name release];
        [_message release];
        [_attributes release];

        _eventIdentifier = nil;
        _date = nil;
        _name = nil;
        _message = nil;
        _attributes = nil;

        _transcript = nil; // weak reference
        _node = NULL;

        [super dealloc];
}

#pragma mark -

- (void) loadSmall {
        if( _loadedSmall || ! _node ) return;

        @synchronized( [self transcript] ) {
                xmlChar *prop = xmlGetProp( (xmlNode *) _node, (xmlChar *) "name" );
                _name = ( prop ? [[NSString allocWithZone:[self zone]] initWithUTF8String:(char *) prop] : nil );
                xmlFree( prop );

                prop = xmlGetProp( (xmlNode *) _node, (xmlChar *) "occurred" );
                _date = ( prop ? [[NSDate allocWithZone:[self zone]] initWithString:[NSString stringWithUTF8String:(char *) prop]] : nil );
                xmlFree( prop );
        }

        _loadedSmall = YES;
}

- (void) loadMessage {
        if( _loadedMessage || ! _node ) return;

        @synchronized( [self transcript] ) {
                xmlNode *subNode = ((xmlNode *) _node) -> children;

                do {
                        if( subNode -> type == XML_ELEMENT_NODE && ! strncmp( "message", (char *) subNode -> name, 6 ) ) {
                                _message = [[NSTextStorage allocWithZone:[self zone]] initWithXHTMLTree:subNode baseURL:nil defaultAttributes:nil];
                                break;
                        }
                } while( ( subNode = subNode -> next ) );
        }

        _loadedMessage = YES;
}

- (void) loadAttributes {
        if( _loadedAttributes || ! _node ) return;

        @synchronized( [self transcript] ) {
                xmlNode *subNode = ((xmlNode *) _node) -> children;
                NSMutableDictionary *attributes = [NSMutableDictionary dictionary];

                do {
                        if( subNode -> type == XML_ELEMENT_NODE && strncmp( "message", (char *) subNode -> name, 6 ) ) { // everything but "message"
                                NSMutableDictionary *properties = [NSMutableDictionary dictionary];
                                xmlAttrPtr prop = NULL;
                                for( prop = subNode -> properties; prop; prop = prop -> next ) {
                                        xmlChar *value = xmlGetProp( subNode, prop -> name );
                                        if( value ) {
                                                [properties setObject:[NSString stringWithUTF8String:(char *) value] forKey:[NSString stringWithUTF8String:(char *) prop -> name]];
                                                xmlFree( value );
                                        }
                                }

                                xmlNode *cnode = subNode -> children;
                                unsigned count = 0;

                                do {
                                        if( cnode && cnode -> type == XML_ELEMENT_NODE ) count++;
                                } while( cnode && ( cnode = cnode -> next ) );

                                id value = nil;
                                if( count > 0 ) {
                                        value = [NSTextStorage attributedStringWithXHTMLTree:subNode baseURL:nil defaultAttributes:nil];
                                } else {
                                        xmlChar *content = xmlNodeGetContent( subNode );
                                        value = [NSString stringWithUTF8String:(char *) content];
                                        xmlFree( content );
                                }

                                if( [properties count] ) {
                                        [properties setObject:value forKey:@"value"];
                                        [attributes setObject:properties forKey:[NSString stringWithUTF8String:(char *) subNode -> name]];
                                } else {
                                        [attributes setObject:value forKey:[NSString stringWithUTF8String:(char *) subNode -> name]];
                                }
                        }
                } while( ( subNode = subNode -> next ) );
        }

        _loadedAttributes = YES;
}

#pragma mark -

- (void *) node {
        return _node;
}

- (void) setNode:(xmlNode *) node {
        _node = node;
}

#pragma mark -

- (JVChatTranscript *) transcript {
        return _transcript;
}

- (NSString *) eventIdentifier {
        return _eventIdentifier;
}

#pragma mark -

- (NSDate *) date {
        [self loadSmall];
        return _date;
}

#pragma mark -

- (NSString *) name {
        [self loadSmall];
        return _name;
}

#pragma mark -

- (NSTextStorage *) message {
        [self loadMessage];
        return _message;
}

- (NSString *) messageAsPlainText {
        return [[self message] string];
}

- (NSString *) messageAsHTML {
        NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:YES], @"IgnoreFonts", [NSNumber numberWithBool:YES], @"IgnoreFontSizes", nil];
        return [[self message] HTMLFormatWithOptions:options];
}

#pragma mark -

- (NSDictionary *) attributes {
        [self loadAttributes];
        return _attributes;
}
@end

#pragma mark -

@implementation JVMutableChatEvent
+ (id) chatEventWithName:(NSString *) name andMessage:(id) message {
        return [[[self alloc] initWithName:name andMessage:message] autorelease];
}

#pragma mark -

- (id) init {
        if( ( self = [super init] ) ) _doc = NULL;
        return self;
}

- (id) initWithName:(NSString *) name andMessage:(id) message {
        if( ( self = [self init] ) ) {
                _loadedMessage = YES;
                _loadedAttributes = YES;
                _loadedSmall = YES;
                [self setDate:[NSDate date]];
                [self setName:name];
                [self setMessage:message];
                [self setEventIdentifier:[NSString locallyUniqueString]];
        }

        return self;
}

- (void) dealloc {
        if( _doc ) xmlFreeDoc( _doc );
        _doc = NULL;

        [super dealloc];
}

#pragma mark -

- (void *) node {
        if( ! _node ) {
                if( _doc ) xmlFreeDoc( _doc );
                _doc = xmlNewDoc( (xmlChar *) "1.0" );

                xmlNodePtr root = xmlNewNode( NULL, (xmlChar *) "event" );
                xmlSetProp( root, (xmlChar *) "id", (xmlChar *) [[self eventIdentifier] UTF8String] );
                xmlSetProp( root, (xmlChar *) "name", (xmlChar *) [[self name] UTF8String] );
                xmlSetProp( root, (xmlChar *) "occurred", (xmlChar *) [[[self date] description] UTF8String] );
                xmlDocSetRootElement( _doc, root );

                xmlDocPtr msgDoc = NULL;
                xmlNodePtr child = NULL;
                const char *msgStr = NULL;

                if( [self message] ) {
                        NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:YES], @"IgnoreFonts", [NSNumber numberWithBool:YES], @"IgnoreFontSizes", nil];
                        NSString *msgValue = [[self message] HTMLFormatWithOptions:options];
                        msgValue = [msgValue stringByStrippingIllegalXMLCharacters];

                        msgStr = [[NSString stringWithFormat:@"<message>%@</message>", msgValue] UTF8String];

                        msgDoc = xmlParseMemory( msgStr, strlen( msgStr ) );
                        child = xmlDocCopyNode( xmlDocGetRootElement( msgDoc ), _doc, 1 );
                        xmlAddChild( root, child );
                        xmlFreeDoc( msgDoc );
                }

                NSEnumerator *kenumerator = [[self attributes] keyEnumerator];
                NSEnumerator *enumerator = [[self attributes] objectEnumerator];
                NSString *key = nil;
                id value = nil;

                while( ( key = [kenumerator nextObject] ) && ( value = [enumerator nextObject] ) ) {
                        msgStr = NULL;

                        if( [value respondsToSelector:@selector( xmlDescriptionWithTagName: )] ) {
                                msgStr = [(NSString *)[value performSelector:@selector( xmlDescriptionWithTagName: ) withObject:key] UTF8String];
                        } else if( [value isKindOfClass:[NSAttributedString class]] ) {
                                NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:YES], @"IgnoreFonts", [NSNumber numberWithBool:YES], @"IgnoreFontSizes", nil];
                                value = [value HTMLFormatWithOptions:options];
                                value = [value stringByStrippingIllegalXMLCharacters];
                                if( [(NSString *)value length] )
                                        msgStr = [[NSString stringWithFormat:@"<%@>%@</%@>", key, value, key] UTF8String];
                        } else if( [value isKindOfClass:[NSString class]] ) {
                                value = [value stringByEncodingXMLSpecialCharactersAsEntities];
                                value = [value stringByStrippingIllegalXMLCharacters];
                                if( [(NSString *)value length] )
                                        msgStr = [[NSString stringWithFormat:@"<%@>%@</%@>", key, value, key] UTF8String];
                        } else if( [value isKindOfClass:[NSData class]] ) {
                                value = [value base64EncodingWithLineLength:0];
                                if( [(NSString *)value length] )
                                        msgStr = [[NSString stringWithFormat:@"<%@ encoding=\"base64\">%@</%@>", key, value, key] UTF8String];
                        }

                        if( ! msgStr ) msgStr = [[NSString stringWithFormat:@"<%@ />", key] UTF8String];

                        msgDoc = xmlParseMemory( msgStr, strlen( msgStr ) );
                        child = xmlDocCopyNode( xmlDocGetRootElement( msgDoc ), _doc, 1 );
                        xmlAddChild( root, child );
                        xmlFreeDoc( msgDoc );
                }

                _node = root;
        }

        return _node;
}

- (void) setNode:(xmlNode *) node {
        if( _doc ) {
                xmlFreeDoc( _doc );
                _doc = NULL;
        }

        _node = node;
}

#pragma mark -

- (void) setDate:(NSDate *) date {
        [self setNode:NULL];
        [_date autorelease];
        _date = [date copyWithZone:[self zone]];
}

- (void) setName:(NSString *) name {
        [self setNode:NULL];
        [_name autorelease];
        _name = [name copyWithZone:[self zone]];
}

#pragma mark -

- (void) setMessage:(id) message {
        [self setNode:NULL];
        if( ! _message ) {
                if( [message isKindOfClass:[NSTextStorage class]] ) _message = [message retain];
                else if( [message isKindOfClass:[NSAttributedString class]] ) _message = [[NSTextStorage alloc] initWithAttributedString:message];
                else if( [message isKindOfClass:[NSString class]] ) _message = [[NSTextStorage alloc] initWithXHTMLFragment:(NSString *)message baseURL:nil defaultAttributes:nil];
        } else if( _message && [message isKindOfClass:[NSAttributedString class]] ) {
                [_message setAttributedString:message];
        } else if( _message && [message isKindOfClass:[NSString class]] ) {
                id string = [NSAttributedString attributedStringWithXHTMLFragment:(NSString *)message baseURL:nil defaultAttributes:nil];
                [_message setAttributedString:string];
        }
}

- (void) setMessageAsPlainText:(NSString *) message {
        [self setMessage:[[[NSAttributedString alloc] initWithString:message] autorelease]];
}

- (void) setMessageAsHTML:(NSString *) message {
        [self setMessage:message];
}

#pragma mark -

- (void) setAttributes:(NSDictionary *) attributes {
        [self setNode:NULL];
        [_attributes autorelease];
        _attributes = [attributes copyWithZone:[self zone]];
}

#pragma mark -

- (void) setEventIdentifier:(NSString *) identifier {
        [self setNode:NULL];
        [_eventIdentifier autorelease];
        _eventIdentifier = [identifier copyWithZone:[self zone]];
}
@end