//

//  ESIRCAccount.m

//  Adium

//

//  Created by Evan Schoenberg on 3/4/06.

//  Copyright 2006 The Adium Team. All rights reserved.

//

#import "ESIRCAccount.h"

#import <Adium/AIHTMLDecoder.h>

#import <Adium/AIChat.h>

#import <Adium/AIContentMessage.h>

#import <Adium/AIListContact.h>

#import <Adium/AIMenuControllerProtocol.h>

#import "AIMessageViewController.h"

#import <AIUtilities/AIMenuAdditions.h>

#import <AIUtilities/AIAttributedStringAdditions.h>

#import <libpurple/irc.h>

#import <libpurple/cmds.h>

#import "SLPurpleCocoaAdapter.h"

@interface SLPurpleCocoaAdapter ()

- (BOOL)attemptPurpleCommandOnMessage:(NSString *)originalMessage fromAccount:(AIAccount *)sourceAccount inChat:(AIChat *)chat;

@end

@interface ESIRCAccount()

- (void)sendRawCommand:(NSString *)command;

- (void)apply:(BOOL)apply operation:(NSString *)operation flag:(NSString *)flag;

@end

static PurpleConversation *fakeConversation(PurpleAccount *account);

@implementation ESIRCAccount

/*!

 * @brief Our explicit formatted UID contains our hostname, so we can differentiate ourself.

 */

- (NSString *)explicitFormattedUID

{

	if (self.host) {

		return [NSString stringWithFormat:@"%@ (%@)", self.host, self.displayName];

	} else {

		return self.displayName;

	}

}

#pragma mark IRC-ism overloads

/*!

 * @brief We always want to autocomplete the UID.

 */

- (BOOL)chatShouldAutocompleteUID:(AIChat *)inChat

{

	return YES;

}

/*!

 * @brief Use the object ID for password name

 *

 * We mess around a lot with the UID. This lets it actually save right.

 */

- (BOOL)useInternalObjectIDForPasswordName

{

	return YES;

}

- (BOOL)openChat:(AIChat *)chat

{

	chat.hideUserIconAndStatus = YES;

	return [super openChat:chat];

}

#pragma mark Command handling

/*!

 * @brief We've connected

 *

 * Send the commands the user wants sent when we do so. Creates a fake conversation to pipe them through.

 */

- (void)didConnect

{

	[super didConnect];

	PurpleConversation *conv = fakeConversation(self.purpleAccount);

	for (NSString *command in [[self preferenceForKey:KEY_IRC_COMMANDS

												group:GROUP_ACCOUNT_STATUS] componentsSeparatedByCharactersInSet:[NSCharacterSet newlineCharacterSet]]) {

		if ([command hasPrefix:@"/"]) {

			command = [command substringFromIndex:1];

		}

		command = [command stringByReplacingOccurrencesOfString:@"$me" withString:self.displayName];

		if (command.length) {

			char *error;

			PurpleCmdStatus cmdStatus = purple_cmd_do_command(conv, [command UTF8String], [command UTF8String], &error);

			if (cmdStatus == PURPLE_CMD_STATUS_NOT_FOUND) {

				// If it's not found, send it as a raw command like we do in chats.

				[self sendRawCommand:command];

			} else if (cmdStatus != PURPLE_CMD_STATUS_OK) {

				// The command failed with something other than "not found" - log it.

				AILogWithSignature(@"Command (%@) failed: %d - %@", command, cmdStatus, [NSString stringWithUTF8String:error]);

			}

		}

	}

	// The fakeConversation was allocated; now free it.

	g_free(conv);

}

/*!

 * @brief Send a raw command to the IRC server.

 */

- (void)sendRawCommand:(NSString *)command

{

	PurpleConnection *connection = purple_account_get_connection(account);

	if (!connection)

		return;

	const char *quote = [command UTF8String];

	irc_cmd_quote(connection->proto_data, NULL, NULL, &quote);	

}

/*!

 * @brief This creates a fake PurpleConversation

 *

 * This fake conversation is used for sending purple_cmd_do_command() messages, which requires

 * a conversation for the command to occur. Free this when finished.

 *

 * This is taken from irchelper.c, the pidgin plugin.

 */

static PurpleConversation *fakeConversation(PurpleAccount *account)

{

	PurpleConversation *conv;

	conv = g_new0(PurpleConversation, 1);

	conv->type = PURPLE_CONV_TYPE_IM;

	/* If we use this then the conversation updated signal is fired and

	 * other plugins might start doing things to our conversation, such as

	 * setting data on it which we would then need to free etc. It's easier

	 * just to be more hacky by setting account directly. */

	/* purple_conversation_set_account(conv, account); */

	conv->account = account;

	return conv;

}

- (NSString *)encodedAttributedStringForSendingContentMessage:(AIContentMessage *)inContentMessage

{

	NSString	*encodedString = nil;

	BOOL		didCommand = [self.purpleAdapter attemptPurpleCommandOnMessage:inContentMessage.message.string

																   fromAccount:(AIAccount *)inContentMessage.source

																	    inChat:inContentMessage.chat];

	NSRange meRange = [inContentMessage.message.string rangeOfString:@"/me " options:NSCaseInsensitiveSearch];

	if (!didCommand || meRange.location == 0) {

		if (meRange.location == 0) {

			inContentMessage.sendContent = NO;

		}

		/* If we're sending a message on an encryption chat (can this even happen on irc?), we can encode the HTML normally, as links will go through fine.

		 * If we're sending a message normally, IRC will drop the title of any link, so we preprocess it to be in the form "title (link)"

		 */

		encodedString = [AIHTMLDecoder encodeHTML:(inContentMessage.chat.isSecure ? inContentMessage.message : [inContentMessage.message attributedStringByConvertingLinksToStrings])

										  headers:NO

										 fontTags:YES

							   includingColorTags:YES

									closeFontTags:YES

										styleTags:YES

					   closeStyleTagsOnFontChange:YES

								   encodeNonASCII:NO

									 encodeSpaces:NO

									   imagesPath:nil

								attachmentsAsText:YES

						onlyIncludeOutgoingImages:NO

								   simpleTagsOnly:YES

								   bodyBackground:NO

							  allowJavascriptURLs:YES];

	}

	if (!didCommand && [inContentMessage.message.string hasPrefix:@"/"]) {

		// Try to send it to the server, if we don't know what it is; definitely don't display.

		[self sendRawCommand:[inContentMessage.message.string substringFromIndex:1]];

		return nil;

	} else {

		return encodedString;

	}

}

#pragma mark Libpurple

- (const char *)protocolPlugin

{

	return "prpl-irc";

}

- (const char *)purpleAccountName

{

	return [[NSString stringWithFormat:@"%@@%@", self.formattedUID, self.host] UTF8String];

}

- (NSString *)defaultUsername

{

	return @"Adium";

}

- (NSString *)defaultRealname

{

	return AILocalizedString(@"Adium User", nil);

}

- (void)configurePurpleAccount

{

	[super configurePurpleAccount];

	purple_account_set_username(self.purpleAccount, self.purpleAccountName);

	// Encoding

	NSString *encoding = [self preferenceForKey:KEY_IRC_ENCODING group:GROUP_ACCOUNT_STATUS] ?: @"UTF-8";

	purple_account_set_string(self.purpleAccount, "encoding", [encoding UTF8String]);

	if (![encoding isEqualToString:@"UTF-8"]) {

		purple_account_set_bool(self.purpleAccount, "autodetect_utf8", TRUE);

	}

	// Use SSL

	BOOL useSSL = [[self preferenceForKey:KEY_IRC_USE_SSL group:GROUP_ACCOUNT_STATUS] boolValue];

	purple_account_set_bool(self.purpleAccount, "ssl", useSSL);

	// Username (for connecting)

	NSString *username = [self preferenceForKey:KEY_IRC_USERNAME group:GROUP_ACCOUNT_STATUS] ?: self.defaultUsername;

	purple_account_set_string(self.purpleAccount, "username", [username UTF8String]);

	// Realname (for connecting)

	NSString *realname = [self preferenceForKey:KEY_IRC_REALNAME group:GROUP_ACCOUNT_STATUS] ?: self.defaultRealname;

	purple_account_set_string(self.purpleAccount, "realname", [realname UTF8String]);

}

/*!

 * @brief Our display name; either retrieve our current nickname, or return our stored one.

 */

- (NSString *)displayName

{

	// Try and get the purple display name, since it changes without telling us.

	if (account) {

		PurpleConnection	*purpleConnection = purple_account_get_connection(account);

		if (purpleConnection) {

			return [NSString stringWithUTF8String:purple_connection_get_display_name(purpleConnection)];

		}

	}

	return self.formattedUID;

}

/*!

 * @brief Re-create the chat's join options.

 */

- (NSDictionary *)extractChatCreationDictionaryFromConversation:(PurpleConversation *)conv

{

	NSMutableDictionary *dict = [NSMutableDictionary dictionary];

	[dict setObject:[NSString stringWithUTF8String:purple_conversation_get_name(conv)] forKey:@"channel"];

	const char *pass = purple_conversation_get_data(conv, "password");

	if (pass)

		[dict setObject: [NSString stringWithUTF8String:pass] forKey:@"password"];

	return dict;

}

#pragma mark Server contacts (NickServ, ChanServ)

/*!

 * @brief Sends a raw command to identify for the nickname

 */

- (void)identifyForName:(NSString *)name password:(NSString *)inPassword

{

	if ([self.host rangeOfString:@"quakenet" options:NSCaseInsensitiveSearch].location != NSNotFound) {

		[self sendRawCommand:[NSString stringWithFormat:@"PRIVMSG Q@CServe.quakenet.org :AUTH %@ %@", name, inPassword]];

	} else if ([self.host rangeOfString:@"undernet" options:NSCaseInsensitiveSearch].location != NSNotFound) {

		[self sendRawCommand:[NSString stringWithFormat:@"PRIVMSG X@channels.undernet.org :LOGIN %@ %@", name, inPassword]];

	} else if ([self.host rangeOfString:@"gamesurge" options:NSCaseInsensitiveSearch].location != NSNotFound) {

		[self sendRawCommand:[NSString stringWithFormat:@"PRIVMSG AuthServ@Services.GameSurge.net :AUTH %@ %@", name, inPassword]];

	} else {

		[self sendRawCommand:[NSString stringWithFormat:@"NICKSERV identify %@ %@", name, inPassword]];	

	}

}

/*!

 * @brief Is this contact a server contact?

 */

BOOL contactUIDIsServerContact(NSString *contactUID)

{

	return (([contactUID caseInsensitiveCompare:@"nickserv"] == NSOrderedSame) ||

			([contactUID caseInsensitiveCompare:@"chanserv"] == NSOrderedSame) ||

			([contactUID rangeOfString:@"-connect" options:(NSBackwardsSearch | NSCaseInsensitiveSearch | NSAnchoredSearch)].location != NSNotFound));

}

/*!

 * @brief Can we send an offline message to this contact?

 *

 * We can only send offline messages to the server contacts, since such a message might cause us to connect

 */

- (BOOL)canSendOfflineMessageToContact:(AIListContact *)inContact

{

	return contactUIDIsServerContact(inContact.UID);

}

/*!

 * @brief Don't autoreply to server contacts (services) or FreeNode's stupidity.

 */

- (BOOL)shouldSendAutoreplyToMessage:(AIContentMessage *)message

{

	return !contactUIDIsServerContact(message.source.UID);

}

/*!

 * @brief Don't log server contacts (services) or FreeNode's stupidity.

 */

- (BOOL)shouldLogChat:(AIChat *)chat

{

	NSString *source = chat.listObject.UID;

	BOOL shouldLog = YES;

	if (source && (([source caseInsensitiveCompare:@"nickserv"] == NSOrderedSame) ||

				   ([source caseInsensitiveCompare:@"chanserv"] == NSOrderedSame) ||

				   ([source rangeOfString:@"-connect" options:(NSBackwardsSearch | NSCaseInsensitiveSearch | NSAnchoredSearch)].location != NSNotFound))) {

		shouldLog = NO;	

	}

	return (shouldLog && [super shouldLogChat:chat]);

}

#pragma mark Chat handling

/*!

 * @brief Allow the chat to close unless we're quitting.

 */

- (BOOL)closeChat:(AIChat*)chat

{

	if(adium.isQuitting)

		return NO;

	else

		return [super closeChat:chat];

}

/*!

 * @brief Do group chats support topics?

 */

- (BOOL)groupChatsSupportTopic

{

	return YES;

}

/*!

 * @brief Our flags in a chat

 */

- (AIGroupChatFlags)flagsInChat:(AIChat *)chat

{

	NSString *ourUID = [NSString stringWithUTF8String:purple_normalize(self.purpleAccount, [self.displayName UTF8String])];

	// XXX Once we don't create a fake contact for ourself, we should do this the right way.

	return [chat flagsForContact:[self contactWithUID:ourUID]];

}

#pragma mark Action Menu

-(NSMenu*)actionMenuForChat:(AIChat*)chat

{

	NSMenu *menu;

	NSArray *listObjects = chat.chatContainer.messageViewController.selectedListObjects;

	AIListObject *listObject = nil;

	if (listObjects.count) {

		listObject = [listObjects objectAtIndex:0];

	}

	menu = [adium.menuController contextualMenuWithLocations:[NSArray arrayWithObjects:

															   [NSNumber numberWithInteger:Context_Contact_GroupChat_ParticipantAction],		

															   [NSNumber numberWithInteger:Context_Contact_Manage],

															   nil]

												forListObject:listObject

													   inChat:chat];

	[menu addItem:[NSMenuItem separatorItem]];

	[menu addItemWithTitle:AILocalizedString(@"Op", nil)

					target:self

					action:@selector(op)

			 keyEquivalent:@""

					   tag:AIRequiresOp];

	[menu addItemWithTitle:AILocalizedString(@"Deop", nil)

					target:self

					action:@selector(deop)

			 keyEquivalent:@""

					   tag:AIRequiresOp];

	[menu addItemWithTitle:AILocalizedString(@"Voice", nil)

					target:self

					action:@selector(voice)

			 keyEquivalent:@""

					   tag:AIRequiresOp];

	[menu addItemWithTitle:AILocalizedString(@"Devoice", nil)

					target:self

					action:@selector(devoice)

			 keyEquivalent:@""

					   tag:AIRequiresOp];

	[menu addItem:[NSMenuItem separatorItem]];

	[menu addItemWithTitle:AILocalizedString(@"Kick", nil)

					target:self

					action:@selector(kick)

			 keyEquivalent:@""

					   tag:AIRequiresHalfop];

	[menu addItemWithTitle:AILocalizedString(@"Ban", nil)

					target:self

					action:@selector(ban)

			 keyEquivalent:@""

					   tag:AIRequiresHalfop];

	[menu addItemWithTitle:AILocalizedString(@"Bankick", nil)

					target:self

					action:@selector(bankick)

			 keyEquivalent:@""

					   tag:AIRequiresHalfop];

	return menu;

}

- (BOOL)validateMenuItem:(NSMenuItem *)menuItem

{

	AIOperationRequirement req = menuItem.tag;

	AIChat *chat = adium.interfaceController.activeChat;

	if (!chat.chatContainer.messageViewController.selectedListObjects.count) {

		return NO;

	}

	AIGroupChatFlags flags = [self flagsInChat:chat];

	switch (req) {

		case AIRequiresHalfop:

			return ((flags & AIGroupChatOp) == AIGroupChatOp || (flags & AIGroupChatHalfOp) == AIGroupChatHalfOp);

			break;

		case AIRequiresOp:

			return ((flags & AIGroupChatOp) == AIGroupChatOp);

			break;

	}

	return NO;

}

#pragma mark Action Menu's Actions

- (void)apply:(BOOL)apply operation:(NSString *)operation flag:(NSString *)flag

{

	AIChat *chat = adium.interfaceController.activeChat;

	NSArray *objects = chat.chatContainer.messageViewController.selectedListObjects;

	NSMutableString *names = [NSMutableString string];

	for (NSUInteger x = 0; x < objects.count; x++) {

		[names appendString:((AIListObject *)[objects objectAtIndex:x]).UID];

		[names appendString:@" "];

		if ((x+1) % 4 == 0 || x+1 == objects.count) {

			if ([operation isEqualToString:@"MODE"]) {

				[self sendRawCommand:[NSString stringWithFormat:@"MODE %@ %@%@ %@",

									  chat.name,

									  (apply ? @"+" : @"-"),

									  [@"" stringByPaddingToLength:(x + 1) % 4 ?: 4

														withString:flag 

												   startingAtIndex:0],

									  names]];

			} else if ([operation isEqualToString:@"KICK"]) {

				[self sendRawCommand:[NSString stringWithFormat:@"KICK %@ %@",

									  chat.name,

									  [names stringByReplacingOccurrencesOfString:@" " withString:@","]]];

			}

			[names setString:@""];

		}

	}

}

- (void)op

{

	[self apply:YES operation:@"MODE" flag:@"o"];

}

- (void)deop

{

	[self apply:NO operation:@"MODE" flag:@"o"];

}

- (void)voice

{

	[self apply:YES operation:@"MODE" flag:@"v"];

}

- (void)devoice

{

	[self apply:NO operation:@"MODE" flag:@"v"];

}

- (void)kick

{

	[self apply:NO operation:@"KICK" flag:nil];

}

- (void)ban

{

	[self apply:YES operation:@"MODE" flag:@"b"];

}

- (void)bankick

{

	[self ban];

	[self kick];

}

@end