#include "problemsdialog.h"
#include "ui_problemsdialog.h"
#include "organizercore.h"
#include <utility.h>
#include <iplugin.h>
#include <iplugindiagnose.h>
#include <QPushButton>
#include <Shellapi.h>


using namespace MOBase;


ProblemsDialog::ProblemsDialog(std::vector<QObject *> pluginObjects, QWidget *parent)
  : QDialog(parent), ui(new Ui::ProblemsDialog), m_PluginObjects(pluginObjects)
{
  ui->setupUi(this);

  runDiagnosis();

  connect(ui->problemsWidget, SIGNAL(itemSelectionChanged()), this, SLOT(selectionChanged()));
  connect(ui->descriptionText, SIGNAL(anchorClicked(QUrl)), this, SLOT(urlClicked(QUrl)));
}


ProblemsDialog::~ProblemsDialog()
{
  delete ui;
}

void ProblemsDialog::runDiagnosis()
{
  ui->problemsWidget->clear();
  for(QObject *pluginObj : m_PluginObjects) {
    IPlugin *plugin = qobject_cast<IPlugin*>(pluginObj);
    if (plugin != nullptr && !plugin->isActive())
      continue;

    IPluginDiagnose *diagnose = qobject_cast<IPluginDiagnose*>(pluginObj);
    if (diagnose == nullptr)
      continue;

    std::vector<unsigned int> activeProblems = diagnose->activeProblems();
    foreach (unsigned int key, activeProblems) {
      QTreeWidgetItem *newItem = new QTreeWidgetItem();
      newItem->setText(0, diagnose->shortDescription(key));
      newItem->setData(0, Qt::UserRole, diagnose->fullDescription(key));

      ui->problemsWidget->addTopLevelItem(newItem);

      if (diagnose->hasGuidedFix(key)) {
        newItem->setText(1, tr("Fix"));
        QPushButton *fixButton = new QPushButton(tr("Fix"));
        fixButton->setProperty("fix", qVariantFromValue(reinterpret_cast<void*>(diagnose)));
        fixButton->setProperty("key", key);
        connect(fixButton, SIGNAL(clicked()), this, SLOT(startFix()));
        ui->problemsWidget->setItemWidget(newItem, 1, fixButton);
      } else {
        newItem->setText(1, tr("No guided fix"));
      }
    }
  }
}

bool ProblemsDialog::hasProblems() const
{
  return ui->problemsWidget->topLevelItemCount() != 0;
}

void ProblemsDialog::selectionChanged()
{
  QString text = ui->problemsWidget->currentItem()->data(0, Qt::UserRole).toString();
  ui->descriptionText->setText(text);
  ui->descriptionText->setLineWrapMode(text.contains('\n') ? QTextEdit::NoWrap : QTextEdit::WidgetWidth);
}

void ProblemsDialog::startFix()
{
  QObject *fixButton = QObject::sender();
  if (fixButton == NULL) {
    qWarning("no button");
    return;
  }
  IPluginDiagnose *plugin = reinterpret_cast<IPluginDiagnose*>(fixButton ->property("fix").value<void*>());
  plugin->startGuidedFix(fixButton ->property("key").toUInt());
  runDiagnosis();
}

void ProblemsDialog::urlClicked(const QUrl &url)
{
  shell::OpenLink(url);
}