#include "downloadstab.h"
#include "downloadlist.h"
#include "downloadlistview.h"
#include "organizercore.h"
#include "ui_mainwindow.h"

DownloadsTab::DownloadsTab(OrganizerCore& core, Ui::MainWindow* mwui)
    : m_core(core),
      ui{mwui->btnRefreshDownloads, mwui->btnQueryDownloadsInfo, mwui->downloadView,
         mwui->showHiddenBox, mwui->downloadFilterEdit}
{
  DownloadList* sourceModel = new DownloadList(m_core, ui.list);

  ui.list->setModel(sourceModel);
  ui.list->setManager(m_core.downloadManager());
  ui.list->setItemDelegate(
      new DownloadProgressDelegate(m_core.downloadManager(), ui.list));

  update();

  m_filter.setEdit(ui.filter);
  m_filter.setList(ui.list);
  m_filter.setSortPredicate([sourceModel](auto&& left, auto&& right) {
    return sourceModel->lessThanPredicate(left, right);
  });

  connect(ui.refresh, &QPushButton::clicked, [&] {
    refresh();
  });
  connect(ui.queryInfos, &QPushButton::clicked, [&] {
    queryInfos();
  });
  connect(ui.list, SIGNAL(installDownload(int)), &m_core, SLOT(installDownload(int)));
  connect(ui.list, SIGNAL(queryInfo(int)), m_core.downloadManager(),
          SLOT(queryInfo(int)));
  connect(ui.list, SIGNAL(queryInfoMd5(int)), m_core.downloadManager(),
          SLOT(queryInfoMd5(int)));
  connect(ui.list, SIGNAL(visitOnNexus(int)), m_core.downloadManager(),
          SLOT(visitOnNexus(int)));
  connect(ui.list, SIGNAL(visitUploaderProfile(int)), m_core.downloadManager(),
          SLOT(visitUploaderProfile(int)));
  connect(ui.list, SIGNAL(openFile(int)), m_core.downloadManager(),
          SLOT(openFile(int)));
  connect(ui.list, SIGNAL(openMetaFile(int)), m_core.downloadManager(),
          SLOT(openMetaFile(int)));
  connect(ui.list, SIGNAL(openInDownloadsFolder(int)), m_core.downloadManager(),
          SLOT(openInDownloadsFolder(int)));
  connect(ui.list, SIGNAL(removeDownload(int, bool)), m_core.downloadManager(),
          SLOT(removeDownload(int, bool)));
  connect(ui.list, SIGNAL(restoreDownload(int)), m_core.downloadManager(),
          SLOT(restoreDownload(int)));
  // The view reports actions by row; translate to DownloadID at the boundary.
  connect(ui.list, &DownloadListView::cancelDownload, this, [this](int row) {
    auto* dm = m_core.downloadManager();
    dm->cancelDownload(dm->downloadIDAtRow(row));
  });
  connect(ui.list, &DownloadListView::pauseDownload, this, [this](int row) {
    auto* dm = m_core.downloadManager();
    dm->pauseDownload(dm->downloadIDAtRow(row));
  });
  connect(ui.list, &DownloadListView::resumeDownload, this, [this](int row) {
    resumeDownload(row);
  });
}

void DownloadsTab::update()
{
  // this means downloadTab initialization hasn't happened yet
  if (ui.list->model() == nullptr) {
    return;
  }

  // set the view attribute and default row sizes
  if (m_core.settings().interface().compactDownloads()) {
    ui.list->setProperty("downloadView", "compact");
    ui.list->setStyleSheet("DownloadListView::item { padding: 4px 2px; }");
  } else {
    ui.list->setProperty("downloadView", "standard");
    ui.list->setStyleSheet("DownloadListView::item { padding: 16px 4px; }");
  }

  ui.list->style()->unpolish(ui.list);
  ui.list->style()->polish(ui.list);
  qobject_cast<DownloadListHeader*>(ui.list->header())->customResizeSections();

  m_core.downloadManager()->refreshList();
}

void DownloadsTab::refresh()
{
  m_core.downloadManager()->refreshList();
}

void DownloadsTab::queryInfos()
{
  if (m_core.settings().network().offlineMode()) {
    if (QMessageBox::warning(nullptr, tr("Query Metadata"),
                             tr("Cannot query metadata while offline mode is enabled. "
                                "Do you want to disable offline mode?"),
                             QMessageBox::Yes | QMessageBox::No) == QMessageBox::Yes) {
      m_core.settings().network().setOfflineMode(false);
    } else {
      return;
    }
  }

  m_core.downloadManager()->queryDownloadListInfo();
}

void DownloadsTab::resumeDownload(int downloadIndex)
{
  m_core.loggedInAction(ui.list, [this, downloadIndex] {
    auto* dm = m_core.downloadManager();
    dm->resumeDownload(dm->downloadIDAtRow(downloadIndex));
  });
}