//

//  AILogViewerWindowController.m

//  Adium

//

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

//

#import "AILogViewerWindowController.h"

#import "AIChatLog.h"

#import "AILogFromGroup.h"

#import "AILogToGroup.h"

#import "AILoggerPlugin.h"

#import "ESRankingCell.h" 

#import "AIXMLChatlogConverter.h"

#import "AILogDateFormatter.h"

#import <Adium/AIAccountControllerProtocol.h>

#import <Adium/AIChatControllerProtocol.h>

#import <Adium/AIContactControllerProtocol.h>

#import <Adium/AIContentControllerProtocol.h>

#import <Adium/AIMenuControllerProtocol.h>

#import <Adium/AIHTMLDecoder.h>

#import <Adium/AIInterfaceControllerProtocol.h>

#import <Adium/AIListContact.h>

#import <Adium/AIMetaContact.h>

#import <Adium/AIService.h>

#import <Adium/AIServiceIcons.h>

#import <Adium/AIUserIcons.h>

#import <Adium/KNShelfSplitView.h>

#import <AIUtilities/AIArrayAdditions.h>

#import <AIUtilities/AIAttributedStringAdditions.h>

#import <AIUtilities/AIDateFormatterAdditions.h>

#import <AIUtilities/AIFileManagerAdditions.h>

#import <AIUtilities/AIImageAdditions.h>

#import <AIUtilities/AIImageTextCell.h>

#import <AIUtilities/AIOutlineViewAdditions.h>

#import <AIUtilities/AISplitView.h>

#import <AIUtilities/AIStringAdditions.h>

#import <AIUtilities/AITableViewAdditions.h>

#import <AIUtilities/AITextAttributes.h>

#import <AIUtilities/AIToolbarUtilities.h>

#import <AIUtilities/AIApplicationAdditions.h>

#import <AIUtilities/AIDividedAlternatingRowOutlineView.h>

#define KEY_LOG_VIEWER_WINDOW_FRAME		@"Log Viewer Frame"

#define KEY_LOG_VIEWER_GROUP_STATE		@"Log Viewer Group State"	//Expand/Collapse state of groups

#define TOOLBAR_LOG_VIEWER				@"Log Viewer Toolbar"

#define MAX_LOGS_TO_SORT_WHILE_SEARCHING	10000	//Max number of logs we will live sort while searching

#define LOG_SEARCH_STATUS_INTERVAL			20	//1/60ths of a second to wait before refreshing search status

#define SEARCH_MENU						AILocalizedString(@"Search Menu",nil)

#define FROM							AILocalizedString(@"From",nil)

#define TO								AILocalizedString(@"To",nil)

#define DATE							AILocalizedString(@"Date",nil)

#define CONTENT							AILocalizedString(@"Content",nil)

#define DELETE							AILocalizedString(@"Delete",nil)

#define DELETEALL						AILocalizedString(@"Delete All",nil)

#define SEARCH							AILocalizedString(@"Search",nil)

#define HIDE_EMOTICONS					AILocalizedString(@"Hide Emoticons",nil)

#define SHOW_EMOTICONS					AILocalizedString(@"Show Emoticons",nil)

#define HIDE_TIMESTAMPS					AILocalizedString(@"Hide Timestamps",nil)

#define SHOW_TIMESTAMPS					AILocalizedString(@"Show Timestamps",nil)

#define IMAGE_EMOTICONS_OFF				@"emoticon32"

#define IMAGE_EMOTICONS_ON				@"emoticon32_transparent"

#define IMAGE_TIMESTAMPS_OFF			@"timestamp32"

#define IMAGE_TIMESTAMPS_ON				@"timestamp32_transparent"

#define	REFRESH_RESULTS_INTERVAL		1.0 //Interval between results refreshes while searching

@interface AILogViewerWindowController ()

- (id)initWithWindowNibName:(NSString *)windowNibName plugin:(id)inPlugin;

- (void)initLogFiltering;

- (void)displayLog:(AIChatLog *)log;

- (void)hilightOccurrencesOfString:(NSString *)littleString inString:(NSMutableAttributedString *)bigString firstOccurrence:(NSRange *)outRange;

- (void)sortCurrentSearchResultsForTableColumn:(NSTableColumn *)tableColumn direction:(BOOL)direction;

- (void)startSearchingClearingCurrentResults:(BOOL)clearCurrentResults;

- (void)buildSearchMenu;

- (NSMenuItem *)_menuItemWithTitle:(NSString *)title forSearchMode:(LogSearchMode)mode;

- (void)_logFilter:(NSString *)searchString searchID:(NSInteger)searchID mode:(LogSearchMode)mode;

- (void)installToolbar;

- (void)updateRankColumnVisibility;

- (void)openLogAtPath:(NSString *)inPath;

- (void)rebuildContactsList;

- (void)filterForContact:(AIListContact *)inContact;

- (void)filterForChatName:(NSString *)chatName withAccount:(AIAccount *)account;

- (void)selectCachedIndex;

- (void)tableViewSelectionDidChangeDelayed;

- (NSAlert *)alertForDeletionOfLogCount:(NSUInteger)logCount;

- (void)_willOpenForContact;

- (void)_didOpenForContact;

- (void)deleteSelection:(id)sender;

@end

@implementation AILogViewerWindowController

static AILogViewerWindowController	*sharedLogViewerInstance = nil;

static NSInteger toArraySort(id itemA, id itemB, void *context);

+ (NSString *)nibName

{

	return @"LogViewer";	

}

+ (id)openForPlugin:(id)inPlugin

{

    if (!sharedLogViewerInstance) {

		sharedLogViewerInstance = [[self alloc] initWithWindowNibName:[self nibName] plugin:inPlugin];

	}

    [sharedLogViewerInstance showWindow:nil];

	return sharedLogViewerInstance;

}

+ (id)openLogAtPath:(NSString *)inPath plugin:(id)inPlugin

{

	[self openForPlugin:inPlugin];

	[sharedLogViewerInstance openLogAtPath:inPath];

	return sharedLogViewerInstance;

}

//Open the log viewer window to a specific contact's logs

+ (id)openForContact:(AIListContact *)inContact plugin:(id)inPlugin

{

    if (!sharedLogViewerInstance) {

		sharedLogViewerInstance = [[self alloc] initWithWindowNibName:[self nibName] plugin:inPlugin];

	}

	[sharedLogViewerInstance _willOpenForContact];

	[sharedLogViewerInstance showWindow:nil];

	[sharedLogViewerInstance filterForContact:inContact];

	[sharedLogViewerInstance _didOpenForContact];

    return sharedLogViewerInstance;

}

+ (id)openForChatName:(NSString *)inChatName withAccount:(AIAccount *)inAccount plugin:(id)inPlugin

{

	if (!sharedLogViewerInstance) {

		sharedLogViewerInstance = [[self alloc] initWithWindowNibName:[self nibName] plugin:inPlugin];

	}

	[sharedLogViewerInstance _willOpenForContact];

	[sharedLogViewerInstance showWindow:nil];

	[sharedLogViewerInstance filterForChatName:inChatName withAccount:inAccount];

	[sharedLogViewerInstance _didOpenForContact];

    return sharedLogViewerInstance;

}

//Returns the window controller if one exists

+ (id)existingWindowController

{

    return sharedLogViewerInstance;

}

//Close the log viewer window

+ (void)closeSharedInstance

{

    if (sharedLogViewerInstance) {

        [sharedLogViewerInstance closeWindow:nil];

    }

}

//init

- (id)initWithWindowNibName:(NSString *)windowNibName plugin:(id)inPlugin

{

	if((self = [super initWithWindowNibName:windowNibName])) {

		plugin = inPlugin;

		selectedColumn = nil;

		activeSearchID = 0;

		searching = NO;

		automaticSearch = YES;

		showEmoticons = NO;

		showTimestamps = YES;

		activeSearchString = nil;

		displayedLogArray = nil;

		windowIsClosing = NO;

		desiredContactsSourceListDeltaX = 0;

		blankImage = [[NSImage alloc] initWithSize:NSMakeSize(16,16)];

		sortDirection = YES;

		searchMode = LOG_SEARCH_CONTENT;

		headerDateFormatter = [[NSDateFormatter localizedDateFormatter] retain];

		currentSearchResults = [[NSMutableArray alloc] init];

		fromArray = [[NSMutableArray alloc] init];

		fromServiceArray = [[NSMutableArray alloc] init];

		logFromGroupDict = [[NSMutableDictionary alloc] init];

		toArray = [[NSMutableArray alloc] init];

		toServiceArray = [[NSMutableArray alloc] init];

		logToGroupDict = [[NSMutableDictionary alloc] init];

		resultsLock = [[NSRecursiveLock alloc] init];

		searchingLock = [[NSLock alloc] init];

		[searchingLock setName:@"LogSearchingLock"];

		contactIDsToFilter = [[NSMutableSet alloc] initWithCapacity:1];

		allContactsIdentifier = [[NSNumber numberWithInteger:-1] retain];

		undoManager = [[NSUndoManager alloc] init];

		currentSearchLock = [[NSLock alloc] init];

		[currentSearchLock setName:@"CurrentLogSearchLock"];

	}

	return self;

}

//dealloc

- (void)dealloc

{

	[filterDate release]; filterDate = nil;

	[currentSearchLock release]; currentSearchLock = nil;

	[resultsLock release];

	[searchingLock release];

	[fromArray release];

	[fromServiceArray release];

	[toArray release];

	[toServiceArray release];

	[currentSearchResults release];

	[selectedColumn release];

	[headerDateFormatter release];

	[displayedLogArray release];

	[blankImage release];

	[activeSearchString release];

	[contactIDsToFilter release];

	[logFromGroupDict release]; logFromGroupDict = nil;

	[logToGroupDict release]; logToGroupDict = nil;

	[filterForAccountName release]; filterForAccountName = nil;

	[horizontalRule release]; horizontalRule = nil;

	[adiumIcon release]; adiumIcon = nil;

	[adiumIconHighlighted release]; adiumIconHighlighted = nil;

	//We loaded	view_DatePicker from a nib manually, so we must release it

	[view_DatePicker release]; view_DatePicker = nil;

	[allContactsIdentifier release];

	[undoManager release]; undoManager = nil;

	[super dealloc];

}

//Init our log filtering tree

- (void)initLogFiltering

{

    NSMutableDictionary		*toDict = [NSMutableDictionary dictionary];

    NSString				*basePath = [AILoggerPlugin logBasePath];

    NSString				*fromUID, *serviceClass;

    //Process each account folder (/Logs/SERVICE.ACCOUNT_NAME/) - sorting by compare: will result in an ordered list

	//first by service, then by account name.

    for (NSString *folderName in [[[NSFileManager defaultManager] contentsOfDirectoryAtPath:basePath error:NULL] sortedArrayUsingSelector:@selector(compare:)]) {

		if (![folderName isEqualToString:@".DS_Store"]) { // avoid the directory info

			AILogFromGroup  *logFromGroup;

			NSMutableSet	*toSetForThisService;

			NSArray         *serviceAndFromUIDArray;

			/* Determine the service and fromUID - should be SERVICE.ACCOUNT_NAME

			 * Check against count to guard in case of old, malformed or otherwise odd folders & whatnot sitting in log base

			 */

			serviceAndFromUIDArray = [folderName componentsSeparatedByString:@"."];

			if ([serviceAndFromUIDArray count] >= 2) {

				serviceClass = [serviceAndFromUIDArray objectAtIndex:0];

				//Use substringFromIndex so we include the rest of the string in the case of a UID with a . in it

				fromUID = [folderName substringFromIndex:([serviceClass length] + 1)]; //One off for the '.'

			} else {

				//Fallback: blank non-nil serviceClass; folderName as the fromUID

				serviceClass = @"";

				fromUID = folderName;

			}

			logFromGroup = [[AILogFromGroup alloc] initWithPath:folderName fromUID:fromUID serviceClass:serviceClass];

			//Store logFromGroup on a key in the form "SERVICE.ACCOUNT_NAME"

			[logFromGroupDict setObject:logFromGroup forKey:folderName];

			//To processing

			if (!(toSetForThisService = [toDict objectForKey:serviceClass])) {

				toSetForThisService = [NSMutableSet set];

				[toDict setObject:toSetForThisService

						   forKey:serviceClass];

			}

			//Add the 'to' for each grouping on this account

			for (AILogToGroup *currentToGroup in [logFromGroup toGroupArray]) {

				NSString	*currentTo;

				if ((currentTo = [currentToGroup to])) {

					//Store currentToGroup on a key in the form "SERVICE.ACCOUNT_NAME/TARGET_CONTACT"

					[logToGroupDict setObject:currentToGroup forKey:[currentToGroup relativePath]];

				}

			}

			[logFromGroup release];

		}

	}

	[self rebuildContactsList];

}

- (void)rebuildContactsList

{

	NSInteger	oldCount = toArray.count;

	[toArray release]; toArray = [[NSMutableArray alloc] initWithCapacity:(oldCount ? oldCount : 20)];

	for (AILogFromGroup *logFromGroup in [logFromGroupDict objectEnumerator]) {

		//Add the 'to' for each grouping on this account

		for (AILogToGroup *currentToGroup in [logFromGroup toGroupArray]) {

			NSString	*currentTo;

			if ((currentTo = [currentToGroup to])) {

				NSString *serviceClass = [currentToGroup serviceClass];

				AIListObject *listObject = ((serviceClass && currentTo) ?

											[adium.contactController existingListObjectWithUniqueID:[AIListObject internalObjectIDForServiceID:serviceClass

																																			 UID:currentTo]] :

											nil);

				if (listObject && [listObject isKindOfClass:[AIListContact class]]) {

					AIListContact *parentContact = [(AIListContact *)listObject parentContact];

					if (![toArray containsObjectIdenticalTo:parentContact]) {

						[toArray addObject:parentContact];

					}

				} else {

					if (![toArray containsObject:currentToGroup]) {

						[toArray addObject:currentToGroup];

					}

				}

			}

		}		

	}

	[toArray sortUsingFunction:toArraySort context:NULL];

	[outlineView_contacts reloadData];

	if (!isOpeningForContact) {

		//If we're opening for a contact, the outline view selection will be changed in a moment anyways

		[self outlineViewSelectionDidChange:nil];

	}

}

- (NSString *)adiumFrameAutosaveName

{

	return KEY_LOG_VIEWER_WINDOW_FRAME;

}

//Setup the window before it is displayed

- (void)windowDidLoad

{

	suppressSearchRequests = YES;

	[super windowDidLoad];

	[plugin pauseIndexing];

	[[self window] setTitle:AILocalizedString(@"Chat Transcript Viewer",nil)];

    [textField_progress setStringValue:@""];

	//Autosave doesn't do anything yet

	[shelf_splitView setAutosaveName:@"LogViewer:Shelf"];

	[shelf_splitView setFrame:[[[self window] contentView] frame]];

	// Pull our main article/display split view out of the nib and position it in the shelf view

	[containingView_results retain];

	[containingView_results removeFromSuperview];

	[shelf_splitView setContentView:containingView_results];

	[containingView_results release];

	[tableView_results accessibilitySetOverrideValue:AILocalizedString(@"Transcripts", nil)

										forAttribute:NSAccessibilityRoleDescriptionAttribute];

	// Pull our source view out of the nib and position it in the shelf view

	[containingView_contactsSourceList retain];

	[containingView_contactsSourceList removeFromSuperview];

	[shelf_splitView setShelfView:containingView_contactsSourceList];

	[outlineView_contacts accessibilitySetOverrideValue:AILocalizedString(@"Contacts", nil)

										forAttribute:NSAccessibilityRoleDescriptionAttribute];

	[containingView_contactsSourceList release];

	//Set emoticon filtering

	showEmoticons = [[adium.preferenceController preferenceForKey:KEY_LOG_VIEWER_EMOTICONS

															  group:PREF_GROUP_LOGGING] boolValue];

	[[toolbarItems objectForKey:@"toggleemoticons"] setLabel:(showEmoticons ? HIDE_EMOTICONS : SHOW_EMOTICONS)];

	[[toolbarItems objectForKey:@"toggleemoticons"] setImage:[NSImage imageNamed:(showEmoticons ? IMAGE_EMOTICONS_ON : IMAGE_EMOTICONS_OFF) forClass:[self class]]];

	// Set timestamp filtering

	showTimestamps = [[adium.preferenceController preferenceForKey:KEY_LOG_VIEWER_TIMESTAMPS

															   group:PREF_GROUP_LOGGING] boolValue];

	[[toolbarItems objectForKey:@"toggletimestamps"] setLabel:(showTimestamps ? HIDE_TIMESTAMPS : SHOW_TIMESTAMPS)];

	[[toolbarItems objectForKey:@"toggletimestamps"] setImage:[NSImage imageNamed:(showTimestamps ? IMAGE_TIMESTAMPS_ON : IMAGE_TIMESTAMPS_OFF) forClass:[self class]]];

	//Toolbar

	[self installToolbar];	

	[outlineView_contacts setSelectionHighlightStyle:NSTableViewSelectionHighlightStyleSourceList];

	AIImageTextCell	*dataCell = [[AIImageTextCell alloc] init];

	NSTableColumn	*tableColumn = [[outlineView_contacts tableColumns] objectAtIndex:0];

	[tableColumn setDataCell:dataCell];

	[tableColumn setEditable:NO];

	[dataCell setFont:[NSFont systemFontOfSize:[NSFont smallSystemFontSize]]];

	[dataCell release];

	// Set the selector for doubleAction

	[outlineView_contacts setDoubleAction:@selector(openChatOnDoubleAction:)];

	//Localize tableView_results column headers

	[[[tableView_results tableColumnWithIdentifier:@"To"] headerCell] setStringValue:TO];

	[[[tableView_results tableColumnWithIdentifier:@"From"] headerCell] setStringValue:FROM];

	[[[tableView_results tableColumnWithIdentifier:@"Date"] headerCell] setStringValue:DATE];

	[self tableViewColumnDidResize:nil];

	[tableView_results sizeLastColumnToFit];

	//Prepare the search controls

	[self buildSearchMenu];

	if ([textView_content respondsToSelector:@selector(setUsesFindPanel:)]) {

		[textView_content setUsesFindPanel:YES];

	}

    //Sort by preference, defaulting to sorting by date

	NSString	*selectedTableColumnPref;

	if ((selectedTableColumnPref = [adium.preferenceController preferenceForKey:KEY_LOG_VIEWER_SELECTED_COLUMN

																		   group:PREF_GROUP_LOGGING])) {

		selectedColumn = [[tableView_results tableColumnWithIdentifier:selectedTableColumnPref] retain];

	}

	if (!selectedColumn) {

		selectedColumn = [[tableView_results tableColumnWithIdentifier:@"Date"] retain];

	}

	[self sortCurrentSearchResultsForTableColumn:selectedColumn direction:YES];

    //Prepare indexing and filter searching

	[plugin prepareLogContentSearching];

    [self initLogFiltering];

    //Begin our initial search

	if (!isOpeningForContact)

		[self setSearchMode:LOG_SEARCH_TO];

    [searchField_logs setStringValue:(activeSearchString ? activeSearchString : @"")];

	suppressSearchRequests = NO;

	if (!isOpeningForContact) {

		//If we're opening for a contact, we'll select it and then begin searching

		[self startSearchingClearingCurrentResults:YES];

	}

	[tableView_results setAutosaveName:@"LogViewerResults"];

	[tableView_results setAutosaveTableColumns:YES];

	[plugin resumeIndexing];

}

-(void)rebuildIndices

{

    //Rebuild the 'global' log indexes

    [logFromGroupDict release]; logFromGroupDict = [[NSMutableDictionary alloc] init];

    [toArray removeAllObjects]; //note: even if there are no logs, the name will remain [bug or feature?]

    [toServiceArray removeAllObjects];

    [fromArray removeAllObjects];

    [fromServiceArray removeAllObjects];

    [self initLogFiltering];

    [tableView_results reloadData];

    [self selectDisplayedLog];

}

//Called as the window closes

- (void)windowWillClose:(id)sender

{

	[super windowWillClose:sender];

	//Set preference for emoticon filtering

	[adium.preferenceController setPreference:[NSNumber numberWithBool:showEmoticons]

										 forKey:KEY_LOG_VIEWER_EMOTICONS

										  group:PREF_GROUP_LOGGING];

	// Set preference for timestamp filtering

	[adium.preferenceController setPreference:[NSNumber numberWithBool:showTimestamps]

																			 forKey:KEY_LOG_VIEWER_TIMESTAMPS

																				group:PREF_GROUP_LOGGING];

	//Set preference for selected column

	[adium.preferenceController setPreference:[selectedColumn identifier]

										 forKey:KEY_LOG_VIEWER_SELECTED_COLUMN

										  group:PREF_GROUP_LOGGING];

    /* Disable the search field.  If we don't disable the search field, it will often try to call its target action

     * after the window has closed (and we are gone).  I'm not sure why this happens, but disabling the field

     * before we close the window down seems to prevent the crash.

	 */

    [searchField_logs setEnabled:NO];

	/* Note that the window is closing so we don't take behaviors which could cause messages to the window after

	 * it was gone, like responding to a logIndexUpdated message

	 */

	windowIsClosing = YES;

    //Abort any in-progress searching and indexing, and wait for their completion

    [self stopSearching];

    [plugin cleanUpLogContentSearching];

	//Reset our column widths if needed

	[activeSearchString release]; activeSearchString = nil;

	[self updateRankColumnVisibility];

	[sharedLogViewerInstance autorelease]; sharedLogViewerInstance = nil;

	[toolbarItems autorelease]; toolbarItems = nil;

}

//Display --------------------------------------------------------------------------------------------------------------

#pragma mark Display

//Update log viewer progress string to reflect current status

- (void)updateProgressDisplay

{

    NSMutableString     *progress = nil;

    NSUInteger					indexNumber, indexTotal;

    BOOL				indexing;

    //We always convey the number of logs being displayed

    [resultsLock lock];

	NSUInteger count = [currentSearchResults count];

    if (activeSearchString && [activeSearchString length]) {

		[shelf_splitView setResizeThumbStringValue:[NSString stringWithFormat:((count != 1) ? 

																			   AILocalizedString(@"%lu matching transcripts",nil) :

																			   AILocalizedString(@"1 matching transcript",nil)),count]];

    } else {

		[shelf_splitView setResizeThumbStringValue:[NSString stringWithFormat:((count != 1) ? 

																			   AILocalizedString(@"%lu transcripts",nil) :

																			   AILocalizedString(@"1 transcript",nil)),count]];

		//We are searching, but there is no active search  string. This indicates we're still opening logs.

		if (searching) {

			progress = [[AILocalizedString(@"Opening transcripts",nil) mutableCopy] autorelease];			

		}

    }

    [resultsLock unlock];

	indexing = [plugin getIndexingProgress:&indexNumber outOf:&indexTotal];

    //Append search progress

    if (activeSearchString && [activeSearchString length]) {

		if (progress) {

			[progress appendString:@" - "];

		} else {

			progress = [NSMutableString string];

		}

		if (searching || indexing) {

			[progress appendString:[NSString stringWithFormat:AILocalizedString(@"Searching for '%@'",nil),activeSearchString]];

		} else {

			[progress appendString:[NSString stringWithFormat:AILocalizedString(@"Search for '%@' complete.",nil),activeSearchString]];			

		}

	}

    //Append indexing progress

    if (indexing) {

		if (progress) {

			[progress appendString:@" - "];

		} else {

			progress = [NSMutableString string];

		}

		[progress appendString:[NSString stringWithFormat:AILocalizedString(@"Indexing %lu of %lu transcripts",nil), indexNumber, indexTotal]];

    }

	if (progress && (searching || indexing || !(activeSearchString && [activeSearchString length]))) {

		[progress appendString:[NSString ellipsis]];	

	}

    //Enable/disable the searching animation

    if (searching || indexing) {

		[progressIndicator startAnimation:nil];

    } else {