widgetdialog.cpp

#include "widgetdialog.h"

#include <QVBoxLayout>
#include <QDialogButtonBox>
#include <QAbstractButton>

using namespace ui;

WidgetDialog::WidgetDialog(QWidget* widget, QWidget* parent,
                           const QStringList& options) : QDialog(parent)
{
    if (widget != 0)
    {
        widget->setParent(this);

        QVBoxLayout* layout = new QVBoxLayout(this);
        layout->addWidget(widget);

        QDialogButtonBox* buttons = new QDialogButtonBox(this);
        layout->addWidget(buttons);

        if (options.isEmpty())
        {
            buttons->addButton(QDialogButtonBox::Ok);
            buttons->addButton(QDialogButtonBox::Cancel);
        }
        else
        {
            Q_FOREACH (QString option, options)
            {
                if (option.toLower() == "ok")
                    buttons->addButton(QDialogButtonBox::Ok);
                else if (option.toLower() == "cancel")
                    buttons->addButton(QDialogButtonBox::Cancel);
                else if (option.toLower() == "yes")
                    buttons->addButton(QDialogButtonBox::Yes);
                else if (option.toLower() == "no")
                    buttons->addButton(QDialogButtonBox::No);
                else
                    buttons->addButton(option, QDialogButtonBox::AcceptRole);
            }

            connect(buttons, SIGNAL(clicked(QAbstractButton*)),
                    this, SLOT(handleClick(QAbstractButton*)));
        }

        connect(buttons, SIGNAL(accepted()), this, SLOT(accept()));
        connect(buttons, SIGNAL(rejected()), this, SLOT(reject()));
        resize(widget->sizeHint());
    }
}

QString WidgetDialog::chosenOption()
{
    return clickedText;
}

void WidgetDialog::handleClick(QAbstractButton* button)
{
    clickedText = button->text();
}

---