/***************************************************************************
                          StandardContentController.m
                          -------------------
    begin                : Tue Jan 20 22:08:40 CST 2004
    copyright            : (C) 2005 by Andrew Ruder
    email                : aeruder@ksu.edu
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#import "Controllers/ContentControllers/StandardChannelController.h"
#import "Controllers/ContentControllers/StandardContentController.h"
#import "Controllers/ContentControllers/StandardQueryController.h"
#import "Controllers/InputController.h"
#import "Controllers/Preferences/ColorPreferencesController.h"
#import "Controllers/Preferences/FontPreferencesController.h"
#import "Controllers/Preferences/PreferencesController.h"
#import "Controllers/ConnectionController.h"
#import "GNUstepOutput.h"
#import "Misc/NSAttributedStringAdditions.h"
#import "Misc/NSColorAdditions.h"

#import <AppKit/NSAttributedString.h>
#import <AppKit/NSFont.h>
#import <AppKit/NSNibLoading.h>
#import <AppKit/NSTextView.h>
#import <AppKit/NSWindow.h>
#import <Foundation/NSArray.h>
#import <Foundation/NSDictionary.h>
#import <Foundation/NSMapTable.h>
#import <Foundation/NSNotification.h>
#import <Foundation/NSNull.h>
#import <AppKit/NSColor.h>
#import <Foundation/NSString.h>
#import <Foundation/NSEnumerator.h>

@interface StandardContentController (PrivateMethods)
- (NSColor *)colorForKey: (NSString *)aKey;
- (void)viewSelected: (NSNotification *)aNotification;
- (void)userOpened: (NSNotification *)aNotification;
- (void)windowWillClose: (NSNotification *)aNotification;
@end

@implementation StandardContentController
+ (Class)masterClass
{
	return Nil;
}
+ (Class)queryClass
{
	return [StandardQueryController class];
}
+ (Class)channelClass
{
	return [StandardChannelController class];
}
- initWithMasterController: (id <MasterController>) aMaster
{
	if (!(self = [super init])) return nil;
	
	masterControllers = [NSMutableArray new];
	nameToChannel = [NSMutableDictionary new];
	nameToQuery = [NSMutableDictionary new];
	nameToBoth = [NSMutableDictionary new];
	nameToPresentation = [NSMutableDictionary new];
	nameToLabel = [NSMutableDictionary new];
	nameToMasterController = [NSMutableDictionary new];
	nameToTyping = [NSMutableDictionary new];
	nameToTitle = [NSMutableDictionary new];
	bothToName = NSCreateMapTable(NSObjectMapKeyCallBacks,
	  NSObjectMapValueCallBacks, 10);
	  
	if (!aMaster) 
		aMaster = AUTORELEASE([[[self class] masterClass] new]); 
	if (!aMaster)
	{
		[self dealloc];
		return nil;
	}
	
	[masterControllers addObject: aMaster];

	channelClass = [[self class] channelClass];
	queryClass = [[self class] queryClass];
	[[NSNotificationCenter defaultCenter] addObserver: self
	  selector: @selector(viewSelected:)
	  name: ContentControllerSelectedNameNotification
	  object: self];
	
	[[NSNotificationCenter defaultCenter] addObserver: self
	  selector: @selector(userOpened:)
	  name: ChannelControllerUserOpenedNotification
	  object: nil];

	[[NSNotificationCenter defaultCenter] addObserver: self
	  selector: @selector(windowWillClose:)
	  name: NSWindowWillCloseNotification
	  object: nil];

	return self;
}
/* Initializes the content controller in a new master
 * controller 
 */
- init
{
	return [self initWithMasterController: nil];
}
/* Returns the corresponding input manager for a view.
 */
- (id <TypingController>)typingControllerForViewController:
   (id <ContentControllerQueryController>)aController
{
	id name;

	if (!aController) return nil;
	name = NSMapGet(bothToName, aController);
	if (!name) return nil;

	return [nameToTyping objectForKey: name];
}
/* Sets the connectioncontroller for this content controller.
 * Does not retain.
 */
- (void)setConnectionController: (ConnectionController *)aController
{
	NSEnumerator *iter;
	id object;

	connectionController = aController;
	iter = [nameToTyping objectEnumerator];
	while ((object = [iter nextObject])) {
		[object setConnectionController: aController];
	}	
}
/* Returns ConnectionController for this content controller.
 */
- (ConnectionController *)connectionController
{
	return connectionController;
}
/* Dealllocate this instance while ensuring that it is removed
 * from the notification pool.
 */
- (void)dealloc
{
	[[NSNotificationCenter defaultCenter] removeObserver: self];

	RELEASE(masterControllers);
	RELEASE(nameToChannel);
	RELEASE(nameToQuery);
	RELEASE(nameToBoth);
	RELEASE(nameToPresentation);
	RELEASE(nameToLabel);
	RELEASE(nameToMasterController);
	RELEASE(nameToTyping);
	RELEASE(nameToTitle);
	NSFreeMapTable(bothToName);
	RELEASE(nickname);
	connectionController = nil;
	lastSelected = nil;

	[super dealloc];
}
/* Returns an array of all master controllers that are used by any of the channels or 
 * queries within this controller
 */
- (NSArray *)masterControllers
{
	return masterControllers;
}
/* Returns the default master controller that queries and channels will be added to.
 * Defaults to the master controller holding the console view.
 */
- (id <MasterController>)primaryMasterController
{
	return [masterControllers objectAtIndex: 0];
}
/* Sets another primary master controller, <var>aController</var>
 */
- (void)setPrimaryMasterController: (id <MasterController>)aController
{
	[masterControllers removeObject: aController];
	[masterControllers insertObject: aController atIndex: 0];
}	
/* Returns the name of the view specified by <var>aController</var>
 */
- (NSString *)nameForViewController: (id <ContentControllerQueryController>)aController
{
	return NSMapGet(bothToName, aController);
}
/* Returns the master controller used by the view with the name 
 * <var>aName</var>
 */
- (id <MasterController>)masterControllerForName: (NSString *)aName
{
	return [nameToMasterController objectForKey: GNUstepOutputLowercase(aName, connectionController)];
}
/* Returns the chat view for the name <var>aName</var>
 */
- (NSTextView *)chatViewForName: (NSString *)aName
{
	return [[nameToBoth objectForKey: GNUstepOutputLowercase(aName, connectionController)] chatView];
}
/* Returns the controller for the name <var>aName</var>
 * This will conform to [(ContentControllerQueryView)] in the case of
 * a query or [(ContentControllerQueryView)] and [(ContentControllerChannelName)]
 * in the case of a channel. 
 */
- (id <ContentControllerQueryController>)viewControllerForName: (NSString *)aName;
{
	return [nameToBoth objectForKey: GNUstepOutputLowercase(aName, connectionController)];
}
/* Returns the type of view for the name <var>aName</var>.  The types is either
 * <var>ContentControllerChannelType</var> or <var>ContentControllerQueryType</var>.
 */
- (NSString *)typeForName: (NSString *)aName
{
	id object = [nameToBoth objectForKey: GNUstepOutputLowercase(aName, connectionController)];
	
	if (!object) return nil;
	
	if ([object conformsToProtocol: @protocol(ContentControllerChannelController)])
	{
		return ContentControllerChannelType;
	}
	
	if ([object conformsToProtocol: @protocol(ContentControllerQueryController)])
	{
		return ContentControllerQueryType;
	}
	
	return nil;
}
/* Returns an array of all chat views of all queries and all channels
 */
- (NSArray *)allChatViews
{
	NSMutableArray *anArray = [NSMutableArray new];
	NSEnumerator *iter;
	id obj;
	
	iter = [[nameToBoth allValues] objectEnumerator];
	while ((obj = [iter nextObject]))
	{
		if ((obj = [obj chatView]))
		{
			[anArray addObject: obj];
		}
	}
	
	return AUTORELEASE(anArray);
}
/* Returns an array of all controllers.
 */
- (NSArray *)allControllers
{
	return [nameToBoth allValues];
}
/* Returns an array of all names.
 */
- (NSArray *)allNames
{
	return [nameToBoth allKeys];
}
/* Will return all chat views of the type <var>aType</var> which can either
 * be <var>ContentControllerChannelType</var> or <var>ContentControllerQueryType</var>
 */ 
- (NSArray *)allChatViewsOfType: (NSString *)aType
{
	NSMutableArray *anArray = [NSMutableArray new];
	NSArray *targetArray;
	NSEnumerator *iter;
	id obj;
	
	if ([aType isEqualToString: ContentControllerChannelType])
	{
		targetArray = [nameToChannel allValues];
	}
	else if ([aType isEqualToString: ContentControllerQueryType])
	{
		targetArray = [nameToQuery allValues];
	}
	else
	{
		return AUTORELEASE(anArray);
	}
	
	iter = [targetArray objectEnumerator];
	while ((obj = [iter nextObject]))
	{
		if ((obj = [obj chatView]))
		{
			[anArray addObject: obj];
		}
	}
	
	return AUTORELEASE(anArray);
}
/* Returns an array of all controllers of a certain type <var>aType</var> which can either
 * be <var>ContentControllerChannelType</var> or <var>ContentControllerQueryType</var>
 */
- (NSArray *)allViewControllersOfType: (NSString *)aType
{
	if ([aType isEqualToString: ContentControllerChannelType])
	{
		return [nameToChannel allValues];
	}
	else if ([aType isEqualToString: ContentControllerQueryType])
	{
		return [nameToQuery allValues];
	}
	
	return AUTORELEASE([NSArray new]);
}
/* Returns array of all the names of a certain type <var>aType</var> which can
 * be either <var>ContentControllerChannelType</var> or <var>ContentControllerQueryType</var>
 */			
- (NSArray *)allNamesOfType: (NSString *)aType
{
	if ([aType isEqualToString: ContentControllerChannelType])
	{
		return [nameToChannel allKeys];
	}
	else if ([aType isEqualToString: ContentControllerQueryType])
	{
		return [nameToQuery allKeys];
	}
	
	return AUTORELEASE([NSArray new]);
}
/* Calls putMessage:in:withEndLine: as [self putMessage: aMessage: in: aName 
 * withEndLine: YES];
 */
- (void)putMessage: aMessage in: (id)aName
{
	[self putMessage: aMessage in: aName withEndLine: YES];
}
/* Puts the message <var>aMessage</var> in <var>aName</var> with an optional
 * endline character appended to the end (specified by <var>hasEnd</var>).
 * <var>aName</var> can be a view conforming to <ContentControllerQueryView>,
 * a NSString with the channel name, a NSAttributedString of the channel name,
 * or an NSArray of any of the above.  If it is nil, it'll put it in the 
 * currently visible channel.
 */
- (void)putMessage: aMessage in: (id)aName 
    withEndLine: (BOOL)hasEnd
{
	id <ContentControllerQueryController> controller = nil;
	id string;
	
	if (!aMessage) return;
	
	if ([aName conformsToProtocol: @protocol(ContentControllerQueryController)])
	{
		controller = aName;
	}
	else if ([aName isKindOfClass: [NSString class]])
	{
		controller = [nameToBoth objectForKey: GNUstepOutputLowercase(aName, connectionController)];
	}
	else if ([aName isKindOfClass: [NSAttributedString class]])
	{
		controller = [nameToBoth objectForKey: 
		    GNUstepOutputLowercase([aName string], connectionController)];
	}
	else if ([aName isKindOfClass: [NSArray class]])
	{
		NSEnumerator *iter;
		id object;
		
		iter = [aName objectEnumerator];
		while ((object = [iter nextObject]))
		{
			[self putMessage: aMessage in: object withEndLine: hasEnd];
		}
		return;
	}
	
	if (controller == nil)
	{
		controller = lastSelected;
	}

	string = [NSMutableAttributedString 
	  attributedStringWithGNUstepOutputPreferences: aMessage];

	[controller appendAttributedString: string];
	
	if (hasEnd)
	{
		[controller appendAttributedString: [NSMutableAttributedString
		  attributedStringWithGNUstepOutputPreferences: @"\n"]];
	}
}
- (void)putMessageInAll: aMessage
{
	[self putMessageInAll: aMessage withEndLine: YES];
}
- (void)putMessageInAll: aMessage
    withEndLine: (BOOL)hasEnd
{
	NSEnumerator *iter;
	id obj;
	
	iter = [[nameToBoth allKeys] objectEnumerator];
	
	while ((obj = [iter nextObject]))
	{
		[self putMessage: aMessage in: obj withEndLine: hasEnd];
	}
}
- (void)putMessageInAll: aMessage
    ofType: (NSString *)aType
{
	[self putMessageInAll: aMessage ofType: aType withEndLine: YES];
}
- (void)putMessageInAll: aMessage
    ofType: (NSString *)aType
    withEndLine: (BOOL)hasEnd
{
	NSArray *targetArray;
	NSEnumerator *iter;
	id obj;
	
	if ([aType isEqualToString: ContentControllerChannelType])
	{
		targetArray = [nameToChannel allKeys];
	}
	else if ([aType isEqualToString: ContentControllerQueryType])
	{
		targetArray = [nameToQuery allKeys];
	}
	else
	{
		return;
	}
	
	iter = [targetArray objectEnumerator];
	while ((obj = [iter nextObject]))
	{
		[self putMessage: aMessage in: obj withEndLine: hasEnd];
	}
}
- (id <ContentControllerQueryController>)addViewControllerOfType: (NSString *)aType 
   withName: (NSString *)aName 
   withLabel: (NSAttributedString *)aLabel 
   inMasterController: (id <MasterController>)aMaster
{
	id controller;
	id name;
	BOOL isQuery, isChannel;
	
	isQuery = [aType isEqualToString: ContentControllerQueryType];
	isChannel = [aType isEqualToString: ContentControllerChannelType];
	
	name = GNUstepOutputLowercase(aName, connectionController);
	
	if ((controller = [nameToBoth objectForKey: name]))
	{
		[self setLabel: aLabel forName: name];
		return controller;
	}
		
	if (!isQuery && !isChannel)
	{
		return nil;
	}
	if (isQuery)
	{
		controller = AUTORELEASE([queryClass new]);
		
		[nameToQuery setObject: controller forKey: name];
	}
	else if (isChannel)
	{
		controller = AUTORELEASE([channelClass new]);
		
		[nameToChannel setObject: controller forKey: name];
	}
		
	[nameToBoth setObject: controller forKey: name];
	[nameToPresentation setObject: aName forKey: name];
	[nameToLabel setObject: aLabel forKey: name];
	
	NSMapInsert(bothToName, controller, name);
	
	if (!aMaster) aMaster = [masterControllers objectAtIndex: 0];
	
	[aMaster addViewController: controller withLabel: aLabel 
	  forContentController: self];
	[nameToMasterController setObject: aMaster forKey: name];
	
	[nameToTyping setObject: 
	 AUTORELEASE([[InputController alloc] 
	  initWithViewController: controller
	  contentController: self]) forKey: name];
	
	return controller;
}
- (void)removeViewControllerWithName: (NSString *)aName
{
	id master;
	id lo;
	id cont;
	
	lo = GNUstepOutputLowercase(aName, connectionController);
	
	master = [nameToMasterController objectForKey: lo];
	
	if (!master)
	{
		return;
	}
	
	cont = [nameToBoth objectForKey: lo];
	if (!cont)
	{
		return;
	}

	[master removeViewController: cont];
	
	[nameToChannel removeObjectForKey: lo];	
	[nameToQuery removeObjectForKey: lo];
	[nameToBoth removeObjectForKey: lo];
	[nameToPresentation removeObjectForKey: lo];
	[nameToLabel removeObjectForKey: lo];
	[nameToTyping removeObjectForKey: lo];
	[nameToMasterController removeObjectForKey: lo];
	[nameToTitle removeObjectForKey: lo];
	NSMapRemove(bothToName, cont);

	AUTORELEASE(masterControllers);
	masterControllers = [[NSMutableArray alloc] 
	  initWithArray: 
	  [nameToMasterController allValues]];

	if (![masterControllers count]) 
	{
		AUTORELEASE(RETAIN(self)); /* Do not want to die yet... */
		[connectionController setContentController: nil];
		connectionController = nil;
	}
}
- (void)renameViewControllerWithName: (NSString *)aName to: (NSString *)newName
{
	id lo1, lo2;
	id obj, which;
	
	lo1 = GNUstepOutputLowercase(aName, connectionController);
	lo2 = GNUstepOutputLowercase(newName, connectionController);
	
	if (![nameToBoth objectForKey: lo1]) return;
	
	if ([lo1 isEqualToString: lo2])
	{
		if (![[nameToPresentation objectForKey: lo1]
		       isEqualToString: newName])
		{
			[nameToPresentation setObject: newName forKey: lo2];
		}
		return;
	}
	
	if ([nameToBoth objectForKey: lo2]) return;
	
	[nameToPresentation setObject: newName forKey: lo2];
	[nameToPresentation removeObjectForKey: lo1];
		
	obj = [nameToBoth objectForKey: lo1];
	which = 
	  ([obj conformsToProtocol: @protocol(ContentControllerChannelController)]) ?
	  nameToChannel : nameToQuery;
		
	[nameToBoth setObject: obj forKey: lo2];
	[which setObject: obj forKey: lo2];
		
	[nameToBoth removeObjectForKey: lo1];
	[which removeObjectForKey: lo1];
	
	NSMapInsert(bothToName, obj, lo2);
		
	[nameToLabel setObject: [nameToLabel objectForKey:
	  lo1] forKey: lo2];
	[nameToLabel removeObjectForKey: lo1];
}
- (NSAttributedString *)labelForName: (NSString *)aName
{
	return [nameToLabel objectForKey: GNUstepOutputLowercase(aName, connectionController)];
}
- (void)setLabel: (NSAttributedString *)aLabel forName: (NSString *)aName
{
	id label;
	id lo;
	id cont;
	id mast;
	
	lo = GNUstepOutputLowercase(aName, connectionController);
	
	if (!(label = RETAIN([nameToLabel objectForKey: lo])))
	{
		return;
	}
	
	if (!(cont = [nameToBoth objectForKey: lo]))
	{
		return;
	}
	
	if (!(mast = [nameToMasterController objectForKey: lo]))
	{
		return;
	}
	
	AUTORELEASE(label);

	if (label == aLabel) return;
	
	[nameToLabel setObject: aLabel forKey: GNUstepOutputLowercase(aName, connectionController)];

	[mast setLabel: aLabel forViewController: cont];
	
	[[NSNotificationCenter defaultCenter]
	 postNotificationName: ContentControllerChangedLabelNotification
	 object: self userInfo: [NSDictionary dictionaryWithObjectsAndKeys:
	  label, @"OldLabel",
	  aLabel, @"Label",
	  cont, @"View",
	  self, @"Content",
	  mast, @"Master",
	  nil]];

	return;
}
- (NSString *)presentationalNameForName: (NSString *)aName
{
	return [nameToPresentation objectForKey: GNUstepOutputLowercase(aName, connectionController)];
}
- (void)setPresentationName: (NSString *)aPresentationName forName: (NSString *)aName
{
	[nameToPresentation setObject: aPresentationName forKey: GNUstepOutputLowercase(aName, connectionController)];
}
- (NSString *)nickname
{
	return nickname;
}
- (void)setNickname: (NSString *)aNickname
{
	if (aNickname == nickname) return;
	
	RELEASE(nickname);
	nickname = RETAIN(aNickname);

	[[NSNotificationCenter defaultCenter]
	 postNotificationName: ContentControllerChangedNicknameNotification
	 object: self userInfo: [NSDictionary dictionaryWithObjectsAndKeys:
	  nickname, @"OldNickname",
	  aNickname, @"Nickname",
	  self, @"Content",
	  nil]];

	return;
}
- (NSString *)titleForViewController: (id <ContentControllerQueryController>)aController
{
	id name;
	id title;
	id server;

	name = NSMapGet(bothToName, aController);
	if (name && (title = [nameToTitle objectForKey: name]))
	{
		return title;
	} 
	else if (!name) 
	{
		return nil;
	}

	server = [connectionController serverString];
	title = [nameToPresentation objectForKey: name];
	if (!server || ![server length])
	{
		return title;
	}
	return [NSString stringWithFormat: _l(@"%@ / %@"), title, server];
}
- (void)setTitle: (NSString *)aTitle
    forViewController: (id <ContentControllerQueryController>)aController
{
	id name;
	
	name = NSMapGet(bothToName, aController);
	if (!name) 
	{
		return;
	}

	if (!aTitle)
	{
		[nameToTitle removeObjectForKey: name];
	} 
	else
	{
		[nameToTitle setObject: aTitle forKey: name];
	}

	[[NSNotificationCenter defaultCenter]
	 postNotificationName: ContentControllerChangedTitleNotification 
	 object: aController 
	 userInfo: [NSDictionary dictionaryWithObjectsAndKeys: 
	  [self titleForViewController: aController], @"Title",
	  self, @"Content",
	  [nameToMasterController objectForKey: name], @"Master",
	  nil]];
}
- (void)bringNameToFront: (NSString *)aName
{
	id <MasterController> master;
	id <ContentControllerQueryController> view;

	master = [nameToMasterController objectForKey: GNUstepOutputLowercase(aName, connectionController)];
	view = [nameToBoth objectForKey: GNUstepOutputLowercase(aName, connectionController)];
	
	if (master) {
		[master bringToFront];
		[master selectViewController: view];
	}
}
@end

@implementation StandardContentController (PrivateMethods)
- (NSColor *)colorForKey: (NSString *)aKey
{
	return [NSColor colorFromEncodedData: [_PREFS_ preferenceForKey:
	  aKey]];
}
- (void)viewSelected: (NSNotification *)aNotification
{
	id name;
	id label;
	
	lastSelected = [[aNotification userInfo] objectForKey: @"View"];

	name = NSMapGet(bothToName, lastSelected);
	if (!name) return;

	label = AUTORELEASE([[NSMutableAttributedString alloc] 
	  initWithString: [[nameToLabel objectForKey: name] string]]);
	[self setLabel: label forName: name];
}
- (void)userOpened: (NSNotification *)aNotification
{
	id chan;
	id view;
	id chanUser;
	id userInfo;
	id name;
	id master;

	userInfo = [aNotification userInfo];
	chanUser = [userInfo objectForKey: @"User"];
	view = [aNotification object];
	chan = [userInfo objectForKey: @"Channel"];

	if (!chan || !view || !chanUser || !(name = NSMapGet(bothToName, view))) 
		return;

	chanUser = [chanUser userName];
	master = [nameToMasterController objectForKey: name];

	[self addViewControllerOfType: ContentControllerQueryType
	  withName: chanUser withLabel: S2AS(chanUser)
	  inMasterController: master]; 

}
- (void)windowWillClose: (NSNotification *)aNotification
{
	NSWindow *win;
	NSEnumerator *iter;
	id object;

	win = [aNotification object];
	iter = [masterControllers objectEnumerator];
	while ((object = [iter nextObject])) 
	{
		if ([object window] == win) 
		{
			NSArray *names;

			names = [nameToMasterController allKeysForObject: object];
			iter = [names objectEnumerator];
			while ((object = [iter nextObject]))
			{
				[self removeViewControllerWithName: object];
			}
		}
	}
}
@end