#include "tfarmcontroller.h"

#include "application.h"

#include "dependedlist.h"

#include "submitpage.h"

#include "tw/message.h"

#include <vector></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);</dependedlist>

  m_remove = new TButton(this, "Remove");

  tconnect<dependedlist>(*(m_remove), this, onRemove);</dependedlist>

}

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

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());</submitpage>

  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));</dependedlist>

  }

  if (!popup) return;

  TFarmController *controller = Application::instance()->getController();

  vector<taskshortinfo> tasks;</taskshortinfo>

  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) {</string>

  TFarmController *controller       = Application::instance()->getController();

  vector<string>::const_iterator it = tasksId.begin();</string>

  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());</submitpage>

  if (submitPage) {

    SubmitPageTask *task = submitPage->getTask();

    if (task) task->setDependencies(m_tasks);

  }

}

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

void DependedList::setList(const std::map<string, string=""> &tasks) {</string,>

  m_depList->clearAll();

  std::map<string, string="">::const_iterator it = tasks.begin();</string,>

  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);</dependedpopup>

  m_cancel = new TButton(this, "Cancel");

  tconnect<dependedpopup>(*(m_cancel), this, TPopup::close);</dependedpopup>

}

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

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;</string>

  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) {</taskshortinfo>

  m_submitList->clearAll();

  vector<taskshortinfo>::const_iterator it = tasks.begin();</taskshortinfo>

  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;

}