// Copyright 2018 Dolphin Emulator Project
// Licensed under GPLv2+
// Refer to the license.txt file included.

#include "DolphinQt/ResourcePackManager.h"

#include <QDesktopServices>
#include <QDialogButtonBox>
#include <QGridLayout>
#include <QHeaderView>
#include <QPushButton>
#include <QTableWidget>
#include <QUrl>

#include "Common/FileUtil.h"
#include "DolphinQt/QtUtils/ModalMessageBox.h"
#include "UICommon/ResourcePack/Manager.h"

ResourcePackManager::ResourcePackManager(QWidget* widget) : QDialog(widget)
{
  CreateWidgets();
  ConnectWidgets();
  RepopulateTable();

  setWindowTitle(tr("Resource Pack Manager"));
  setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint);

  resize(QSize(900, 600));
}

void ResourcePackManager::CreateWidgets()
{
  auto* layout = new QGridLayout;

  m_table_widget = new QTableWidget;

  m_open_directory_button = new QPushButton(tr("Open Directory..."));
  m_change_button = new QPushButton(tr("Install"));
  m_remove_button = new QPushButton(tr("Remove"));
  m_refresh_button = new QPushButton(tr("Refresh"));
  m_priority_up_button = new QPushButton(tr("Up"));
  m_priority_down_button = new QPushButton(tr("Down"));

  auto* buttons = new QDialogButtonBox(QDialogButtonBox::Ok);

  connect(buttons, &QDialogButtonBox::accepted, this, &QDialog::accept);

  layout->addWidget(m_table_widget, 0, 0, 7, 1);
  layout->addWidget(m_open_directory_button, 0, 1);
  layout->addWidget(m_change_button, 1, 1);
  layout->addWidget(m_remove_button, 2, 1);
  layout->addWidget(m_refresh_button, 3, 1);
  layout->addWidget(m_priority_up_button, 4, 1);
  layout->addWidget(m_priority_down_button, 5, 1);

  layout->addWidget(buttons, 7, 1, Qt::AlignRight);
  setLayout(layout);
  setLayout(layout);
}

void ResourcePackManager::ConnectWidgets()
{
  connect(m_open_directory_button, &QPushButton::pressed, this,
          &ResourcePackManager::OpenResourcePackDir);
  connect(m_refresh_button, &QPushButton::pressed, this, &ResourcePackManager::Refresh);
  connect(m_change_button, &QPushButton::pressed, this, &ResourcePackManager::Change);
  connect(m_remove_button, &QPushButton::pressed, this, &ResourcePackManager::Remove);
  connect(m_priority_up_button, &QPushButton::pressed, this, &ResourcePackManager::PriorityUp);
  connect(m_priority_down_button, &QPushButton::pressed, this, &ResourcePackManager::PriorityDown);

  connect(m_table_widget, &QTableWidget::itemSelectionChanged, this,
          &ResourcePackManager::SelectionChanged);

  connect(m_table_widget, &QTableWidget::itemDoubleClicked, this,
          &ResourcePackManager::ItemDoubleClicked);
}

void ResourcePackManager::OpenResourcePackDir()
{
  QDesktopServices::openUrl(
      QUrl::fromLocalFile(QString::fromStdString(File::GetUserPath(D_RESOURCEPACK_IDX))));
}

void ResourcePackManager::RepopulateTable()
{
  m_table_widget->clear();
  m_table_widget->setColumnCount(6);

  m_table_widget->setHorizontalHeaderLabels({QStringLiteral(""), tr("Name"), tr("Version"),
                                             tr("Description"), tr("Author"), tr("Website")});

  auto* header = m_table_widget->horizontalHeader();

  for (int i = 0; i < 4; i++)
    header->setSectionResizeMode(i, QHeaderView::ResizeToContents);

  header->setStretchLastSection(true);

  int size = static_cast<int>(ResourcePack::GetPacks().size());

  m_table_widget->setSelectionBehavior(QAbstractItemView::SelectRows);
  m_table_widget->setSelectionMode(QAbstractItemView::SingleSelection);

  m_table_widget->setRowCount(size);
  m_table_widget->setIconSize(QSize(32, 32));

  for (int i = 0; i < size; i++)
  {
    const auto& pack = ResourcePack::GetPacks()[size - 1 - i];
    auto* manifest = pack.GetManifest();

    auto* logo_item = new QTableWidgetItem;
    auto* name_item = new QTableWidgetItem(QString::fromStdString(manifest->GetName()));
    auto* version_item = new QTableWidgetItem(QString::fromStdString(manifest->GetVersion()));
    auto* author_item = new QTableWidgetItem(
        QString::fromStdString(manifest->GetAuthors().value_or("Unknown author")));
    auto* description_item =
        new QTableWidgetItem(QString::fromStdString(manifest->GetDescription().value_or("")));
    auto* website_item =
        new QTableWidgetItem(QString::fromStdString(manifest->GetWebsite().value_or("")));

    QPixmap logo;

    logo.loadFromData(reinterpret_cast<const uchar*>(pack.GetLogo().data()),
                      (int)pack.GetLogo().size());

    logo_item->setIcon(QIcon(logo));

    QFont link_font = website_item->font();

    link_font.setUnderline(true);

    website_item->setFont(link_font);
    website_item->setForeground(QBrush(Qt::blue));
    website_item->setData(Qt::UserRole, website_item->text());

    for (auto* item :
         {logo_item, name_item, version_item, author_item, description_item, website_item})
    {
      item->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);

      if (ResourcePack::IsInstalled(pack))
      {
        item->setBackgroundColor(QColor(Qt::green));

        auto font = item->font();
        font.setBold(true);
        item->setFont(font);
      }
    }

    m_table_widget->setItem(i, 0, logo_item);
    m_table_widget->setItem(i, 1, name_item);
    m_table_widget->setItem(i, 2, version_item);
    m_table_widget->setItem(i, 3, description_item);
    m_table_widget->setItem(i, 4, author_item);
    m_table_widget->setItem(i, 5, website_item);
  }

  SelectionChanged();
}

// Revert the indicies as to be more intuitive for users
int ResourcePackManager::GetResourcePackIndex(QTableWidgetItem* item) const
{
  return m_table_widget->rowCount() - 1 - item->row();
}

void ResourcePackManager::Change()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  if (ResourcePack::IsInstalled(ResourcePack::GetPacks()[GetResourcePackIndex(items[0])]))
  {
    Uninstall();
  }
  else
  {
    Install();
  }
}

void ResourcePackManager::Install()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  auto& item = ResourcePack::GetPacks()[GetResourcePackIndex(items[0])];

  bool success = item.Install(File::GetUserPath(D_USER_IDX));

  if (!success)
  {
    ModalMessageBox::critical(
        this, tr("Error"),
        tr("Failed to install pack: %1").arg(QString::fromStdString(item.GetError())));
  }

  RepopulateTable();
}

void ResourcePackManager::Uninstall()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  auto& item = ResourcePack::GetPacks()[GetResourcePackIndex(items[0])];

  bool success = item.Uninstall(File::GetUserPath(D_USER_IDX));

  if (!success)
  {
    ModalMessageBox::critical(
        this, tr("Error"),
        tr("Failed to uninstall pack: %1").arg(QString::fromStdString(item.GetError())));
  }

  RepopulateTable();
}

void ResourcePackManager::Remove()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  ModalMessageBox box(this);
  box.setWindowTitle(tr("Confirmation"));
  box.setText(tr("Are you sure you want to delete this pack?"));
  box.setIcon(QMessageBox::Warning);
  box.setStandardButtons(QMessageBox::Yes | QMessageBox::Abort);

  if (box.exec() != QMessageBox::Yes)
    return;

  Uninstall();
  File::Delete(ResourcePack::GetPacks()[GetResourcePackIndex(items[0])].GetPath());
  RepopulateTable();
}

void ResourcePackManager::PriorityDown()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  auto row = GetResourcePackIndex(items[0]);

  if (items[0]->row() >= m_table_widget->rowCount())
    return;

  auto& pack = ResourcePack::GetPacks()[row];
  std::string path = pack.GetPath();

  row--;

  ResourcePack::Remove(pack);
  ResourcePack::Add(path, row);

  RepopulateTable();

  m_table_widget->selectRow(row == 0 ? m_table_widget->rowCount() - 1 : row);
}

void ResourcePackManager::PriorityUp()
{
  auto items = m_table_widget->selectedItems();

  if (items.empty())
    return;

  auto row = GetResourcePackIndex(items[0]);

  if (items[0]->row() == 0)
    return;

  auto& pack = ResourcePack::GetPacks()[row];
  std::string path = pack.GetPath();

  row++;

  ResourcePack::Remove(pack);
  ResourcePack::Add(path, items[0]->row() == m_table_widget->rowCount() ? -1 : row);

  RepopulateTable();

  m_table_widget->selectRow(row == m_table_widget->rowCount() - 1 ? 0 : row);
}

void ResourcePackManager::Refresh()
{
  ResourcePack::Init();
  RepopulateTable();
}

void ResourcePackManager::SelectionChanged()
{
  auto items = m_table_widget->selectedItems();

  const bool has_selection = !items.empty();

  if (has_selection)
  {
    m_change_button->setText(
        ResourcePack::IsInstalled(ResourcePack::GetPacks()[GetResourcePackIndex(items[0])]) ?
            tr("Uninstall") :
            tr("Install"));
  }

  for (auto* item : {m_change_button, m_remove_button})
    item->setEnabled(has_selection);

  m_priority_down_button->setEnabled(has_selection &&
                                     items[0]->row() < m_table_widget->rowCount() - 1);
  m_priority_up_button->setEnabled(has_selection && items[0]->row() != 0);
}

void ResourcePackManager::ItemDoubleClicked(QTableWidgetItem* item)
{
  auto item_data = item->data(Qt::UserRole);

  if (item_data.isNull())
    return;

  QDesktopServices::openUrl(QUrl(item_data.toString()));
}