/*
 * Copyright 2002-2006, project beam (http://sourceforge.net/projects/beam).
 * All rights reserved. Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Oliver Tappe <beam@hirschkaefer.de>
 */

#include "TextViewCompleter.h"

#include <Looper.h>
#include <TextControl.h>
#include <stdio.h>

#include "AutoCompleterDefaultImpl.h"


// #pragma mark - TextViewWrapper


TextViewCompleter::TextViewWrapper::TextViewWrapper(BTextView* textView)
	:
	fTextView(textView)
{
}


void
TextViewCompleter::TextViewWrapper::GetEditViewState(BString& text,
	int32* caretPos)
{
	if (fTextView && fTextView->LockLooper()) {
		text = fTextView->Text();
		if (caretPos) {
			int32 end;
			fTextView->GetSelection(caretPos, &end);
		}
		fTextView->UnlockLooper();
	}
}


void
TextViewCompleter::TextViewWrapper::SetEditViewState(const BString& text,
	int32 caretPos, int32 selectionLength)
{
	if (fTextView && fTextView->LockLooper()) {
		fTextView->SetText(text.String(), text.Length());
		fTextView->Select(caretPos, caretPos + selectionLength);
		fTextView->ScrollToSelection();
		fTextView->UnlockLooper();
	}
}


BRect
TextViewCompleter::TextViewWrapper::GetAdjustmentFrame()
{
	BRect frame = fTextView->Bounds();
	frame = fTextView->ConvertToScreen(frame);
	frame.InsetBy(0, -3);
	return frame;
}


// #pragma mark -


TextViewCompleter::TextViewCompleter(BTextView* textView, ChoiceModel* model,
		PatternSelector* patternSelector)
	:
	BAutoCompleter(new TextViewWrapper(textView), model,
		new BDefaultChoiceView(), patternSelector),
	BMessageFilter(B_KEY_DOWN),
	fTextView(textView),
	fModificationsReported(false)
{
	fTextView->AddFilter(this);
}


TextViewCompleter::~TextViewCompleter()
{
	fTextView->RemoveFilter(this);
}


void
TextViewCompleter::SetModificationsReported(bool reported)
{
	fModificationsReported = reported;
}


void
TextViewCompleter::TextModified(bool updateChoices)
{
	EditViewStateChanged(updateChoices);
}


filter_result
TextViewCompleter::Filter(BMessage* message, BHandler** target)
{
	const char* bytes;
	int32 modifiers;
	if ((!target || message->FindString("bytes", &bytes) != B_OK
			|| message->FindInt32("modifiers", &modifiers) != B_OK)
		|| (modifiers & (B_CONTROL_KEY | B_COMMAND_KEY | B_OPTION_KEY
			| B_SHIFT_KEY)) != 0) {
		return B_DISPATCH_MESSAGE;
	}

	switch (bytes[0]) {
		case B_UP_ARROW:
			SelectPrevious();
			// Insert the current choice into the text view, so the user can
			// continue typing. This will not trigger another evaluation, since
			// we don't invoke EditViewStateChanged().
			ApplyChoice(false);
			return B_SKIP_MESSAGE;
		case B_DOWN_ARROW:
			SelectNext();
			// See above.
			ApplyChoice(false);
			return B_SKIP_MESSAGE;
		case B_PAGE_UP:
		{
			int32 index = SelectedChoiceIndex() - CountVisibleChoices();
			index = max_c(index, 0);
			Select(index);
			ApplyChoice(false);
			return B_SKIP_MESSAGE;
		}
		case B_PAGE_DOWN:
		{
			int32 index = SelectedChoiceIndex() + CountVisibleChoices();
			index = min_c(index, CountChoices() - 1);
			Select(index);
			ApplyChoice(false);
			return B_SKIP_MESSAGE;
		}

		case B_ESCAPE:
			CancelChoice();
			return B_DISPATCH_MESSAGE;
		case B_RETURN:
			if (IsChoiceSelected()) {
				ApplyChoice();
				EditViewStateChanged();
			} else
				CancelChoice();
			return B_DISPATCH_MESSAGE;
		case B_TAB: {
			// make sure that the choices-view is closed when tabbing out:
			CancelChoice();
			return B_DISPATCH_MESSAGE;
		}
		default:
			if (!fModificationsReported) {
				// dispatch message to textview manually...
				Looper()->DispatchMessage(message, *target);
				// ...and propagate the new state to the auto-completer:
				EditViewStateChanged();
				return B_SKIP_MESSAGE;
			}
			return B_DISPATCH_MESSAGE;
	}
}