// ================================================================================
// ==      This file is a part of Turbo Badger. (C) 2011-2014, Emil Segerås      ==
// ==                     See tb_core.h for more information.                    ==
// ================================================================================

#include "tb_select.h"
#include "tb_menu_window.h"
#include "tb_widgets_listener.h"
#include "tb_language.h"
#include <assert.h>
#include <stdlib.h>

namespace tb {

/** TBSimpleLayoutItemWidget is a item containing a layout with the following:
	-TBSkinImage showing the item image.
	-TBTextField showing the item string.
	-TBSkinImage showing the arrow for items with a submenu.
	It also handles submenu events. */

class TBSimpleLayoutItemWidget : public TBLayout, private TBWidgetListener
{
public:
	TBSimpleLayoutItemWidget(TBID image, TBSelectItemSource *source, const char *str);
	~TBSimpleLayoutItemWidget();
	virtual bool OnEvent(const TBWidgetEvent &ev);
private:
	TBSelectItemSource *m_source;
	TBTextField m_textfield;
	TBSkinImage m_image;
	TBSkinImage m_image_arrow;
	TBMenuWindow *m_menu; ///< Points to the submenu window if opened
	virtual void OnWidgetDelete(TBWidget *widget);
	void OpenSubMenu();
	void CloseSubMenu();
};

// == TBSimpleLayoutItemWidget ==============================================================================

TBSimpleLayoutItemWidget::TBSimpleLayoutItemWidget(TBID image, TBSelectItemSource *source, const char *str)
	: m_source(source)
	, m_menu(nullptr)
{
	SetSkinBg(TBIDC("TBSelectItem"));
	SetLayoutDistribution(LAYOUT_DISTRIBUTION_AVAILABLE);
	SetPaintOverflowFadeout(false);

	if (image)
	{
		m_image.SetSkinBg(image);
		m_image.SetIgnoreInput(true);
		AddChild(&m_image);
	}

	m_textfield.SetText(str);
	m_textfield.SetTextAlign(TB_TEXT_ALIGN_LEFT);
	m_textfield.SetIgnoreInput(true);
	AddChild(&m_textfield);

	if (source)
	{
		m_image_arrow.SetSkinBg(TBIDC("arrow.right"));
		m_image_arrow.SetIgnoreInput(true);
		AddChild(&m_image_arrow);
	}
}

TBSimpleLayoutItemWidget::~TBSimpleLayoutItemWidget()
{
	if (m_image_arrow.GetParent())
		RemoveChild(&m_image_arrow);
	RemoveChild(&m_textfield);
	if (m_image.GetParent())
		RemoveChild(&m_image);
	CloseSubMenu();
}

bool TBSimpleLayoutItemWidget::OnEvent(const TBWidgetEvent &ev)
{
	if (m_source && ev.type == EVENT_TYPE_CLICK && ev.target == this)
	{
		OpenSubMenu();
		return true;
	}
	return false;
}

void TBSimpleLayoutItemWidget::OnWidgetDelete(TBWidget *widget)
{
	assert(widget == m_menu);
	CloseSubMenu();
}

void TBSimpleLayoutItemWidget::OpenSubMenu()
{
	if (m_menu)
		return;

	// Open a new menu window for the submenu with this widget as target
	m_menu = new TBMenuWindow(this, TBIDC("submenu"));
	if (m_menu)
	{
		SetState(WIDGET_STATE_SELECTED, true);
		m_menu->AddListener(this);
		m_menu->Show(m_source, TBPopupAlignment(TB_ALIGN_RIGHT), -1);
	}
}

void TBSimpleLayoutItemWidget::CloseSubMenu()
{
	if (!m_menu)
		return;

	SetState(WIDGET_STATE_SELECTED, false);
	m_menu->RemoveListener(this);
	if (!m_menu->GetIsDying())
		m_menu->Close();
	m_menu = nullptr;
}

// == TBSelectItemViewer ==============================================================================

void TBSelectItemViewer::SetSource(TBSelectItemSource *source)
{
	if (m_source == source)
		return;

	if (m_source)
		m_source->m_viewers.Remove(this);
	m_source = source;
	if (m_source)
		m_source->m_viewers.AddLast(this);

	OnSourceChanged();
}

// == TBSelectItemSource ====================================================================================

TBSelectItemSource::~TBSelectItemSource()
{
    TBLinkListOf<TBSelectItemViewer>::Iterator iter = m_viewers.IterateForward();
    while (TBSelectItemViewer *viewer = iter.GetAndStep())
    {
        viewer->SetSource(nullptr);
    }

    m_viewers.RemoveAll();

	// If this assert trig, you are deleting a model that's still set on some
	// TBSelect widget. That might be dangerous.
    assert(!m_viewers.HasLinks());
}

bool TBSelectItemSource::Filter(int index, const char *filter)
{
	const char *str = GetItemString(index);
	if (str && stristr(str, filter))
		return true;
	return false;
}

TBWidget *TBSelectItemSource::CreateItemWidget(int index, TBSelectItemViewer *viewer)
{
	const char *string = GetItemString(index);
	TBSelectItemSource *sub_source = GetItemSubSource(index);
	TBID image = GetItemImage(index);
	if (sub_source || image)
	{
		if (TBSimpleLayoutItemWidget *itemwidget = new TBSimpleLayoutItemWidget(image, sub_source, string))
			return itemwidget;
	}
	else if (string && *string == '-')
	{
		if (TBSeparator *separator = new TBSeparator)
		{
			separator->SetGravity(WIDGET_GRAVITY_ALL);
			separator->SetSkinBg(TBIDC("TBSelectItem.separator"));
			return separator;
		}
	}
	else if (TBTextField *textfield = new TBTextField)
	{
		textfield->SetSkinBg("TBSelectItem");
		textfield->SetText(string);
		textfield->SetTextAlign(TB_TEXT_ALIGN_LEFT);
		return textfield;
	}
	return nullptr;
}

void TBSelectItemSource::InvokeItemChanged(int index, TBSelectItemViewer *exclude_viewer)
{
	TBLinkListOf<TBSelectItemViewer>::Iterator iter = m_viewers.IterateForward();
	while (TBSelectItemViewer *viewer = iter.GetAndStep())
		if (viewer != exclude_viewer)
			viewer->OnItemChanged(index);
}

void TBSelectItemSource::InvokeItemAdded(int index)
{
	TBLinkListOf<TBSelectItemViewer>::Iterator iter = m_viewers.IterateForward();
	while (TBSelectItemViewer *viewer = iter.GetAndStep())
		viewer->OnItemAdded(index);
}

void TBSelectItemSource::InvokeItemRemoved(int index)
{
	TBLinkListOf<TBSelectItemViewer>::Iterator iter = m_viewers.IterateForward();
	while (TBSelectItemViewer *viewer = iter.GetAndStep())
		viewer->OnItemRemoved(index);
}

void TBSelectItemSource::InvokeAllItemsRemoved()
{
	TBLinkListOf<TBSelectItemViewer>::Iterator iter = m_viewers.IterateForward();
	while (TBSelectItemViewer *viewer = iter.GetAndStep())
		viewer->OnAllItemsRemoved();
}

}; // namespace tb
