#include "textlist.h"
#include "tw/colors.h"
#include "tw/event.h"
#include "tw/scrollbar.h"
#include "tw/keycodes.h"

#include <vector>
#include <map>
#include <string>

using namespace std;
using namespace TwConsts;

namespace
{
const int rowHeight = 20; // 15
};

//==============================================================================

class TTextList::Data
{
public:
	Data(TWidget *w) : m_w(w), m_selAction(0), m_dblClickAction(0), m_scrollbar(new TScrollbar(w)), m_yoffset(0)
	{
	}

	~Data()
	{
		if (m_selAction)
			delete m_selAction;

		if (m_dblClickAction)
			delete m_dblClickAction;

		std::map<string, TTextListItem *>::iterator it = m_items.begin();
		for (; it != m_items.end(); ++it)
			delete it->second;
	}

	int posToItem(const TPoint &p);
	void updateScrollBarStatus()
	{
		if (!m_scrollbar)
			return;
		unsigned int ly = m_w->getLy();

		if ((m_items.size() * rowHeight) > ly) {
			m_scrollbar->setValue(m_yoffset, 0, (m_items.size() + 1) * rowHeight - ly, ly);
			m_scrollbar->show(); //m_scrollbar->setValue(m_yoffset,0, yrange-ly, ly);
		} else {
			m_yoffset = 0;
			m_scrollbar->hide(); //m_scrollbar->setValue(0,0, 0, 0);
		}
		m_scrollbar->invalidate();
	}

	TWidget *m_w;
	std::map<string, TTextListItem *> m_items;
	vector<string> m_selectedItems;

	TGenericTextListAction *m_selAction;
	TGenericTextListAction *m_dblClickAction;
	TScrollbar *m_scrollbar;
	int m_yoffset;
};

//------------------------------------------------------------------------------

int TTextList::Data::posToItem(const TPoint &p)
{
	/*  TDimension d = getSize();
  int i = getItemCount()-1;

  10 + 
  for(int y=10; y<d.ly && i>=0; y+=rowHeight, --i)
*/

	int y = m_w->getSize().ly - p.y + m_yoffset;
	int item = y / rowHeight;
	return item;
}

//==============================================================================

TTextListItem::TTextListItem(const string &id, const string &caption)
	: m_id(id), m_caption(caption)
{
}

//==============================================================================

TTextList::TTextList(TWidget *parent, string name)
	: TWidget(parent, name), m_data(0)
{
	m_data = new Data(this);
	m_data->m_scrollbar->setAction(new TScrollbarAction<TTextList>(this, &TTextList::scrollTo));
}

//------------------------------------------------------------------------------

TTextList::~TTextList()
{
	delete m_data;
}

//------------------------------------------------------------------------------

void TTextList::addItem(TTextListItem *item)
{
	std::map<string, TTextListItem *>::iterator it = m_data->m_items.find(item->getId());

	if (it == m_data->m_items.end()) {
		m_data->m_items.insert(make_pair(item->getId(), item));
		m_data->updateScrollBarStatus();
	}
}

//------------------------------------------------------------------------------

void TTextList::removeItem(const string &itemId)
{
	std::map<string, TTextListItem *>::iterator it = m_data->m_items.find(itemId);
	if (it != m_data->m_items.end()) {
		m_data->m_items.erase(it);
		m_data->updateScrollBarStatus();
	}
}

//------------------------------------------------------------------------------

void TTextList::clearAll()
{
	m_data->m_items.clear();
	m_data->m_selectedItems.clear();
	invalidate();
	m_data->updateScrollBarStatus();
}

//------------------------------------------------------------------------------

int TTextList::getItemCount() const
{
	return m_data->m_items.size();
}

//------------------------------------------------------------------------------

TTextListItem *TTextList::getItem(int i) const
{
	if (i >= 0 && i < (int)m_data->m_items.size()) {
		std::map<string, TTextListItem *>::iterator it = m_data->m_items.begin();
		advance(it, i);
		return it->second;
	}
	return 0;
}

//------------------------------------------------------------------------------

int TTextList::itemToIndex(const string &itemId)
{
	std::map<string, TTextListItem *>::iterator it = m_data->m_items.find(itemId);

	if (it == m_data->m_items.end())
		return -1;
	else
		return distance(m_data->m_items.begin(), it);
}

//------------------------------------------------------------------------------

int TTextList::getSelectedItemCount() const
{
	return m_data->m_selectedItems.size();
}

//------------------------------------------------------------------------------

TTextListItem *TTextList::getSelectedItem(int i) const
{
	if (i < 0 || (int)m_data->m_selectedItems.size() <= i)
		return 0;

	string itemId = m_data->m_selectedItems[i];
	std::map<string, TTextListItem *>::iterator it = m_data->m_items.find(itemId);
	if (it != m_data->m_items.end())
		return it->second;
	else
		return 0;
}

//------------------------------------------------------------------------------

string TTextList::getSelectedItemId(int i) const
{
	if ((int)m_data->m_selectedItems.size() <= i)
		return "";
	assert(i >= 0 && i < (int)m_data->m_selectedItems.size());
	return m_data->m_selectedItems[i];
}
//------------------------------------------------------------------------------

void TTextList::select(int i, bool on)
{
	assert(i >= 0 && i < (int)m_data->m_items.size());

	std::map<string, TTextListItem *>::iterator it = m_data->m_items.begin();
	advance(it, i);

	string id = it->first;
	vector<string>::iterator it2 =
		find(m_data->m_selectedItems.begin(), m_data->m_selectedItems.end(), id);

	if (on) {
		if (it2 == m_data->m_selectedItems.end())
			m_data->m_selectedItems.push_back(id);
	} else {
		if (it2 != m_data->m_selectedItems.end())
			m_data->m_selectedItems.erase(it2);
	}

	if (m_data->m_selAction)
		m_data->m_selAction->sendCommand(i);

	invalidate();
}

//------------------------------------------------------------------------------

void TTextList::select(const string &itemId, bool on)
{
	std::map<string, TTextListItem *>::iterator it =
		m_data->m_items.find(itemId);

	if (it != m_data->m_items.end()) {
		int i = distance(m_data->m_items.begin(), it);
		select(i, on);
	}
}

//------------------------------------------------------------------------------

void TTextList::unselectAll()
{
	m_data->m_selectedItems.clear();
	invalidate();
}

//------------------------------------------------------------------------------

bool TTextList::isSelected(int i) const
{
	assert(i >= 0 && i < (int)m_data->m_items.size());

	std::map<string, TTextListItem *>::iterator it = m_data->m_items.begin();
	advance(it, i);

	string id = it->first;

	vector<string>::iterator it2 =
		find(m_data->m_selectedItems.begin(), m_data->m_selectedItems.end(), id);

	return it2 != m_data->m_selectedItems.end();
}

//------------------------------------------------------------------------------

bool TTextList::isSelected(const string &item) const
{
	assert(false);
	return false;
}

//------------------------------------------------------------------------------

void TTextList::scrollTo(int y)
{
	y = (y / rowHeight) * rowHeight;
	if (m_data->m_yoffset == y)
		return;
	m_data->m_yoffset = y;
	m_data->updateScrollBarStatus();
	invalidate();
}

//------------------------------------------------------------------------------

void TTextList::draw()
{
	drawRect(TRect(TPoint(0, 0), getSize()));
	TDimension d = getSize();

	int y = d.ly - 1 - rowHeight;
	for (int i = m_data->m_yoffset / rowHeight; i < getItemCount() && y >= 0; ++i, y -= rowHeight) {
		if (isSelected(i)) {
			setColor(Blue, 2);
			fillRect(2, y - 2, getSize().lx - 1 - 2, y + rowHeight - 1 - 2);
		}

		setColor(Black);
		drawText(TPoint(10, y), getItem(i)->getCaption());
	}
}

//------------------------------------------------------------------------------

void TTextList::setSelAction(TGenericTextListAction *action)
{
	if (m_data->m_selAction)
		delete m_data->m_selAction;

	m_data->m_selAction = action;
}

//------------------------------------------------------------------------------

void TTextList::setDblClickAction(TGenericTextListAction *action)
{
	if (m_data->m_dblClickAction)
		delete m_data->m_dblClickAction;

	m_data->m_dblClickAction = action;
}

//------------------------------------------------------------------------------

void TTextList::configureNotify(const TDimension &d)
{
	m_data->m_scrollbar->setGeometry(d.lx - 20, 1, d.lx - 2, d.ly - 2);
	m_data->updateScrollBarStatus();
}

//------------------------------------------------------------------------------

void TTextList::leftButtonDown(const TMouseEvent &e)
{
	int i = m_data->posToItem(e.m_pos);
	if (i >= 0 && i < (int)m_data->m_items.size()) {
		if (!e.isShiftPressed())
			unselectAll();
		select(i, !isSelected(i));
	}
}

//------------------------------------------------------------------------------

void TTextList::leftButtonDoubleClick(const TMouseEvent &e)
{
	int i = m_data->posToItem(e.m_pos);
	if (i >= 0 && i < (int)m_data->m_items.size()) {
		if (m_data->m_dblClickAction)
			m_data->m_dblClickAction->sendCommand(i);
	}
}

//------------------------------------------------------------------------------

void TTextList::keyDown(int key, unsigned long mod, const TPoint &pos)
{
	if ((key == TK_UpArrow) || (key == TK_DownArrow)) {
		int lastSelected = -1;
		for (int i = 0; i < getItemCount(); ++i)
			if (isSelected(i))
				lastSelected = i;
		if (lastSelected == -1)
			return;
		int newSelected = (key == TK_UpArrow) ? lastSelected - 1 : lastSelected + 1;

		if (newSelected >= 0 && newSelected < (int)m_data->m_items.size()) {
			//   if (!e.isShiftPressed())
			unselectAll();
			select(newSelected, !isSelected(newSelected));
		}
	} else
		TWidget::keyDown(key, mod, pos);
}