/*
 * Copyright (c) 1998-2007 Matthijs Hollemans
 * All rights reserved. Distributed under the terms of the MIT License.
 */

#include "GrepListView.h"

#include <Catalog.h>
#include <MenuItem.h>
#include <Path.h>
#include <Window.h>

#include "Model.h"

ResultItem::ResultItem(const entry_ref& ref)
	: BStringItem("", 0, false),
	  ref(ref)
{
	BEntry entry(&ref);
	BPath path(&entry);
	SetText(path.Path());
}


GrepListView::GrepListView()
	: BOutlineListView("SearchResults",
		B_MULTIPLE_SELECTION_LIST,
		B_WILL_DRAW | B_NAVIGABLE)
{
	_CreatePopUpMenu();
}


void
GrepListView::AttachedToWindow()
{
	fContextMenu->SetTargetForItems(Window());

	BOutlineListView::AttachedToWindow();
}


void
GrepListView::MouseDown(BPoint where)
{
	int32 index = IndexOf(where);
	if (index >= 0) {
		if (Window() != NULL && Window()->CurrentMessage() != NULL) {
			uint32 buttons = Window()->CurrentMessage()->FindInt32("buttons");
			if ((buttons & B_SECONDARY_MOUSE_BUTTON) != 0) {
				BListItem* item = ItemAt(index);
				bool clickOnSelection = item != NULL && item->IsSelected();
				if (CurrentSelection() < 0 || !clickOnSelection)
					Select(index);
				if (CurrentSelection() >= 0)
					fContextMenu->Go(ConvertToScreen(where), true, true, true);
				return;
			}
		}
	}
	BOutlineListView::MouseDown(where);
}


ResultItem*
GrepListView::FindItem(const entry_ref& ref, int32* _index) const
{
	int32 count = FullListCountItems();
	for (int32 i = 0; i < count; i++) {
		ResultItem* item = dynamic_cast<ResultItem*>(FullListItemAt(i));
		if (item == NULL)
			continue;
		if (item->ref == ref) {
			*_index = i;
			return item;
		}
	}
	*_index = -1;
	return NULL;
}


ResultItem*
GrepListView::RemoveResults(const entry_ref& ref, bool completeItem)
{
	int32 index;
	ResultItem* item = FindItem(ref, &index);
	if (item == NULL)
		return NULL;

	// remove all the sub items
	while (true) {
		BListItem* subItem = FullListItemAt(index + 1);
		if (subItem && subItem->OutlineLevel() > 0)
			delete RemoveItem(index + 1);
		else
			break;
	}

	if (completeItem) {
		// remove file item itself
		delete RemoveItem(index);
		item = NULL;
	}

	return item;
}


void
GrepListView::_CreatePopUpMenu()
{
	fContextMenu = new BPopUpMenu("ContextMenu", false, false);

	#undef B_TRANSLATION_CONTEXT
	#define B_TRANSLATION_CONTEXT "GrepWindow"
	fContextMenu->AddItem(new BMenuItem(B_TRANSLATE("Select all"),
		new BMessage(MSG_SELECT_ALL), 'A', B_SHIFT_KEY));

	fContextMenu->AddItem(new BMenuItem(B_TRANSLATE("Trim to selection"),
		new BMessage(MSG_TRIM_SELECTION), 'T'));

	fContextMenu->AddSeparatorItem();

	fContextMenu->AddItem(new BMenuItem(B_TRANSLATE("Open selection"),
		new BMessage(MSG_OPEN_SELECTION), 'O'));

	fContextMenu->AddItem(new BMenuItem(B_TRANSLATE("Show files in Tracker"),
		new BMessage(MSG_SELECT_IN_TRACKER), 'K'));

	fContextMenu->AddItem(new BMenuItem(B_TRANSLATE("Copy text to clipboard"),
		new BMessage(MSG_COPY_TEXT), 'B'));
}