#include "VacuumDialog.h"
#include "ui_VacuumDialog.h"
#include "sqlitedb.h"

#include <QTreeWidgetItem>

VacuumDialog::VacuumDialog(DBBrowserDB* _db, QWidget* parent) :
    QDialog(parent),
    ui(new Ui::VacuumDialog),
    db(_db)
{
    // Create UI
    ui->setupUi(this);

    // Show warning if DB is dirty
    ui->labelSavepointWarning->setVisible(db->getDirty());

    // Populate list of objects to compact. We just support vacuuming the main schema here.
    for(auto it=db->schemata.constBegin();it!=db->schemata.constEnd();++it)
    {
        QTreeWidgetItem* item = new QTreeWidgetItem(ui->treeDatabases);
        item->setText(0, it.key());
        item->setIcon(0, QIcon(QString(":icons/database")));
        ui->treeDatabases->addTopLevelItem(item);
    }

    // Select the first item which should always be the main schema
    ui->treeDatabases->setCurrentItem(ui->treeDatabases->topLevelItem(0));
}

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

void VacuumDialog::accept()
{
    if(ui->treeDatabases->selectedItems().count() == 0)
        return QDialog::reject();

    QApplication::setOverrideCursor(Qt::WaitCursor);

    // Commit all changes first
    db->releaseAllSavepoints();

    // Loop through all selected databases and vacuum them individually
    QList<QTreeWidgetItem*> selection = ui->treeDatabases->selectedItems();
    for(const QTreeWidgetItem* item : selection)
        db->executeSQL(QString("VACUUM %1;").arg(sqlb::escapeIdentifier(item->text(0))), false);

    QApplication::restoreOverrideCursor();
    QDialog::accept();
}