Files
sqlitebrowser/src/MainWindow.cpp
2019-08-27 21:55:20 +02:00

3235 lines
138 KiB
C++

#include "MainWindow.h"
#include "ui_MainWindow.h"
#include "Application.h"
#include "EditIndexDialog.h"
#include "AboutDialog.h"
#include "EditTableDialog.h"
#include "ImportCsvDialog.h"
#include "ExportDataDialog.h"
#include "Settings.h"
#include "PreferencesDialog.h"
#include "EditDialog.h"
#include "sqlitetablemodel.h"
#include "SqlExecutionArea.h"
#include "VacuumDialog.h"
#include "DbStructureModel.h"
#include "version.h"
#include "sqlite.h"
#include "CipherDialog.h"
#include "ExportSqlDialog.h"
#include "SqlUiLexer.h"
#include "FileDialog.h"
#include "FilterTableHeader.h"
#include "RemoteDock.h"
#include "RemoteDatabase.h"
#include "FindReplaceDialog.h"
#include "RunSql.h"
#include "ExtendedTableWidget.h"
#include <chrono>
#include <QFile>
#include <QTextStream>
#include <QWhatsThis>
#include <QMessageBox>
#include <QStandardItemModel>
#include <QPersistentModelIndex>
#include <QDragEnterEvent>
#include <QScrollBar>
#include <QMimeData>
#include <QDesktopServices>
#include <QXmlStreamReader>
#include <QXmlStreamWriter>
#include <QInputDialog>
#include <QProgressDialog>
#include <QTextEdit>
#include <QClipboard>
#include <QShortcut>
#include <QUrlQuery>
#include <QDataStream> // This include seems to only be necessary for the Windows build
#include <QPrinter>
#include <QPrintPreviewDialog>
#include <QToolButton>
#ifdef Q_OS_MACX //Needed only on macOS
#include <QOpenGLWidget>
#endif
#include <limits>
const int MainWindow::MaxRecentFiles;
// These are needed for reading and writing object files
QDataStream& operator>>(QDataStream& ds, sqlb::ObjectIdentifier& objid)
{
// Read in the item
QVariant v;
ds >> v;
// If it is a string list, we can treat it as an object identifier. If it isn't, we assume it's just a
// single string and use interpret it as the table name in the main schema. This is done for backwards
// compatability with old project file formats.
QStringList str = v.toStringList();
if(str.isEmpty())
{
objid = sqlb::ObjectIdentifier("main", v.toString().toStdString());
} else {
objid.setSchema(str.first().toStdString());
if(str.size() >= 2)
objid.setName(str.last().toStdString());
}
return ds;
}
// This is a temporary helper function. Delete it once we clean up the project file loading.
static std::vector<sqlb::SortedColumn> toSortOrderVector(int index, Qt::SortOrder mode)
{
std::vector<sqlb::SortedColumn> vector;
vector.emplace_back(index, mode == Qt::AscendingOrder ? sqlb::Ascending : sqlb::Descending);
return vector;
}
MainWindow::MainWindow(QWidget* parent)
: QMainWindow(parent),
ui(new Ui::MainWindow),
db(),
m_remoteDb(new RemoteDatabase),
editDock(new EditDialog(this)),
plotDock(new PlotDock(this)),
remoteDock(new RemoteDock(this)),
findReplaceDialog(new FindReplaceDialog(this)),
execute_sql_worker(nullptr),
isProjectModified(false)
{
ui->setupUi(this);
init();
activateFields(false);
updateRecentFileActions();
}
MainWindow::~MainWindow()
{
delete m_remoteDb;
delete ui;
}
void MainWindow::init()
{
// Load window settings
tabifyDockWidget(ui->dockLog, ui->dockPlot);
tabifyDockWidget(ui->dockLog, ui->dockSchema);
tabifyDockWidget(ui->dockLog, ui->dockRemote);
#ifdef Q_OS_MACX
// Add OpenGL Context for macOS
QOpenGLWidget *ogl = new QOpenGLWidget(this);
ui->verticalLayout->addWidget(ogl);
ogl->setHidden(true);
#endif
// Automatic update check
#ifdef CHECKNEWVERSION
connect(m_remoteDb, &RemoteDatabase::networkReady, [this]() {
// Check for a new version if automatic update check aren't disabled in the settings dialog
if(Settings::getValue("checkversion", "enabled").toBool())
m_remoteDb->fetch("https://download.sqlitebrowser.org/currentrelease", RemoteDatabase::RequestTypeNewVersionCheck);
});
#endif
// Connect SQL logging and database state setting to main window
connect(&db, &DBBrowserDB::dbChanged, this, &MainWindow::dbState, Qt::QueuedConnection);
connect(&db, &DBBrowserDB::sqlExecuted, this, &MainWindow::logSql, Qt::QueuedConnection);
connect(&db, &DBBrowserDB::requestCollation, this, &MainWindow::requestCollation);
// Initialise table browser first
ui->tableBrowser->init(&db);
// Set project modified flag when the settings in the table browser were changed
connect(ui->tableBrowser, &TableBrowser::projectModified, this, [this]() {
isProjectModified = true;
});
connect(ui->tableBrowser->model(), &SqliteTableModel::dataChanged, this, &MainWindow::dataTableSelectionChanged);
connect(ui->tableBrowser, &TableBrowser::selectionChanged, this, &MainWindow::dataTableSelectionChanged);
connect(ui->tableBrowser, &TableBrowser::selectionChangedByDoubleClick, this, &MainWindow::doubleClickTable);
connect(ui->tableBrowser, &TableBrowser::updatePlot, this, &MainWindow::attachPlot);
connect(ui->tableBrowser, &TableBrowser::createView, this, &MainWindow::saveAsView);
connect(ui->tableBrowser, &TableBrowser::requestFileOpen, this, [this](const QString& file) {
fileOpen(file);
});
connect(ui->tableBrowser, &TableBrowser::statusMessageRequested, ui->statusbar, [this](const QString& message) {
ui->statusbar->showMessage(message);
});
m_currentTabTableModel = ui->tableBrowser->model();
// Set up DB structure tab
dbStructureModel = new DbStructureModel(db, this);
connect(&db, &DBBrowserDB::structureUpdated, this, [this]() {
QString old_table = QString::fromStdString(ui->tableBrowser->currentlyBrowsedTableName().name());
dbStructureModel->reloadData();
populateStructure(old_table);
});
ui->dbTreeWidget->setModel(dbStructureModel);
ui->dbTreeWidget->setColumnWidth(DbStructureModel::ColumnName, 300);
ui->dbTreeWidget->setColumnHidden(DbStructureModel::ColumnObjectType, true);
ui->dbTreeWidget->setColumnHidden(DbStructureModel::ColumnSchema, true);
// Set up DB schema dock
ui->treeSchemaDock->setModel(dbStructureModel);
ui->treeSchemaDock->setColumnHidden(DbStructureModel::ColumnObjectType, true);
ui->treeSchemaDock->setColumnHidden(DbStructureModel::ColumnSchema, true);
// Set up the table combo box in the Browse Data tab
ui->tableBrowser->setStructure(dbStructureModel);
// Create docks
ui->dockEdit->setWidget(editDock);
ui->dockPlot->setWidget(plotDock);
ui->dockRemote->setWidget(remoteDock);
// Set up edit dock
editDock->setReadOnly(true);
// Restore window geometry
restoreGeometry(Settings::getValue("MainWindow", "geometry").toByteArray());
// Save default and restore window state
defaultWindowState = saveState();
restoreState(Settings::getValue("MainWindow", "windowState").toByteArray());
// Save default and restore open tab order if the openTabs setting is saved.
defaultOpenTabs = saveOpenTabs();
restoreOpenTabs(Settings::getValue("MainWindow", "openTabs").toString());
// Restore dock state settings
ui->comboLogSubmittedBy->setCurrentIndex(ui->comboLogSubmittedBy->findText(Settings::getValue("SQLLogDock", "Log").toString()));
// Add keyboard shortcuts
QShortcut* shortcutBrowseRefreshF5 = new QShortcut(QKeySequence("F5"), this);
connect(shortcutBrowseRefreshF5, &QShortcut::activated, this, &MainWindow::refresh);
QShortcut* shortcutBrowseRefreshCtrlR = new QShortcut(QKeySequence("Ctrl+R"), this);
connect(shortcutBrowseRefreshCtrlR, &QShortcut::activated, this, &MainWindow::refresh);
// Add print shortcut for the DB Structure tab (dbTreeWidget) with context to the widget, so other print shortcuts aren't eclipsed.
QShortcut* shortcutPrint = new QShortcut(QKeySequence(QKeySequence::Print), ui->dbTreeWidget, nullptr, nullptr, Qt::WidgetShortcut);
connect(shortcutPrint, &QShortcut::activated, this, &MainWindow::printDbStructure);
// 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], &QAction::triggered, this, &MainWindow::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->actionEditBrowseTable);
popupTableMenu->addAction(ui->editModifyObjectAction);
popupTableMenu->addAction(ui->editDeleteObjectAction);
popupTableMenu->addSeparator();
popupTableMenu->addAction(ui->actionEditCopyCreateStatement);
popupTableMenu->addAction(ui->actionExportCsvPopup);
popupSchemaDockMenu = new QMenu(this);
popupSchemaDockMenu->addAction(ui->actionPopupSchemaDockBrowseTable);
popupSchemaDockMenu->addSeparator();
popupSchemaDockMenu->addAction(ui->actionDropQualifiedCheck);
popupSchemaDockMenu->addAction(ui->actionEnquoteNamesCheck);
popupOpenDbMenu = new QMenu(this);
popupOpenDbMenu->addAction(ui->fileOpenAction);
popupOpenDbMenu->addAction(ui->fileOpenReadOnlyAction);
ui->fileOpenActionPopup->setMenu(popupOpenDbMenu);
popupSaveSqlFileMenu = new QMenu(this);
popupSaveSqlFileMenu->addAction(ui->actionSqlSaveFile);
popupSaveSqlFileMenu->addAction(ui->actionSqlSaveFileAs);
ui->actionSqlSaveFilePopup->setMenu(popupSaveSqlFileMenu);
popupSaveSqlResultsMenu = new QMenu(this);
popupSaveSqlResultsMenu->addAction(ui->actionSqlResultsExportCsv);
popupSaveSqlResultsMenu->addAction(ui->actionSqlResultsSaveAsView);
ui->actionSqlResultsSave->setMenu(popupSaveSqlResultsMenu);
qobject_cast<QToolButton*>(ui->toolbarSql->widgetForAction(ui->actionSqlResultsSave))->setPopupMode(QToolButton::InstantPopup);
// Add menu item for log dock
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"));
// Add menu item for plot dock
ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockPlot->toggleViewAction());
ui->viewMenu->actions().at(1)->setShortcut(QKeySequence(tr("Ctrl+D")));
ui->viewMenu->actions().at(1)->setIcon(QIcon(":/icons/log_dock"));
// Add menu item for schema dock
ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockSchema->toggleViewAction());
ui->viewMenu->actions().at(2)->setShortcut(QKeySequence(tr("Ctrl+I")));
ui->viewMenu->actions().at(2)->setIcon(QIcon(":/icons/log_dock"));
// Add menu item for edit dock
ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockEdit->toggleViewAction());
ui->viewMenu->actions().at(3)->setShortcut(QKeySequence(tr("Ctrl+E")));
ui->viewMenu->actions().at(3)->setIcon(QIcon(":/icons/log_dock"));
// Add menu item for plot dock
ui->viewMenu->insertAction(ui->viewDBToolbarAction, ui->dockRemote->toggleViewAction());
ui->viewMenu->actions().at(4)->setIcon(QIcon(":/icons/log_dock"));
// Set checked state if toolbar is visible
ui->viewDBToolbarAction->setChecked(!ui->toolbarDB->isHidden());
ui->viewExtraDBToolbarAction->setChecked(!ui->toolbarExtraDB->isHidden());
ui->viewProjectToolbarAction->setChecked(!ui->toolbarProject->isHidden());
// Add separator between docks and toolbars
ui->viewMenu->insertSeparator(ui->viewDBToolbarAction);
// Connect the tabCloseRequested to the actual closeTab function.
// This must be done before the connections for checking the actions in the View menu so
// they are updated accordingly.
connect(ui->mainTab, &QTabWidget::tabCloseRequested, this, &MainWindow::closeTab);
// Add entries for toggling the visibility of main tabs
for (QWidget* widget : {ui->structure, ui->browser, ui->pragmas, ui->query}) {
QAction* action = ui->viewMenu->addAction(QIcon(":/icons/tab"), widget->accessibleName());
action->setCheckable(true);
action->setChecked(ui->mainTab->indexOf(widget) != -1);
connect(action, &QAction::toggled, [=](bool show) { toggleTabVisible(widget, show); });
// Connect tabCloseRequested for setting checked the appropiate menu entry.
// Note these are called after the actual tab is closed only because they are connected
// after connecting closeTab.
connect(ui->mainTab, &QTabWidget::tabCloseRequested, [=](int /*index*/) {
action->setChecked(ui->mainTab->indexOf(widget) != -1);
});
}
ui->viewMenu->addSeparator();
QAction* resetLayoutAction = ui->viewMenu->addAction(tr("Reset Window Layout"));
resetLayoutAction->setShortcut(QKeySequence(tr("Alt+0")));
connect(resetLayoutAction, &QAction::triggered, [=]() {
restoreState(defaultWindowState);
restoreOpenTabs(defaultOpenTabs);
});
// Set Alt+[1-4] shortcuts for opening the corresponding tab in that position.
// Note that it is safe to call setCurrentIndex with a tab that is currently closed,
// since setCurrentIndex does nothing in that case.
QShortcut* setTab1Shortcut = new QShortcut(QKeySequence("Alt+1"), this);
connect(setTab1Shortcut, &QShortcut::activated, [this]() { ui->mainTab->setCurrentIndex(0); });
QShortcut* setTab2Shortcut = new QShortcut(QKeySequence("Alt+2"), this);
connect(setTab2Shortcut, &QShortcut::activated, [this]() { ui->mainTab->setCurrentIndex(1); });
QShortcut* setTab3Shortcut = new QShortcut(QKeySequence("Alt+3"), this);
connect(setTab3Shortcut, &QShortcut::activated, [this]() { ui->mainTab->setCurrentIndex(2); });
QShortcut* setTab4Shortcut = new QShortcut(QKeySequence("Alt+4"), this);
connect(setTab4Shortcut, &QShortcut::activated, [this]() { ui->mainTab->setCurrentIndex(3); });
// If we're not compiling in SQLCipher, hide its FAQ link in the help menu
#ifndef ENABLE_SQLCIPHER
ui->actionSqlCipherFaq->setVisible(false);
#endif
// Set statusbar fields
statusBusyLabel = new QLabel(ui->statusbar);
statusBusyLabel->setEnabled(false);
statusBusyLabel->setVisible(false);
statusBusyLabel->setToolTip(tr("The database is currenctly busy."));
ui->statusbar->addPermanentWidget(statusBusyLabel);
statusStopButton = new QToolButton(ui->statusbar);
statusStopButton->setVisible(false);
statusStopButton->setIcon(QIcon(":icons/cancel"));
statusStopButton->setToolTip(tr("Click here to interrupt the currently running query."));
statusStopButton->setMaximumSize(ui->statusbar->geometry().height() - 6, ui->statusbar->geometry().height() - 6);
statusStopButton->setAutoRaise(true);
ui->statusbar->addPermanentWidget(statusStopButton);
statusEncryptionLabel = new QLabel(ui->statusbar);
statusEncryptionLabel->setEnabled(false);
statusEncryptionLabel->setVisible(false);
statusEncryptionLabel->setText(tr("Encrypted"));
statusEncryptionLabel->setToolTip(tr("Database is encrypted using SQLCipher"));
ui->statusbar->addPermanentWidget(statusEncryptionLabel);
statusReadOnlyLabel = new QLabel(ui->statusbar);
statusReadOnlyLabel->setEnabled(false);
statusReadOnlyLabel->setVisible(false);
statusReadOnlyLabel->setText(tr("Read only"));
statusReadOnlyLabel->setToolTip(tr("Database file is read only. Editing the database is disabled."));
ui->statusbar->addPermanentWidget(statusReadOnlyLabel);
statusEncodingLabel = new QLabel(ui->statusbar);
statusEncodingLabel->setEnabled(false);
statusEncodingLabel->setText("UTF-8");
statusEncodingLabel->setToolTip(tr("Database encoding"));
ui->statusbar->addPermanentWidget(statusEncodingLabel);
// When changing the text of the toolbar actions, also automatically change their icon text and their tooltip text
connect(ui->editModifyObjectAction, &QAction::changed, [=]() {
ui->editModifyObjectAction->setIconText(ui->editModifyObjectAction->text());
ui->editModifyObjectAction->setToolTip(ui->editModifyObjectAction->text());
});
connect(ui->editDeleteObjectAction, &QAction::changed, [=]() {
ui->editDeleteObjectAction->setIconText(ui->editDeleteObjectAction->text());
ui->editDeleteObjectAction->setToolTip(ui->editDeleteObjectAction->text());
});
// When clicking the interrupt query button in the status bar, ask SQLite to interrupt the current query
connect(statusStopButton, &QToolButton::clicked, [this]() {
db.interruptQuery();
});
// Connect some more signals and slots
connect(editDock, &EditDialog::recordTextUpdated, this, &MainWindow::updateRecordText);
connect(ui->dbTreeWidget->selectionModel(), &QItemSelectionModel::currentChanged, this, &MainWindow::changeTreeSelection);
connect(ui->dockEdit, &QDockWidget::visibilityChanged, this, &MainWindow::toggleEditDock);
connect(m_remoteDb, SIGNAL(openFile(QString)), this, SLOT(fileOpen(QString)));
connect(m_remoteDb, &RemoteDatabase::gotCurrentVersion, this, &MainWindow::checkNewVersion);
connect(ui->actionDropQualifiedCheck, &QAction::toggled, dbStructureModel, &DbStructureModel::setDropQualifiedNames);
connect(ui->actionEnquoteNamesCheck, &QAction::toggled, dbStructureModel, &DbStructureModel::setDropEnquotedNames);
connect(&db, &DBBrowserDB::databaseInUseChanged, this, &MainWindow::updateDatabaseBusyStatus);
ui->actionDropQualifiedCheck->setChecked(Settings::getValue("SchemaDock", "dropQualifiedNames").toBool());
ui->actionEnquoteNamesCheck->setChecked(Settings::getValue("SchemaDock", "dropEnquotedNames").toBool());
connect(ui->tableBrowser->model(), &SqliteTableModel::finishedFetch, [this](){
auto& settings = ui->tableBrowser->settings(ui->tableBrowser->currentlyBrowsedTableName());
plotDock->updatePlot(ui->tableBrowser->model(), &settings, true, false);
});
connect(ui->actionSqlStop, &QAction::triggered, [this]() {
if(execute_sql_worker && execute_sql_worker->isRunning())
execute_sql_worker->stop();
});
// Connect tool pragmas
connect(ui->actionIntegrityCheck, &QAction::triggered, [this]() {
runSqlNewTab("PRAGMA integrity_check;", ui->actionIntegrityCheck->text());
});
connect(ui->actionQuickCheck, &QAction::triggered, [this]() {
runSqlNewTab("PRAGMA quick_check;", ui->actionQuickCheck->text());
});
connect(ui->actionForeignKeyCheck, &QAction::triggered, [this]() {
runSqlNewTab("PRAGMA foreign_key_check;", ui->actionForeignKeyCheck->text());
});
connect(ui->actionOptimize, &QAction::triggered, [this]() {
runSqlNewTab("PRAGMA optimize;", ui->actionOptimize->text());
});
// Action for switching the table via the Database Structure tab
connect(ui->actionPopupSchemaDockBrowseTable, &QAction::triggered, [this]() {
sqlb::ObjectIdentifier obj(ui->treeSchemaDock->model()->data(ui->treeSchemaDock->currentIndex().sibling(ui->treeSchemaDock->currentIndex().row(), DbStructureModel::ColumnSchema), Qt::EditRole).toString().toStdString(),
ui->treeSchemaDock->model()->data(ui->treeSchemaDock->currentIndex().sibling(ui->treeSchemaDock->currentIndex().row(), DbStructureModel::ColumnName), Qt::EditRole).toString().toStdString());
switchToBrowseDataTab(obj);
refresh(); // Required in case the Browse Data tab already was the active main tab
});
// Set other window settings
setAcceptDrops(true);
setWindowTitle(QApplication::applicationName());
// Add the documentation of shortcuts, which aren't otherwise visible in the user interface, to some buttons.
addShortcutsTooltip(ui->actionDbPrint);
addShortcutsTooltip(ui->actionSqlOpenTab);
addShortcutsTooltip(ui->actionSqlPrint);
addShortcutsTooltip(ui->actionExecuteSql, {shortcutBrowseRefreshF5->key(), shortcutBrowseRefreshCtrlR->key()});
addShortcutsTooltip(ui->actionSqlExecuteLine);
addShortcutsTooltip(ui->actionSqlFind);
addShortcutsTooltip(ui->actionSqlFindReplace);
addShortcutsTooltip(ui->actionSqlToggleComment);
// Load all settings
reloadSettings();
#ifndef ENABLE_SQLCIPHER
// Only show encryption menu action when SQLCipher support is enabled
ui->actionEncryption->setVisible(false);
#endif
/* Remove all the '&' signs from the dock titles. On at least Windows and
* OSX, Qt doesn't seem to support them properly, so they end up being
* visible instead of creating a keyboard shortcut
*/
ui->dockEdit->setWindowTitle(ui->dockEdit->windowTitle().remove('&'));
ui->dockLog->setWindowTitle(ui->dockLog->windowTitle().remove('&'));
ui->dockPlot->setWindowTitle(ui->dockPlot->windowTitle().remove('&'));
ui->dockSchema->setWindowTitle(ui->dockSchema->windowTitle().remove('&'));
ui->dockRemote->setWindowTitle(ui->dockRemote->windowTitle().remove('&'));
}
bool MainWindow::fileOpen(const QString& fileName, bool openFromProject, bool readOnly)
{
bool retval = false;
QString wFile = fileName;
// QFile::exist will produce error message if passed empty string.
// Test string length before usage w/ QFile to silence warning
if (wFile.isEmpty() || !QFile::exists(wFile))
{
wFile = FileDialog::getOpenFileName(
OpenDatabaseFile,
this,
tr("Choose a database file")
#ifndef Q_OS_MAC // Filters on OS X are buggy
, FileDialog::getSqlDatabaseFileFilter()
#endif
);
}
// catch situation where user has canceled file selection from dialog
if(!wFile.isEmpty() && QFile::exists(wFile) )
{
// Close the database. If the user didn't want to close it, though, stop here
if (db.isOpen())
if(!fileClose())
return false;
// Try opening it as a project file first
if(loadProject(wFile, readOnly))
{
retval = true;
} else {
// No project file; so it should be a database file
if(db.open(wFile, readOnly))
{
// Close all open but empty SQL tabs
for(int i=ui->tabSqlAreas->count()-1;i>=0;i--)
{
if(qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i))->getSql().trimmed().isEmpty())
closeSqlTab(i, true);
}
statusEncodingLabel->setText(db.getPragma("encoding"));
statusEncryptionLabel->setVisible(db.encrypted());
statusReadOnlyLabel->setVisible(db.readOnly());
setCurrentFile(wFile);
if(!openFromProject) {
currentProjectFilename.clear();
addToRecentFilesMenu(wFile, readOnly);
}
openSqlTab(true);
if(ui->mainTab->currentWidget() == ui->browser)
populateTable();
else if(ui->mainTab->currentWidget() == ui->pragmas)
loadPragmas();
retval = true;
} else {
QMessageBox::warning(this, qApp->applicationName(), tr("Could not open database file.\nReason: %1").arg(db.lastError()));
return false;
}
}
}
return retval;
}
void MainWindow::fileNew()
{
QString fileName = FileDialog::getSaveFileName(
CreateDatabaseFile,
this,
tr("Choose a filename to save under"),
FileDialog::getSqlDatabaseFileFilter());
if(!fileName.isEmpty())
{
if(QFile::exists(fileName))
QFile::remove(fileName);
db.create(fileName);
setCurrentFile(fileName);
addToRecentFilesMenu(fileName);
statusEncodingLabel->setText(db.getPragma("encoding"));
statusEncryptionLabel->setVisible(false);
statusReadOnlyLabel->setVisible(false);
populateTable();
openSqlTab(true);
createTable();
}
}
void MainWindow::fileNewInMemoryDatabase()
{
db.create(":memory:");
setCurrentFile(tr("In-Memory database"));
statusEncodingLabel->setText(db.getPragma("encoding"));
statusEncryptionLabel->setVisible(false);
statusReadOnlyLabel->setVisible(false);
populateTable();
openSqlTab(true);
createTable();
}
void MainWindow::populateStructure(const QString& old_table)
{
// Refresh the structure tab
ui->dbTreeWidget->setRootIndex(dbStructureModel->index(1, 0)); // Show the 'All' part of the db structure
ui->dbTreeWidget->expandToDepth(0);
ui->treeSchemaDock->setRootIndex(dbStructureModel->index(1, 0)); // Show the 'All' part of the db structure
ui->treeSchemaDock->expandToDepth(0);
// Refresh the browse data tab
ui->tableBrowser->setStructure(dbStructureModel, old_table);
// Cancel here if no database is opened
if(!db.isOpen())
return;
// Update table and column names for syntax highlighting
SqlUiLexer::QualifiedTablesMap qualifiedTablesMap;
for(const auto& it : db.schemata)
{
SqlUiLexer::TablesAndColumnsMap tablesToColumnsMap;
objectMap tab = db.getBrowsableObjects(it.first);
for(const auto& jt : tab)
{
QString objectname = QString::fromStdString(jt.second->name());
sqlb::FieldInfoList fi = jt.second->fieldInformation();
for(const sqlb::FieldInfo& f : fi)
tablesToColumnsMap[objectname].push_back(QString::fromStdString(f.name));
}
qualifiedTablesMap[QString::fromStdString(it.first)] = tablesToColumnsMap;
}
SqlTextEdit::sqlLexer->setTableNames(qualifiedTablesMap);
ui->editLogApplication->reloadKeywords();
ui->editLogUser->reloadKeywords();
for(int i=0;i<ui->tabSqlAreas->count();i++)
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i))->getEditor()->reloadKeywords();
// Resize SQL column to fit contents
ui->dbTreeWidget->resizeColumnToContents(DbStructureModel::ColumnSQL);
ui->treeSchemaDock->resizeColumnToContents(DbStructureModel::ColumnSQL);
// Resize also the Name column in the Dock since it has usually
// short content and there is little space there.
ui->treeSchemaDock->resizeColumnToContents(DbStructureModel::ColumnName);
}
void MainWindow::populateTable()
{
// Early exit if the Browse Data tab isn't visible as there is no need to update it in this case
if(ui->mainTab->currentWidget() != ui->browser)
return;
QApplication::setOverrideCursor(Qt::WaitCursor);
ui->tableBrowser->updateTable();
QApplication::restoreOverrideCursor();
}
bool MainWindow::fileClose()
{
// Stop any running SQL statements before closing the database
if(execute_sql_worker && execute_sql_worker->isRunning())
{
if(QMessageBox::warning(this, qApp->applicationName(),
tr("You are still executing SQL statements. When closing the database now the execution will be stopped. maybe "
"leaving the database in an incosistent state. Are you sure you want to close the database?"),
QMessageBox::Yes, QMessageBox::Cancel | QMessageBox::Default | QMessageBox::Escape) == QMessageBox::Cancel)
return false;
execute_sql_worker->stop();
execute_sql_worker->wait();
}
// Close the database but stop the closing process here if the user pressed the cancel button in there
if(!closeProject())
return false;
setWindowTitle(QApplication::applicationName());
loadPragmas();
statusEncryptionLabel->setVisible(false);
statusReadOnlyLabel->setVisible(false);
// Reset the table browser of the Browse Data tab
ui->tableBrowser->reset();
// Clear edit dock
editDock->setCurrentIndex(QModelIndex());
activateFields(false);
// Clear the SQL Log
ui->editLogApplication->clear();
ui->editLogUser->clear();
ui->editLogErrorLog->clear();
// Remove completion and highlighting for identifiers
SqlTextEdit::sqlLexer->setTableNames(SqlUiLexer::QualifiedTablesMap());
for(int i=0; i < ui->tabSqlAreas->count(); i++)
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i))->getEditor()->reloadKeywords();
return true;
}
void MainWindow::closeEvent( QCloseEvent* event )
{
if(closeFiles())
{
Settings::setValue("MainWindow", "geometry", saveGeometry());
Settings::setValue("MainWindow", "windowState", saveState());
Settings::setValue("MainWindow", "openTabs", saveOpenTabs());
Settings::setValue("SQLLogDock", "Log", ui->comboLogSubmittedBy->currentText());
Settings::setValue("SchemaDock", "dropQualifiedNames", ui->actionDropQualifiedCheck->isChecked());
Settings::setValue("SchemaDock", "dropEnquotedNames", ui->actionEnquoteNamesCheck->isChecked());
SqlExecutionArea::saveState();
QMainWindow::closeEvent(event);
} else {
event->ignore();
}
}
bool MainWindow::closeFiles()
{
bool ignoreUnattachedBuffers = false;
// Ask for saving all modified open SQL files in their files and all the unattached tabs in a project file.
for(int i=0; i<ui->tabSqlAreas->count(); i++)
// Ask for saving and comply with cancel answer.
if(!askSaveSqlTab(i, ignoreUnattachedBuffers))
return false;
return closeProject();
}
bool MainWindow::closeProject()
{
if(!currentProjectFilename.isEmpty() && isProjectModified) {
QMessageBox::StandardButton reply = QMessageBox::question
(nullptr,
QApplication::applicationName(),
tr("Do you want to save the changes made to the project file %1?").
arg(QFileInfo(currentProjectFilename).fileName()),
QMessageBox::Save | QMessageBox::No | QMessageBox::Cancel);
switch(reply) {
case QMessageBox::Save:
saveProject();
break;
case QMessageBox::Cancel:
return false;
default:
break;
}
}
return db.close();
}
void MainWindow::attachPlot(ExtendedTableWidget* tableWidget, SqliteTableModel* model, BrowseDataTableSettings* settings, bool keepOrResetSelection)
{
plotDock->updatePlot(model, settings, true, keepOrResetSelection);
// Disconnect previous connection
disconnect(plotDock, SIGNAL(pointsSelected(int,int)), nullptr, nullptr);
if(tableWidget) {
// Connect plot selection to the current table results widget.
connect(plotDock, SIGNAL(pointsSelected(int,int)), tableWidget, SLOT(selectTableLines(int, int)));
connect(tableWidget, &ExtendedTableWidget::destroyed, plotDock, &PlotDock::resetPlot);
}
}
void MainWindow::refresh()
{
// What the Refresh function does depends on the currently active tab. This way the keyboard shortcuts (F5 and Ctrl+R)
// always perform some meaningful task; they just happen to be context dependent in the function they trigger.
QWidget* currentTab = ui->mainTab->currentWidget();
if (currentTab == ui->structure) {
// Refresh the schema
db.updateSchema();
} else if (currentTab == ui->browser) {
// Refresh the schema and reload the current table
db.updateSchema();
populateTable();
} else if (currentTab == ui->pragmas) {
// Reload pragma values
loadPragmas();
} else if (currentTab == ui->query) {
// (Re-)Run the current SQL query
executeQuery();
}
}
void MainWindow::createTable()
{
EditTableDialog dialog(db, sqlb::ObjectIdentifier(), true, this);
if(dialog.exec())
{
populateTable();
}
}
void MainWindow::createIndex()
{
EditIndexDialog dialog(db, sqlb::ObjectIdentifier(), true, this);
if(dialog.exec())
populateTable();
}
void MainWindow::compact()
{
VacuumDialog dialog(&db, this);
dialog.exec();
}
void MainWindow::deleteObject()
{
// Get name and type of object to delete
sqlb::ObjectIdentifier name(ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnSchema), Qt::EditRole).toString().toStdString(),
ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnName), Qt::EditRole).toString().toStdString());
QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnObjectType), Qt::EditRole).toString();
// Due to different grammar in languages (e.g. gender or declension), each message must be given separately to translation.
QString message;
if (type == "table")
message = tr("Are you sure you want to delete the table '%1'?\nAll data associated with the table will be lost.");
else if (type == "view")
message = tr("Are you sure you want to delete the view '%1'?");
else if (type == "trigger")
message = tr("Are you sure you want to delete the trigger '%1'?");
else if (type == "index")
message = tr("Are you sure you want to delete the index '%1'?");
// Ask user if he really wants to delete that table
if(QMessageBox::warning(this, QApplication::applicationName(), message.arg(QString::fromStdString(name.name())),
QMessageBox::Yes | QMessageBox::No, QMessageBox::No) == QMessageBox::Yes)
{
// Delete the table
QString statement = QString("DROP %1 %2;").arg(type.toUpper()).arg(QString::fromStdString(name.toString()));
if(!db.executeSQL(statement))
{
if (type == "table")
message = tr("Error: could not delete the table.");
else if (type == "view")
message = tr("Error: could not delete the view.");
else if (type == "trigger")
message = tr("Error: could not delete the trigger.");
else if (type == "index")
message = tr("Error: could not delete the index.");
QString error = tr("Message from database engine:\n%1").arg(db.lastError());
QMessageBox::warning(this, QApplication::applicationName(), message + " " + error);
} else {
populateTable();
changeTreeSelection();
}
}
}
void MainWindow::editObject()
{
if(!ui->dbTreeWidget->selectionModel()->hasSelection())
return;
// Get name and type of the object to edit
sqlb::ObjectIdentifier name(ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnSchema), Qt::EditRole).toString().toStdString(),
ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnName), Qt::EditRole).toString().toStdString());
QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnObjectType), Qt::EditRole).toString();
if(type == "table")
{
// For a safe and possibly complex table modification we must follow the steps documented in
// https://www.sqlite.org/lang_altertable.html
// Paragraph (first procedure): Making Other Kinds Of Table Schema Changes
QString foreign_keys = db.getPragma("foreign_keys");
if (foreign_keys == "1") {
if(db.getDirty() && QMessageBox::question(this,
QApplication::applicationName(),
tr("Editing the table requires to save all pending changes now.\nAre you sure you want to save the database?"),
QMessageBox::Save | QMessageBox::Default,
QMessageBox::Cancel | QMessageBox::Escape) != QMessageBox::Save)
return;
// Commit all changes so the foreign_keys can be effective.
fileSave();
db.setPragma("foreign_keys", "0");
}
EditTableDialog dialog(db, name, false, this);
bool ok = dialog.exec();
// If foreign_keys were enabled, we must commit or rollback the transaction so the foreign_keys pragma can be restored.
if (foreign_keys == "1") {
if (!db.querySingleValueFromDb(QString("PRAGMA %1.foreign_key_check").arg(QString::fromStdString(sqlb::escapeIdentifier(name.schema())))).isNull()) {
// Raise warning for accepted modification. When rejected, warn user also since we know now that the table has problems,
// but it wasn't our fault.
if (ok)
QMessageBox::warning(this, QApplication::applicationName(),
tr("Error checking foreign keys after table modification. The changes will be reverted."));
else
QMessageBox::warning(this, QApplication::applicationName(),
tr("This table did not pass a foreign-key check.<br/>"
"You should run 'Tools | Foreign-Key Check' and fix the reported issues."));
db.revertAll();
} else {
// Commit all changes so the foreign_keys can be effective.
fileSave();
}
db.setPragma("foreign_keys", foreign_keys);
}
if(ok) {
ui->tableBrowser->clearFilters();
populateTable();
}
} else if(type == "index") {
EditIndexDialog dialog(db, name, false, this);
if(dialog.exec())
populateTable();
}
}
void MainWindow::helpWhatsThis()
{
QWhatsThis::enterWhatsThisMode ();
}
void MainWindow::helpAbout()
{
AboutDialog dialog(this);
dialog.exec();
}
void MainWindow::updateRecordText(const QPersistentModelIndex& idx, const QByteArray& text, bool isBlob)
{
m_currentTabTableModel->setTypedData(idx, isBlob, text);
}
void MainWindow::toggleEditDock(bool visible)
{
if (!visible) {
// Update main window
ui->tableBrowser->setFocus();
} else {
// fill edit dock with actual data
editDock->setCurrentIndex(ui->tableBrowser->currentIndex());
}
}
void MainWindow::doubleClickTable(const QModelIndex& index)
{
// Cancel on invalid index
if (!index.isValid()) {
return;
}
// * Don't allow editing of other objects than tables and editable views
bool isEditingAllowed = !db.readOnly() && m_currentTabTableModel == ui->tableBrowser->model() &&
ui->tableBrowser->model()->isEditable();
// Enable or disable the Apply, Null, & Import buttons in the Edit Cell
// dock depending on the value of the "isEditingAllowed" bool above
editDock->setReadOnly(!isEditingAllowed);
editDock->setCurrentIndex(index);
// Show the edit dock
ui->dockEdit->setVisible(true);
// Set focus on the edit dock
editDock->setFocus();
}
void MainWindow::dataTableSelectionChanged(const QModelIndex& index)
{
// Cancel on invalid index
if(!index.isValid()) {
editDock->setCurrentIndex(QModelIndex());
return;
}
bool editingAllowed = !db.readOnly() && (m_currentTabTableModel == ui->tableBrowser->model()) &&
ui->tableBrowser->model()->isEditable();
// Don't allow editing of other objects than tables and editable views
editDock->setReadOnly(!editingAllowed);
// If the Edit Cell dock is visible, load the new value into it
if (editDock->isVisible()) {
editDock->setCurrentIndex(index);
}
}
/*
* I'm still not happy how the results are represented to the user
* right now you only see the result of the last executed statement.
* A better experience would be tabs on the bottom with query results
* for all the executed statements.
*/
void MainWindow::executeQuery()
{
// Make sure a database is opened. This is necessary because we allow opened SQL editor tabs even if no database is loaded. Hitting F5 or similar
// then might call this function.
if(!db.isOpen())
return;
// Check if other task is still running and stop it if necessary
if(execute_sql_worker && execute_sql_worker->isRunning())
{
// Ask the user and do nothing if he/she doesn't want to interrupt the running query
if(QMessageBox::warning(this, qApp->applicationName(),
tr("You are already executing SQL statements. Do you want to stop them in order to execute the current "
"statements instead? Note that this might leave the database in an inconsistent state."),
QMessageBox::Yes, QMessageBox::Cancel | QMessageBox::Default | QMessageBox::Escape) == QMessageBox::Cancel)
return;
// Stop the running query
execute_sql_worker->stop();
execute_sql_worker->wait();
}
// Get current SQL tab and editor
SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
SqlTextEdit* editor = sqlWidget->getEditor();
auto* current_tab = ui->tabSqlAreas->currentWidget();
const QString tabName = ui->tabSqlAreas->tabText(ui->tabSqlAreas->currentIndex()).remove('&');
// Remove any error indicators
editor->clearErrorIndicators();
// Determine execution mode: execute all, execute selection or execute current line
enum executionMode
{
All,
Selection,
Line
} mode;
if(sender() && sender()->objectName() == "actionSqlExecuteLine")
mode = Line;
else if(!sqlWidget->getSelectedSql().isEmpty())
mode = Selection;
else
mode = All;
// Get SQL code to execute. This depends on the execution mode.
int execute_from_position = 0; // Where we want to start the execution in the query string
int execute_to_position = 0; // Where we roughly want to end the execution in the query string
switch(mode)
{
case Selection:
{
// Start and end positions are start and end positions from the selection
int execute_from_line, execute_from_index, execute_to_line, execute_to_index;
editor->getSelection(&execute_from_line, &execute_from_index, &execute_to_line, &execute_to_index);
execute_from_position = editor->positionFromLineIndex(execute_from_line, execute_from_index);
execute_to_position = editor->positionFromLineIndex(execute_to_line, execute_to_index);
db.logSQL(tr("-- EXECUTING SELECTION IN '%1'\n--").arg(tabName), kLogMsg_User);
} break;
case Line:
{
// Start position is the first character of the current line, except for those cases where we're in the middle of a
// statement which started on one the previous line. In that case the start position is actually a bit earlier. For
// the end position we set the last character of the current line. If the statement(s) continue(s) into the next line,
// SQLite will execute it/them anyway and we'll stop afterwards.
int execute_from_line, dummy;
editor->getCursorPosition(&execute_from_line, &dummy);
execute_from_position = editor->positionFromLineIndex(execute_from_line, 0);
// Need to set the end position here before adjusting the start line
int execute_to_line = execute_from_line;
int execute_to_index = editor->text(execute_to_line).length() - 1; // The -1 compensates for the line break at the end of the line
execute_to_position = editor->positionFromLineIndex(execute_to_line, execute_to_index);
QByteArray firstPartEntireSQL = sqlWidget->getSql().toUtf8().left(execute_from_position);
if(firstPartEntireSQL.lastIndexOf(';') != -1)
execute_from_position -= firstPartEntireSQL.length() - firstPartEntireSQL.lastIndexOf(';') - 1;
db.logSQL(tr("-- EXECUTING LINE IN '%1'\n--").arg(tabName), kLogMsg_User);
} break;
case All:
{
// Start position is the first byte, end position the last.
// Note that we use byte positions that might differ from character positions.
execute_to_position = editor->length();
db.logSQL(tr("-- EXECUTING ALL IN '%1'\n--").arg(tabName), kLogMsg_User);
} break;
}
// Prepare a lambda function for logging the results of a query
auto query_logger = [this, sqlWidget, editor](bool ok, const QString& status_message, int from_position, int to_position) {
int execute_from_line, execute_from_index;
editor->lineIndexFromPosition(from_position, &execute_from_line, &execute_from_index);
// Special case: if the start position is at the end of a line, then move to the beggining of next line.
// Otherwise for the typical case, the line reference is one less than expected.
// Note that execute_from_index uses character positions and not byte positions, so text().length() must be used.
if (editor->text(execute_from_line).length() == execute_from_index+1) {
execute_from_line++;
execute_from_index = 0;
}
// If there was an error highlight the erroneous SQL statement
if(!ok)
{
int end_of_current_statement_line, end_of_current_statement_index;
editor->lineIndexFromPosition(to_position, &end_of_current_statement_line, &end_of_current_statement_index);
editor->setErrorIndicator(execute_from_line, execute_from_index, end_of_current_statement_line, end_of_current_statement_index);
editor->setCursorPosition(execute_from_line, execute_from_index);
}
// Log the query and the result message.
// The query takes the last placeholder as it may itself contain the sequence '%' + number.
QString query = editor->text(from_position, to_position);
QString log_message = "-- " + tr("At line %1:").arg(execute_from_line+1) + "\n" + query.trimmed() + "\n-- " + tr("Result: %1").arg(status_message);
db.logSQL(log_message, kLogMsg_User);
log_message = tr("Result: %2").arg(status_message) + "\n" + tr("At line %1:").arg(execute_from_line+1) + "\n" + query.trimmed();
// Update the execution area
sqlWidget->finishExecution(log_message, ok);
};
// Get the statement(s) to execute. When in selection mode crop the query string at exactly the end of the selection to make sure SQLite has
// no chance to execute any further.
QString sql = sqlWidget->getSql();
if(mode == Selection)
sql = sql.left(execute_to_position);
// Prepare the SQL worker to run the query. We set the context of each signal-slot connection to the current SQL execution area.
// This means that if the tab is closed all these signals are automatically disconnected so the lambdas won't be called for a not
// existing execution area.
execute_sql_worker.reset(new RunSql(db, sql, execute_from_position, execute_to_position, true));
connect(execute_sql_worker.get(), &RunSql::structureUpdated, sqlWidget, [this]() {
db.updateSchema();
}, Qt::QueuedConnection);
connect(execute_sql_worker.get(), &RunSql::statementErrored, sqlWidget, [query_logger, this, sqlWidget](const QString& status_message, int from_position, int to_position) {
sqlWidget->getModel()->reset();
ui->actionSqlResultsSave->setEnabled(false);
ui->actionSqlResultsSaveAsView->setEnabled(false);
attachPlot(sqlWidget->getTableResult(), sqlWidget->getModel());
query_logger(false, status_message, from_position, to_position);
}, Qt::QueuedConnection);
connect(execute_sql_worker.get(), &RunSql::statementExecuted, sqlWidget, [query_logger, this, sqlWidget](const QString& status_message, int from_position, int to_position) {
sqlWidget->getModel()->reset();
ui->actionSqlResultsSave->setEnabled(false);
ui->actionSqlResultsSaveAsView->setEnabled(false);
attachPlot(sqlWidget->getTableResult(), sqlWidget->getModel());
query_logger(true, status_message, from_position, to_position);
execute_sql_worker->startNextStatement();
}, Qt::QueuedConnection);
connect(execute_sql_worker.get(), &RunSql::statementReturnsRows, sqlWidget, [query_logger, this, sqlWidget](const QString& query, int from_position, int to_position, qint64 time_in_ms_so_far) {
auto time_start = std::chrono::high_resolution_clock::now();
ui->actionSqlResultsSave->setEnabled(true);
ui->actionSqlResultsSaveAsView->setEnabled(!db.readOnly());
auto * model = sqlWidget->getModel();
model->setQuery(query);
// Wait until the initial loading of data (= first chunk and row count) has been performed
auto conn = std::make_shared<QMetaObject::Connection>();
*conn = connect(model, &SqliteTableModel::finishedFetch, [=]() {
// Disconnect this connection right now. This avoids calling this slot multiple times
disconnect(*conn);
attachPlot(sqlWidget->getTableResult(), sqlWidget->getModel());
connect(sqlWidget->getTableResult()->selectionModel(), &QItemSelectionModel::currentChanged, this, &MainWindow::dataTableSelectionChanged);
connect(sqlWidget->getTableResult(), &QTableView::doubleClicked, this, &MainWindow::doubleClickTable);
auto time_end = std::chrono::high_resolution_clock::now();
auto time_in_ms = std::chrono::duration_cast<std::chrono::milliseconds>(time_end-time_start);
query_logger(true, tr("%1 rows returned in %2ms").arg(model->rowCount()).arg(time_in_ms.count()+time_in_ms_so_far), from_position, to_position);
execute_sql_worker->startNextStatement();
});
}, Qt::QueuedConnection);
connect(execute_sql_worker.get(), &RunSql::confirmSaveBeforePragmaOrVacuum, sqlWidget, [this]() {
if(QMessageBox::question(nullptr, QApplication::applicationName(),
tr("Setting PRAGMA values or vacuuming will commit your current transaction.\nAre you sure?"),
QMessageBox::Yes | QMessageBox::Default,
QMessageBox::No | QMessageBox::Escape) == QMessageBox::No)
execute_sql_worker->stop();
}, Qt::BlockingQueuedConnection);
connect(execute_sql_worker.get(), &RunSql::finished, sqlWidget, [this, current_tab, sqlWidget]() {
// We work with a pointer to the current tab here instead of its index because the user might reorder the tabs in the meantime
ui->tabSqlAreas->setTabIcon(ui->tabSqlAreas->indexOf(current_tab), QIcon());
// We don't need to check for the current SQL tab here because two concurrently running queries are not allowed
ui->actionSqlExecuteLine->setEnabled(true);
ui->actionExecuteSql->setEnabled(true);
ui->actionSqlStop->setEnabled(false);
sqlWidget->getEditor()->setReadOnly(false);
// Show Done message
if(sqlWidget->inErrorState())
sqlWidget->getStatusEdit()->setPlainText(tr("Execution finished with errors.") + "\n" + sqlWidget->getStatusEdit()->toPlainText());
else
sqlWidget->getStatusEdit()->setPlainText(tr("Execution finished without errors.") + "\n" + sqlWidget->getStatusEdit()->toPlainText());
});
// Add an hourglass icon to the current tab to indicate that there's a running execution in there.
// NOTE It's a bit hack-ish but we don't use this icon just as a signal to the user but also check for it in various places to check whether a
// specific SQL tab is currently running a query or not.
ui->tabSqlAreas->setTabIcon(ui->tabSqlAreas->currentIndex(), QIcon(":icons/hourglass"));
// Deactivate the buttons to start a query and activate the button to stop the query
ui->actionSqlExecuteLine->setEnabled(false);
ui->actionExecuteSql->setEnabled(false);
ui->actionSqlStop->setEnabled(true);
// Make the SQL editor widget read-only. We do this because the error indicators would be misplaced if the user changed the SQL text during execution
sqlWidget->getEditor()->setReadOnly(true);
// Start the execution
execute_sql_worker->start();
}
void MainWindow::mainTabSelected(int /*tabindex*/)
{
editDock->setReadOnly(true);
if(ui->mainTab->currentWidget() == ui->browser)
{
m_currentTabTableModel = ui->tableBrowser->model();
populateTable();
} else if(ui->mainTab->currentWidget() == ui->pragmas) {
loadPragmas();
} else if(ui->mainTab->currentWidget() == ui->query) {
SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if (sqlWidget) {
m_currentTabTableModel = sqlWidget->getModel();
dataTableSelectionChanged(sqlWidget->getTableResult()->currentIndex());
}
}
}
void MainWindow::importTableFromCSV()
{
QStringList file_filter;
file_filter << FILE_FILTER_CSV
<< FILE_FILTER_TSV
<< FILE_FILTER_DSV
<< FILE_FILTER_TXT
<< FILE_FILTER_ALL;
QStringList wFiles = FileDialog::getOpenFileNames(
OpenCSVFile,
this,
tr("Choose text files"),
file_filter.join(";;"));
QStringList validFiles;
for(const auto& file : wFiles) {
if (QFile::exists(file))
validFiles.append(file);
}
if (!validFiles.isEmpty())
{
ImportCsvDialog dialog(validFiles, &db, this);
if (dialog.exec())
populateTable();
}
}
void MainWindow::exportTableToCSV()
{
// Get the current table name if we are in the Browse Data tab
sqlb::ObjectIdentifier current_table;
if(ui->mainTab->currentWidget() == ui->structure)
{
QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnObjectType)).toString();
if(type == "table" || type == "view")
{
QString schema = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnSchema)).toString();
QString name = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnName)).toString();
current_table = sqlb::ObjectIdentifier(schema.toStdString(), name.toStdString());
}
} else if(ui->mainTab->currentWidget() == ui->browser) {
current_table = ui->tableBrowser->currentlyBrowsedTableName();
}
// Open dialog
ExportDataDialog dialog(db, ExportDataDialog::ExportFormatCsv, this, "", current_table);
dialog.exec();
}
void MainWindow::exportTableToJson()
{
// Get the current table name if we are in the Browse Data tab
sqlb::ObjectIdentifier current_table;
if(ui->mainTab->currentWidget() == ui->structure)
{
QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnObjectType)).toString();
if(type == "table" || type == "view")
{
QString schema = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnSchema)).toString();
QString name = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnName)).toString();
current_table = sqlb::ObjectIdentifier(schema.toStdString(), name.toStdString());
}
} else if(ui->mainTab->currentWidget() == ui->browser) {
current_table = ui->tableBrowser->currentlyBrowsedTableName();
}
// Open dialog
ExportDataDialog dialog(db, ExportDataDialog::ExportFormatJson, this, "", current_table);
dialog.exec();
}
void MainWindow::dbState( bool dirty )
{
ui->fileSaveAction->setEnabled(dirty);
ui->fileRevertAction->setEnabled(dirty);
ui->fileAttachAction->setEnabled(!dirty);
//ui->actionEncryption->setEnabled(!dirty);
}
void MainWindow::fileSave()
{
if(db.isOpen())
{
if(!db.releaseAllSavepoints())
{
QMessageBox::warning(this, QApplication::applicationName(), tr("Error while saving the database file. This means that not all changes to the database were "
"saved. You need to resolve the following error first.\n\n%1").arg(db.lastError()));
}
}
}
void MainWindow::fileRevert()
{
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.currentFile());
if(QMessageBox::question(this, QApplication::applicationName(), msg, QMessageBox::Yes | QMessageBox::Default, QMessageBox::No | QMessageBox::Escape) == QMessageBox::Yes)
{
db.revertAll();
populateTable();
}
}
}
void MainWindow::exportDatabaseToSQL()
{
QString current_table;
if(ui->mainTab->currentWidget() == ui->browser)
current_table = QString::fromStdString(ui->tableBrowser->currentlyBrowsedTableName().name());
ExportSqlDialog dialog(&db, this, current_table);
dialog.exec();
}
void MainWindow::importDatabaseFromSQL()
{
QStringList file_filter;
file_filter << FILE_FILTER_SQL
<< FILE_FILTER_TXT
<< FILE_FILTER_ALL;
// Get file name to import
QString fileName = FileDialog::getOpenFileName(
OpenSQLFile,
this,
tr("Choose a file to import"),
file_filter.join(";;"));
// Cancel when file doesn't exist
if(!QFile::exists(fileName))
return;
// If there is already a database file opened ask the user whether 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 = FileDialog::getSaveFileName(
CreateDatabaseFile,
this,
tr("Choose a filename to save under"),
FileDialog::getSqlDatabaseFileFilter());
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;
}
// Create the new file and open it in the browser
db.create(newDbFile);
closeProject();
fileOpen(newDbFile);
}
// Defer foreign keys. Just deferring them instead of disabling them should work fine because in the import we only expect CREATE and INSERT
// statements which unlike in the Edit Table dialog shouldn't trigger any problems.
QString foreignKeysOldSettings = db.getPragma("defer_foreign_keys");
db.setPragma("defer_foreign_keys", "1");
// Open, read, execute and close file
QApplication::setOverrideCursor(Qt::WaitCursor);
QFile f(fileName);
f.open(QIODevice::ReadOnly);
bool ok = db.executeMultiSQL(f.readAll(), newDbFile.size() == 0);
// Restore cursor before asking the user to accept the message
QApplication::restoreOverrideCursor();
if(!ok)
QMessageBox::warning(this, QApplication::applicationName(), tr("Error importing data: %1").arg(db.lastError()));
else if(db.getPragma("foreign_keys") == "1" && !db.querySingleValueFromDb(QString("PRAGMA foreign_key_check")).isNull())
QMessageBox::warning(this, QApplication::applicationName(), tr("Import completed. Some foreign key constraints are violated. Please fix them before saving."));
else
QMessageBox::information(this, QApplication::applicationName(), tr("Import completed."));
f.close();
// Restore the former foreign key settings
db.setPragma("defer_foreign_keys", foreignKeysOldSettings);
// Refresh views
db.updateSchema();
populateTable();
}
void MainWindow::openPreferences()
{
PreferencesDialog dialog(this);
if(dialog.exec())
reloadSettings();
}
//** Db Tree Context Menu
void MainWindow::createTreeContextMenu(const QPoint &qPoint)
{
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));
}
//** DB Schema Dock Context Menu
void MainWindow::createSchemaDockContextMenu(const QPoint &qPoint)
{
bool enable_browse_table = false;
if(ui->treeSchemaDock->selectionModel()->hasSelection())
{
QString type = ui->treeSchemaDock->model()->data(ui->treeSchemaDock->currentIndex().sibling(ui->treeSchemaDock->currentIndex().row(), DbStructureModel::ColumnObjectType), Qt::EditRole).toString();
if(type == "table" || type == "view")
enable_browse_table = true;
}
ui->actionPopupSchemaDockBrowseTable->setEnabled(enable_browse_table);
popupSchemaDockMenu->exec(ui->treeSchemaDock->mapToGlobal(qPoint));
}
void MainWindow::changeTreeSelection()
{
// Just assume first that something's selected that can not be edited at all
ui->editDeleteObjectAction->setEnabled(false);
ui->editModifyObjectAction->setEnabled(false);
ui->actionEditBrowseTable->setEnabled(false);
if(!ui->dbTreeWidget->currentIndex().isValid())
return;
// Change the text and tooltips of the actions
QString type = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 1)).toString();
if (type.isEmpty())
{
ui->editDeleteObjectAction->setIcon(QIcon(":icons/table_delete"));
ui->editModifyObjectAction->setIcon(QIcon(":icons/table_modify"));
} else {
ui->editDeleteObjectAction->setIcon(QIcon(QString(":icons/%1_delete").arg(type)));
ui->editModifyObjectAction->setIcon(QIcon(QString(":icons/%1_modify").arg(type)));
}
if (type == "view") {
ui->editDeleteObjectAction->setText(tr("Delete View"));
ui->editModifyObjectAction->setText(tr("Modify View"));
} else if(type == "trigger") {
ui->editDeleteObjectAction->setText(tr("Delete Trigger"));
ui->editModifyObjectAction->setText(tr("Modify Trigger"));
} else if(type == "index") {
ui->editDeleteObjectAction->setText(tr("Delete Index"));
ui->editModifyObjectAction->setText(tr("Modify Index"));
} else if(type == "table") {
ui->editDeleteObjectAction->setText(tr("Delete Table"));
ui->editModifyObjectAction->setText(tr("Modify Table"));
} else {
// Nothing to do for other types. Set the buttons not visible and return.
ui->editDeleteObjectAction->setVisible(false);
ui->editModifyObjectAction->setVisible(false);
return;
}
ui->editDeleteObjectAction->setVisible(true);
ui->editModifyObjectAction->setVisible(true);
// Activate actions
if(type == "table" || type == "index")
{
ui->editDeleteObjectAction->setEnabled(!db.readOnly());
ui->editModifyObjectAction->setEnabled(!db.readOnly());
} else if(type == "view" || type == "trigger") {
ui->editDeleteObjectAction->setEnabled(!db.readOnly());
}
if(type == "table" || type == "view")
{
ui->actionEditBrowseTable->setEnabled(true);
ui->actionExportCsvPopup->setEnabled(true);
}
}
void MainWindow::openRecentFile()
{
QAction *action = qobject_cast<QAction *>(sender());
if (action)
{
QString file = action->data().toString();
bool read_only = false;
if(file.startsWith("[ro]")) // Check if file is in read-only
{
file = file.mid(4);
read_only = true;
}
fileOpen(file, false, read_only);
}
}
void MainWindow::updateRecentFileActions()
{
// Get recent files list from settings
QStringList files = Settings::getValue("General", "recentFileList").toStringList();
// Check if files still exist and remove any non-existent file
for(int i=0;i<files.size();i++)
{
// Remove preceding "[ro]" mark for check
QString file = files.at(i);
if(file.startsWith("[ro]"))
file = file.mid(4);
if(!QFileInfo::exists(file))
{
files.removeAt(i);
i--;
}
}
// Store updated list
Settings::setValue("General", "recentFileList", files);
int numRecentFiles = qMin(files.size(), MaxRecentFiles);
for (int i = 0; i < numRecentFiles; ++i) {
QString file = files[i];
bool read_only = false;
if(file.startsWith("[ro]")) // Check if file is in read-only
{
file = file.mid(4);
read_only = true;
}
QString text = tr("&%1 %2%3").arg(i + 1).arg(QDir::toNativeSeparators(file)).arg(read_only ? tr(" (read only)") : "");
recentFileActs[i]->setText(text);
recentFileActs[i]->setData(files[i]);
recentFileActs[i]->setVisible(true);
// Add shortcut for opening the file using the keyboard. However, if the application is configured to store
// more than nine recently opened files don't set shortcuts for the later ones which wouldn't be single digit anymore.
if(i < 9)
recentFileActs[i]->setShortcut(QKeySequence(Qt::CTRL + (Qt::Key_1+i)));
}
for (int j = numRecentFiles; j < MaxRecentFiles; ++j)
recentFileActs[j]->setVisible(false);
recentSeparatorAct->setVisible(numRecentFiles > 0);
}
void MainWindow::setCurrentFile(const QString &fileName)
{
setWindowFilePath(fileName);
setWindowTitle(QApplication::applicationName() + " - " + QDir::toNativeSeparators(fileName));
activateFields(true);
dbState(db.getDirty());
}
void MainWindow::addToRecentFilesMenu(const QString& filename, bool read_only)
{
QFileInfo info(filename);
QString path = info.absoluteFilePath();
if(read_only)
path = "[ro]" + path;
QStringList files = Settings::getValue("General", "recentFileList").toStringList();
files.removeAll(path);
files.prepend(path);
while (files.size() > MaxRecentFiles)
files.removeLast();
Settings::setValue("General", "recentFileList", files);
for(QWidget* widget : QApplication::topLevelWidgets()) {
MainWindow *mainWin = qobject_cast<MainWindow *>(widget);
if (mainWin)
mainWin->updateRecentFileActions();
}
}
void MainWindow::dragEnterEvent(QDragEnterEvent *event)
{
if( event->mimeData()->hasFormat("text/uri-list") )
event->acceptProposedAction();
}
void MainWindow::dropEvent(QDropEvent *event)
{
QList<QUrl> urls = event->mimeData()->urls();
if( urls.isEmpty() )
return;
QString fileName = urls.first().toLocalFile();
if(!fileName.isEmpty()) {
// If there is no open database, the only possible option is to open the file.
if (!db.isOpen()) {
fileOpen(fileName);
return;
}
bool ok;
const QString open = tr("Open Database or Project");
const QString attach = tr("Attach Database...");
const QString import = tr("Import CSV file(s)...");
QString action = QInputDialog::getItem(this,
qApp->applicationName(),
tr("Select the action to apply to the dropped file(s). <br/>"
"Note: only 'Import' will process more than one file.", "", urls.count()),
{open, attach, import},
0,
false,
&ok);
if(ok) {
if (action == open) {
fileOpen(fileName);
} else if (action == attach) {
fileAttach(fileName);
} else if (action == import) {
QStringList validFiles;
for(const auto& url : urls) {
if (QFile::exists(url.toLocalFile()))
validFiles.append(url.toLocalFile());
}
ImportCsvDialog dialog(validFiles, &db, this);
if (dialog.exec())
populateTable();
}
}
}
}
void MainWindow::activateFields(bool enable)
{
bool write = !db.readOnly();
bool tempDb = db.currentFile() == ":memory:";
ui->tableBrowser->setEnabled(enable);
ui->fileCloseAction->setEnabled(enable);
ui->fileAttachAction->setEnabled(enable);
ui->fileCompactAction->setEnabled(enable && write);
ui->fileExportJsonAction->setEnabled(enable);
ui->fileExportCSVAction->setEnabled(enable);
ui->fileExportSQLAction->setEnabled(enable);
ui->fileImportCSVAction->setEnabled(enable && write);
ui->editCreateTableAction->setEnabled(enable && write);
ui->editCreateIndexAction->setEnabled(enable && write);
ui->actionDbPrint->setEnabled(enable);
ui->scrollAreaWidgetContents->setEnabled(enable);
ui->buttonBoxPragmas->setEnabled(enable && write);
ui->actionExecuteSql->setEnabled(enable);
ui->actionLoadExtension->setEnabled(enable);
ui->actionSqlExecuteLine->setEnabled(enable);
ui->actionSaveProject->setEnabled(enable && !tempDb);
ui->actionSaveProjectAs->setEnabled(enable && !tempDb);
ui->actionSaveAll->setEnabled(enable && !tempDb);
ui->actionEncryption->setEnabled(enable && write && !tempDb);
ui->actionIntegrityCheck->setEnabled(enable);
ui->actionQuickCheck->setEnabled(enable);
ui->actionForeignKeyCheck->setEnabled(enable);
ui->actionOptimize->setEnabled(enable);
ui->dockEdit->setEnabled(enable);
ui->dockPlot->setEnabled(enable);
if(!enable)
ui->actionSqlResultsSave->setEnabled(false);
remoteDock->enableButtons();
}
void MainWindow::resizeEvent(QResizeEvent*)
{
ui->tableBrowser->updateRecordsetLabel();
}
void MainWindow::loadPragmas()
{
pragmaValues.autovacuum = db.getPragma("auto_vacuum").toInt();
pragmaValues.automatic_index = db.getPragma("automatic_index").toInt();
pragmaValues.checkpoint_fullsync = db.getPragma("checkpoint_fullfsync").toInt();
pragmaValues.foreign_keys = db.getPragma("foreign_keys").toInt();
pragmaValues.fullfsync = db.getPragma("fullfsync").toInt();
pragmaValues.ignore_check_constraints = db.getPragma("ignore_check_constraints").toInt();
pragmaValues.journal_mode = db.getPragma("journal_mode").toUpper();
pragmaValues.journal_size_limit = db.getPragma("journal_size_limit").toInt();
pragmaValues.locking_mode = db.getPragma("locking_mode").toUpper();
pragmaValues.max_page_count = db.getPragma("max_page_count").toInt();
pragmaValues.page_size = db.getPragma("page_size").toInt();
pragmaValues.recursive_triggers = db.getPragma("recursive_triggers").toInt();
pragmaValues.secure_delete = db.getPragma("secure_delete").toInt();
pragmaValues.synchronous = db.getPragma("synchronous").toInt();
pragmaValues.temp_store = db.getPragma("temp_store").toInt();
pragmaValues.user_version = db.getPragma("user_version").toInt();
pragmaValues.wal_autocheckpoint = db.getPragma("wal_autocheckpoint").toInt();
pragmaValues.case_sensitive_like = db.getPragma("case_sensitive_like").toInt();
updatePragmaUi();
}
void MainWindow::updatePragmaUi()
{
ui->comboboxPragmaAutoVacuum->setCurrentIndex(pragmaValues.autovacuum);
ui->checkboxPragmaAutomaticIndex->setChecked(pragmaValues.automatic_index);
ui->checkboxPragmaCheckpointFullFsync->setChecked(pragmaValues.checkpoint_fullsync);
ui->checkboxPragmaForeignKeys->setChecked(pragmaValues.foreign_keys);
ui->checkboxPragmaFullFsync->setChecked(pragmaValues.fullfsync);
ui->checkboxPragmaIgnoreCheckConstraints->setChecked(pragmaValues.ignore_check_constraints);
ui->comboboxPragmaJournalMode->setCurrentIndex(ui->comboboxPragmaJournalMode->findText(pragmaValues.journal_mode, Qt::MatchFixedString));
ui->spinPragmaJournalSizeLimit->setValue(pragmaValues.journal_size_limit);
ui->comboboxPragmaLockingMode->setCurrentIndex(ui->comboboxPragmaLockingMode->findText(pragmaValues.locking_mode, Qt::MatchFixedString));
ui->spinPragmaMaxPageCount->setValue(pragmaValues.max_page_count);
ui->comboPragmaPageSize->setCurrentIndex(ui->comboPragmaPageSize->findText(QString::number(pragmaValues.page_size), Qt::MatchFixedString));
ui->checkboxPragmaRecursiveTriggers->setChecked(pragmaValues.recursive_triggers);
ui->checkboxPragmaSecureDelete->setChecked(pragmaValues.secure_delete);
ui->comboboxPragmaSynchronous->setCurrentIndex(pragmaValues.synchronous);
ui->comboboxPragmaTempStore->setCurrentIndex(pragmaValues.temp_store);
ui->spinPragmaUserVersion->setValue(pragmaValues.user_version);
ui->spinPragmaWalAutoCheckpoint->setValue(pragmaValues.wal_autocheckpoint);
ui->checkboxPragmaCaseSensitiveLike->setChecked(pragmaValues.case_sensitive_like);
}
void MainWindow::savePragmas()
{
if( db.getDirty() )
{
QString msg = tr("Setting PRAGMA values will commit your current transaction.\nAre you sure?");
if(QMessageBox::question(this, QApplication::applicationName(), msg, QMessageBox::Yes | QMessageBox::Default, QMessageBox::No | QMessageBox::Escape) == QMessageBox::No)
{
return; // abort
}
}
db.setPragma("auto_vacuum", ui->comboboxPragmaAutoVacuum->currentIndex(), pragmaValues.autovacuum);
db.setPragma("automatic_index", ui->checkboxPragmaAutomaticIndex->isChecked(), pragmaValues.automatic_index);
db.setPragma("checkpoint_fullfsync", ui->checkboxPragmaCheckpointFullFsync->isChecked(), pragmaValues.checkpoint_fullsync);
db.setPragma("foreign_keys", ui->checkboxPragmaForeignKeys->isChecked(), pragmaValues.foreign_keys);
db.setPragma("fullfsync", ui->checkboxPragmaFullFsync->isChecked(), pragmaValues.fullfsync);
db.setPragma("ignore_check_constraints", ui->checkboxPragmaIgnoreCheckConstraints->isChecked(), pragmaValues.ignore_check_constraints);
db.setPragma("journal_mode", ui->comboboxPragmaJournalMode->currentText().toUpper(), pragmaValues.journal_mode);
db.setPragma("journal_size_limit", ui->spinPragmaJournalSizeLimit->value(), pragmaValues.journal_size_limit);
db.setPragma("locking_mode", ui->comboboxPragmaLockingMode->currentText().toUpper(), pragmaValues.locking_mode);
db.setPragma("max_page_count", ui->spinPragmaMaxPageCount->value(), pragmaValues.max_page_count);
db.setPragma("page_size", ui->comboPragmaPageSize->currentText().toInt(), pragmaValues.page_size);
db.setPragma("recursive_triggers", ui->checkboxPragmaRecursiveTriggers->isChecked(), pragmaValues.recursive_triggers);
db.setPragma("secure_delete", ui->checkboxPragmaSecureDelete->isChecked(), pragmaValues.secure_delete);
db.setPragma("synchronous", ui->comboboxPragmaSynchronous->currentIndex(), pragmaValues.synchronous);
db.setPragma("temp_store", ui->comboboxPragmaTempStore->currentIndex(), pragmaValues.temp_store);
db.setPragma("user_version", ui->spinPragmaUserVersion->value(), pragmaValues.user_version);
db.setPragma("wal_autocheckpoint", ui->spinPragmaWalAutoCheckpoint->value(), pragmaValues.wal_autocheckpoint);
db.setPragma("case_sensitive_like", ui->checkboxPragmaCaseSensitiveLike->isChecked(), pragmaValues.case_sensitive_like);
isProjectModified = true;
updatePragmaUi();
}
void MainWindow::logSql(const QString& sql, int msgtype)
{
if(msgtype == kLogMsg_User)
{
ui->editLogUser->append(sql + "\n");
ui->editLogUser->verticalScrollBar()->setValue(ui->editLogUser->verticalScrollBar()->maximum());
} else if(msgtype == kLogMsg_App) {
ui->editLogApplication->append(sql + "\n");
ui->editLogApplication->verticalScrollBar()->setValue(ui->editLogApplication->verticalScrollBar()->maximum());
} else if(msgtype == kLogMsg_ErrorLog) {
ui->editLogErrorLog->append(sql + "\n");
ui->editLogErrorLog->verticalScrollBar()->setValue(ui->editLogErrorLog->verticalScrollBar()->maximum());
}
}
// Ask user to save the buffer in the specified tab index.
// ignoreUnattachedBuffers is used to store answer about buffers not linked to files, so user is only asked once about them.
// Return true unless user wants to cancel the invoking action.
bool MainWindow::askSaveSqlTab(int index, bool& ignoreUnattachedBuffers)
{
SqlExecutionArea* sqlExecArea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(index));
if(sqlExecArea->getEditor()->isModified()) {
if(sqlExecArea->fileName().isEmpty() && !ignoreUnattachedBuffers) {
// Once the project is saved, remaining SQL tabs will not be modified, so this is only expected to be asked once.
QString message = currentProjectFilename.isEmpty() ?
tr("Do you want to save the changes made to SQL tabs in a new project file?") :
tr("Do you want to save the changes made to SQL tabs in the project file %1?").
arg(QFileInfo(currentProjectFilename).fileName());
QMessageBox::StandardButton reply = QMessageBox::question(nullptr,
QApplication::applicationName(),
message,
QMessageBox::Save | QMessageBox::No | QMessageBox::Cancel);
switch(reply) {
case QMessageBox::Save:
saveProject();
break;
case QMessageBox::Cancel:
return false;
default:
ignoreUnattachedBuffers = true;
break;
}
} else if(!sqlExecArea->fileName().isEmpty()) {
QMessageBox::StandardButton reply =
QMessageBox::question(nullptr,
QApplication::applicationName(),
tr("Do you want to save the changes made to the SQL file %1?").
arg(QFileInfo(sqlExecArea->fileName()).fileName()),
QMessageBox::Save | QMessageBox::No | QMessageBox::Cancel);
switch(reply) {
case QMessageBox::Save:
saveSqlFile(index);
break;
case QMessageBox::Cancel:
return false;
default:
break;
}
}
}
return true;
}
void MainWindow::closeSqlTab(int index, bool force)
{
// Don't close last tab
if(ui->tabSqlAreas->count() == 1 && !force)
return;
// Check if we're still executing statements from this tab and stop them before proceeding
if(!ui->tabSqlAreas->tabIcon(index).isNull())
{
if(QMessageBox::warning(this, qApp->applicationName(), tr("The statements in this tab are still executing. Closing the tab will stop the "
"execution. This might leave the database in an inconsistent state. Are you sure "
"you want to close the tab?"),
QMessageBox::Yes,
QMessageBox::Cancel | QMessageBox::Default | QMessageBox::Escape) == QMessageBox::Cancel)
return;
execute_sql_worker->stop();
execute_sql_worker->wait();
}
// Ask for saving and comply with cancel answer.
bool ignoreUnattachedBuffers = false;
if (!askSaveSqlTab(index, ignoreUnattachedBuffers))
return;
// Remove the tab and delete the widget
QWidget* w = ui->tabSqlAreas->widget(index);
ui->tabSqlAreas->removeTab(index);
delete w;
}
int MainWindow::openSqlTab(bool resetCounter)
{
static int tabNumber = 0;
if(resetCounter)
tabNumber = 0;
// Create new tab, add it to the tab widget and select it
SqlExecutionArea* w = new SqlExecutionArea(db, this);
int index = ui->tabSqlAreas->addTab(w, QString("SQL %1").arg(++tabNumber));
ui->tabSqlAreas->setCurrentIndex(index);
w->setFindFrameVisibility(ui->actionSqlFind->isChecked());
// Disable the find dialog in the SQL tabs, since the shortcut
// would interfere with the search bar and it'd be anyway redundant.
w->getEditor()->setEnabledFindDialog(false);
w->getEditor()->setFocus();
connect(w, &SqlExecutionArea::findFrameVisibilityChanged, ui->actionSqlFind, &QAction::setChecked);
// Connect now the find shortcut to the editor with widget context, so it isn't ambiguous with other Scintilla Widgets.
QShortcut* shortcutFind = new QShortcut(ui->actionSqlFind->shortcut(), w->getEditor(), nullptr, nullptr, Qt::WidgetShortcut);
connect(shortcutFind, &QShortcut::activated, ui->actionSqlFind, &QAction::toggle);
return index;
}
void MainWindow::changeSqlTab(int index)
{
// Instead of figuring out if there are some execution results in the new tab and which statement was used to generate them,
// we just disable the export buttons in the toolbar.
ui->actionSqlResultsSave->setEnabled(false);
// Check if the new tab is currently running a query or not
if(ui->tabSqlAreas->tabIcon(index).isNull())
{
// Not running a query
ui->actionSqlExecuteLine->setEnabled(db.isOpen());
ui->actionExecuteSql->setEnabled(db.isOpen());
ui->actionSqlStop->setEnabled(false);
} else {
// Running a query
ui->actionSqlExecuteLine->setEnabled(false);
ui->actionExecuteSql->setEnabled(false);
ui->actionSqlStop->setEnabled(true);
}
}
void MainWindow::openSqlFile()
{
QString file = FileDialog::getOpenFileName(
OpenSQLFile,
this,
tr("Select SQL file to open"),
tr("Text files(*.sql *.txt);;All files(*)"));
if(QFile::exists(file))
{
// Decide whether to open a new tab or take the current one
int index;
SqlExecutionArea* current_tab = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if(current_tab && current_tab->getSql().isEmpty() && current_tab->getModel()->rowCount() == 0)
index = ui->tabSqlAreas->currentIndex();
else
index = openSqlTab();
SqlExecutionArea* sqlarea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(index));
sqlarea->openFile(file);
QFileInfo fileinfo(file);
ui->tabSqlAreas->setTabText(index, fileinfo.fileName());
}
}
void MainWindow::saveSqlFile(int tabIndex)
{
SqlExecutionArea* sqlarea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(tabIndex));
if(!sqlarea)
return;
// If this SQL file hasn't been saved before open the Save As dialog. Otherwise just use the old file name for saving
if(sqlarea->fileName().isEmpty())
{
saveSqlFileAs();
} else {
sqlarea->saveFile(sqlarea->fileName());
}
}
void MainWindow::saveSqlFile()
{
saveSqlFile(ui->tabSqlAreas->currentIndex());
}
void MainWindow::saveSqlFileAs()
{
SqlExecutionArea* sqlarea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if(!sqlarea)
return;
QStringList file_filter;
file_filter << FILE_FILTER_SQL
<< FILE_FILTER_TXT
<< FILE_FILTER_ALL;
QString file = FileDialog::getSaveFileName(
CreateSQLFile,
this,
tr("Select file name"),
file_filter.join(";;"));
if(!file.isEmpty())
{
sqlarea->saveFile(file);
QFileInfo fileinfo(file);
ui->tabSqlAreas->setTabText(ui->tabSqlAreas->currentIndex(), fileinfo.fileName());
}
}
void MainWindow::saveSqlResultsAsCsv()
{
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget())->saveAsCsv();
}
void MainWindow::saveSqlResultsAsView()
{
saveAsView(qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget())->getModel()->query());
}
void MainWindow::loadExtension()
{
QStringList file_filter;
file_filter << FILE_FILTER_DYN
<< FILE_FILTER_ALL;
QString file = FileDialog::getOpenFileName(
OpenExtensionFile,
this,
tr("Select extension file"),
file_filter.join(";;"));
if(file.isEmpty())
return;
if(db.loadExtension(file))
QMessageBox::information(this, QApplication::applicationName(), tr("Extension successfully loaded."));
else
QMessageBox::warning(this, QApplication::applicationName(), tr("Error loading extension: %1").arg(db.lastError()));
}
void MainWindow::reloadSettings()
{
// Set data browser font
ui->tableBrowser->reloadSettings();
switch (static_cast<Settings::AppStyle>(Settings::getValue("General", "appStyle").toInt())) {
case Settings::FollowDesktopStyle :
qApp->setStyleSheet("");
break;
case Settings::DarkStyle :
QFile f(":qdarkstyle/style.qss");
if (!f.exists()) {
QMessageBox::warning(this, qApp->applicationName(),
tr("Could not open find resource file: %1").arg(f.fileName()));
} else {
f.open(QFile::ReadOnly | QFile::Text);
QTextStream ts(&f);
qApp->setStyleSheet(ts.readAll());
}
break;
}
setToolButtonStyle(static_cast<Qt::ToolButtonStyle>(Settings::getValue("General", "toolbarStyle").toInt()));
ui->dbToolbar->setToolButtonStyle(static_cast<Qt::ToolButtonStyle>(Settings::getValue("General", "toolbarStyleStructure").toInt()));
ui->toolbarSql->setToolButtonStyle(static_cast<Qt::ToolButtonStyle>(Settings::getValue("General", "toolbarStyleSql").toInt()));
// Set prefetch sizes for lazy population of table models
for(int i=0;i<ui->tabSqlAreas->count();++i)
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i))->reloadSettings();
// Prepare log font
QFont logfont("Monospace");
logfont.setStyleHint(QFont::TypeWriter);
logfont.setPointSize(Settings::getValue("log", "fontsize").toInt());
// Set font for SQL logs and edit dialog
ui->editLogApplication->reloadSettings();
ui->editLogUser->reloadSettings();
ui->editLogErrorLog->reloadSettings();
ui->editLogApplication->setFont(logfont);
ui->editLogUser->setFont(logfont);
ui->editLogErrorLog->setFont(logfont);
editDock->reloadSettings();
// Load extensions
db.loadExtensionsFromSettings();
// Refresh view
dbStructureModel->reloadData();
populateStructure();
populateTable();
// Hide or show the remote dock as needed
bool showRemoteActions = Settings::getValue("remote", "active").toBool();
ui->viewMenu->actions().at(4)->setVisible(showRemoteActions);
if(!showRemoteActions)
ui->dockRemote->setHidden(true);
// Update the remote database connection settings
m_remoteDb->reloadSettings();
// Reload remote dock settings
remoteDock->reloadSettings();
sqlb::setIdentifierQuoting(static_cast<sqlb::escapeQuoting>(Settings::getValue("editor", "identifier_quotes").toInt()));
}
void MainWindow::checkNewVersion(const QString& versionstring, const QString& url)
{
// versionstring contains a major.minor.patch version string
QStringList versiontokens = versionstring.split(".");
if(versiontokens.size() < 3)
return;
int major = versiontokens[0].toInt();
int minor = versiontokens[1].toInt();
int patch = versiontokens[2].toInt();
bool newversion = false;
if(major > MAJOR_VERSION)
newversion = true;
else if(major == MAJOR_VERSION)
{
if(minor > MINOR_VERSION)
newversion = true;
else if(minor == MINOR_VERSION)
{
if(patch > PATCH_VERSION)
newversion = true;
}
}
if(newversion)
{
int ignmajor = Settings::getValue("checkversion", "ignmajor").toInt();
int ignminor = Settings::getValue("checkversion", "ignminor").toInt();
int ignpatch = Settings::getValue("checkversion", "ignpatch").toInt();
// check if the user doesn't care about the current update
if(!(ignmajor == major && ignminor == minor && ignpatch == patch))
{
QMessageBox msgBox;
QPushButton *idontcarebutton = msgBox.addButton(tr("Don't show again"), QMessageBox::ActionRole);
msgBox.addButton(QMessageBox::Ok);
msgBox.setTextFormat(Qt::RichText);
msgBox.setWindowTitle(tr("New version available."));
msgBox.setText(tr("A new DB Browser for SQLite version is available (%1.%2.%3).<br/><br/>"
"Please download at <a href='%4'>%4</a>.").arg(major).arg(minor).arg(patch).
arg(url));
msgBox.exec();
if(msgBox.clickedButton() == idontcarebutton)
{
// save that the user don't want to get bothered about this update
Settings::setValue("checkversion", "ignmajor", major);
Settings::setValue("checkversion", "ignminor", minor);
Settings::setValue("checkversion", "ignpatch", patch);
}
}
}
}
void MainWindow::on_actionWiki_triggered()
{
QDesktopServices::openUrl(QUrl("https://github.com/sqlitebrowser/sqlitebrowser/wiki"));
}
// 'Help | Bug Report...' link will set an appropiate body, add the system information and set the label 'bug' automatically to the issue
void MainWindow::on_actionBug_report_triggered()
{
const QString version = Application::versionString();
const QString os = QSysInfo::prettyProductName();
const QString kernelType = QSysInfo::kernelType();
const QString kernelVersion = QSysInfo::kernelVersion();
const QString arch = QSysInfo::currentCpuArchitecture();
const QString built_for = QSysInfo::buildAbi();
QString sqlite_version, sqlcipher_version;
DBBrowserDB::getSqliteVersion(sqlite_version, sqlcipher_version);
if(sqlcipher_version.isNull())
sqlite_version = QString("SQLite Version ") + sqlite_version;
else
sqlite_version = QString("SQLCipher Version ") + sqlcipher_version + QString(" (based on SQLite %1)").arg(sqlite_version);
const QString body =
QString("Details for the issue\n"
"--------------------\n\n"
"#### What did you do?\n\n\n"
"#### What did you expect to see?\n\n\n"
"#### What did you see instead?\n\n\n"
"Useful extra information\n"
"-------------------------\n"
"> DB4S v%1 [built for %2] on %3 (%4/%5) [%6]\n"
"> using %7\n"
"> and Qt %8")
.arg(version, built_for, os, kernelType, kernelVersion, arch, sqlite_version, QT_VERSION_STR);
QUrlQuery query;
query.addQueryItem("labels", "bug");
query.addQueryItem("body", body);
QUrl url("https://github.com/sqlitebrowser/sqlitebrowser/issues/new");
url.setQuery(query);
QDesktopServices::openUrl(url);
}
// 'Help | Feature Request...' link will set an appropiate body and add the label 'enhancement' automatically to the issue
void MainWindow::on_actionFeature_Request_triggered()
{
QUrlQuery query;
// Add the label enhancement and use the Feature request template that
// we have in GitHub.
query.addQueryItem("labels", "enhancement");
query.addQueryItem("template", "Feature_request.md");
QUrl url("https://github.com/sqlitebrowser/sqlitebrowser/issues/new");
url.setQuery(query);
QDesktopServices::openUrl(url);
}
void MainWindow::on_actionSqlCipherFaq_triggered()
{
QDesktopServices::openUrl(QUrl("https://discuss.zetetic.net/c/sqlcipher/sqlcipher-faq"));
}
void MainWindow::on_actionWebsite_triggered()
{
QDesktopServices::openUrl(QUrl("https://sqlitebrowser.org"));
}
void MainWindow::on_actionDonatePatreon_triggered()
{
QDesktopServices::openUrl(QUrl("https://www.patreon.com/bePatron?u=11578749"));
}
static void loadBrowseDataTableSettings(BrowseDataTableSettings& settings, QXmlStreamReader& xml)
{
// TODO Remove this in the near future. This file format was only created temporarily by the nightlies from the late 3.11 development period.
if(xml.attributes().hasAttribute("sort_order_index"))
{
int sortOrderIndex = xml.attributes().value("sort_order_index").toInt();
Qt::SortOrder sortOrderMode = static_cast<Qt::SortOrder>(xml.attributes().value("sort_order_mode").toInt());
settings.query.setOrderBy(toSortOrderVector(sortOrderIndex, sortOrderMode));
}
settings.showRowid = xml.attributes().value("show_row_id").toInt();
settings.encoding = xml.attributes().value("encoding").toString();
settings.plotXAxis = xml.attributes().value("plot_x_axis").toString();
settings.unlockViewPk = xml.attributes().value("unlock_view_pk").toString();
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "table") {
if(xml.name() == "sort")
{
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "sort")
{
if(xml.name() == "column")
{
int index = xml.attributes().value("index").toInt();
int mode = xml.attributes().value("mode").toInt();
settings.query.orderBy().emplace_back(index, mode == Qt::AscendingOrder ? sqlb::Ascending : sqlb::Descending);
xml.skipCurrentElement();
}
}
} else if(xml.name() == "column_widths") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "column_widths") {
if (xml.name() == "column") {
int index = xml.attributes().value("index").toInt();
settings.columnWidths[index] = xml.attributes().value("value").toInt();
xml.skipCurrentElement();
}
}
} else if(xml.name() == "filter_values") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "filter_values") {
if (xml.name() == "column") {
int index = xml.attributes().value("index").toInt();
settings.filterValues[index] = xml.attributes().value("value").toString();
xml.skipCurrentElement();
}
}
} else if(xml.name() == "conditional_formats") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "conditional_formats") {
if (xml.name() == "column") {
int index = xml.attributes().value("index").toInt();
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "column") {
if(xml.name() == "format") {
settings.condFormats[index].emplace_back(xml.attributes().value("condition").toString(),
QColor(xml.attributes().value("foreground").toString()),
QColor(xml.attributes().value("background").toString()),
settings.encoding);
xml.skipCurrentElement();
}
}
}
}
} else if(xml.name() == "display_formats") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "display_formats") {
if (xml.name() == "column") {
int index = xml.attributes().value("index").toInt();
settings.displayFormats[index] = xml.attributes().value("value").toString();
xml.skipCurrentElement();
}
}
} else if(xml.name() == "hidden_columns") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "hidden_columns") {
if (xml.name() == "column") {
int index = xml.attributes().value("index").toInt();
settings.hiddenColumns[index] = xml.attributes().value("value").toInt();
xml.skipCurrentElement();
}
}
} else if(xml.name() == "plot_y_axes") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "plot_y_axes") {
QString yAxisName;
PlotDock::PlotSettings yAxisSettings;
if (xml.name() == "y_axis") {
yAxisName = xml.attributes().value("name").toString();
yAxisSettings.lineStyle = xml.attributes().value("line_style").toInt();
yAxisSettings.pointShape = xml.attributes().value("point_shape").toInt();
yAxisSettings.colour = QColor (xml.attributes().value("colour").toString());
yAxisSettings.active = xml.attributes().value("active").toInt();
xml.skipCurrentElement();
}
settings.plotYAxes[yAxisName] = yAxisSettings;
}
}
}
}
bool MainWindow::loadProject(QString filename, bool readOnly)
{
// Show the open file dialog when no filename was passed as parameter
if(filename.isEmpty())
{
filename = FileDialog::getOpenFileName(
OpenProjectFile,
this,
tr("Choose a project file to open"),
tr("DB Browser for SQLite project file (*.sqbpro)"));
}
if(!filename.isEmpty())
{
QFile file(filename);
file.open(QFile::ReadOnly | QFile::Text);
QXmlStreamReader xml(&file);
xml.readNext(); // token == QXmlStreamReader::StartDocument
xml.readNext(); // name == sqlb_project
if(xml.name() != "sqlb_project")
return false;
isProjectModified = false;
addToRecentFilesMenu(filename, readOnly);
QString currentTable;
while(!xml.atEnd() && !xml.hasError())
{
// Read next token
QXmlStreamReader::TokenType token = xml.readNext();
// Handle element start
if(token == QXmlStreamReader::StartElement)
{
if(xml.name() == "db")
{
// Read only?
if(xml.attributes().hasAttribute("readonly") && xml.attributes().value("readonly").toInt())
readOnly = true;
// DB file
QString dbfilename = xml.attributes().value("path").toString();
if(!QFile::exists(dbfilename)) {
dbfilename = QFileInfo(filename).absolutePath() + QDir::separator() + dbfilename;
// New DB filename is pending to be saved
isProjectModified = true;
}
fileOpen(dbfilename, true, readOnly);
ui->dbTreeWidget->collapseAll();
// PRAGMAs
if(xml.attributes().hasAttribute("foreign_keys"))
db.setPragma("foreign_keys", xml.attributes().value("foreign_keys").toString());
if(xml.attributes().hasAttribute("case_sensitive_like"))
db.setPragma("case_sensitive_like", xml.attributes().value("case_sensitive_like").toString());
if(xml.attributes().hasAttribute("temp_store"))
db.setPragma("temp_store", xml.attributes().value("temp_store").toString());
if(xml.attributes().hasAttribute("wal_autocheckpoint"))
db.setPragma("wal_autocheckpoint", xml.attributes().value("wal_autocheckpoint").toString());
if(xml.attributes().hasAttribute("synchronous"))
db.setPragma("synchronous", xml.attributes().value("synchronous").toString());
loadPragmas();
} else if(xml.name() == "attached") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "attached")
{
if(xml.name() == "db")
{
db.attach(xml.attributes().value("path").toString(), xml.attributes().value("schema").toString());
xml.skipCurrentElement();
}
}
} else if(xml.name() == "window") {
// Window settings
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "window")
{
if(xml.name() == "main_tabs") {
// Currently open tabs
restoreOpenTabs(xml.attributes().value("open").toString());
// Currently selected open tab
ui->mainTab->setCurrentIndex(xml.attributes().value("current").toString().toInt());
xml.skipCurrentElement();
} else if(xml.name() == "current_tab") {
// Currently selected tab (3.11 or older format, first restore default open tabs)
restoreOpenTabs(defaultOpenTabs);
ui->mainTab->setCurrentIndex(xml.attributes().value("id").toString().toInt());
xml.skipCurrentElement();
}
}
} else if(xml.name() == "tab_structure") {
// Database Structure tab settings
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "tab_structure")
{
if(xml.name() == "column_width")
{
// Tree view column widths
ui->dbTreeWidget->setColumnWidth(xml.attributes().value("id").toString().toInt(),
xml.attributes().value("width").toString().toInt());
xml.skipCurrentElement();
} else if(xml.name() == "expanded_item") {
// Tree view expanded items
int parent = xml.attributes().value("parent").toString().toInt();
QModelIndex idx;
if(parent == -1)
idx = ui->dbTreeWidget->model()->index(xml.attributes().value("id").toString().toInt(), 0);
else
idx = ui->dbTreeWidget->model()->index(xml.attributes().value("id").toString().toInt(), 0, ui->dbTreeWidget->model()->index(parent, 0));
ui->dbTreeWidget->expand(idx);
xml.skipCurrentElement();
}
}
} else if(xml.name() == "tab_browse") {
// Browse Data tab settings
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "tab_browse")
{
if(xml.name() == "current_table")
{
// Currently selected table
currentTable = xml.attributes().value("name").toString();
xml.skipCurrentElement();
} else if(xml.name() == "default_encoding") {
// Default text encoding
ui->tableBrowser->setDefaultEncoding(xml.attributes().value("codec").toString());
xml.skipCurrentElement();
} else if(xml.name() == "browsetable_info") {
// This tag is only found in old project files. In newer versions (>= 3.11) it is replaced by a new implementation.
// We still support loading it though we might decide to drop that support later. But for now we show a warning to the
// user when loading an old file.
if(!Settings::getValue("idontcare", "projectBrowseTable").toBool())
{
QMessageBox msgBox;
QPushButton* idontcarebutton = msgBox.addButton(tr("Don't show again"), QMessageBox::ActionRole);
msgBox.addButton(QMessageBox::Ok);
msgBox.setTextFormat(Qt::RichText);
msgBox.setWindowTitle(qApp->applicationName());
msgBox.setText(tr("This project file is using an old file format because it was created using DB Browser for SQLite "
"version 3.10 or lower. Loading this file format is still fully supported but we advice you to convert "
"all your project files to the new file format because support for older formats might be dropped "
"at some point in the future. You can convert your files by simply opening and re-saving them."));
msgBox.exec();
if(msgBox.clickedButton() == idontcarebutton)
Settings::setValue("idontcare", "projectBrowseTable", true);
}
QString attrData = xml.attributes().value("data").toString();
QByteArray temp = QByteArray::fromBase64(attrData.toUtf8());
QDataStream stream(temp);
QMap<sqlb::ObjectIdentifier, BrowseDataTableSettings> settings;
stream >> settings;
for(auto it=settings.begin();it!=settings.end();++it)
ui->tableBrowser->setSettings(it.key(), it.value());
xml.skipCurrentElement();
} else if(xml.name() == "browse_table_settings") {
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "browse_table_settings") {
if (xml.name() == "table") {
sqlb::ObjectIdentifier tableIdentifier =
sqlb::ObjectIdentifier (xml.attributes().value("schema").toString().toStdString(),
xml.attributes().value("name").toString().toStdString());
BrowseDataTableSettings settings;
loadBrowseDataTableSettings(settings, xml);
ui->tableBrowser->setSettings(tableIdentifier, settings);
}
}
}
}
} else if(xml.name() == "tab_sql") {
// Close all open tabs first
for(int i=ui->tabSqlAreas->count()-1;i>=0;i--)
closeSqlTab(i, true);
// Execute SQL tab data
while(xml.readNext() != QXmlStreamReader::EndElement && xml.name() != "tab_sql")
{
if(xml.name() == "sql")
{
// SQL editor tab
int index = openSqlTab();
ui->tabSqlAreas->setTabText(index, xml.attributes().value("name").toString());
SqlTextEdit* sqlEditor = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(index))->getEditor();
sqlEditor->setText(xml.readElementText());
sqlEditor->setModified(false);
} else if(xml.name() == "current_tab") {
// Currently selected tab
ui->tabSqlAreas->setCurrentIndex(xml.attributes().value("id").toString().toInt());
xml.skipCurrentElement();
}
}
}
}
}
file.close();
currentProjectFilename = filename;
if(ui->mainTab->currentWidget() == ui->browser) {
if (!currentTable.isEmpty())
{
sqlb::ObjectIdentifier obj;
if(!obj.fromSerialised(currentTable.toStdString()))
{
// This is an old project file format which doesn't yet contain serialised table identifiers. This means
// we have to try our best to unserialise this one manually. The only problem is when the name of an
// attached database or of a table contains a dot character. In that case the name becomes ambigious and
// we just try to split it at the first dot. I don't think it affects many (if any) project files. But if
// it turn out to be wrong, we can always add a loop here which checks for any possible combination of schema
// and table name whether an object with that combination exists.
// TODO: Delete this code in the future when we don't expect there to be any project files in the old format anymore.
if(currentTable.contains('.'))
{
obj.setSchema(currentTable.left(currentTable.indexOf('.')).toStdString());
obj.setName(currentTable.mid(currentTable.indexOf('.')+1).toStdString());
} else {
obj.setName(currentTable.toStdString());
}
}
switchToBrowseDataTab(obj);
}
populateTable(); // Refresh view
}
return !xml.hasError();
} else {
// No project was opened
return false;
}
}
static void saveDbTreeState(const QTreeView* tree, QXmlStreamWriter& xml, QModelIndex index = QModelIndex(), int parent_row = -1)
{
for(int i=0;i<tree->model()->rowCount(index);i++)
{
if(tree->isExpanded(tree->model()->index(i, 0, index)))
{
xml.writeStartElement("expanded_item");
xml.writeAttribute("id", QString::number(i));
xml.writeAttribute("parent", QString::number(parent_row));
xml.writeEndElement();
}
saveDbTreeState(tree, xml, tree->model()->index(i, 0, index), i);
}
}
static void saveBrowseDataTableSettings(const BrowseDataTableSettings& object, QXmlStreamWriter& xml)
{
xml.writeAttribute("show_row_id", QString::number(object.showRowid));
xml.writeAttribute("encoding", object.encoding);
xml.writeAttribute("plot_x_axis", object.plotXAxis);
xml.writeAttribute("unlock_view_pk", object.unlockViewPk);
xml.writeStartElement("sort");
for(const auto& column : object.query.orderBy())
{
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(column.column));
xml.writeAttribute("mode", QString::number(column.direction));
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("column_widths");
for(auto iter=object.columnWidths.constBegin(); iter!=object.columnWidths.constEnd(); ++iter) {
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(iter.key()));
xml.writeAttribute("value", QString::number(iter.value()));
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("filter_values");
for(auto iter=object.filterValues.constBegin(); iter!=object.filterValues.constEnd(); ++iter) {
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(iter.key()));
xml.writeAttribute("value", iter.value());
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("conditional_formats");
for(auto iter=object.condFormats.constBegin(); iter!=object.condFormats.constEnd(); ++iter) {
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(iter.key()));
for(auto format : iter.value()) {
xml.writeStartElement("format");
xml.writeAttribute("condition", format.filter());
xml.writeAttribute("background", format.backgroundColor().name());
xml.writeAttribute("foreground", format.foregroundColor().name());
xml.writeEndElement();
}
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("display_formats");
for(auto iter=object.displayFormats.constBegin(); iter!=object.displayFormats.constEnd(); ++iter) {
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(iter.key()));
xml.writeAttribute("value", iter.value());
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("hidden_columns");
for(auto iter=object.hiddenColumns.constBegin(); iter!=object.hiddenColumns.constEnd(); ++iter) {
xml.writeStartElement("column");
xml.writeAttribute("index", QString::number(iter.key()));
xml.writeAttribute("value", QString::number(iter.value()));
xml.writeEndElement();
}
xml.writeEndElement();
xml.writeStartElement("plot_y_axes");
for(auto iter=object.plotYAxes.constBegin(); iter!=object.plotYAxes.constEnd(); ++iter) {
PlotDock::PlotSettings plotSettings = iter.value();
xml.writeStartElement("y_axis");
xml.writeAttribute("name", iter.key());
xml.writeAttribute("line_style", QString::number(plotSettings.lineStyle));
xml.writeAttribute("point_shape", QString::number(plotSettings.pointShape));
xml.writeAttribute("colour", plotSettings.colour.name());
xml.writeAttribute("active", QString::number(plotSettings.active));
xml.writeEndElement();
}
xml.writeEndElement();
}
QString MainWindow::saveProject(const QString& currentFilename)
{
QString filename;
if(currentFilename.isEmpty()) {
QString basePathName = db.currentFile();
// Remove database suffix
basePathName.chop(QFileInfo(basePathName).suffix().size()+1);
filename = FileDialog::getSaveFileName(
CreateProjectFile,
this,
tr("Choose a filename to save under"),
FILE_FILTER_SQLPRJ,
basePathName);
} else
filename = currentFilename;
if(!filename.isEmpty())
{
// Make sure the file has got a .sqbpro ending
if(!filename.endsWith(FILE_EXT_SQLPRJ_DEFAULT, Qt::CaseInsensitive))
filename.append(FILE_EXT_SQLPRJ_DEFAULT);
QFile file(filename);
bool opened = file.open(QFile::WriteOnly | QFile::Text);
if(!opened) {
QMessageBox::warning(this, qApp->applicationName(),
tr("Could not open project file for writing.\nReason: %1").arg(file.errorString()));
return QString();
}
QApplication::setOverrideCursor(Qt::WaitCursor);
QXmlStreamWriter xml(&file);
xml.writeStartDocument();
xml.writeStartElement("sqlb_project");
// Database file name
xml.writeStartElement("db");
xml.writeAttribute("path", db.currentFile());
xml.writeAttribute("readonly", QString::number(db.readOnly()));
xml.writeAttribute("foreign_keys", db.getPragma("foreign_keys"));
xml.writeAttribute("case_sensitive_like", db.getPragma("case_sensitive_like"));
xml.writeAttribute("temp_store", db.getPragma("temp_store"));
xml.writeAttribute("wal_autocheckpoint", db.getPragma("wal_autocheckpoint"));
xml.writeAttribute("synchronous", db.getPragma("synchronous"));
xml.writeEndElement();
// Attached databases
xml.writeStartElement("attached");
QString sql("PRAGMA database_list;");
db.logSQL(sql, kLogMsg_App);
sqlite3_stmt* db_vm;
if(sqlite3_prepare_v2(db.get("project").get(), sql.toUtf8(), sql.toUtf8().length(), &db_vm, nullptr) == SQLITE_OK)
{
while(sqlite3_step(db_vm) == SQLITE_ROW)
{
QString schema(QString::fromUtf8(reinterpret_cast<const char*>(sqlite3_column_text(db_vm, 1))));
if(schema != "main" && schema != "temp")
{
QString path(QString::fromUtf8(reinterpret_cast<const char*>(sqlite3_column_text(db_vm, 2))));
xml.writeStartElement("db");
xml.writeAttribute("schema", schema);
xml.writeAttribute("path", path);
xml.writeEndElement();
}
}
sqlite3_finalize(db_vm);
}
xml.writeEndElement();
// Window settings
xml.writeStartElement("window");
xml.writeStartElement("main_tabs"); // Currently open tabs
xml.writeAttribute("open", saveOpenTabs());
xml.writeAttribute("current", QString::number(ui->mainTab->currentIndex()));
xml.writeEndElement();
xml.writeEndElement();
// Database Structure tab settings
xml.writeStartElement("tab_structure");
for(int i=0;i<ui->dbTreeWidget->model()->columnCount();i++) // Widths of tree view columns
{
xml.writeStartElement("column_width");
xml.writeAttribute("id", QString::number(i));
xml.writeAttribute("width", QString::number(ui->dbTreeWidget->columnWidth(i)));
xml.writeEndElement();
}
saveDbTreeState(ui->dbTreeWidget, xml); // Expanded tree items
xml.writeEndElement();
// Browse Data tab settings
xml.writeStartElement("tab_browse");
xml.writeStartElement("current_table"); // Currently selected table
xml.writeAttribute("name", QString::fromStdString(ui->tableBrowser->currentlyBrowsedTableName().toSerialised()));
xml.writeEndElement();
xml.writeStartElement("default_encoding"); // Default encoding for text stored in tables
xml.writeAttribute("codec", ui->tableBrowser->defaultEncoding());
xml.writeEndElement();
xml.writeStartElement("browse_table_settings");
const auto settings = ui->tableBrowser->allSettings();
for(auto tableIt=settings.constBegin(); tableIt!=settings.constEnd(); ++tableIt) {
xml.writeStartElement("table");
xml.writeAttribute("schema", QString::fromStdString(tableIt.key().schema()));
xml.writeAttribute("name", QString::fromStdString(tableIt.key().name()));
saveBrowseDataTableSettings(tableIt.value(), xml);
xml.writeEndElement();
}
// </browse_table_settings>
xml.writeEndElement();
// </tab_browse>
xml.writeEndElement();
// Execute SQL tab data
xml.writeStartElement("tab_sql");
for(int i=0;i<ui->tabSqlAreas->count();i++) // All SQL tabs content
{
SqlExecutionArea* sqlArea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i));
xml.writeStartElement("sql");
xml.writeAttribute("name", ui->tabSqlAreas->tabText(i));
xml.writeCharacters(sqlArea->getSql());
sqlArea->getEditor()->setModified(false);
xml.writeEndElement();
}
xml.writeStartElement("current_tab"); // Currently selected tab
xml.writeAttribute("id", QString::number(ui->tabSqlAreas->currentIndex()));
xml.writeEndElement();
xml.writeEndElement();
xml.writeEndElement();
xml.writeEndDocument();
file.close();
addToRecentFilesMenu(filename);
isProjectModified = false;
QApplication::restoreOverrideCursor();
}
return filename;
}
void MainWindow::saveProject()
{
currentProjectFilename = saveProject(currentProjectFilename);
}
void MainWindow::saveProjectAs()
{
currentProjectFilename = saveProject(QString());
}
void MainWindow::fileAttach(const QString& fileName)
{
QString file;
if (fileName.isEmpty()) {
// Get file name of database to attach
file = FileDialog::getOpenFileName(
OpenDatabaseFile,
this,
tr("Choose a database file"),
FileDialog::getSqlDatabaseFileFilter());
} else
file = fileName;
if(!QFile::exists(file))
return;
// Attach it
db.attach(file);
isProjectModified = true;
}
void MainWindow::editEncryption()
{
#ifdef ENABLE_SQLCIPHER
CipherDialog cipherDialog(this, true);
if(cipherDialog.exec())
{
// Show progress dialog even though we can't provide any detailed progress information but this
// process might take some time.
QProgressDialog progress(this);
progress.setCancelButton(nullptr);
progress.setWindowModality(Qt::ApplicationModal);
progress.show();
qApp->processEvents();
// Apply all unsaved changes
bool ok = db.releaseAllSavepoints();
qApp->processEvents();
// Create the new file first or it won't work
if(ok)
{
QFile file(db.currentFile() + ".enctemp");
file.open(QFile::WriteOnly);
file.close();
}
CipherSettings cipherSettings = cipherDialog.getCipherSettings();
// Attach a new database using the new settings
qApp->processEvents();
if(ok)
ok = db.executeSQL(QString("ATTACH DATABASE '%1' AS sqlitebrowser_edit_encryption KEY %2;").arg(db.currentFile() + ".enctemp").arg(cipherSettings.getPassword()),
false, false);
qApp->processEvents();
if(ok)
ok = db.executeSQL(QString("PRAGMA sqlitebrowser_edit_encryption.cipher_page_size = %1").arg(cipherSettings.getPageSize()), false, false);
if(ok)
ok = db.executeSQL(QString("PRAGMA sqlitebrowser_edit_encryption.cipher_hmac_algorithm = %1").arg(cipherSettings.getHmacAlgorithm()), false, false);
if(ok)
ok = db.executeSQL(QString("PRAGMA sqlitebrowser_edit_encryption.cipher_kdf_algorithm = %1").arg(cipherSettings.getKdfAlgorithm()), false, false);
if(ok)
ok = db.executeSQL(QString("PRAGMA sqlitebrowser_edit_encryption.kdf_iter = %1").arg(cipherSettings.getKdfIterations()), false, false);
// Export the current database to the new one
qApp->processEvents();
if(ok)
ok = db.executeSQL("SELECT sqlcipher_export('sqlitebrowser_edit_encryption');", false, false);
// Set user version of the new database
qApp->processEvents();
if (ok)
ok = db.executeSQL(QString("PRAGMA sqlitebrowser_edit_encryption.user_version = %1;").arg(db.getPragma("user_version").toInt()), false, false);
// We need to detach the database before proceeding
qApp->processEvents();
if (ok)
ok = db.executeSQL("DETACH sqlitebrowser_edit_encryption;", false, false);
// Check for errors
qApp->processEvents();
if(ok)
{
// No errors: Then close the current database, switch names, open the new one and if that succeeded delete the old one
fileClose();
QFile::rename(db.currentFile(), db.currentFile() + ".enctempold");
QFile::rename(db.currentFile() + ".enctemp", db.currentFile());
if(fileOpen(db.currentFile()))
QFile::remove(db.currentFile() + ".enctempold");
} else {
QMessageBox::warning(this, qApp->applicationName(), db.lastError());
}
}
#endif
}
void MainWindow::switchToBrowseDataTab(sqlb::ObjectIdentifier tableToBrowse)
{
// If no table name was provided get the currently selected table fromt he structure tab
if(tableToBrowse.isEmpty())
{
// Cancel here if there is no selection
if(!ui->dbTreeWidget->selectionModel()->hasSelection())
return;
tableToBrowse.setSchema(ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnSchema), Qt::EditRole).toString().toStdString());
tableToBrowse.setName(ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), DbStructureModel::ColumnName), Qt::EditRole).toString().toStdString());
}
ui->tableBrowser->setCurrentTable(tableToBrowse);
if (ui->mainTab->indexOf(ui->browser) == -1)
ui->mainTab->addTab(ui->browser, ui->browser->accessibleName());
ui->mainTab->setCurrentWidget(ui->browser);
}
void MainWindow::copyCurrentCreateStatement()
{
// Cancel if no field is currently selected
if(!ui->dbTreeWidget->selectionModel()->hasSelection())
return;
// Get the CREATE statement from the Schema column
QString stmt = ui->dbTreeWidget->model()->data(ui->dbTreeWidget->currentIndex().sibling(ui->dbTreeWidget->currentIndex().row(), 3), Qt::EditRole).toString();
// Copy the statement to the global application clipboard
QApplication::clipboard()->setText(stmt);
}
void MainWindow::fileOpenReadOnly()
{
// Redirect to 'standard' fileOpen(), with the read only flag set
fileOpen(QString(), false, true);
}
void MainWindow::requestCollation(const QString& name, int eTextRep)
{
QMessageBox::StandardButton reply = QMessageBox::question(
this,
tr("Collation needed! Proceed?"),
tr("A table in this database requires a special collation function '%1' "
"that this application can't provide without further knowledge.\n"
"If you choose to proceed, be aware bad things can happen to your database.\n"
"Create a backup!").arg(name), QMessageBox::Yes | QMessageBox::No);
if(reply == QMessageBox::Yes) {
auto pDb = db.get(tr("creating collation"));
sqlite3_create_collation(pDb.get(), name.toUtf8(), eTextRep, nullptr, collCompare);
}
}
void MainWindow::renameSqlTab(int index)
{
QString new_name = QInputDialog::getText(this,
qApp->applicationName(),
tr("Set a new name for the SQL tab. Use the '&&' character to allow using the following character as a keyboard shortcut."),
QLineEdit::EchoMode::Normal,
ui->tabSqlAreas->tabText(index));
if(!new_name.isNull()) // Don't do anything if the Cancel button was clicked
ui->tabSqlAreas->setTabText(index, new_name);
}
void MainWindow::setFindFrameVisibility(bool show)
{
// Set the find frame visibility for all tabs, but leave the
// current as the last, to retain there the focus.
for(int i=0;i<ui->tabSqlAreas->count();i++)
if (i != ui->tabSqlAreas->currentIndex())
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i))->setFindFrameVisibility(show);
if (ui->tabSqlAreas->count()>0)
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget())->setFindFrameVisibility(show);
}
void MainWindow::openFindReplaceDialog()
{
// The slot for the shortcut must discover which sqltexedit widget has the focus and then open its dialog.
SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if (sqlWidget)
sqlWidget->getEditor()->openFindReplaceDialog();
}
void MainWindow::toggleSqlBlockComment()
{
// The slot for the shortcut must discover which sqltexedit widget has the focus
SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if (sqlWidget)
sqlWidget->getEditor()->toggleBlockComment();
}
void MainWindow::openSqlPrintDialog()
{
// The slot for the shortcut must discover which sqltexedit widget has the focus and then open its dialog.
SqlExecutionArea* sqlWidget = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->currentWidget());
if (sqlWidget)
sqlWidget->getEditor()->openPrintDialog();
}
void MainWindow::saveAsView(QString query)
{
// Let the user select a name for the new view and make sure it doesn't already exist
QString name;
while(true)
{
name = QInputDialog::getText(this, qApp->applicationName(), tr("Please specify the view name")).trimmed();
if(name.isNull())
return;
if(db.getObjectByName(sqlb::ObjectIdentifier("main", name.toStdString())) != nullptr)
QMessageBox::warning(this, qApp->applicationName(), tr("There is already an object with that name. Please choose a different name."));
else
break;
}
// Create the view
if(db.executeSQL(QString("CREATE VIEW %1 AS %2;").arg(sqlb::escapeIdentifier(name)).arg(query)))
QMessageBox::information(this, qApp->applicationName(), tr("View successfully created."));
else
QMessageBox::warning(this, qApp->applicationName(), tr("Error creating view: %1").arg(db.lastError()));
}
void MainWindow::runSqlNewTab(const QString& query, const QString& title)
{
QString message = tr("This action will open a new SQL tab for running:") +
QString("<br/><tt>%1</tt><p/>").arg(query) +
tr("Press Help for opening the corresponding SQLite reference page.");
QString windowTitle = title;
windowTitle.remove('&');
switch (QMessageBox::information(this, windowTitle, message, QMessageBox::Ok | QMessageBox::Default, QMessageBox::Cancel | QMessageBox::Escape, QMessageBox::Help))
{
case QMessageBox::Ok: {
if (ui->mainTab->indexOf(ui->query) == -1)
ui->mainTab->addTab(ui->query, ui->query->accessibleName());
ui->mainTab->setCurrentWidget(ui->query);
int index = openSqlTab();
ui->tabSqlAreas->setTabText(index, title);
qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(index))->getEditor()->setText(query);
executeQuery();
break;
}
case QMessageBox::Help: {
QString anchor = query.toLower();
anchor.replace(" ", "_").chop(1);
QDesktopServices::openUrl(QUrl(QString("https://www.sqlite.org/pragma.html#") + anchor));
break;
}
default:
return;
}
}
void MainWindow::printDbStructure ()
{
const QTreeView* treeView = ui->dbTreeWidget;
const QAbstractItemModel* model = treeView->model();
const int rowCount = model->rowCount(treeView->rootIndex());
const int columnCount = model->columnCount(treeView->rootIndex());
QString strStream;
QTextStream out(&strStream);
out << "<html><meta http-equiv=\"Content-Type\" content=\"text/html; charset=utf-8\">"
<< QString("<title>%1</title>").arg(treeView->windowTitle())
<< "<style type=\"text/css\">pre {white-space: pre-wrap;}</style></head><body bgcolor=\"#FFFFFF\">";
for (int row = 0; row < rowCount; row++) {
QModelIndex headerIndex = model->index(row, 0, treeView->rootIndex());
QString strData = model->data(headerIndex).toString().toHtmlEscaped();
out << QString("<h1>%1</h1>").arg(strData);
// Open a new table for each group of objects
out << "<table border=1 cellspacing=0 cellpadding=2><thead><tr bgcolor=\"#F0F0F0\">";
for (int column = 0; column < columnCount; column++) {
// Headers
if (!treeView->isColumnHidden(column))
out << QString("<th>%1</th>").arg(model->headerData(column, Qt::Horizontal).toString().toHtmlEscaped());
}
out << "</tr></thead>";
for (int column = 0; column < columnCount; column++) {
QModelIndex groupIndex = model->index(row, column, treeView->rootIndex());
// A row for the object name
for (int rowChild = 0; rowChild < model->rowCount(groupIndex); rowChild++) {
QModelIndex objectIndex = model->index(rowChild, column, groupIndex);
out << "<tr>";
for (int column2 = 0; column2 < columnCount; column2++) {
if (!treeView->isColumnHidden(column2)) {
QModelIndex cellIndex = model->index(rowChild, column2, groupIndex);
QString header_data = model->data(cellIndex).toString().toHtmlEscaped();
if (column2 != DbStructureModel::ColumnSQL)
out << QString("<td><h2>%1</h2></td>").arg((!header_data.isEmpty()) ? header_data : QString("&nbsp;"));
else
out << QString("<td><pre>%1</pre></td>").arg((!header_data.isEmpty()) ? header_data : QString("&nbsp;"));
}
}
out << "</tr>";
// One row for each object's fields
for (int rowChild2 = 0; rowChild2 < model->rowCount(objectIndex); rowChild2++) {
out << "<tr>";
for (int column2 = 0; column2 < columnCount; column2++) {
if (!treeView->isColumnHidden(column2)) {
QModelIndex fieldIndex = model->index(rowChild2, column2, objectIndex);
QString field_data = model->data(fieldIndex).toString().toHtmlEscaped();
out << QString("<td>%1</td>").arg((!field_data.isEmpty()) ? field_data : QString("&nbsp;"));
}
}
out << "</tr>";
}
}
}
out << "</table>";
}
out << "</body></html>";
QTextDocument *document = new QTextDocument();
document->setHtml(strStream);
QPrinter printer;
printer.setDocName(treeView->windowTitle());
QPrintPreviewDialog *dialog = new QPrintPreviewDialog(&printer);
connect(dialog, &QPrintPreviewDialog::paintRequested, [&](QPrinter *previewPrinter) {
document->print(previewPrinter);
});
dialog->exec();
delete dialog;
delete document;
}
void MainWindow::updateDatabaseBusyStatus(bool busy, const QString& user)
{
statusBusyLabel->setText(tr("Busy (%1)").arg(user));
statusBusyLabel->setVisible(busy);
statusStopButton->setVisible(busy);
}
void MainWindow::closeTab(int index)
{
ui->mainTab->removeTab(index);
}
void MainWindow::toggleTabVisible(QWidget* tabWidget, bool show)
{
if (show)
ui->mainTab->addTab(tabWidget, tabWidget->accessibleName());
else
ui->mainTab->removeTab(ui->mainTab->indexOf(tabWidget));
}
void MainWindow::restoreOpenTabs(QString tabs)
{
// Split the tab list, skiping the empty parts so the empty string turns to an empty list
// and not a list of one empty string.
QStringList tabList = tabs.split(' ', QString::SkipEmptyParts);
// Clear the tabs and then add them in the order specified by the setting.
// Use the accessibleName attribute for restoring the tab label.
if (!tabList.isEmpty()) {
// Avoid flickering while clearing and adding tabs.
ui->mainTab->setUpdatesEnabled(false);
ui->mainTab->clear();
for (QString objectName : tabList) {
for (QWidget* widget : {ui->structure, ui->browser, ui->pragmas, ui->query})
if (widget->objectName() == objectName) {
ui->mainTab->addTab(widget, widget->accessibleName());
break;
}
}
ui->mainTab->setUpdatesEnabled(true);
// Force the update of the View menu toggable entries
// (it doesn't seem to be a better way)
ui->mainTab->tabCloseRequested(-1);
}
}
QString MainWindow::saveOpenTabs()
{
QString openTabs;
for (int i=0; i < ui->mainTab->count(); i++)
openTabs.append(ui->mainTab->widget(i)->objectName() + ' ');
openTabs.chop(1);
return openTabs;
}
void MainWindow::showStatusMessage5s(QString message)
{
ui->statusbar->showMessage(message, 5000);
}
void MainWindow::saveAll()
{
for(int i=0; i<ui->tabSqlAreas->count(); i++) {
SqlExecutionArea* sqlExecArea = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(i));
if(sqlExecArea->getEditor()->isModified() && !sqlExecArea->fileName().isEmpty())
saveSqlFile(i);
}
if(!currentProjectFilename.isEmpty())
saveProject();
fileSave();
}
void MainWindow::showContextMenuSqlTabBar(const QPoint& pos)
{
// Don't show context menu if the mouse click was outside of all the tabs
int tab = ui->tabSqlAreas->tabBar()->tabAt(pos);
if(tab == -1)
return;
// Prepare all menu actions
QAction* actionRename = new QAction(this);
actionRename->setText(tr("Rename Tab"));
connect(actionRename, &QAction::triggered, [this, tab]() {
renameSqlTab(tab);
});
QAction* actionDuplicate = new QAction(this);
actionDuplicate->setText(tr("Duplicate Tab"));
connect(actionDuplicate, &QAction::triggered, [this, tab]() {
QString tab_name = ui->tabSqlAreas->tabText(tab).remove("&").remove(QRegExp(" \\(\\d+\\)$"));
QString new_tab_name;
for(int i=1;;i++)
{
new_tab_name = tab_name + QString(" (%1)").arg(i);
bool name_already_exists = false;
for(int j=0;j<ui->tabSqlAreas->count();j++)
{
if(ui->tabSqlAreas->tabText(j).remove("&") == new_tab_name)
{
name_already_exists = true;
break;
}
}
if(!name_already_exists)
break;
}
int new_tab = openSqlTab();
ui->tabSqlAreas->setTabText(new_tab, new_tab_name);
SqlExecutionArea* old_area = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(tab));
SqlExecutionArea* new_area = qobject_cast<SqlExecutionArea*>(ui->tabSqlAreas->widget(new_tab));
new_area->setSql(old_area->getSql());
});
QAction* actionClose = new QAction(this);
actionClose->setText(tr("Close Tab"));
connect(actionClose, &QAction::triggered, [this, tab]() {
closeSqlTab(tab);
});
// Show menu
QMenu* menuTabs = new QMenu(this);
menuTabs->addAction(actionRename);
menuTabs->addAction(actionDuplicate);
menuTabs->addAction(actionClose);
menuTabs->exec(ui->tabSqlAreas->mapToGlobal(pos));
}