260 lines
7 KiB
C++
260 lines
7 KiB
C++
#include "DatabaseWindow.h"
|
|
#include "util.h"
|
|
#include "OpenDatabase.h"
|
|
#include "MasterController.h"
|
|
#include "TaskExecutor.h"
|
|
#include <QAction>
|
|
#include <QApplication>
|
|
#include <QFileDialog>
|
|
#include <QMenuBar>
|
|
#include <QMessageBox>
|
|
#include <QMetaMethod>
|
|
#include <QStandardPaths>
|
|
#include <QTableView>
|
|
|
|
#include "EditTableWidget.h"
|
|
#include "CodeGenerator.h"
|
|
#include "QueryTool.h"
|
|
|
|
namespace pg = Pgsql;
|
|
|
|
|
|
DatabaseWindow::DatabaseWindow(MasterController *master, QWidget *parent)
|
|
: QMainWindow(parent)
|
|
, m_masterController(master)
|
|
{
|
|
connect(&loadWatcher, &QFutureWatcher<LoadCatalog::Result>::finished,
|
|
this, &DatabaseWindow::catalogLoaded);
|
|
|
|
m_tabWidget = new QTabWidget(this);
|
|
setCentralWidget(m_tabWidget);
|
|
|
|
createActions();
|
|
initMenus();
|
|
|
|
QMetaObject::connectSlotsByName(this);
|
|
}
|
|
|
|
DatabaseWindow::~DatabaseWindow() = default;
|
|
|
|
void DatabaseWindow::addPage(QWidget* page, QString caption)
|
|
{
|
|
m_tabWidget->addTab(page, caption);
|
|
m_tabWidget->setCurrentWidget(page);
|
|
}
|
|
|
|
void DatabaseWindow::setTabCaptionForWidget(QWidget *widget, const QString &caption, const QString &hint)
|
|
{
|
|
auto index = m_tabWidget->indexOf(widget);
|
|
m_tabWidget->setTabText(index, caption);
|
|
m_tabWidget->setTabToolTip(index, hint);
|
|
}
|
|
|
|
void DatabaseWindow::setTabIcon(QWidget *widget, const QString &iconname)
|
|
{
|
|
auto index = m_tabWidget->indexOf(widget);
|
|
auto n = ":/icons/16x16/" + iconname;
|
|
m_tabWidget->setTabIcon(index, QIcon(n));
|
|
}
|
|
|
|
void DatabaseWindow::newCreateTablePage()
|
|
{
|
|
auto w = new EditTableWidget(m_database, this);
|
|
m_tabWidget->addTab(w, "Create table");
|
|
}
|
|
|
|
void DatabaseWindow::newCodeGenPage(QString query, std::shared_ptr<const Pgsql::Result> dbres)
|
|
{
|
|
// TODO should this call be this direct or should it go through module system
|
|
// yes it should otherwise context cannot properly setup toolbar and menu!!!
|
|
// auto cgtab = new CodeGenerator(context(), pluginModule(), this);
|
|
// cgtab->Init(m_database->catalog(), query, dbres);
|
|
// addPage(cgtab, "Codegen");
|
|
//
|
|
}
|
|
|
|
void DatabaseWindow::setConfig(const ConnectionConfig &config)
|
|
{
|
|
m_config = config;
|
|
try {
|
|
QString title = "pglab - ";
|
|
title += m_config.name().c_str();
|
|
setWindowTitle(title);
|
|
|
|
auto f = TaskExecutor::run(new LoadCatalog(m_config));
|
|
loadWatcher.setFuture(f);
|
|
|
|
} catch (std::runtime_error &ex) {
|
|
QMessageBox::critical(this, "Error reading database",
|
|
QString::fromUtf8(ex.what()));
|
|
|
|
close();
|
|
}
|
|
}
|
|
|
|
|
|
void DatabaseWindow::createActions()
|
|
{
|
|
{
|
|
QIcon icon;
|
|
icon.addFile(QString::fromUtf8(":/icons/about.png"), QSize(), QIcon::Normal, QIcon::On);
|
|
actionAbout = new QAction(icon, tr("About"), this);
|
|
}
|
|
{
|
|
QIcon icon;
|
|
icon.addFile(QString::fromUtf8(":/icons/page_white_delete.png"), QSize(), QIcon::Normal, QIcon::On);
|
|
auto action = actionClose = new QAction(icon, tr("Close"), this);
|
|
action->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_W));
|
|
}
|
|
{
|
|
QIcon icon;
|
|
icon.addFile(QString::fromUtf8(":/icons/new_query_tab.png"), QSize(), QIcon::Normal, QIcon::On);
|
|
auto action = actionNewSql = new QAction(icon, tr("New Query"), this);
|
|
action->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_N));
|
|
}
|
|
{
|
|
QIcon icon;
|
|
icon.addFile(QString::fromUtf8(":/icons/folder.png"), QSize(), QIcon::Normal, QIcon::On);
|
|
auto action = actionOpenSql = new QAction(icon, tr("Open Query"), this);
|
|
action->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_O));
|
|
}
|
|
|
|
// {
|
|
// auto ca = makeContextAction<QueryTool>(tr("Save SQL"), &QueryTool::save);
|
|
// ca->setShortcut(QKeySequence(Qt::CTRL + Qt::Key_S));
|
|
// ca->setMenuLocation(MenuPath("File/Save"));
|
|
// ca->setToolbarLocation(ToolbarLocation("main", "save"));
|
|
// // how we tell the system we want this to become a menu button with this as it's default action
|
|
// registerContextAction(ca);
|
|
// }
|
|
// {
|
|
// auto ca = makeContextAction<QueryTool>(tr("Save SQL as"), &QueryTool::saveAs);
|
|
// ca->setMenuLocation(MenuPath("File/Save"));
|
|
// ca->setToolbarLocation(ToolbarLocation("main", "save"));
|
|
// // how we tell the system we want this to become a secondary action for the previous button?
|
|
// registerContextAction(ca);
|
|
// }
|
|
// {
|
|
// auto ca = makeContextAction<QueryTool>(tr("Save copy of SQL as"), &QueryTool::saveCopyAs);
|
|
// ca->setMenuLocation(MenuPath("File/Save"));
|
|
// ca->setToolbarLocation(ToolbarLocation("main", "save"));
|
|
// // how we tell the system we want this to become a secondary action for the previous button?
|
|
// registerContextAction(ca);
|
|
// }
|
|
|
|
}
|
|
|
|
void DatabaseWindow::initMenus()
|
|
{
|
|
// TODO construct menubar
|
|
auto seperator = new QAction(this);
|
|
seperator->setSeparator(true);
|
|
|
|
auto mb = new QMenuBar(this);
|
|
menuFile = mb->addMenu(tr("File"));
|
|
menuFile->addActions({
|
|
actionNewSql,
|
|
actionOpenSql,
|
|
seperator,
|
|
seperator,
|
|
actionClose
|
|
});
|
|
|
|
menuHelp = mb->addMenu(tr("Help"));
|
|
menuHelp->addActions({
|
|
seperator,
|
|
actionAbout
|
|
});
|
|
|
|
|
|
|
|
|
|
setMenuBar(mb);
|
|
}
|
|
|
|
|
|
void DatabaseWindow::on_actionClose_triggered()
|
|
{
|
|
m_tabWidget->tabCloseRequested(m_tabWidget->currentIndex());
|
|
}
|
|
|
|
void DatabaseWindow::on_actionNewSql_triggered()
|
|
{
|
|
auto *ct = new QueryTool(m_database, this);
|
|
addPage(ct, "");
|
|
ct->newdoc();
|
|
}
|
|
|
|
void DatabaseWindow::on_actionOpenSql_triggered()
|
|
{
|
|
QString home_dir = QStandardPaths::locate(QStandardPaths::HomeLocation, "", QStandardPaths::LocateDirectory);
|
|
QString file_name = QFileDialog::getOpenFileName(this,
|
|
tr("Open sql query"), home_dir, tr("SQL files (*.sql *.txt)"));
|
|
if ( ! file_name.isEmpty()) {
|
|
auto *ct = new QueryTool(m_database, this);
|
|
addPage(ct, "");
|
|
if (!ct->load(file_name)) {
|
|
// TODO load has failed remove widget or never add it?
|
|
}
|
|
}
|
|
}
|
|
|
|
void DatabaseWindow::catalogLoaded()
|
|
{
|
|
try {
|
|
m_database = loadWatcher.future().result();
|
|
|
|
for (auto f : { "user", "pg_catalog", "information_schema" }) {
|
|
// TODO open inspector windows
|
|
}
|
|
|
|
|
|
newCreateTablePage();
|
|
} catch (const OpenDatabaseException &ex) {
|
|
QMessageBox::critical(this, "Error reading database", ex.text());
|
|
close();
|
|
}
|
|
}
|
|
|
|
void DatabaseWindow::on_actionAbout_triggered()
|
|
{
|
|
QMessageBox::about(this, "pgLab 0.1", tr(
|
|
"Copyrights 2016-2019, Eelke Klein, All Rights Reserved.\n"
|
|
"\n"
|
|
"The program is provided AS IS with NO WARRANTY OF ANY KIND, "
|
|
"INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS "
|
|
"FOR A PARTICULAR PURPOSE.\n"
|
|
"\n"
|
|
"This program is dynamically linked with Qt 5.12 Copyright (C) 2018 "
|
|
"The Qt Company Ltd. https://www.qt.io/licensing/. \n"
|
|
"\n"
|
|
"Icons by fatcow http://www.fatcow.com/free-icons provided under Creative Commons "
|
|
"attribution 3.0 license."
|
|
));
|
|
|
|
}
|
|
|
|
void DatabaseWindow::on_actionShow_connection_manager_triggered()
|
|
{
|
|
m_masterController->showConnectionManager();
|
|
}
|
|
|
|
void DatabaseWindow::on_actionCopy_triggered()
|
|
{
|
|
// What should be copied?
|
|
|
|
QWidget *w = QApplication::focusWidget();
|
|
QTableView *tv = dynamic_cast<QTableView*>(w);
|
|
if (tv) {
|
|
copySelectionToClipboard(tv);
|
|
}
|
|
else {
|
|
const QMetaObject *meta = w->metaObject();
|
|
int i = meta->indexOfSlot("copy");
|
|
if (i != -1) {
|
|
QMetaMethod method = meta->method(i);
|
|
method.invoke(w, Qt::AutoConnection);
|
|
}
|
|
}
|
|
}
|
|
|