: QMainWindow(parent),

ui(new Ui::MainWindow),

m_browseTableModel(new SqliteTableModel(this, &db, PreferencesDialog::getSettingsValue("db", "prefetchsize").toInt())),

sqliteHighlighterLogUser(0),

sqliteHighlighterLogApp(0),

editWin(new EditDialog(this)),

gotoValidator(new QIntValidator(0, 0, this))

{

ui->setupUi(this);

init();

activateFields(false);

updateRecentFileActions();

{

// Init the SQL log dock

db.mainWindow = this;

// Set the validator for the goto line edit

ui->editGoto->setValidator(gotoValidator);

// Set up DB models

ui->dataTable->setModel(m_browseTableModel);

// Set up DB structure tab

dbStructureModel = new DbStructureModel(ui->dbTreeWidget);

ui->dbTreeWidget->setModel(dbStructureModel);

ui->dbTreeWidget->setColumnHidden(1, true);

ui->dbTreeWidget->setColumnWidth(0, 300);

// Create the actions for the recently opened dbs list

for(int i = 0; i < MaxRecentFiles; ++i) {

recentFileActs[i] = new QAction(this);

recentFileActs[i]->setVisible(false);

connect(recentFileActs[i], SIGNAL(triggered()), this, SLOT(openRecentFile()));

}

for(int i = 0; i < MaxRecentFiles; ++i)

ui->fileMenu->insertAction(ui->fileExitAction, recentFileActs[i]);

recentSeparatorAct = ui->fileMenu->insertSeparator(ui->fileExitAction);

// Create popup menus

popupTableMenu = new QMenu(this);

popupTableMenu->addAction(ui->editModifyTableAction);

popupTableMenu->addAction(ui->editDeleteObjectAction);

popupTableMenu->addSeparator();

popupTableMenu->addAction(ui->actionExportCsvPopup);

// Set state of checkable menu actions

ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockLog->toggleViewAction());

ui->viewMenu->actions().at(0)->setShortcut(QKeySequence(tr("Ctrl+L")));

ui->viewMenu->actions().at(0)->setIcon(QIcon(":/icons/log_dock"));

ui->viewDBToolbarAction->setChecked(!ui->toolbarDB->isHidden());

// Plot view menu

ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockPlot->toggleViewAction());

ui->viewMenu->actions().at(1)->setShortcut(QKeySequence(tr("Ctrl+P")));

ui->viewMenu->actions().at(1)->setIcon(QIcon(":/icons/log_dock"));

// Set statusbar fields

statusEncodingLabel = new QLabel(ui->statusbar);

statusEncodingLabel->setEnabled(false);

statusEncodingLabel->setText("UTF-8");

statusEncodingLabel->setToolTip(tr("Database encoding"));

ui->statusbar->addPermanentWidget(statusEncodingLabel);

// Connect some more signals and slots

connect(ui->dataTable->filterHeader(), SIGNAL(filterChanged(int,QString)), this, SLOT(setRecordsetLabel()));

connect(ui->dataTable->filterHeader(), SIGNAL(sectionClicked(int)), this, SLOT(browseTableHeaderClicked(int)));

connect(ui->dataTable->verticalScrollBar(), SIGNAL(valueChanged(int)), this, SLOT(setRecordsetLabel()));

connect(ui->dataTable->horizontalHeader(), SIGNAL(sectionResized(int,int,int)), this, SLOT(updateBrowseDataColumnWidth(int,int,int)));

connect(editWin, SIGNAL(goingAway()), this, SLOT(editWinAway()));

connect(editWin, SIGNAL(updateRecordText(int, int, QByteArray)), this, SLOT(updateRecordText(int, int, QByteArray)));

connect(ui->dbTreeWidget->selectionModel(), SIGNAL(currentChanged(QModelIndex,QModelIndex)), this, SLOT(changeTreeSelection()));

// Load window settings

tabifyDockWidget(ui->dockPlot, ui->dockLog);

restoreGeometry(PreferencesDialog::getSettingsValue("MainWindow", "geometry").toByteArray());

restoreState(PreferencesDialog::getSettingsValue("MainWindow", "windowState").toByteArray());

ui->comboLogSubmittedBy->setCurrentIndex(ui->comboLogSubmittedBy->findText(PreferencesDialog::getSettingsValue("SQLLogDock", "Log").toString()));

ui->splitterForPlot->restoreState(PreferencesDialog::getSettingsValue("PlotDock", "splitterSize").toByteArray());

// plot widgets

ui->treePlotColumns->setSelectionMode(QAbstractItemView::NoSelection);

// Set other window settings

setAcceptDrops(true);

setWindowTitle(QApplication::applicationName());

// Load all settings

reloadSettings();

#ifdef CHECKNEWVERSION

// Check for a new release version, usually only enabled on windows

m_NetworkManager = new QNetworkAccessManager(this);

QObject::connect(m_NetworkManager, SIGNAL(finished(QNetworkReply*)), this, SLOT(httpresponse(QNetworkReply*)));

QUrl url("https://raw.github.com/sqlitebrowser/sqlitebrowser/master/currentrelease");

m_NetworkManager->get(QNetworkRequest(url));

#endif

{

for(SqlTextEdit::FieldCompleterModelMap::iterator it = completerModelsFields.begin();

it != completerModelsFields.end();

++it)

{

delete *it;

}

completerModelsFields.clear();

{

bool retval = false;

QString wFile = fileName;

if (!QFile::exists(wFile))

{

wFile = QFileDialog::getOpenFileName(

this,

tr("Choose a database file"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString());

}

if(QFile::exists(wFile) )

{

fileClose();

if(db.open(wFile))

{

statusEncodingLabel->setText(db.getPragma("encoding"));

setCurrentFile(wFile);

retval = true;

} else {

// Failed opening file; so it might be a SQLiteBrowser project file

return loadProject(wFile);

}

loadExtensionsFromSettings();

populateStructure();

resetBrowser();

if(ui->mainTab->currentIndex() == 2)

loadPragmas();

openSqlTab(true);

}

return retval;

{

QString fileName = QFileDialog::getSaveFileName(this, tr("Choose a filename to save under"), PreferencesDialog::getSettingsValue("db", "defaultlocation").toString());

if(!fileName.isEmpty())

{

if(QFile::exists(fileName))

QFile::remove(fileName);

db.create(fileName);

setCurrentFile(fileName);

statusEncodingLabel->setText(db.getPragma("encoding"));

loadExtensionsFromSettings();

populateStructure();

resetBrowser();

openSqlTab(true);

createTable();

}

{

completerModelTables.clear();

clearCompleterModelsFields();

// Refresh the structure tab

db.updateSchema();

dbStructureModel->reloadData(&db);

ui->dbTreeWidget->expandToDepth(0);

if(!db.isOpen())

return;

QStringList tblnames = db.getBrowsableObjectNames();

sqliteHighlighterLogUser->setTableNames(tblnames);

sqliteHighlighterLogApp->setTableNames(tblnames);

// setup models for sqltextedit autocomplete

completerModelTables.setRowCount(tblnames.count());

completerModelTables.setColumnCount(1);

objectMap tab = db.getBrowsableObjects();

int row = 0;

for(objectMap::ConstIterator it=tab.begin(); it!=tab.end(); ++it, ++row)

{

QString sName = it.value().getname();

QStandardItem* item = new QStandardItem(sName);

item->setIcon(QIcon(QString(":icons/%1").arg(it.value().gettype())));

completerModelTables.setItem(row, 0, item);

// If it is a table add the field Nodes

if((*it).gettype() == "table" || (*it).gettype() == "view")

{

QStandardItemModel* tablefieldmodel = new QStandardItemModel();

tablefieldmodel->setRowCount((*it).fldmap.count());

tablefieldmodel->setColumnCount(1);

int fldrow = 0;

for(int i=0; i < (*it).fldmap.size(); ++i, ++fldrow)

{

QString fieldname = (*it).fldmap.at(i)->name();

QStandardItem* fldItem = new QStandardItem(fieldname);

fldItem->setIcon(QIcon(":/icons/field"));

tablefieldmodel->setItem(fldrow, 0, fldItem);

}

completerModelsFields.insert(sName.toLower(), tablefieldmodel);

}

}

for(int i=0; i < ui->tabSqlAreas->count(); ++i)

{

SqlExecutionArea* sqlarea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i));

sqlarea->setTableNames(tblnames);

sqlarea->getEditor()->setDefaultCompleterModel(&completerModelTables);

sqlarea->getEditor()->insertFieldCompleterModels(completerModelsFields);

}

{

// Remove the model-view link if the table name is empty in order to remove any data from the view

if(tablename.isEmpty())

{

ui->dataTable->setModel(0);

if(qobject_cast<FilterTableHeader*>(ui->dataTable->horizontalHeader()))

qobject_cast<FilterTableHeader*>(ui->dataTable->horizontalHeader())->generateFilters(0);

return;

}

QApplication::setOverrideCursor(Qt::WaitCursor);

// Set model

ui->dataTable->setModel(m_browseTableModel);

// Set new table

m_browseTableModel->setTable(tablename);

ui->dataTable->setColumnHidden(0, true);

// Restore column widths

QMap<QString, QMap<int, int> >::ConstIterator colWidthsIt;

if((colWidthsIt = browseTableColumnWidths.constFind(tablename)) != browseTableColumnWidths.constEnd())

{

// There are some column widths stored for this table

for(QMap<int, int>::ConstIterator it=colWidthsIt.value().constBegin();it!=colWidthsIt.value().constEnd();++it)

ui->dataTable->setColumnWidth(it.key(), it.value());

} else {

// There aren't any column widths stored for this table yet, so set default widths

for(int i=1;i<m_browseTableModel->columnCount();i++)

ui->dataTable->setColumnWidth(i, ui->dataTable->horizontalHeader()->defaultSectionSize());

}

// Reset sorting

curBrowseOrderByIndex = 0;

curBrowseOrderByMode = Qt::AscendingOrder;

m_browseTableModel->sort(curBrowseOrderByIndex, curBrowseOrderByMode);

ui->dataTable->filterHeader()->setSortIndicator(curBrowseOrderByIndex, curBrowseOrderByMode);

// Update the filter row

qobject_cast<FilterTableHeader*>(ui->dataTable->horizontalHeader())->generateFilters(m_browseTableModel->columnCount());

// Activate the add and delete record buttons and editing only if a table has been selected

bool is_table = db.getObjectByName(tablename).gettype() == "table";

ui->buttonNewRecord->setEnabled(is_table);

ui->buttonDeleteRecord->setEnabled(is_table);

ui->dataTable->setEditTriggers(is_table ? QAbstractItemView::SelectedClicked | QAbstractItemView::AnyKeyPressed | QAbstractItemView::EditKeyPressed : QAbstractItemView::NoEditTriggers);

// Set the recordset label

setRecordsetLabel();

// Reset the edit dialog

if(editWin)

editWin->reset();

// update plot

updatePlot(m_browseTableModel);

QApplication::restoreOverrideCursor();

{

const QString sCurrentTable = ui->comboBrowseTable->currentText();

ui->comboBrowseTable->clear();

const objectMap& tab = db.getBrowsableObjects();

// fill a objmap which is sorted by table/view names

QMap<QString, DBBrowserObject> objmap;

for(objectMap::ConstIterator i=tab.begin();i!=tab.end();++i)

{

objmap[i.value().getname()] = i.value();;

}

// finaly fill the combobox in sorted order

for(QMap<QString, DBBrowserObject>::ConstIterator it=objmap.constBegin();

it!=objmap.constEnd();

++it)

{

ui->comboBrowseTable->addItem(

QIcon(QString(":icons/%1").arg((*it).gettype())),

(*it).getname());

}

setRecordsetLabel();

int pos = ui->comboBrowseTable->findText(sCurrentTable);

pos = pos == -1 ? 0 : pos;

ui->comboBrowseTable->setCurrentIndex(pos);

curBrowseOrderByIndex = 0;

curBrowseOrderByMode = Qt::AscendingOrder;

m_browseTableModel->sort(curBrowseOrderByIndex, curBrowseOrderByMode);

populateTable(ui->comboBrowseTable->currentText());

{

db.close();

setWindowTitle(QApplication::applicationName());

resetBrowser();

populateStructure();

loadPragmas();

// Delete the model for the Browse tab and create a new one

delete m_browseTableModel;

m_browseTableModel = new SqliteTableModel(this, &db, PreferencesDialog::getSettingsValue("db", "prefetchsize").toInt());

connect(ui->dataTable->filterHeader(), SIGNAL(filterChanged(int,QString)), m_browseTableModel, SLOT(updateFilter(int,QString)));

// Remove all stored column widths for the browse data table

browseTableColumnWidths.clear();

// Manually update the recordset label inside the Browse tab now

setRecordsetLabel();

// Reset the plot dock model

updatePlot(0);

activateFields(false);

ui->buttonLogClear->click();

for(int i=ui->tabSqlAreas->count()-1;i>=0;i--)

closeSqlTab(i, true);

{

db.close();

PreferencesDialog::setSettingsValue("MainWindow", "geometry", saveGeometry());

PreferencesDialog::setSettingsValue("MainWindow", "windowState", saveState());

PreferencesDialog::setSettingsValue("SQLLogDock", "Log", ui->comboLogSubmittedBy->currentText());

PreferencesDialog::setSettingsValue("PlotDock", "splitterSize", ui->splitterForPlot->saveState());

clearCompleterModelsFields();

QMainWindow::closeEvent(event);

{

int row = m_browseTableModel->rowCount();

if(m_browseTableModel->insertRow(row))

{

selectTableLine(row);

} else {

QMessageBox::warning( this, QApplication::applicationName(), tr("Error adding record:\n") + db.lastErrorMessage);

}

{

if(ui->dataTable->selectionModel()->hasSelection())

{

int old_row = ui->dataTable->currentIndex().row();

while(ui->dataTable->selectionModel()->hasSelection())

{

if(!m_browseTableModel->removeRow(ui->dataTable->selectionModel()->selectedIndexes().first().row()))

{

QMessageBox::warning(this, QApplication::applicationName(), tr("Error deleting record:\n%1").arg(db.lastErrorMessage));

break;

}

}

populateTable(ui->comboBrowseTable->currentText());

if(old_row > m_browseTableModel->totalRowCount())

old_row = m_browseTableModel->totalRowCount();

selectTableLine(old_row);

} else {

QMessageBox::information( this, QApplication::applicationName(), tr("Please select a record first"));

}

{

// Are there even that many lines?

if(lineToSelect >= m_browseTableModel->totalRowCount())

return;

QApplication::setOverrideCursor( Qt::WaitCursor );

// Make sure this line has already been fetched

while(lineToSelect >= m_browseTableModel->rowCount() && m_browseTableModel->canFetchMore())

m_browseTableModel->fetchMore();

// Select it

ui->dataTable->clearSelection();

ui->dataTable->selectRow(lineToSelect);

ui->dataTable->scrollTo(ui->dataTable->currentIndex());

QApplication::restoreOverrideCursor();

{

int curRow = ui->dataTable->currentIndex().row();

curRow -= 100;

if(curRow < 0)

curRow = 0;

selectTableLine(curRow);

{

int curRow = ui->dataTable->currentIndex().row();

curRow += 100;

if(curRow >= m_browseTableModel->totalRowCount())

curRow = m_browseTableModel->totalRowCount() - 1;

selectTableLine(curRow);

{

int row = ui->editGoto->text().toInt();

if(row <= 0)

row = 1;

if(row > m_browseTableModel->totalRowCount())

row = m_browseTableModel->totalRowCount();

selectTableLine(row - 1);

ui->editGoto->setText(QString::number(row));

{

// Get all the numbers, i.e. the number of the first row and the last row as well as the total number of rows

int from = ui->dataTable->verticalHeader()->visualIndexAt(0) + 1;

int to = ui->dataTable->verticalHeader()->visualIndexAt(ui->dataTable->height()) - 1;

int total = m_browseTableModel->totalRowCount();

if(to == -2)

to = total;

// Update the validator of the goto row field

gotoValidator->setRange(0, total);

// Update the label showing the current position

ui->labelRecordset->setText(tr("%1 - %2 of %3").arg(from).arg(to).arg(total));

{

if (!db.isOpen()){

QMessageBox::information( this, QApplication::applicationName(), tr("There is no database opened. Please open or create a new database file."));

return;

}

EditTableDialog dialog(&db, "", this);

if(dialog.exec())

{

populateStructure();

resetBrowser();

}

{

if (!db.isOpen()){

QMessageBox::information( this, QApplication::applicationName(), tr("There is no database opened. Please open or create a new database file."));

return;

}

CreateIndexDialog dialog(&db, this);

if(dialog.exec())

{

populateStructure();

resetBrowser();

}

{

VacuumDialog dialog(&db, this);

if(dialog.exec())

{

populateStructure();

resetBrowser();

}

{

// Get name and type of object to delete

QString table = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 0)).toString();

QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 1)).toString();

// Ask user if he really wants to delete that table

if(QMessageBox::warning(this, QApplication::applicationName(), tr("Are you sure you want to delete the %1 '%2'?\nAll data associated with the %1 will be lost.").arg(type).arg(table),

QMessageBox::Yes, QMessageBox::No | QMessageBox::Default | QMessageBox::Escape) == QMessageBox::Yes)

{

// Delete the table

QString statement = QString("DROP %1 `%2`;").arg(type.toUpper()).arg(table);

if(!db.executeSQL( statement))

{

QString error = tr("Error: could not delete the %1. Message from database engine:\n%2").arg(type).arg(db.lastErrorMessage);

QMessageBox::warning(this, QApplication::applicationName(), error);

} else {

populateStructure();

resetBrowser();

}

}

{

if (!db.isOpen()){

QMessageBox::information( this, QApplication::applicationName(), tr("There is no database opened."));

return;

}

if(!ui->dbTreeWidget->selectionModel()->hasSelection()){

return;

}

QString tableToEdit = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 0)).toString();

EditTableDialog dialog(&db, tableToEdit, this);

if(dialog.exec())

{

populateStructure();

resetBrowser();

}

{

AboutDialog dialog(this);

dialog.exec();

{

editWin->hide();

activateWindow();

ui->dataTable->setCurrentIndex(ui->dataTable->currentIndex().sibling(editWin->getCurrentRow(), editWin->getCurrentCol()));

{

editWin->loadText(index.data(Qt::EditRole).toByteArray(), index.row(), index.column());

editWin->show();

{

if(!index.isValid())

return;

// Don't allow editing of other objects than tables

if(db.getObjectByName(ui->comboBrowseTable->currentText()).gettype() != "table")

return;

editText(index);

{

SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());

// Get SQL code to execute. This depends on the button that's been pressed

QString query;

bool singleStep = false;

if(sender()->objectName() == "actionSqlExecuteLine")

{

query = sqlWidget->getEditor()->document()->toPlainText().mid(sqlWidget->getEditor()->textCursor().block().position());

singleStep = true;

} else {

// if a part of the query is selected, we will only execute this part

query = sqlWidget->getSelectedSql();

if(query.isEmpty())

query = sqlWidget->getSql();

}

if (query.isEmpty())

return;

//log the query

db.logSQL(query, kLogMsg_User);

sqlite3_stmt *vm;

QByteArray utf8Query = query.toUtf8();

const char *tail = utf8Query.data();

int sql3status = 0;

QString statusMessage;

bool modified = false;

bool wasdirty = db.getDirty();

// there is no choice, we have to start a transaction before

// we create the prepared statement, otherwise every executed

// statement will get committed after the prepared statement

// gets finalized, see http://www.sqlite.org/lang_transaction.html

db.setRestorePoint();

//Accept multi-line queries, by looping until the tail is empty

QElapsedTimer timer;

timer.start();

do

{

const char* qbegin = tail;

sql3status = sqlite3_prepare_v2(db._db,tail,utf8Query.length(),

&vm, &tail);

QString queryPart = QString::fromUtf8(qbegin, tail - qbegin);

if (sql3status == SQLITE_OK){

sql3status = sqlite3_step(vm);

sqlite3_finalize(vm);

// SQLite returns SQLITE_DONE when a valid SELECT statement was executed but returned no results. To run into the branch that updates

// the status message and the table view anyway manipulate the status value here.

if(queryPart.trimmed().startsWith("SELECT", Qt::CaseInsensitive) && sql3status == SQLITE_DONE)

sql3status = SQLITE_ROW;

switch(sql3status)

{

case SQLITE_ROW:

{

sqlWidget->getModel()->setQuery(queryPart);

if(sqlWidget->getModel()->valid())

{

statusMessage = tr("%1 Rows returned from: %2 (took %3ms)").arg(

sqlWidget->getModel()->totalRowCount()).arg(queryPart.trimmed()).arg(timer.elapsed());

sqlWidget->enableSaveButton(true);

sql3status = SQLITE_OK;

}

else

{

statusMessage = tr("Error executing query: %1").arg(queryPart);

sql3status = SQLITE_ERROR;

}

}

case SQLITE_DONE:

case SQLITE_OK:

{

if( !queryPart.trimmed().startsWith("SELECT", Qt::CaseInsensitive) )

{

modified = true;

statusMessage = tr("Query executed successfully: %1 (took %2ms)").arg(queryPart.trimmed()).arg(timer.elapsed());

}

break;

}

default:

statusMessage = QString::fromUtf8((const char*)sqlite3_errmsg(db._db)) +

": " + queryPart;

break;

}

timer.restart();

// Stop after the first full statement if we're in single step mode

if(singleStep)

break;

} else {

statusMessage = QString::fromUtf8((const char*)sqlite3_errmsg(db._db)) +

": " + queryPart;

}

} while( tail && *tail != 0 && (sql3status == SQLITE_OK || sql3status == SQLITE_DONE));

sqlWidget->finishExecution(statusMessage);

updatePlot(sqlWidget->getModel());

if(!modified && !wasdirty)

db.revert(); // better rollback, if the logic is not enough we can tune it.

{

if(tabindex == 0)

{

populateStructure();

} else if(tabindex == 1) {

populateStructure();

resetBrowser();

} else if(tabindex == 2) {

loadPragmas();

}

{

QString wFile = QFileDialog::getOpenFileName(

this,

tr("Choose a text file"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString(),

tr("Text files(*.csv *.txt);;All files(*)"));

if (QFile::exists(wFile) )

{

ImportCsvDialog dialog(wFile, &db, this);

if(dialog.exec())

{

populateStructure();

resetBrowser();

QMessageBox::information(this, QApplication::applicationName(), tr("Import completed"));

}

}

{

// Get the current table name if we are in the Browse Data tab

QString current_table;

if(ui->mainTab->currentIndex() == 0)

current_table = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 0)).toString();

else if(ui->mainTab->currentIndex() == 1)

current_table = ui->comboBrowseTable->currentText();

// Open dialog

ExportCsvDialog dialog(&db, this, "", current_table);

dialog.exec();

{

ui->fileSaveAction->setEnabled(dirty);

ui->fileRevertAction->setEnabled(dirty);

{

if(db.isOpen())

db.saveAll();

{

if (db.isOpen()){

QString msg = tr("Are you sure you want to undo all changes made to the database file '%1' since the last save?").arg(db.curDBFilename);

if(QMessageBox::question(this, QApplication::applicationName(), msg, QMessageBox::Yes | QMessageBox::Default, QMessageBox::No | QMessageBox::Escape) == QMessageBox::Yes)

{

db.revertAll();

populateStructure();

resetBrowser();

}

}

{

QString fileName = QFileDialog::getSaveFileName(

this,

tr("Choose a filename to export"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString(),

tr("Text files(*.sql *.txt)"));

if(fileName.size())

{

if(!db.dump(fileName))

QMessageBox::warning(this, QApplication::applicationName(), tr("Export cancelled or failed."));

else

QMessageBox::information(this, QApplication::applicationName(), tr("Export completed."));

}

{

// Get file name to import

QString fileName = QFileDialog::getOpenFileName(

this,

tr("Choose a file to import"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString(),

tr("Text files(*.sql *.txt);;All files(*)"));

// Cancel when file doesn't exist

if(!QFile::exists(fileName))

return;

// If there is already a database file opened ask the user wether to import into this one or a new one. If no DB is opened just ask for a DB name directly

QString newDbFile;

if((db.isOpen() && QMessageBox::question(this,

QApplication::applicationName(),

tr("Do you want to create a new database file to hold the imported data?\n"

"If you answer no we will attempt to import the data in the SQL file to the current database."),

QMessageBox::Yes, QMessageBox::No) == QMessageBox::Yes) || !db.isOpen())

{

newDbFile = QFileDialog::getSaveFileName(

this,

tr("Choose a filename to save under"),

PreferencesDialog::getSettingsValue("db", "defaultlocation").toString());

if(QFile::exists(newDbFile))

{

QMessageBox::information(this, QApplication::applicationName(), tr("File %1 already exists. Please choose a different name.").arg(newDbFile));

return;

} else if(newDbFile.size() == 0) {

return;

}

db.create(newDbFile);

loadExtensionsFromSettings();

}

// Open, read, execute and close file

QApplication::setOverrideCursor(Qt::WaitCursor);

QFile f(fileName);

f.open(QIODevice::ReadOnly);

if(!db.executeMultiSQL(f.readAll(), newDbFile.size() == 0))

QMessageBox::warning(this, QApplication::applicationName(), tr("Error importing data: %1").arg(db.lastErrorMessage));

else

QMessageBox::information(this, QApplication::applicationName(), tr("Import completed."));

f.close();

QApplication::restoreOverrideCursor();

// Resfresh window when importing into an existing DB or - when creating a new file - just open it correctly

if(newDbFile.size())

{

fileOpen(newDbFile);

} else {

populateStructure();

resetBrowser();

}

{

PreferencesDialog dialog(this);

if(dialog.exec())

reloadSettings();

{

delete sqliteHighlighterLogApp;

delete sqliteHighlighterLogUser;

sqliteHighlighterLogApp = new SQLiteSyntaxHighlighter(ui->editLogApplication->document());

sqliteHighlighterLogUser = new SQLiteSyntaxHighlighter(ui->editLogUser->document());

{

if(!ui->dbTreeWidget->selectionModel()->hasSelection())

return;

QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 1)).toString();

if(type == "table" || type == "view" || type == "trigger" || type == "index")

popupTableMenu->exec(ui->dbTreeWidget->mapToGlobal(qPoint));

{

// Just assume first that something's selected that can not be edited at all

ui->editDeleteObjectAction->setEnabled(false);

ui->editModifyTableAction->setEnabled(false);

if(!ui->dbTreeWidget->currentIndex().isValid())

return;

// Change the text of the actions

QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 1)).toString();

ui->editDeleteObjectAction->setIcon(QIcon(QString(":icons/%1_delete").arg(type)));

if(type == "view")

ui->editDeleteObjectAction->setText(tr("Delete View"));

else if(type == "trigger")

ui->editDeleteObjectAction->setText(tr("Delete Trigger"));

else if(type == "index")

ui->editDeleteObjectAction->setText(tr("Delete Index"));

else

ui->editDeleteObjectAction->setText(tr("Delete Table"));

// Activate actions

if(type == "table")

{

ui->editDeleteObjectAction->setEnabled(true);

ui->editModifyTableAction->setEnabled(true);

} else if(type == "view" || type == "trigger" || type == "index") {

ui->editDeleteObjectAction->setEnabled(true);

}

ui->actionExportCsvPopup->setEnabled(type == "table" || type == "view");

{

QAction *action = qobject_cast<QAction *>(sender());

if (action)

fileOpen(action->data().toString());

{

QStringList files = PreferencesDialog::getSettingsValue("General", "recentFileList").toStringList();

int numRecentFiles = qMin(files.size(), (int)MaxRecentFiles);

for (int i = 0; i < numRecentFiles; ++i) {

QString text = tr("&%1 %2").arg(i + 1).arg(files[i]);

recentFileActs[i]->setText(text);

recentFileActs[i]->setData(files[i]);

recentFileActs[i]->setVisible(true);

}

for (int j = numRecentFiles; j < MaxRecentFiles; ++j)

recentFileActs[j]->setVisible(false);

recentSeparatorAct->setVisible(numRecentFiles > 0);

{

setWindowFilePath(fileName);

setWindowTitle( QApplication::applicationName() +" - "+fileName);

activateFields(true);

QStringList files = PreferencesDialog::getSettingsValue("General", "recentFileList").toStringList();

files.removeAll(fileName);

files.prepend(fileName);

while (files.size() > MaxRecentFiles)

files.removeLast();

PreferencesDialog::setSettingsValue("General", "recentFileList", files);

foreach (QWidget *widget, QApplication::topLevelWidgets()) {

MainWindow *mainWin = qobject_cast<MainWindow *>(widget);

if (mainWin)

mainWin->updateRecentFileActions();

}