/* 

  * Adium is the legal property of its developers, whose names are listed in the copyright file included

  * with this source distribution.

  * 

  * This program is free software; you can redistribute it and/or modify it under the terms of the GNU

  * General Public License as published by the Free Software Foundation; either version 2 of the License,

  * or (at your option) any later version.

  * 

  * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even

  * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General

  * Public License for more details.

  * 

  * You should have received a copy of the GNU General Public License along with this program; if not,

  * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

  */

 #import "AIMessageViewController.h"

 #import "AIMessageTabViewItem.h"

 #import "AITwitterAccount.h"

 #import "AITwitterURLHandler.h"

 #import "AIURLHandlerPlugin.h"

 #import <AIUtilities/AIURLAdditions.h>

 #import <AIUtilities/AIAttributedStringAdditions.h>

 #import <AIUtilities/AIStringAdditions.h>

 #import <Adium/AIMessageEntryTextView.h>

 #import <Adium/AIAccount.h>

 #import <Adium/AIChat.h>

 #import <Adium/AIService.h>

 #import <Adium/AIAccountControllerProtocol.h>

 #import <Adium/AIChatControllerProtocol.h>

 #import <Adium/AIInterfaceControllerProtocol.h>

 #import <Adium/AIContentControllerProtocol.h>

 @implementation AITwitterURLHandler

 /*!

  * @brief Install the plugin

  *

  * This plugin handles links in the format "twitterreply://account@username?status=(sid)" where the account as a provided user is optional.

  */

 - (void)installPlugin

 {

 	[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(urlRequest:) name:AIURLHandleNotification object:nil];

 }

 /*!

  * @brief Uninstall plugin

  */

 - (void)uninstallPlugin

 {

 	[[NSNotificationCenter defaultCenter] removeObserver:self];

 }

 /*!

  * @brief A reply link was licked

  *

  * Parse the reply link and set up as appropriate.

  */

 - (void)urlRequest:(NSNotification *)notification

 {

 	NSString *urlString = notification.object;

 	NSURL *url = [NSURL URLWithString:urlString];

 	if (![url.scheme isEqualToString:@"twitterreply"]) {

 		return;

 	}

 	NSString *inUser = [url host];

 	NSString *inAction = [url queryArgumentForKey:@"action" withDelimiter:@"&"] ?: @"reply";

 	NSString *inTweet = [url queryArgumentForKey:@"status" withDelimiter:@"&"];

 	NSString *inDM = [url queryArgumentForKey:@"dm" withDelimiter:@"&"];

 	NSString *inMessage = [url queryArgumentForKey:@"message" withDelimiter:@"&"];

 	NSString *inAccount = [url user];

 	AILogWithSignature(@"Twitter Reply requested: %@", url);

 	NSArray		*accountArray = adium.accountController.accounts;

 	AITwitterAccount	*account = nil;

 	BOOL		exactMatchForInternalID = NO;

 	// Look for an account with the given internalObjectID

 	for(AIAccount *tempAccount in accountArray) {

 		if (![tempAccount isKindOfClass:[AITwitterAccount class]]) {

 			continue;

 		}

 		account = (AITwitterAccount *)tempAccount;

 		if([tempAccount.internalObjectID isEqualToString:inAccount]) {

 			exactMatchForInternalID = YES;

 			break;

 		}

 	}

 	if(!account) {

 		// No exact match. Fail.

 		return;

 	}

 	if ([inAction isEqualToString:@"retweet"]) {

 		[account retweetTweet:inTweet];

 	} else if ([inAction isEqualToString:@"reply"]) {

 		AIChat *timelineChat = [adium.chatController existingChatWithName:account.timelineChatName

 																onAccount:account];

 		if (!timelineChat) {

 			// Timeline chat isn't already open. Open it.

 			timelineChat = [adium.chatController chatWithName:account.timelineChatName

 												   identifier:nil

 													onAccount:account

 											 chatCreationInfo:nil];

 		}

 		if (!timelineChat.isOpen) {

 			[adium.interfaceController openChat:timelineChat];

 		}

 		[adium.interfaceController setActiveChat:timelineChat];

 		AIMessageEntryTextView *textView = ((AIMessageTabViewItem *)[timelineChat valueForProperty:@"MessageTabViewItem"]).messageViewController.textEntryView;

 		// Insert the @reply text

 		NSString *prefix = [NSString stringWithFormat:@"@%@ ", inUser];

 		if (![textView.string hasPrefix:prefix]) {

 			NSMutableAttributedString *newString;

 			if (textView.attributedString.length > 0){

 				newString = [[[textView.attributedString attributedSubstringFromRange:NSMakeRange(0, 1)] mutableCopy] autorelease];

 				[newString replaceCharactersInRange:NSMakeRange(0, 1) withString:prefix];

 			}

 			else

 				newString = [[[NSMutableAttributedString alloc] initWithString:prefix attributes:[adium.contentController defaultFormattingAttributes]] autorelease];

 			[newString appendAttributedString:textView.attributedString];

 			[textView setAttributedString:newString];

 			// Shift the selected range over by the length of our prefix string

 			NSRange selectedRange = textView.selectedRange;

 			[textView setSelectedRange:NSMakeRange(selectedRange.location + prefix.length, selectedRange.length)];

 		}

 		// Make the text view have focus

 		[[adium.interfaceController windowForChat:timelineChat] makeFirstResponder:textView];

 		[timelineChat setValue:inTweet forProperty:@"TweetInReplyToStatusID" notify:NotifyNow];

 		[timelineChat setValue:inUser forProperty:@"TweetInReplyToUserID" notify:NotifyNow];

 		[timelineChat setValue:@"@" forProperty:@"Character Counter Prefix" notify:NotifyNow];

 		AILogWithSignature(@"Flagging chat %@ to in_reply_to_status_id = %@", timelineChat, inTweet);

 		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSTextDidChangeNotification object:textView];

 		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(textDidChange:) name:NSTextDidChangeNotification object:textView];

 	} else if ([inAction isEqualToString:@"favorite"]) {

 		[account toggleFavoriteTweet:inTweet];

 	} else if ([inAction isEqualToString:@"destroy"] && exactMatchForInternalID) {

 		if (inTweet && inMessage) {

 			// Confirm if the user wants to delete this tweet.

 			if (NSRunAlertPanel(AILocalizedString(@"Delete Tweet?", nil),

 								AILocalizedString(@"Are you sure you want to delete the tweet:\n\n\"%@\"\n\nThis action cannot be undone.", nil),

 								AILocalizedString(@"Delete", nil), AILocalizedString(@"Cancel", nil), nil,

 								[inMessage stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding]) == NSAlertDefaultReturn) {

 				[account destroyTweet:inTweet];

 			}

 		} else if (inDM && inMessage) {

 			// Confirm if the user wants to delete this DM.

 			if (NSRunAlertPanel(AILocalizedString(@"Delete Direct Message?", nil),

 								AILocalizedString(@"Are you sure you want to delete the direct message:\n\n\"%@\"\n\nThis action cannot be undone.", nil),

 								AILocalizedString(@"Delete", nil), AILocalizedString(@"Cancel", nil), nil,

 								[inMessage stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding]) == NSAlertDefaultReturn) {

 				[account destroyDirectMessage:inDM forUser:inUser];

 			}			

 		}

 	}

 }

 - (void)textDidChange:(NSNotification *)notification

 {

 	AIMessageEntryTextView *textView = [notification object];

 	AIChat *chat = textView.chat;

 	if(![chat valueForProperty:@"TweetInReplyToStatusID"] || ![chat valueForProperty:@"TweetInReplyToUserID"]) {

 		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSTextDidChangeNotification object:textView];

 		return;

 	}

 	NSString *contents = [textView string];

 	BOOL keepTweetValues = YES;

 	NSRange usernameRange = [contents rangeOfString:[NSString stringWithFormat:@"@%@", [chat valueForProperty:@"TweetInReplyToUserID"]]];

 	if (usernameRange.location == NSNotFound) {

 		keepTweetValues = NO;

 	}

 	if (!keepTweetValues) {

 		AILogWithSignature(@"Removing in_reply_to_status_id from chat %@", chat);

 		[chat setValue:nil forProperty:@"TweetInReplyToStatusID" notify:NotifyNow];

 		[chat setValue:nil forProperty:@"TweetInReplyToUserID" notify:NotifyNow];

 		[chat setValue:nil forProperty:@"Character Counter Prefix" notify:NotifyNow];

 		[[NSNotificationCenter defaultCenter] removeObserver:self name:NSTextDidChangeNotification object:textView];

 	}

 }

 @end