/*
 * CPTableHeaderView.j
 * AppKit
 *
 * Created by Ross Boucher.
 * Copyright 2009 280 North, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

@import "CPTableColumn.j"
@import "CPTableView.j"
@import "CPView.j"
 
var CPThemeStatePressed = CPThemeState("pressed");

@implementation _CPTableColumnHeaderView : CPView
{
    CPTextField _textField;
}

- (void)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self)
    {   
        _textField = [[CPTextField alloc] initWithFrame:CGRectMake(5, 1, CGRectGetWidth([self bounds]) - 5, CGRectGetHeight([self bounds]) - 1)];
        [_textField setAutoresizingMask:CPViewWidthSizable|CPViewHeightSizable];
        [_textField setTextColor: [CPColor colorWithHexString: @"333333"]];
        [_textField setValue:[CPFont boldSystemFontOfSize:12.0] forThemeAttribute:@"font"];
        [_textField setValue:CPLeftTextAlignment forThemeAttribute:@"alignment"];
        [_textField setVerticalAlignment:CPCenterVerticalTextAlignment];
        [_textField setValue:CGSizeMake(0,1) forThemeAttribute:@"text-shadow-offset"];
	    [_textField setValue:[CPColor whiteColor] forThemeAttribute:@"text-shadow-color"];


        [self addSubview:_textField];
    }
    
    return self;
}

- (void)layoutSubviews
{
    var themeState = [self themeState];

    if((themeState & CPThemeStateSelected) && (themeState & CPThemeStatePressed))
        [self setBackgroundColor:[CPColor colorWithPatternImage:CPAppKitImage("tableview-headerview-highlighted-pressed.png", CGSizeMake(1.0, 22.0))]];
    else if (themeState == CPThemeStateSelected)
        [self setBackgroundColor:[CPColor colorWithPatternImage:CPAppKitImage("tableview-headerview-highlighted.png", CGSizeMake(1.0, 22.0))]];
    else if (themeState == CPThemeStatePressed)
        [self setBackgroundColor:[CPColor colorWithPatternImage:CPAppKitImage("tableview-headerview-pressed.png", CGSizeMake(1.0, 22.0))]];
    else 
        [self setBackgroundColor:[CPColor colorWithPatternImage:CPAppKitImage("tableview-headerview.png", CGSizeMake(1.0, 22.0))]];
}

- (void)setStringValue:(CPString)string
{
    [_textField setStringValue:string];
}

- (CPString)stringValue
{
    return [_textField stringValue];
}

- (void)textField
{
    return _textField;
}

- (void)sizeToFit
{
    [_textField sizeToFit];
}

- (void)setValue:(id)aValue forThemeAttribute:(id)aKey
{
    [_textField setValue:aValue forThemeAttribute:aKey];
}
@end

@implementation CPTableHeaderView : CPView
{
    int _resizedColumn @accessors(readonly, property=resizedColumn);
    int _draggedColumn @accessors(readonly, property=draggedColumn);
    int _pressedColumn @accessors(readonly, property=pressedColumn);
    
    float _draggedDistance @accessors(readonly, property=draggedDistance);
    float _lastLocation;
    float _columnOldWidth;
    
    CPTableView _tableView @accessors(property=tableView);
}

- (void)initWithFrame:(CGRect)aFrame
{
    self = [super initWithFrame:aFrame];

    if (self)
    {
        _resizedColumn = CPNotFound;
        _draggedColumn = CPNotFound;
        _pressedColumn = CPNotFound;
        _draggedDistance = 0.0;
        _lastLocation = nil;
        _columnOldWidth = nil;
        
        [self setBackgroundColor:[CPColor colorWithPatternImage:CPAppKitImage("tableview-headerview.png", CGSizeMake(1.0, 22.0))]];
    }

    return self;
}

- (int)columnAtPoint:(CGPoint)aPoint
{
    if (!CGRectContainsPoint([self bounds], aPoint))
        return CPNotFound;

    // at this point, we can essentially ignore height, because all columns have equal heights
    // and that height is equal to the height of our own bounds, which we know we are inside of

    var index = 0,
        count = [[_tableView tableColumns] count],
        tableSpacing = [_tableView intercellSpacing],
        tableColumns = [_tableView tableColumns],
        leftOffset = 0,
        pointX = aPoint.x;

    for (; index < count; index++)
    {
        var width = [tableColumns[index] width] + tableSpacing.width;

        if (pointX >= leftOffset && pointX < leftOffset + width)
            return index;

        leftOffset += width;
    }

    return CPNotFound;
}

- (CGRect)headerRectOfColumn:(int)aColumnIndex
{
    var tableColumns = [_tableView tableColumns],
        tableSpacing = [_tableView intercellSpacing],
        bounds = [self bounds];

    if (aColumnIndex < 0 || aColumnIndex > [tableColumns count])
        [CPException raise:"invalid" reason:"tried to get headerRectOfColumn: on invalid column"];

    // UPDATE COLUMN RANGES ?
    if (_tableView._dirtyTableColumnRangeIndex !== CPNotFound)
        [_tableView _recalculateTableColumnRanges];
        
    var tableRange = _tableView._tableColumnRanges[aColumnIndex];
    bounds.origin.x = tableRange.location;
    bounds.size.width = tableRange.length;
    
    return bounds;
}

- (CPRect)_resizeRectBeforeColumn:(CPInteger)column
{
    if (!([_tableView._tableColumns[column] resizingMask] & CPTableColumnUserResizingMask))
        return CGRectMakeZero();
        
    var rect = [self headerRectOfColumn:column];

    rect.origin.x -= 10;
    rect.size.width = 20;

    return rect;
}

- (void)_setPressedColumn:(CPInteger)column
{
    if (_pressedColumn != CPNotFound)
    {
        var headerView = [_tableView._tableColumns[_pressedColumn] headerView];
        [headerView unsetThemeState:CPThemeStatePressed];
    }    
    
    if (column != CPNotFound)
    {
        var headerView = [_tableView._tableColumns[column] headerView];
        [headerView setThemeState:CPThemeStatePressed];
    }
    
    _pressedColumn = column;
}

- (void)mouseDown:(CPEvent)theEvent
{
    var mouseLocation = [self convertPoint:[theEvent locationInWindow] fromView:nil],
        clickedColumn = [self columnAtPoint:mouseLocation];    
    
    if (clickedColumn == -1)
        return;
    
    [_tableView _sendDelegateDidMouseDownInHeader:clickedColumn];
    
    var resizeLocation = CGPointMake(mouseLocation.x + 10, mouseLocation.y),
        resizedColumn = [self columnAtPoint:resizeLocation] - 1;
    
    // 2 different tracking methods: one for resizing/stop-resizing, another one for selection/reordering
    if ([_tableView allowsColumnResizing] 
        && resizedColumn >= 0 
        && CGRectContainsPoint([self _resizeRectBeforeColumn:(resizedColumn + 1)], mouseLocation))
    {
        _resizedColumn = resizedColumn;
        [_tableView._tableColumns[_resizedColumn] setDisableResizingPosting:YES];
        [self trackResizeWithEvent:theEvent];
    }
    else
    {
        [self _setPressedColumn:clickedColumn];
        [self trackMouseWithEvent:theEvent];
    }    
}

- (void)trackMouseWithEvent:(CPEvent)theEvent
{
    var type = [theEvent type];
    
    if (type == CPLeftMouseUp)
    {
        var location = [self convertPoint:[theEvent locationInWindow] fromView:nil],
            clickedColumn = [self columnAtPoint:location];

        [self _setPressedColumn:CPNotFound];
        
        if (clickedColumn == -1)
            return;    
        
        [_tableView _sendDelegateDidClickColumn:clickedColumn];
        
        if ([_tableView allowsColumnSelection])
        {        
            if ([theEvent modifierFlags] & CPCommandKeyMask)
            {
                if ([_tableView isColumnSelected:clickedColumn])
                    [_tableView deselectColumn:clickedColumn];
                else if ([_tableView allowsMultipleSelection] == YES)
                    [_tableView selectColumnIndexes:[CPIndexSet indexSetWithIndex:clickedColumn]  byExtendingSelection:YES];
            }
            else if ([theEvent modifierFlags] & CPShiftKeyMask)
            {
            // should be from clickedColumn to lastClickedColum with extending:(direction == previous selection)
                var selectedIndexes = [_tableView selectedColumnIndexes],
                    startColumn = MIN(clickedColumn, [selectedIndexes lastIndex]),
                    endColumn = MAX(clickedColumn, [selectedIndexes firstIndex]);
            
                [_tableView selectColumnIndexes:[CPIndexSet indexSetWithIndexesInRange:CPMakeRange(startColumn, endColumn - startColumn + 1)] byExtendingSelection:YES];
            }
            else
                [_tableView selectColumnIndexes:[CPIndexSet indexSetWithIndex:clickedColumn] byExtendingSelection:NO];
        }
        return;
    }
/*
    else if (type & CPLeftMouseDragged && [_tableView allowsColumnREordering])
    {
        // Start dragging here   
        [[CPCursor closedHandCursor] set];
        return;
    }
*/        
    [CPApp setTarget:self selector:@selector(trackMouseWithEvent:) forNextEventMatchingMask:CPLeftMouseDraggedMask | CPLeftMouseUpMask | CPLeftMouseDownMask untilDate:nil inMode:nil dequeue:YES];
}

- (void)trackResizeWithEvent:(CPEvent)anEvent
{
    var location = [self convertPoint:[anEvent locationInWindow] fromView:nil],
        tableColumn = [[_tableView tableColumns] objectAtIndex:_resizedColumn],
        type = [anEvent type];

    if (_lastLocation == nil) _lastLocation = location;
    if (_columnOldWidth == nil) _columnOldWidth = [tableColumn width];
        
    if (type === CPLeftMouseUp)
    {   
        [self _updateResizeCursor:anEvent];
                
        [tableColumn _postDidResizeNotificationWithOldWidth:_columnOldWidth];
        [tableColumn setDisableResizingPosting:NO];
        
        _resizedColumn = CPNotFound;
        _lastLocation = nil;
        _columnOldWidth = nil;     
        return;
    }            
    else if (type === CPLeftMouseDragged)
    {
        var newWidth = [tableColumn width] + location.x - _lastLocation.x;
        
        if (newWidth >= [tableColumn minWidth])
        {
            [tableColumn setWidth:newWidth];
            // FIXME: there has to be a better way to do this...
            // We should refactor the auto resizing crap.
            // We need to figure out the exact cocoa behavior here though. 
            [_tableView resizeWithOldSuperviewSize:[_tableView bounds]];
            _lastLocation = location;
                        
            [[CPCursor resizeLeftRightCursor] set];
            [self setNeedsLayout];
            [self setNeedsDisplay:YES];
        }
        else
            [[CPCursor resizeRightCursor] set];
    }
    
    [CPApp setTarget:self selector:@selector(trackResizeWithEvent:) forNextEventMatchingMask:CPLeftMouseDraggedMask | CPLeftMouseUpMask untilDate:nil inMode:nil dequeue:YES];
}

- (void)_updateResizeCursor:(CPEvent)theEvent
{
    var mouseLocation = [self convertPoint:[theEvent locationInWindow] fromView:nil];
    
    var mouseOverLocation = CGPointMake(mouseLocation.x + 10, mouseLocation.y),
        overColumn = [self columnAtPoint:mouseOverLocation];
    
    var isInside = (overColumn > 0 && CGRectContainsPoint([self _resizeRectBeforeColumn:overColumn], mouseLocation));
    if (isInside)
    {
        var column = [[_tableView tableColumns] objectAtIndex:overColumn - 1];
        if ([column width] == [column minWidth])
            [[CPCursor resizeRightCursor] set];
        else
            [[CPCursor resizeLeftRightCursor] set];
    }
    else
        [[CPCursor arrowCursor] set];
}

- (void)viewDidMoveToWindow
{
    if ([_tableView allowsColumnResizing])
        [[self window] setAcceptsMouseMovedEvents:YES];
}

- (void)mouseEntered:(CPEvent)theEvent
{   
    [self _updateResizeCursor:theEvent];
}

- (void)mouseMoved:(CPEvent)theEvent
{
    [self _updateResizeCursor:theEvent];
}

- (void)mouseExited:(CPEvent)theEvent
{
    // FIXME: we should use CPCursor push/pop (if previous currentCursor != arrow).
    [[CPCursor arrowCursor] set];
}

- (void)layoutSubviews
{
    var tableColumns = [_tableView tableColumns],
        count = [tableColumns count];    
    
    for (var i = 0; i < count; i++) 
    {
        var column = [tableColumns objectAtIndex:i],
            headerView = [column headerView];
        
        var frame = [self headerRectOfColumn:i];
        frame.size.height -= 0.5;
        if (i > 0) 
        {
            frame.origin.x += 0.5;
            frame.size.width -= 1;
        }
        
        [headerView setFrame:frame];
        
        if([headerView superview] != self)
            [self addSubview:headerView];
    }
}

- (void)drawRect:(CGRect)aRect
{
    var context = [[CPGraphicsContext currentContext] graphicsPort],
        exposedColumnIndexes = [_tableView columnIndexesInRect:aRect],
        columnsArray = [],
        tableColumns = [_tableView tableColumns],
        exposedTableColumns = _tableView._exposedColumns,
        firstIndex = [exposedTableColumns firstIndex],
        exposedRange = CPMakeRange(firstIndex, [exposedTableColumns lastIndex] - firstIndex + 1);

    CGContextSetLineWidth(context, 1);
    CGContextSetStrokeColor(context, [_tableView gridColor]);

    [exposedColumnIndexes getIndexes:columnsArray maxCount:-1 inIndexRange:exposedRange];

    var columnArrayIndex = 0,
        columnArrayCount = columnsArray.length,
        columnMaxX;
        
    CGContextBeginPath(context);
    for(; columnArrayIndex < columnArrayCount; columnArrayIndex++)
    {
        // grab each column rect and add vertical lines
        var columnIndex = columnsArray[columnArrayIndex],
            columnToStroke = [self headerRectOfColumn:columnIndex];
            
        columnMaxX = CGRectGetMaxX(columnToStroke);
        
        CGContextMoveToPoint(context, ROUND(columnMaxX) + 0.5, ROUND(CGRectGetMinY(columnToStroke)));
        CGContextAddLineToPoint(context, ROUND(columnMaxX) + 0.5, ROUND(CGRectGetMaxY(columnToStroke)));
    }
    
    CGContextClosePath(context);
    CGContextStrokePath(context);
        
/*
    var maxY = CGRectGetMaxY([self bounds]);
    // draw normal gradient for remaining space
    if (supportsCanvasGradient)
    {
        aRect.origin.x = columnMaxX - 0.5;
        aRect.size.width -= columnMaxX;
        CGContextBeginPath(context);
        CGContextAddRect(context, CGRectMake(columnMaxX + 1, 0, CGRectGetMaxX([self bounds]) - columnMaxX, CGRectGetHeight([self bounds])));
        CGContextClosePath(context);    
        CGContextDrawLinearGradient(context, [_CPTableColumnHeaderView headerGradient], CGPointMake(0,0), CGPointMake(0, maxY - 1),0);
    }   
   
    // Draw bottom line
    CGContextBeginPath(context);    
    CGContextMoveToPoint(context, 0, maxY - 0.5);
    CGContextAddLineToPoint(context, CGRectGetMaxX([self bounds]), maxY - 0.5);
    CGContextClosePath(context);
    CGContextStrokePath(context);
*/   
}

@end