//
//  PBGitSidebar.m
//  GitX
//
//  Created by Pieter de Bie on 9/8/09.
//  Copyright 2009 __MyCompanyName__. All rights reserved.
//

#import "PBGitSidebarController.h"
#import "PBSourceViewItem.h"

@interface PBGitSidebarController ()

- (void)populateList;

@end

@implementation PBGitSidebarController
@synthesize items;

- (id)initWithRepository:(PBGitRepository *)theRepository superController:(PBGitWindowController *)controller
{
	self = [super initWithRepository:theRepository superController:controller];
	[sourceView setDelegate:self];
	items = [NSMutableArray array];

	return self;
}

- (void)awakeFromNib
{
	[super awakeFromNib];
	window.contentView = self.view;
	[self populateList];
}

#pragma mark NSOutlineView delegate methods
- (BOOL)outlineView:(NSOutlineView *)outlineView isGroupItem:(id)item
{
	return [item isGroupItem];
}

- (void)outlineView:(NSOutlineView *)outlineView willDisplayCell:(id)cell forTableColumn:(NSTableColumn *)tableColumn item:(id)item
{
	[cell setImage:[item icon]];
}

- (BOOL)outlineView:(NSOutlineView *)outlineView shouldSelectItem:(id)item
{
	return ![item isGroupItem];
}

//
// The next two methods are necessary to hide the triangle for uncollapsible items
// That is, items which should always be displayed, such as the action items.
//
- (BOOL)outlineView:(NSOutlineView *)outlineView shouldCollapseItem:(id)item
{
	return !([item isUncollapsible]);
}

- (void)outlineView:(NSOutlineView *)outlineView willDisplayOutlineCell:(id)cell forTableColumn:(NSTableColumn *)tableColumn item:(id)item
{
	[cell setTransparent:[item isUncollapsible]];
} 

- (void)populateList
{
	PBSourceViewItem *actions = [PBSourceViewItem groupItemWithTitle:@"Actions"];

	PBSourceViewItem *branches = [PBSourceViewItem groupItemWithTitle:@"Branches"];
	PBSourceViewItem *remotes = [PBSourceViewItem groupItemWithTitle:@"Remotes"];
	PBSourceViewItem *tags = [PBSourceViewItem groupItemWithTitle:@"Tags"];
	PBSourceViewItem *custom = [PBSourceViewItem groupItemWithTitle:@"Custom"];

	for (PBGitRevSpecifier *rev in repository.branches)
	{
		if (![rev isSimpleRef]) {
			[custom addChild:[PBSourceViewItem itemWithRevSpec:rev]];
			continue;
		}
		NSArray *pathComponents = [[rev simpleRef] componentsSeparatedByString:@"/"];
		if ([[pathComponents objectAtIndex:1] isEqualToString:@"heads"])
			[branches addRev:rev toPath:[pathComponents subarrayWithRange:NSMakeRange(2, [pathComponents count] - 2)]];
		else if ([[rev simpleRef] hasPrefix:@"refs/tags/"])
			[tags addRev:rev toPath:[pathComponents subarrayWithRange:NSMakeRange(2, [pathComponents count] - 2)]];
		else if ([[rev simpleRef] hasPrefix:@"refs/remotes/"])
			[remotes addRev:rev toPath:[pathComponents subarrayWithRange:NSMakeRange(2, [pathComponents count] - 2)]];
		
	}

	[items addObject:actions];

	[items addObject:branches];
	[items addObject:remotes];
	[items addObject:tags];
	[items addObject:custom];

	[sourceView reloadData];
	[sourceView expandItem:branches expandChildren:YES];
	[sourceView expandItem:actions];

	NSAssert(actions == [sourceView itemAtRow:0], @"First item is not the Action");
	[sourceView reloadItem:nil reloadChildren:YES];

}

#pragma mark NSOutlineView Datasource methods

- (id)outlineView:(NSOutlineView *)outlineView child:(NSInteger)index ofItem:(id)item
{
	if (!item)
		return [items objectAtIndex:index];

	return [[(PBSourceViewItem *)item children] objectAtIndex:index];
}

- (BOOL)outlineView:(NSOutlineView *)outlineView isItemExpandable:(id)item
{
	return [[(PBSourceViewItem *)item children] count];
}

- (NSInteger)outlineView:(NSOutlineView *)outlineView numberOfChildrenOfItem:(id)item
{
	if (!item)
		return [items count];

	return [[(PBSourceViewItem *)item children] count];
}

- (id)outlineView:(NSOutlineView *)outlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
{
	return [(PBSourceViewItem *)item title];
}

@end