

#include "batchserversviewer.h"
#include "toonzqt/menubarcommand.h"
#include "menubarcommandids.h"
#include "floatingpanelcommand.h"
#include "tfarmstuff.h"
#include "batches.h"
#include "tsystem.h"

#include <QGridLayout>
#include <QLabel>
#include <QComboBox>
#include <QPushButton>
#include <QSettings>
#include <QMouseEvent>
#include <QMenu>

using namespace DVGui;
using namespace TFarmStuff;

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

class MyListItem final : public QListWidgetItem {
public:
  QString m_id;

  MyListItem(const QString &id, const QString &name, QListWidget *w)
      : QListWidgetItem(name, w), m_id(id) {}
};

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

FarmServerListView::FarmServerListView(QWidget *parent) : QListWidget(parent) {
  setFrameStyle(QFrame::StyledPanel);
}

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

void FarmServerListView::update() {
  try {
    while (count()) delete takeItem(0);

    TFarmController *controller = getTFarmController();

    std::vector<ServerIdentity> servers;
    controller->getServers(servers);
    std::vector<ServerIdentity>::iterator it = servers.begin();
    for (; it != servers.end(); ++it) {
      ServerIdentity sid = *it;
      new MyListItem(sid.m_id, sid.m_name, this);
    }
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
  }
}

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

void FarmServerListView::activate() {
  MyListItem *item = (MyListItem *)currentItem();
  if (!item) return;

  TFarmController *controller = getTFarmController();
  try {
    controller->activateServer(item->m_id);
    BatchesController::instance()->update();
    static_cast<BatchServersViewer *>(parentWidget())->updateSelected();
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
  }
}

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

void FarmServerListView::deactivate() {
  MyListItem *item = (MyListItem *)currentItem();
  if (!item) return;

  TFarmController *controller = getTFarmController();
  try {
    controller->deactivateServer(item->m_id);
    BatchesController::instance()->update();
    static_cast<BatchServersViewer *>(parentWidget())->updateSelected();
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
  }
}
//-----------------------------------------------------------------------------

void FarmServerListView::openContextMenu(const QPoint &p) {
  MyListItem *item = (MyListItem *)currentItem();
  if (!item) return;

  m_menu.reset(new QMenu(this));

  TFarmController *controller = getTFarmController();
  ServerState state;
  try {
    state = controller->queryServerState2(item->m_id);
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
    return;
  }

  QAction *action;
  if (state == Offline) {
    action = new QAction(tr("Activate"), this);
    connect(action, SIGNAL(triggered(bool)), this, SLOT(activate()));
  } else {
    action = new QAction(tr("Deactivate"), this);
    connect(action, SIGNAL(triggered(bool)), this, SLOT(deactivate()));
  }

  m_menu->addAction(action);
  m_menu->exec(p);
}

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

void FarmServerListView::mousePressEvent(QMouseEvent *event) {
  QListWidget::mousePressEvent(event);

  if (event->button() == Qt::RightButton) openContextMenu(event->globalPos());
}

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

void BatchServersViewer::updateServerInfo(const QString &id) {
  if (id == "") {
    m_name->setText("");
    m_ip->setText("");
    m_port->setText("");
    m_state->setText("");
    m_tasks->setText("");
    m_cpu->setText("");
    m_mem->setText("");
    return;
  }

  if (id == "localhost") {
    m_name->setText(TSystem::getHostName());
    m_ip->setText("");
    m_port->setText("");
    m_state->setText("");
    m_tasks->setText("");
    m_cpu->setText(QString::number(TSystem::getProcessorCount()));

#ifdef _WIN32
    // Please observe that the commented value is NOT the same reported by
    // tfarmserver...
    MEMORYSTATUSEX buff;
    buff.dwLength = sizeof(MEMORYSTATUSEX);
    GlobalMemoryStatusEx(&buff);

    unsigned long mem = buff.ullTotalPhys >> 30;
    m_mem->setText(QString::number(mem, 'g', 2) + " GB");
#else
    double mem = (double)TSystem::getFreeMemorySize(false) / (1024 * 1024);
    m_mem->setText(QString::number(mem, 'g', 2) + " GB");
#endif

    return;
  }

  TFarmController *controller = getTFarmController();
  ServerInfo info;

  try {
    controller->queryServerInfo(id, info);
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
  }

  switch (info.m_state) {
  case Ready:
    m_state->setText("Ready");
    break;
  case Busy:
    m_state->setText("Busy");
    break;
  case NotResponding:
    m_state->setText("Not Responding");
    break;
  case Down:
    m_state->setText("Down");
    break;
  case Offline:
    m_state->setText("Offline");
    break;
  case ServerUnknown:
    m_state->setText("");
    m_name->setText("");
    m_ip->setText("");
    m_port->setText("");
    m_tasks->setText("");
    m_cpu->setText("");
    m_mem->setText("");
    return;
  }

  m_name->setText(info.m_name);
  m_ip->setText(info.m_ipAddress);
  m_port->setText(info.m_portNumber);
  if (info.m_currentTaskId == "")
    m_tasks->setText("");
  else {
    TFarmTask task;
    try {
      controller->queryTaskInfo(info.m_currentTaskId, task);
      m_tasks->setText("<" + task.m_id + "> " + task.m_name);
    } catch (TException &e) {
      m_tasks->setText("");
      DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
    }
  }

  if (info.m_state != Down) {
    m_cpu->setText(QString::number(info.m_cpuCount));
    double mem = (double)info.m_totPhysMem / (1024 * 1024);
    m_mem->setText(QString::number(mem, 'g', 2) + " GB");
  } else {
    m_cpu->setText("");
    m_mem->setText("");
  }
}

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

void BatchServersViewer::updateSelected() {
  MyListItem *item = (MyListItem *)m_serverList->currentItem();
  onCurrentItemChanged(item);
}

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

void BatchServersViewer::onCurrentItemChanged(QListWidgetItem *_item) {
  MyListItem *item = (MyListItem *)_item;
  updateServerInfo(item ? item->m_id : QString());
}

//=============================================================================
/*! \class BatchServersViewer
                \brief The BatchServersViewer class is a ...

                Inherits \b QFrame.
*/

static LineEdit *create(QGridLayout *layout, const QString &name, int &row,
                        bool readOnly = true) {
  layout->addWidget(new QLabel(name), row, 0, Qt::AlignRight);
  LineEdit *ret = new LineEdit();
  ret->setReadOnly(readOnly);
  layout->addWidget(ret, row++, 1);
  return ret;
}

BatchServersViewer::BatchServersViewer(QWidget *parent, Qt::WindowFlags flags)
    : QFrame(parent) {
  int row = 0;

  setFrameStyle(QFrame::StyledPanel);
  setObjectName("OnePixelMarginFrame");

  QGridLayout *layout = new QGridLayout();
  layout->setMargin(15);
  layout->setSpacing(8);

  layout->addWidget(new QLabel(tr("Process with:")), row, 0, Qt::AlignRight);
  m_processWith = new QComboBox();
  QStringList type;
  type << tr("Local") << tr("Render Farm");
  m_processWith->addItems(type);
  layout->addWidget(m_processWith, row++, 1);
  connect(m_processWith, SIGNAL(currentIndexChanged(int)), this,
          SLOT(onProcessWith(int)));

  m_farmRootField = ::create(layout, tr("Farm Global Root:"), row, false);

  QString path = QSettings().value("TFarmGlobalRoot").toString();
  if (path != "") {
    m_farmRootField->setText(path);
    TFarmStuff::setGlobalRoot(TFilePath(path.toStdWString()));
  }
  connect(m_farmRootField, SIGNAL(editingFinished()), this, SLOT(setGRoot()));

  m_serverList = new FarmServerListView(this);
  connect(m_serverList,
          SIGNAL(currentItemChanged(QListWidgetItem *, QListWidgetItem *)),
          this, SLOT(onCurrentItemChanged(QListWidgetItem *)));
  // connect(m_serverList, SIGNAL(itemPressed(QListWidgetItem *)), this,
  // SLOT(onCurrentItemChanged(QListWidgetItem *)));

  layout->addWidget(m_serverList, row, 0, 18, 2);
  row += 18 + 1;

  m_name  = ::create(layout, tr("Name:"), row);
  m_ip    = ::create(layout, tr("IP Address:"), row);
  m_port  = ::create(layout, tr("Port Number:"), row);
  m_tasks = ::create(layout, tr("Tasks:"), row);
  m_state = ::create(layout, tr("State:"), row);
  m_cpu   = ::create(layout, tr("Number of CPU:"), row);
  m_mem   = ::create(layout, tr("Physical Memory:"), row);

  setLayout(layout);
}

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

BatchServersViewer::~BatchServersViewer() {}

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

void BatchServersViewer::setGRoot() {
  TFarmStuff::setGlobalRoot(TFilePath(m_farmRootField->text().toStdWString()));
  QSettings().setValue("TFarmGlobalRoot", m_farmRootField->text());
}

//=============================================================================
// vinz: GRootEnvVarPopup togliere?

void BatchServersViewer::onProcessWith(int index) {
  if (index == 0) {
    BatchesController::instance()->setController(0);
    m_serverList->clear();
    m_serverId = "localhost";
    return;
  }

  bool connected = false;
  try {
    connected = TFarmStuff::testConnectionToController();
  } catch (TMissingGRootEnvironmentVariable &) {
    DVGui::warning(
        QString(tr("In order to use the render farm you have to define the "
                   "Farm Global Root first.")));
    m_processWith->setCurrentIndex(0);
    return;
  } catch (TMissingGRootFolder &) {
    DVGui::warning(
        tr("The Farm Global Root folder doesn't exist\nPlease create this "
           "folder before using the render farm."));
    m_processWith->setCurrentIndex(0);
    return;
  } catch (TException &e) {
    DVGui::warning(QString::fromStdString(::to_string(e.getMessage())));
    m_processWith->setCurrentIndex(0);
    return;
  }

  if (!TFarmStuff::testConnectionToController()) {
    QString hostName, addr;
    int port;
    TFarmStuff::getControllerData(hostName, addr, port);

    QString msg(tr("Unable to connect to the ToonzFarm Controller\n \
  The Controller should run on %1 at port %2\n \
  Please start the Controller before using the ToonzFarm")
                    .arg(hostName)
                    .arg(QString::number(port)));

    DVGui::warning(msg);
    m_processWith->setCurrentIndex(0);
    return;
  }

  BatchesController::instance()->setController(getTFarmController());

  m_serverList->update();

  // m_serversListView->update();
  // if (m_serversListView->getItemCount() > 0)
  //  m_serversListView->select(0, true);
}

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

OpenFloatingPanel openBatchServersCommand(MI_OpenBatchServers, "BatchServers",
                                          QObject::tr("Batch Servers"));
