/*
 Copyright (c) 2006, Greg Hulands <ghulands@mac.com>
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without modification, 
 are permitted provided that the following conditions are met:
 
 Redistributions of source code must retain the above copyright notice, this list 
 of conditions and the following disclaimer.
 
 Redistributions in binary form must reproduce the above copyright notice, this 
 list of conditions and the following disclaimer in the documentation and/or other 
 materials provided with the distribution.
 
 Neither the name of Greg Hulands nor the names of its contributors may be used to 
 endorse or promote products derived from this software without specific prior 
 written permission.
 
 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 
 EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 
 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 
 SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 
 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 
 TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 
 BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY 
 WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#import "KTLog.h"
#import <stdarg.h>

@interface KTLogger (Private)

- (id)init;
+ (id)sharedLogger;
- (void)logFile:(char *)file lineNumber:(int)line loggingDomain:(NSString *)domain loggingLevel:(int)level message:(NSString *)log;
- (void)setLoggingLevel:(KTLoggingLevel)level forDomain:(NSString *)domain;

@end

NSString *KTLogKeyPrefix = @"KTLoggingLevel.";
NSString *KTLogWildcardDomain = @"*";

static KTLogger *_sharedLogger = nil;
static unsigned long long KTLogMaximumLogSize = 5242880; // 5MB
static BOOL KTLogToConsole = YES;
static id _loggingDelegate = nil;

static NSString *KTLevelMap[] = {
	@"Off",
	@"FATAL",
	@"ERROR",
	@"WARN",
	@"INFO",
	@"DEBUG"
};

@implementation KTLogger


+ (id)sharedLogger
{
	if (nil == _sharedLogger)
	{
		_sharedLogger = [[KTLogger alloc] init];
	}
	return _sharedLogger;
}

- (id)init
{
	if ((self = [super init]))
	{
		myLock = [[NSLock alloc] init];
		myLoggingLevels = [[NSMutableArray array] retain];
		// load in from user defaults
		NSUserDefaults *ud = [NSUserDefaults standardUserDefaults];
		NSEnumerator *e = [[ud dictionaryRepresentation] keyEnumerator];
		NSString *key;
		
		while ((key = [e nextObject]))
		{
			if ([key hasPrefix:KTLogKeyPrefix])
			{
				NSMutableDictionary *d = [NSMutableDictionary dictionary];
				NSString *domain = [key substringFromIndex:[KTLogKeyPrefix length]];
				[d setObject:domain forKey:@"domain"];
				[d setObject:[ud objectForKey:key] forKey:@"level"];
				[myLoggingLevels addObject:d];
			}
		}
		
		NSNumber *con = [ud objectForKey:@"KTLogToConsole"];
		if (con)
		{
			KTLogToConsole = [con boolValue];
		}
		NSNumber *size = [ud objectForKey:@"KTLogFileSize"];
		if (size)
		{
			KTLogMaximumLogSize = [size unsignedLongLongValue];
		}
	}
	return self;
}

- (id)retain { return self; }
- (id)autorelease { return self; }
- (oneway void)release { }

+ (void)setDelegate:(id)delegate
{
	if ([delegate respondsToSelector:@selector(logger:logged:)])
	{
		[_loggingDelegate autorelease];
		_loggingDelegate = [delegate retain];
	}
}

+ (void)setMaximumLogSize:(unsigned long long)bytes
{
	KTLogMaximumLogSize = bytes;
	[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithUnsignedLongLong:bytes]
											  forKey:@"KTLogFileSize"];
}

+ (void)setLogToConsole:(BOOL)flag
{
	KTLogToConsole = flag;
	[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithBool:flag]
											  forKey:@"KTLogToConsole"];
}

+ (void)setLoggingLevel:(KTLoggingLevel)level forDomain:(NSString *)domain
{
	[[KTLogger sharedLogger] setLoggingLevel:level forDomain:domain];
}

- (NSMutableDictionary *)recordForDomain:(NSString *)domain
{
	NSEnumerator *e = [myLoggingLevels objectEnumerator];
	NSMutableDictionary *cur;
	
	while ((cur = [e nextObject]))
	{
		if ([[cur objectForKey:@"domain"] isEqualToString:domain])
			return cur;
	}
	cur = [NSMutableDictionary dictionary];
	[cur setObject:domain forKey:@"domain"];
	[cur setObject:[NSNumber numberWithInt:KTLogOff] forKey:@"level"];
	[myLoggingLevels addObject:cur];
	return cur;
}

- (int)loggingLevelForDomain:(NSString *)domain
{
	NSNumber *level = [[NSUserDefaults standardUserDefaults] objectForKey:[KTLogKeyPrefix stringByAppendingString:domain]];
	
	if (level)
	{
		return [level intValue];
	}
	else if ([domain isEqualToString:KTLogWildcardDomain])
	{
		return KTLogOff;	// wildcard defaults to off
	}
	return DEFAULT_LEVEL;
}

- (void)setLoggingLevel:(KTLoggingLevel)level forDomain:(NSString *)domain
{
	[myLock lock];
	int currentLevel = [self loggingLevelForDomain:domain];
	
	if (currentLevel != level)
	{
		[[self recordForDomain:domain] setObject:[NSNumber numberWithInt:level]
										  forKey:@"level"];
		[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithInt:level]
												  forKey:[KTLogKeyPrefix stringByAppendingString:domain]];
	}
	[myLock unlock];
}

- (NSString *)logfileName
{
	NSString *logPath = [[NSString stringWithFormat:@"%@", NSHomeDirectory()] stringByAppendingPathComponent:@"Library/Logs/"];
	NSFileManager *fm = [NSFileManager defaultManager];
	BOOL isDir;
	
	if (!([fm fileExistsAtPath:logPath isDirectory:&isDir] && isDir))
	{
		if (![fm createDirectoryAtPath:logPath attributes:nil])
		{
			NSLog(@"Failed to create log directory: %@", logPath);
		}
	}
	
	NSString *processName = [[NSProcessInfo processInfo] processName];
	NSString *logName = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.ktlog", processName]];
	
	if (![fm fileExistsAtPath:logName])
	{
		if (![fm createFileAtPath:logName contents:[NSData data] attributes:nil])
		{
			NSLog(@"Failed to create log file at: %@", logName);
		}
	}
	
	return logName;
}

- (void)rotateLogs
{
	NSString *logPath = [[NSString stringWithFormat:@"%@", NSHomeDirectory()] stringByAppendingPathComponent:@"Library/Logs/"];
	NSFileManager *fm = [NSFileManager defaultManager];

	[myLog closeFile];
	[myLog release];
	myLog = nil;
	
	NSString *processName = [[NSProcessInfo processInfo] processName];
	int i = 0;
	NSString *logName = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%d.ktlog", processName, i]];
	
	while ([fm fileExistsAtPath:logName])
	{
		i++;
		logName = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%d.ktlog", processName, i]];
	}
	
	i++;
	NSString *from;
	NSString *to;
	
	while (i > 0)
	{
		from = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%d.ktlog", processName, i - 1]];
		to = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%d.ktlog", processName, i]];
		
		[fm movePath:from toPath:to handler:nil];
		i--;
	}
	
	from = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.ktlog", processName]];
	to = [logPath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%d.ktlog", processName, i]];
	[fm movePath:from toPath:to handler:nil];
}


// Support for logFile: ... assumes that lock has been acquired.
- (void)_logFile:(char *)file
	  lineNumber:(int)line
   loggingDomain:(NSString *)domain
	loggingLevel:(int)level
		  thread:(NSString *)thread
		 message:(NSString *)log
{
	NSDate *now = [NSDate date];
	NSString *filename = [NSString stringWithCString:file];
	NSNumber *lineNumber = [NSNumber numberWithInt:line];
	NSNumber *thisLevel = [NSNumber numberWithInt:level];

	NSDictionary *rec = [NSDictionary dictionaryWithObjectsAndKeys:now, @"t", filename, @"f", lineNumber, @"n", thisLevel, @"l", domain, @"d", log, @"m", thread, @"th", nil];
	NSData *recData = [NSArchiver archivedDataWithRootObject:rec];

	if (!myLog)
	{
		// need to get the log file handle
		myLog = [[NSFileHandle fileHandleForWritingAtPath:[self logfileName]] retain];
	}

	[myLog seekToEndOfFile];

	unsigned len = CFSwapInt32HostToLittle([recData length]);
	NSMutableData *entry = [NSMutableData data];
	[entry appendBytes:&len length:sizeof(unsigned)];
	[entry appendData:recData];

	[myLog writeData:entry];

	if (KTLogToConsole)
	{
		NSProcessInfo *pi = [NSProcessInfo processInfo];
		NSString *processName = [pi processName];
		NSString *nowDescription = [now descriptionWithCalendarFormat:@"%Y-%m-%d %H:%M:%S.%F"
															 timeZone:nil
															   locale:nil];
		NSString *logLevelString = (level >= 0 && level <= 5) ? KTLevelMap[level] : @"UNKNOWN";
		
		NSString *console = [NSString stringWithFormat:@"%@ %@[%d][%@:%@][%@:%d] %@\n",
			nowDescription, processName, [pi processIdentifier], logLevelString, domain,
			[filename lastPathComponent], line, log];
		const char *utf8String = [console UTF8String];
		fprintf(stderr, "%s", utf8String);
	}
	
	if (_loggingDelegate)
	{
		[_loggingDelegate logger:self logged:rec];
	}
	NSDictionary *logAttribs = [[NSFileManager defaultManager] fileAttributesAtPath:[self logfileName] traverseLink:YES];
	if ([[logAttribs objectForKey:NSFileSize] unsignedLongLongValue] > KTLogMaximumLogSize)
	{
		[self rotateLogs];
	}
}


- (void)logFile:(char *)file
	 lineNumber:(int)line
  loggingDomain:(NSString *)domain
   loggingLevel:(int)level
		 thread:(NSString *)thread
		message:(NSString *)log
{
	[myLock lock];
	
	if (	level <= [self loggingLevelForDomain:KTLogWildcardDomain]
		||	level <= [self loggingLevelForDomain:domain])	// only log statement whose level is at or below my threshold
	{
		[self _logFile:file lineNumber:line loggingDomain:domain loggingLevel:level thread:thread message:log];
	}
	[myLock unlock];
}

// Similar to above, but with a format and arguments.  Don't construct the string unless we want to use it.
- (void)logFile:(char *)file
	 lineNumber:(int)line
  loggingDomain:(NSString *)domain
   loggingLevel:(int)level
		 thread:(NSString *)thread
		 format:(NSString *)log
	  arguments:(va_list)argList
{
	[myLock lock];
	
	if (	level <= [self loggingLevelForDomain:KTLogWildcardDomain]
		||	level <= [self loggingLevelForDomain:domain])	// only log statement whose level is at or below my threshold
	{
		NSString *message = [[[NSString alloc] initWithFormat:log arguments:argList] autorelease];
		[self _logFile:file
			lineNumber:line
		 loggingDomain:domain
		  loggingLevel:level
				thread:thread
			   message:message];
	}
	[myLock unlock];
}	

// Similar to above, but with a format
- (void)logFile:(char *)file
	 lineNumber:(int)line
  loggingDomain:(NSString *)domain
   loggingLevel:(int)level
		 thread:(NSString *)thread
		 format:(NSString *)log, ...
{
	va_list ap;
	va_start(ap, log);
	
	[self logFile:file
	   lineNumber:line
	loggingDomain:domain
	 loggingLevel:level
		   thread:thread
		   format:log
		arguments:ap];
	
	va_end(ap);
}

// Class method to log.  Accepts a format.  Thread argument is generated here, not passed in.

+ (void)logFile:(char *)file
	 lineNumber:(int)line
  loggingDomain:(NSString *)domain
   loggingLevel:(int)level
		 format:(NSString *)log, ...
{
	va_list ap;
	va_start(ap, log);
	
	[[KTLogger sharedLogger] logFile:file
						  lineNumber:line
					   loggingDomain:domain
						loggingLevel:level
							  thread:[NSString stringWithFormat:@"0x%06x",[NSThread currentThread]]
							 format:log
						   arguments:ap];

	va_end(ap);
}

+ (NSArray *)entriesWithLogFile:(NSString *)file
{
	NSFileHandle *log = [NSFileHandle fileHandleForReadingAtPath:file];
	NSMutableArray *entries = [NSMutableArray array];
	
	@try {
		// keep going until we throw an exception for being out of bounds
		while (1)
		{
			unsigned len;
			NSData *lenData = [log readDataOfLength:sizeof(unsigned)];
			[lenData getBytes:&len];
			len = CFSwapInt32LittleToHost(len);
			NSData *archive = [log readDataOfLength:len];
			NSDictionary *record = [NSUnarchiver unarchiveObjectWithData:archive];
			[entries addObject:record];
		}
	} 
	@catch (NSException *e) {
	
	}
	[log closeFile];
	
	return entries;
}

#pragma mark -
#pragma mark Configuration UI

- (IBAction)configure:(id)sender
{
	if (!oPanel)
	{
		[NSBundle loadNibNamed:@"KTLog" owner:self];
	}
	[oDomains reloadData];
	[oPanel makeKeyAndOrderFront:self];
}

- (void)awakeFromNib
{
	NSPopUpButtonCell *cell = [[NSPopUpButtonCell alloc] initTextCell:@"" pullsDown:NO];
	[cell setBordered:NO];
	[cell removeAllItems];
	[cell addItemsWithTitles:[self levelNames]];
	[[oDomains tableColumnWithIdentifier:@"level"] setDataCell:cell];
	[cell release];
	
	[oDomains setDataSource:self];
}

+ (IBAction)configure:(id)sender
{
	[[KTLogger sharedLogger] configure:sender];
}

- (NSArray *)levelNames
{
	return [NSArray arrayWithObjects:KTLevelMap count:KTLogDebug + 1];
}

- (NSArray *)domains
{
	return [[myLoggingLevels retain] autorelease];
}

- (void)setLogToConsole:(BOOL)flag
{
	KTLogToConsole = flag;
	[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithBool:flag]
											  forKey:@"KTLogToConsole"];
}

- (BOOL)logToConsole
{
	return KTLogToConsole;
}

- (void)logToConsoleChanged:(id)sender
{
	[self setLogToConsole:[sender state] == NSOnState];
}

#pragma mark -
#pragma mark NSTableView Datasource

- (int)numberOfRowsInTableView:(NSTableView *)aTableView
{
	return [myLoggingLevels count];
}

- (id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex
{
	NSString *ident = [aTableColumn identifier];
	NSDictionary *rec = [myLoggingLevels objectAtIndex:rowIndex];
	if ([ident isEqualToString:@"domain"])
	{
		return [rec objectForKey:@"domain"];
	}
	else
	{
		return [rec objectForKey:@"level"];
	}
}

- (void)tableView:(NSTableView *)aTableView setObjectValue:(id)anObject forTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex
{
	NSString *ident = [aTableColumn identifier];
	NSMutableDictionary *rec = [myLoggingLevels objectAtIndex:rowIndex];
	if ([ident isEqualToString:@"domain"])
	{
		[rec setObject:anObject forKey:@"domain"];
	}
	else
	{
		[rec setObject:anObject forKey:@"level"];
	}
	
	[[NSUserDefaults standardUserDefaults] setObject:[rec objectForKey:@"level"]
											  forKey:[KTLogKeyPrefix stringByAppendingString:[rec objectForKey:@"domain"]]];
}

- (IBAction)addDomain:(id)sender
{
	NSMutableDictionary *cur = [NSMutableDictionary dictionary];
	[cur setObject:@"New Domain" forKey:@"domain"];
	[cur setObject:[NSNumber numberWithInt:KTLogOff] forKey:@"level"];
	[myLoggingLevels addObject:cur];
	[oDomains reloadData];
}
@end