#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();

}