19564e033c
Idiomatic C++11 code. Amends 2690822428deec4f0c08f4d118d69a7c6036369e, which, however, inherited the issue from older code. Pick-to: 6.9 6.8 Change-Id: Iba1fb9874bd93b4a560c33e3ecf62ecaa96d8bda Reviewed-by: Volker Hilsheimer <volker.hilsheimer@qt.io>
131 lines
3.7 KiB
C++
131 lines
3.7 KiB
C++
// Copyright (C) 2016 The Qt Company Ltd.
|
|
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
|
|
|
|
#include "connectionwidget.h"
|
|
|
|
#include <QAction>
|
|
#include <QHeaderView>
|
|
#include <QSqlDatabase>
|
|
#include <QTreeWidget>
|
|
#include <QVBoxLayout>
|
|
|
|
ConnectionWidget::ConnectionWidget(QWidget *parent)
|
|
: QWidget(parent)
|
|
, tree(new QTreeWidget(this))
|
|
{
|
|
QVBoxLayout *layout = new QVBoxLayout(this);
|
|
layout->setContentsMargins({});
|
|
tree->setHeaderLabels(QStringList(tr("Database")));
|
|
tree->header()->setStretchLastSection(true);
|
|
QAction *refreshAction = new QAction(tr("Refresh"), tree);
|
|
metaDataAction = new QAction(tr("Show Schema"), tree);
|
|
connect(refreshAction, &QAction::triggered, this, &ConnectionWidget::refresh);
|
|
connect(metaDataAction, &QAction::triggered, this, &ConnectionWidget::showMetaData);
|
|
tree->addAction(refreshAction);
|
|
tree->addAction(metaDataAction);
|
|
tree->setContextMenuPolicy(Qt::ActionsContextMenu);
|
|
|
|
layout->addWidget(tree);
|
|
|
|
connect(tree, &QTreeWidget::itemActivated,
|
|
this, &ConnectionWidget::onItemActivated);
|
|
connect(tree, &QTreeWidget::currentItemChanged,
|
|
this, &ConnectionWidget::onCurrentItemChanged);
|
|
}
|
|
|
|
ConnectionWidget::~ConnectionWidget()
|
|
= default;
|
|
|
|
void ConnectionWidget::refresh()
|
|
{
|
|
const auto qDBCaption = [](const QSqlDatabase &db)
|
|
{
|
|
QString nm = db.driverName() + QLatin1Char(':');
|
|
if (!db.userName().isEmpty())
|
|
nm += db.userName() + QLatin1Char('@');
|
|
nm += db.databaseName();
|
|
return nm;
|
|
};
|
|
|
|
tree->clear();
|
|
const QStringList connectionNames = QSqlDatabase::connectionNames();
|
|
|
|
bool gotActiveDb = false;
|
|
for (const auto &connectionName : connectionNames) {
|
|
QTreeWidgetItem *root = new QTreeWidgetItem(tree);
|
|
QSqlDatabase db = QSqlDatabase::database(connectionName, false);
|
|
root->setText(0, qDBCaption(db));
|
|
if (connectionName == activeDb) {
|
|
gotActiveDb = true;
|
|
setActive(root);
|
|
}
|
|
if (db.isOpen()) {
|
|
QStringList tables = db.tables();
|
|
for (int t = 0; t < tables.count(); ++t) {
|
|
QTreeWidgetItem *table = new QTreeWidgetItem(root);
|
|
table->setText(0, tables.at(t));
|
|
}
|
|
}
|
|
}
|
|
if (!gotActiveDb) {
|
|
activeDb = connectionNames.value(0);
|
|
setActive(tree->topLevelItem(0));
|
|
}
|
|
|
|
tree->doItemsLayout(); // HACK
|
|
}
|
|
|
|
QSqlDatabase ConnectionWidget::currentDatabase() const
|
|
{
|
|
return QSqlDatabase::database(activeDb);
|
|
}
|
|
|
|
void ConnectionWidget::setActive(QTreeWidgetItem *item)
|
|
{
|
|
const auto qSetBold = [](QTreeWidgetItem *item, bool bold)
|
|
{
|
|
QFont font = item->font(0);
|
|
font.setBold(bold);
|
|
item->setFont(0, font);
|
|
};
|
|
|
|
for (int i = 0; i < tree->topLevelItemCount(); ++i) {
|
|
if (tree->topLevelItem(i)->font(0).bold())
|
|
qSetBold(tree->topLevelItem(i), false);
|
|
}
|
|
|
|
if (!item)
|
|
return;
|
|
|
|
qSetBold(item, true);
|
|
activeDb = QSqlDatabase::connectionNames().value(tree->indexOfTopLevelItem(item));
|
|
}
|
|
|
|
void ConnectionWidget::onItemActivated(QTreeWidgetItem *item)
|
|
{
|
|
if (!item)
|
|
return;
|
|
|
|
if (!item->parent()) {
|
|
setActive(item);
|
|
} else {
|
|
setActive(item->parent());
|
|
emit tableActivated(item->text(0));
|
|
}
|
|
}
|
|
|
|
void ConnectionWidget::showMetaData()
|
|
{
|
|
QTreeWidgetItem *cItem = tree->currentItem();
|
|
if (!cItem || !cItem->parent())
|
|
return;
|
|
setActive(cItem->parent());
|
|
emit metaDataRequested(cItem->text(0));
|
|
}
|
|
|
|
void ConnectionWidget::onCurrentItemChanged(QTreeWidgetItem *current)
|
|
{
|
|
metaDataAction->setEnabled(current && current->parent());
|
|
}
|
|
|