#include "gui/messagebox.h"

#include "gui/iconthemefactory.h"

#include <QDialogButtonBox>
#include <QtGlobal>
#include <QPushButton>
#include <QDialogButtonBox>
#include <QStyle>
#include <QApplication>

MessageBox::MessageBox() {
}

void MessageBox::iconify(QDialogButtonBox *button_box) {
  foreach (QAbstractButton *button, button_box->buttons()) {
    button->setIcon(iconForRole(button_box->standardButton(button)));
  }
}

QIcon MessageBox::iconForRole(QDialogButtonBox::StandardButton button) {
  switch (button) {
    case QMessageBox::Ok:
      return IconThemeFactory::instance()->fromTheme("dialog-ok");

    case QMessageBox::Yes:
    case QMessageBox::YesToAll:
      return IconThemeFactory::instance()->fromTheme("dialog-yes");

    case QMessageBox::No:
    case QMessageBox::NoToAll:
      return IconThemeFactory::instance()->fromTheme("dialog-no");

    default:
      return QIcon();
  }
}

QIcon MessageBox::iconForStatus(QMessageBox::Icon status) {
  switch (status) {
    case QMessageBox::Information:
      return IconThemeFactory::instance()->fromTheme("help-about");

    case QMessageBox::Warning:
      return IconThemeFactory::instance()->fromTheme("dialog-warning");

    case QMessageBox::Critical:
      return IconThemeFactory::instance()->fromTheme("dialog-error");

    case QMessageBox::Question:
      return IconThemeFactory::instance()->fromTheme("dialog-question");

    case QMessageBox::NoIcon:
    default:
      return QIcon();
  }
}

QMessageBox::StandardButton MessageBox::show(QWidget *parent,
                                             QMessageBox::Icon icon,
                                             const QString &title,
                                             const QString &text,
                                             QMessageBox::StandardButtons buttons,
                                             QMessageBox::StandardButton defaultButton) {
  // Create and find needed components.
  QMessageBox msg_box(parent);
  QDialogButtonBox *button_box = msg_box.findChild<QDialogButtonBox*>();

  // Initialize message box properties.
  msg_box.setWindowTitle(title);
  msg_box.setText(text);
  msg_box.setStandardButtons(buttons);
  msg_box.setDefaultButton(defaultButton);

  iconify(button_box);

  // Setup status icon.
  int icon_size = qApp->style()->pixelMetric(QStyle::PM_MessageBoxIconSize, 0, &msg_box);
  msg_box.setIconPixmap(iconForStatus(icon).pixmap(icon_size,
                                                   icon_size));

  // Display it.
  if (msg_box.exec() == -1) {
    return QMessageBox::Cancel;
  }
  else {
    return msg_box.standardButton(msg_box.clickedButton());
  }
}