#include "ExportSqlDialog.h"
#include "ui_ExportSqlDialog.h"
#include "sqlitedb.h"
#include "FileDialog.h"
#include "Settings.h"

#include <QFile>
#include <QMessageBox>

enum WhatComboEntries
{
 ExportEverything,
 ExportSchemaOnly,
 ExportDataOnly,
};

ExportSqlDialog::ExportSqlDialog(DBBrowserDB* db, QWidget* parent, const QString& selection)
 : QDialog(parent),
 ui(new Ui::ExportSqlDialog),
 pdb(db)
{
 // Create UI
 ui->setupUi(this);

 // Load settings
 ui->checkColNames->setChecked(Settings::getValue("exportsql", "insertcolnames").toBool());
 ui->checkMultiple->setChecked(Settings::getValue("exportsql", "insertmultiple").toBool());
 ui->comboOldSchema->setCurrentIndex(Settings::getValue("exportsql", "oldschema").toInt());

 // Get list of tables to export
 const auto objects = pdb->schemata["main"].equal_range("table");
 for(auto it=objects.first;it!=objects.second;++it)
 ui->listTables->addItem(new QListWidgetItem(QIcon(QString(":icons/%1").arg(QString::fromStdString(sqlb::Object::typeToString(it->second->type())))), QString::fromStdString(it->second->name())));

 // Sort list of tables and select the table specified in the
 // selection parameter or all tables if table not specified
 ui->listTables->model()->sort(0);
 if(selection.isEmpty())
 {
 for (int i = 0; i < ui->listTables->count(); ++i)
 ui->listTables->item(i)->setSelected(true);
 }
 else
 {
 QList<QListWidgetItem*> items = ui->listTables->findItems(selection, Qt::MatchExactly);
 ui->listTables->setCurrentItem(items.at(0));
 }
 ui->listTables->setFocus();
}

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

void ExportSqlDialog::doSelectAll()
{
	for (int i = 0; i < ui->listTables->count(); ++i)
		ui->listTables->item(i)->setSelected(true);
}

void ExportSqlDialog::doDeselectAll()
{
	for (int i = 0; i < ui->listTables->count(); ++i)
		ui->listTables->item(i)->setSelected(false);
}

void ExportSqlDialog::accept()
{
 QList<QListWidgetItem*> selectedItems = ui->listTables->selectedItems();
 if(selectedItems.isEmpty())
 {
 QMessageBox::warning(this, QApplication::applicationName(),
 tr("Please select at least one table."));
 return;
 }

 // Try to find a default file name
 QString defaultFileName;
 if(selectedItems.count() == 1) // One table -> Suggest table name
 defaultFileName = selectedItems.at(0)->text() + FILE_EXT_SQL_DEFAULT;
 else if(selectedItems.count() == ui->listTables->count()) // All tables -> Suggest database name
 defaultFileName = pdb->currentFile() + FILE_EXT_SQL_DEFAULT;;

 QString fileName = FileDialog::getSaveFileName(
 CreateSQLFile,
 this,
 tr("Choose a filename to export"),
 FILE_FILTER_SQL,
 defaultFileName);
 if(fileName.isEmpty())
 return;

 // Save settings
 Settings::setValue("exportsql", "insertcolnames", ui->checkColNames->isChecked());
 Settings::setValue("exportsql", "insertmultiple", ui->checkMultiple->isChecked());
 Settings::setValue("exportsql", "oldschema", ui->comboOldSchema->currentIndex());

 QStringList tables;
 for(const QListWidgetItem* item : ui->listTables->selectedItems())
 tables.push_back(item->text());

 // Check what to export. The indices here depend on the order of the items in the combobox in the ui file
 bool exportSchema = ui->comboWhat->currentIndex() == ExportEverything || ui->comboWhat->currentIndex() == ExportSchemaOnly;
 bool exportData = ui->comboWhat->currentIndex() == ExportEverything || ui->comboWhat->currentIndex() == ExportDataOnly;
 bool keepSchema = ui->comboOldSchema->currentIndex() == 0;

 // Perform actual export
 bool dumpOk = pdb->dump(fileName,
 tables,
 ui->checkColNames->isChecked(),
 ui->checkMultiple->isChecked(),
 exportSchema,
 exportData,
 keepSchema);
 if (dumpOk)
 QMessageBox::information(this, QApplication::applicationName(), tr("Export completed."));
 else
 QMessageBox::warning(this, QApplication::applicationName(), tr("Export cancelled or failed."));

 QDialog::accept();
}

void ExportSqlDialog::whatChanged(int index)
{
 // Only show the combobox for deciding what to do with the old schema when importing into an existing database when we're
 // actually exporting the schema here
 if(index != ExportDataOnly)
 ui->comboOldSchema->setVisible(true);
 else
 ui->comboOldSchema->setVisible(false);
}