#include "tfarmcontroller.h"
#include "application.h"
#include "dependedlist.h"
#include "submitpage.h"

#include "tw/message.h"

#include <vector>

using namespace std;

DependedList::DependedList(TWidget *parent)
	: TWidget(parent)
{
	m_depList = new TTextList(this);

	m_add = new TButton(this, "Add");
	tconnect<DependedList>(*(m_add), this, onAdd);

	m_remove = new TButton(this, "Remove");
	tconnect<DependedList>(*(m_remove), this, onRemove);
}

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

void DependedList::configureNotify(const TDimension &size)
{
	const int leftSize = size.lx / 3;
	m_depList->setGeometry(0, 0, leftSize, size.ly - 1);

	TDimension buttonSize(60, 15);
	int x = leftSize + 30;
	int y0 = size.ly - 10;

	m_add->setGeometry(x, y0 - buttonSize.ly, x + buttonSize.lx, y0);
	y0 -= (buttonSize.ly + 10);
	m_remove->setGeometry(x, y0 - buttonSize.ly, x + buttonSize.lx, y0);
}

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

void DependedList::onRemove()
{
	if (!m_depList->getSelectedItem(0))
		return;

	int count = m_depList->getSelectedItemCount();
	int i = 0;
	for (; i < count; ++i) {
		string id = m_depList->getSelectedItemId(i);
		m_tasks.erase(m_tasks.find(id));
		m_depList->removeItem(id);
	}

	m_depList->invalidate();

	SubmitPage *submitPage = dynamic_cast<SubmitPage *>(getParent());
	if (submitPage) {
		SubmitPageTask *task = submitPage->getTask();
		if (task)
			task->setDependencies(m_tasks);
	}
}

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

void DependedList::clearAll()
{
	m_depList->clearAll();
	m_tasks.erase(m_tasks.begin(), m_tasks.end());
}

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

void DependedList::onAdd()
{
	static DependedPopup *popup = 0;

	if (!popup) {
		popup = new DependedPopup(this);

		popup->setOkAction(
			new TDependedPopupAction<DependedList>(this, &DependedList::AddItems));
	}

	if (!popup)
		return;

	TFarmController *controller = Application::instance()->getController();
	vector<TaskShortInfo> tasks;
	try {
		controller->getTasks("", tasks);
	} catch (TException &e) {
		TMessage::error(toString(e.getMessage()));
		return;
	}

	popup->setList(tasks);

	TDimension d = TMainshell::getMainshell()->getSize();
#ifdef WIN32
	HDC hdc = GetDC(0);
	d.lx = GetDeviceCaps(hdc, HORZRES);
	d.ly = GetDeviceCaps(hdc, VERTRES);
	ReleaseDC(0, hdc);
#endif

	d -= popup->getSize();
	popup->popup(TPoint(d.lx / 2, d.ly / 2));
	popup->setCaption("Tasks Submitted");
}

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

void DependedList::AddItems(const vector<string> &tasksId)
{
	TFarmController *controller = Application::instance()->getController();
	vector<string>::const_iterator it = tasksId.begin();
	for (; it != tasksId.end(); ++it)
		if (m_tasks.end() == m_tasks.find(*it)) {
			try {
				string parentId, name;
				TaskState status;
				controller->queryTaskShortInfo(*it, parentId, name, status);
				string label = "<" + *it + "> " + name;
				m_depList->addItem(new TTextListItem(*it, label));
				m_tasks.insert(make_pair(*it, label));
			} catch (TException &e) {
				TMessage::error(toString(e.getMessage()));
			}
		}
	m_depList->invalidate();

	SubmitPage *submitPage = dynamic_cast<SubmitPage *>(getParent());
	if (submitPage) {
		SubmitPageTask *task = submitPage->getTask();
		if (task)
			task->setDependencies(m_tasks);
	}
}

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

void DependedList::setList(const std::map<string, string> &tasks)
{
	m_depList->clearAll();
	std::map<string, string>::const_iterator it = tasks.begin();
	for (; it != tasks.end(); ++it)
		m_depList->addItem(new TTextListItem((*it).first, (*it).second));
	m_tasks = tasks;
	invalidate();
}

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

DependedPopup::DependedPopup(TWidget *parent)
	: TModalPopup(parent, "DependedList"), m_okAction(0)
{
	m_submitList = new TTextList(this);

	m_ok = new TButton(this, "Ok");
	tconnect<DependedPopup>(*(m_ok), this, onOk);

	m_cancel = new TButton(this, "Cancel");
	tconnect<DependedPopup>(*(m_cancel), this, TPopup::close);
}

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

void DependedPopup::configureNotify(const TDimension &size)
{
	const int bottomSize = 35;
	m_submitList->setGeometry(0, bottomSize, size.lx - 1, size.ly - 1);

	TDimension buttonSize(60, 15);
	int x1 = 30;
	int x2 = size.lx - x1;
	int y0 = bottomSize - 10;

	m_ok->setGeometry(x1, y0 - buttonSize.ly, x1 + buttonSize.lx, y0);
	m_cancel->setGeometry(x2 - buttonSize.lx, y0 - buttonSize.ly, x2, y0);
}

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

TDimension DependedPopup::getPreferredSize() const
{
	return TDimension(400, 300);
}

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

void DependedPopup::onOk()
{
	assert(m_okAction);
	vector<string> tasks;
	if (m_submitList->getSelectedItem(0)) {
		int count = m_submitList->getSelectedItemCount();
		int i = 0;
		for (; i < count; ++i)
			tasks.push_back(m_submitList->getSelectedItemId(i));
	}
	try {
		m_okAction->sendCommand(tasks);
		close();
	} catch (TException &e) {
		TMessage::error(toString(e.getMessage()));
	}
}

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

void DependedPopup::setList(const vector<TaskShortInfo> &tasks)
{
	m_submitList->clearAll();

	vector<TaskShortInfo>::const_iterator it = tasks.begin();
	for (; it != tasks.end(); ++it) {
		string label = "<" + (*it).m_id + "> " + (*it).m_name;
		m_submitList->addItem(new TTextListItem((*it).m_id, label));
	}
}

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

void DependedPopup::setOkAction(TGenericDependedPopupAction *action)
{
	m_okAction = action;
}