//
//  $Id$
//
//  SPTableInfo.m
//  sequel-pro
//
//  Created by Ben Perry on Jun 6, 2008
//
//  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
//
//  More info at <http://code.google.com/p/sequel-pro/>

#import <MCPKit/MCPKit.h>

#import "SPTableInfo.h"
#import "ImageAndTextCell.h"
#import "TableDocument.h"
#import "TablesList.h"
#import "SPTableData.h"
#import "SPStringAdditions.h"

@interface SPTableInfo (PrivateAPI)

- (NSString *)_getUserDefinedDateStringFromMySQLDate:(NSString *)mysqlDate;

@end

@implementation SPTableInfo

- (id)init
{
	if ((self = [super init])) {
		info = [[NSMutableArray alloc] init];
	}
	
	return self;
}

- (void)awakeFromNib
{
	[[NSNotificationCenter defaultCenter] addObserver:self
											 selector:@selector(tableChanged:) 
												 name:NSTableViewSelectionDidChangeNotification 
											   object:tableList];
	
	[info addObject:NSLocalizedString(@"TABLE INFORMATION", @"header for table info pane")];
	[infoTable reloadData];
}

- (void)dealloc
{
	[[NSNotificationCenter defaultCenter] removeObserver:self];
	
	[info release];
	
	[super dealloc];
}

- (void)tableChanged:(NSNotification *)notification
{
	NSDictionary *tableStatus;
	NSNumberFormatter *numberFormatter = [[[NSNumberFormatter alloc] init] autorelease];	
	[numberFormatter setNumberStyle:NSNumberFormatterDecimalStyle];

	[info removeAllObjects];

	// For views, no information can be displayed.
	if ([tableListInstance tableType] == SP_TABLETYPE_VIEW) {
		[info addObject:@"VIEW INFORMATION"];
		[info addObject:@"no information available"];
		[infoTable reloadData];
		return;
	}

	if ([[tableListInstance tableName] isEqualToString:@""]) {
		[info addObject:@"INFORMATION"];
		[info addObject:@"multiple selection"];
		[infoTable reloadData];
		return;
	} 

	// Get TABLE information
	if ([tableListInstance tableType] == SP_TABLETYPE_TABLE) {

		[info addObject:@"TABLE INFORMATION"];

		if ([tableListInstance tableName]) {
			if ([[tableListInstance tableName] isEqualToString:@""]) {
				[info addObject:@"multiple tables"];
			} 
			else {
				// Retrieve the table status information via the data cache
				tableStatus = [tableDataInstance statusValues];

				// Check for errors
				if (![tableStatus count]) {
					[info addObject:@"error occurred"];
					return;
				}

				// Check for 'Create_time' == NULL
				if (![[tableStatus objectForKey:@"Create_time"] isNSNull]) {

					// Add the creation date to the infoTable
					[info addObject:[NSString stringWithFormat:@"created: %@", [self _getUserDefinedDateStringFromMySQLDate:[tableStatus objectForKey:@"Create_time"]]]];
				}

				// Check for 'Update_time' == NULL - InnoDB tables don't have an update time
				if (![[tableStatus objectForKey:@"Update_time"] isNSNull]) {

					// Add the update date to the infoTable
					[info addObject:[NSString stringWithFormat:@"updated: %@", [self _getUserDefinedDateStringFromMySQLDate:[tableStatus objectForKey:@"Update_time"]]]];
				}

				// Check for 'Rows' == NULL - information_schema database doesn't report row count for it's tables
				if (![[tableStatus objectForKey:@"Rows"] isNSNull]) {
					[info addObject:[NSString stringWithFormat:[[tableStatus objectForKey:@"RowsCountAccurate"] boolValue] ? @"rows: %@" : @"rows: ~%@",
						[numberFormatter stringFromNumber:[NSNumber numberWithLongLong:[[tableStatus objectForKey:@"Rows"] longLongValue]]]]];
				} 

				[info addObject:[NSString stringWithFormat:@"size: %@", [NSString stringForByteSize:[[tableStatus objectForKey:@"Data_length"] longLongValue]]]];
				[info addObject:[NSString stringWithFormat:@"encoding: %@", [tableDataInstance tableEncoding]]];

				if (![[tableStatus objectForKey:@"Auto_increment"] isNSNull]) {
					[info addObject:[NSString stringWithFormat:@"auto_increment: %@",
						[numberFormatter stringFromNumber:[NSNumber numberWithLongLong:[[tableStatus objectForKey:@"Auto_increment"] longLongValue]]]]];
				}
			}
		}
	}
	
	// Get PROC/FUNC information
	else if ([tableListInstance tableType] == SP_TABLETYPE_PROC || [tableListInstance tableType] == SP_TABLETYPE_FUNC) {

		if ([tableListInstance tableType] == SP_TABLETYPE_PROC)
			[info addObject:@"PROCEDURE INFORMATION"];
		else
			[info addObject:@"FUNCTION INFORMATION"];
			
		if ([tableListInstance tableName]) {
			if ([[tableListInstance tableName] isEqualToString:@""]) {
				[info addObject:@"multiple tables"];
			} else {
				// Retrieve the table status information via the data cache
				tableStatus = [tableDataInstance statusValues];

				// Check for errors
				if (![tableStatus count]) {
					[info addObject:@"error occurred"];
					return;
				}

				// Add the creation date to the infoTable
				// Check for 'CREATED' == NULL
				if (![[tableStatus objectForKey:@"CREATED"] isNSNull]) {

					// Add the creation date to the infoTable
					[info addObject:[NSString stringWithFormat:@"created: %@", [self _getUserDefinedDateStringFromMySQLDate:[tableStatus objectForKey:@"CREATED"]]]];
				}

				// Check for 'LAST_ALTERED'
				if (![[tableStatus objectForKey:@"LAST_ALTERED"] isNSNull]) {
				
					// Add the update date to the infoTable
					[info addObject:[NSString stringWithFormat:@"updated: %@", [self _getUserDefinedDateStringFromMySQLDate:[tableStatus objectForKey:@"LAST_ALTERED"]]]];
				}

				// Check for 'SQL ACCESS' and deterministic
				if (![[tableStatus objectForKey:@"SQL_DATA_ACCESS"] isNSNull] && ![[tableStatus objectForKey:@"IS_DETERMINISTIC"] isNSNull]) {
					[info addObject:[NSString stringWithFormat:@"data access: %@ (%@)", [tableStatus objectForKey:@"SQL_DATA_ACCESS"], ([[tableStatus objectForKey:@"IS_DETERMINISTIC"] isEqualToString:@"YES"]) ? @"deterministic" : @"non-deterministic"]];
				}

				if ([tableListInstance tableType] == SP_TABLETYPE_FUNC) {
					// Check for 'DTD_IDENTIFIER'
					if (![[tableStatus objectForKey:@"DTD_IDENTIFIER"] isNSNull]) {
						[info addObject:[NSString stringWithFormat:@"return type: %@", [tableStatus objectForKey:@"DTD_IDENTIFIER"]]];
					}
				}

				// Check for 'SECURITY_TYPE'
				if (![[tableStatus objectForKey:@"SECURITY_TYPE"] isNSNull]) {
					[info addObject:[NSString stringWithFormat:@"execution privilege: %@", [tableStatus objectForKey:@"SECURITY_TYPE"]]];
				}

				// Check for 'DEFINER'
				if (![[tableStatus objectForKey:@"DEFINER"] isNSNull]) {
					[info addObject:[NSString stringWithFormat:@"definer: %@", [tableStatus objectForKey:@"DEFINER"]]];
				}

			}
		}
	}

	[infoTable reloadData];

}

#pragma mark -
#pragma mark TableView datasource methods

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

- (id)tableView:(NSTableView *)aTableView objectValueForTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex
{
	return [info objectAtIndex:rowIndex];
}

- (CGFloat)tableView:(NSTableView *)tableView heightOfRow:(NSInteger)row
{
	return (row == 0 ? 25 : [tableView rowHeight]);
}


- (BOOL)tableView:(NSTableView *)aTableView shouldSelectRow:(int)rowIndex
{
	// row 1 and 6 should be editable - ie be able to rename the table and change the auto_increment value.
	return NO;//(rowIndex == 1 || rowIndex == 6 );
}

- (BOOL)tableView:(NSTableView *)aTableView isGroupRow:(int)row
{
	// This makes the top row (TABLE INFORMATION) have the diff styling
	return (row == 0);	
}

- (void)tableView:(NSTableView *)aTableView willDisplayCell:(id)aCell forTableColumn:(NSTableColumn *)aTableColumn row:(int)rowIndex
{
	if ((rowIndex > 0) && [[aTableColumn identifier] isEqualToString:@"info"]) {
		[(ImageAndTextCell*)aCell setImage:[NSImage imageNamed:@"table-property"]];
		[(ImageAndTextCell*)aCell setIndentationLevel:1];
		[(ImageAndTextCell*)aCell setDrawsBackground:NO];
	} else {
		[(ImageAndTextCell*)aCell setImage:nil];
		[(ImageAndTextCell*)aCell setIndentationLevel:0];
		//[(ImageAndTextCell*)aCell setDrawsBackground:YES];
		//[(ImageAndTextCell*)aCell setBackgroundColor:[NSColor colorWithDeviceRed:0.894 green:0.917 blue:0.945 alpha:1]];
	}
}

@end

@implementation SPTableInfo (PrivateAPI)

- (NSString *)_getUserDefinedDateStringFromMySQLDate:(NSString *)mysqlDate
{
	// Setup our data formatter
	NSDateFormatter *dateFormatter = [[[NSDateFormatter alloc] init] autorelease];
	
	[dateFormatter setFormatterBehavior:NSDateFormatterBehavior10_4];
	
	[dateFormatter setDateStyle:NSDateFormatterShortStyle];
	[dateFormatter setTimeStyle:NSDateFormatterNoStyle];
	
	// Convert our string date from the result to an NSDate.
	NSDate *updateDate = [NSDate dateWithNaturalLanguageString:mysqlDate];
	
	return [dateFormatter stringFromDate:updateDate];
}

@end