XMPPMessage.m

//
//  XMPPMessage.m
//  Jabber
//
//  Created by David Chisnall on 20/08/2004.
//  Copyright 2004 __MyCompanyName__. All rights reserved.
//

#import "XMPPMessage.h"
#include <wchar.h>
#include <wctype.h>
#import <AppKit/AppKit.h>
#import <EtoileXML/ETXMLParser.h>
#import <EtoileXML/ETXMLString.h>
#import <EtoileXML/ETXMLWriter.h>
#import "XMPPMessageStanzaFactory.h"
#import "XMPPError.h"
#import "NSAttributedString+HTML-IM.h"
#import <EtoileFoundation/EtoileFoundation.h>

NSDictionary * MESSAGE_TYPES;

@implementation XMPPMessage
+ (void) initialize
{
	MESSAGE_TYPES = [[NSDictionary alloc] initWithObjectsAndKeys:
		[NSNumber numberWithInt:MESSAGE_TYPE_CHAT], @"chat",
		[NSNumber numberWithInt:MESSAGE_TYPE_MESSAGE], @"message",
		[NSNumber numberWithInt:MESSAGE_TYPE_ERROR], @"error",
		[NSNumber numberWithInt:MESSAGE_TYPE_GROUPCHAT], @"groupchat",
		nil];
}
	
+ (XMPPMessage*) messageWithBody:(id)aBody for:(JID*)aRecipient withSubject:(NSString*)aSubject type:(message_type_t)aType
{
    return [[XMPPMessage alloc] initWithBody:aBody for:aRecipient withSubject:aSubject type:aType];
}

- (XMPPMessage*) initWithBody:(id)aBody for:(JID*)aRecipient withSubject:(NSString*)aSubject type:(message_type_t)aType
{
	if((self = [super init])==nil){
		return nil;
	}
#ifndef DNDEBUG
	ETLog(@"Body (%@) %@", [aBody class], aBody);
#endif
	if([aBody isKindOfClass:[NSString class]])
	{
		body = [aBody stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
	}
	else if([aBody isKindOfClass:[NSAttributedString class]])
	{
		body = [[aBody stringValueWithExpandedLinks] stringByTrimmingCharactersInSet: [NSCharacterSet whitespaceAndNewlineCharacterSet]];
		html = aBody;
	}
	correspondent = aRecipient;
	subject = aSubject;
	type = aType;
	direction = out;
	return self;
}

- (id) initWithXMLParser: (ETXMLParser*)aParser
                     key: (id) aKey
{
	self = [super initWithXMLParser: aParser
	                            key: aKey];
	if (nil == self)
	{
		return nil;
	}
	unknownAttributes = [[NSMutableDictionary alloc] init];
	timestamps = [[NSMutableArray alloc] init];
	body = @"";
	shouldDisplay = YES;
	return self;
}

- (void) beginWritingToXMLWriter: (ETXMLWriter*)xmlWriter
{
	NSMutableDictionary * attributes = [[NSMutableDictionary alloc] init];
	
	switch(type)
	{
		case MESSAGE_TYPE_CHAT:
			[attributes setValue:@"chat" forKey:@"type"];
			break;
		case MESSAGE_TYPE_ERROR:
			[attributes setValue:@"error" forKey:@"type"];
			break;
		case MESSAGE_TYPE_GROUPCHAT:
			[attributes setValue:@"groupchat" forKey:@"type"];
		case MESSAGE_TYPE_MESSAGE:
		case MESSAGE_TYPE_SPECIAL:
			break;
	}
	if(direction == out)
	{
		[attributes setValue:[correspondent jidString] forKey:@"to"];
	}
	else
	{
		[attributes setValue:[correspondent jidString] forKey:@"from"];
	}
	[xmlWriter startElement: @"message" attributes: attributes];

	if(subject != nil)
	{
		[xmlWriter startElement: @"subject"];
		[xmlWriter characters: subject];
		[xmlWriter endElement];
	}
	if(body != nil)
	{
		[xmlWriter startElement: @"body"];
		[xmlWriter characters: body];
		[xmlWriter endElement];
	}
	if(html != nil)
	{
		[html writeXHTMLIMToXMLWriter: xmlWriter];
	}
}

- (void) finishWritingToXMLWriter: (ETXMLWriter*) xmlWriter
{
	[xmlWriter endElement];
}

- (void)writeToXMLWriter: (ETXMLWriter*)xmlWriter
{
	[self beginWritingToXMLWriter: (ETXMLWriter*) xmlWriter];
	[self finishWritingToXMLWriter: (ETXMLWriter*) xmlWriter];
}


- (JID*) correspondent
{
	return correspondent;
}

- (NSString*) subject
{
	return subject;
}

- (void) setSubject:(NSString*)aSubject
{
	subject = aSubject;
}
- (NSString*) body
{
	return body;
}

- (void) setBody:(NSString*)aBody
{
	body = aBody;
}


- (NSAttributedString*) HTMLBody
{
	if(html != nil)
	{
		return html;
	}
	return [[NSAttributedString alloc] initWithString:body];
}

- (BOOL) in
{
	if(direction == in)
	{
		return YES;
	}
	return NO;
}
- (message_type_t) type
{
	return type;
}
- (XMPPTimestamp*) timestamp
{
	return [timestamps lastObject];
}
- (XMPPError*) error
{
	return error;
}
- (NSComparisonResult) compareByTimestamp:(XMPPMessage*)_other
{
	return [[self timestamp] compare:[_other timestamp]];
}
- (void)characters:(NSString *)_chars
{
	//Ignore cdata
}
- (void)startElement:(NSString *)aName
		  attributes:(NSDictionary*)attributes
{
	if([aName isEqualToString:@"message"])
	{
		correspondent = [JID jidWithString:[attributes objectForKey:@"from"]];
		direction = in;
		type = [[MESSAGE_TYPES objectForKey:[attributes objectForKey:@"type"]] intValue];
	}
	else 
	{
		XMPPMessageStanzaFactory * factory = [XMPPMessageStanzaFactory sharedStazaFactory];
		NSString * xmlns = [attributes objectForKey:@"xmlns"];
		Class handler = [factory handlerForTag:aName inNamespace:xmlns];
		NSString * elementKey = [factory valueForTag:aName inNamespace:xmlns];
		[[[handler alloc] initWithXMLParser:parser
						key:elementKey] startElement:aName
					 attributes:attributes];
	}
}
- (void)endElement:(NSString *)aName
{
	if([aName isEqualToString:@"message"])
	{
		[[parser parentHandler] addChild:self forKey:key];
		[parser popContentHandler];
	}
	else
	{
		ETLog(@"End of %@ tag received while parsing a message.  This probably indicates a bug.", aName);
	}


}


- (void) setParser:(id) XMLParser
{
	parser = XMLParser;
}

- (void) addbody:(NSString*)aBody
{
	body = [aBody stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
}
- (void) addsubject:(NSString*)aSubject
{
	subject = aSubject;
}
- (void) addtimestamp:(XMPPTimestamp*)aTimestamp
{
	[timestamps addObject:aTimestamp];
	[timestamps sortUsingSelector:@selector(compare:)];
}
- (void) addhtml:(NSAttributedString*)anAttributedString
{
	html = anAttributedString;
}
- (void) adderror:(XMPPError*)anError
{
	error = anError;
}

- (void) setShouldDisplay:(BOOL)aFlag
{
	shouldDisplay = aFlag;
}
- (BOOL) shouldDisplay
{
	return shouldDisplay;
}
@end