Update copyright headers
[qt:qt.git] / demos / sqlbrowser / connectionwidget.cpp
1 /****************************************************************************
2 **
3 ** Copyright (C) 2015 The Qt Company Ltd.
4 ** Contact: http://www.qt.io/licensing/
5 **
6 ** This file is part of the demonstration applications of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:LGPL$
9 ** Commercial License Usage
10 ** Licensees holding valid commercial Qt licenses may use this file in
11 ** accordance with the commercial license agreement provided with the
12 ** Software or, alternatively, in accordance with the terms contained in
13 ** a written agreement between you and The Qt Company. For licensing terms
14 ** and conditions see http://www.qt.io/terms-conditions. For further
15 ** information use the contact form at http://www.qt.io/contact-us.
16 **
17 ** GNU Lesser General Public License Usage
18 ** Alternatively, this file may be used under the terms of the GNU Lesser
19 ** General Public License version 2.1 or version 3 as published by the Free
20 ** Software Foundation and appearing in the file LICENSE.LGPLv21 and
21 ** LICENSE.LGPLv3 included in the packaging of this file. Please review the
22 ** following information to ensure the GNU Lesser General Public License
23 ** requirements will be met: https://www.gnu.org/licenses/lgpl.html and
24 ** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
25 **
26 ** As a special exception, The Qt Company gives you certain additional
27 ** rights. These rights are described in The Qt Company LGPL Exception
28 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
29 **
30 ** GNU General Public License Usage
31 ** Alternatively, this file may be used under the terms of the GNU
32 ** General Public License version 3.0 as published by the Free Software
33 ** Foundation and appearing in the file LICENSE.GPL included in the
34 ** packaging of this file.  Please review the following information to
35 ** ensure the GNU General Public License version 3.0 requirements will be
36 ** met: http://www.gnu.org/copyleft/gpl.html.
37 **
38 ** $QT_END_LICENSE$
39 **
40 ****************************************************************************/
41
42 #include "connectionwidget.h"
43
44 #include <QtGui>
45 #include <QtSql>
46
47 ConnectionWidget::ConnectionWidget(QWidget *parent)
48     : QWidget(parent)
49 {
50     QVBoxLayout *layout = new QVBoxLayout(this);
51     tree = new QTreeWidget(this);
52     tree->setObjectName(QLatin1String("tree"));
53     tree->setHeaderLabels(QStringList(tr("database")));
54     tree->header()->setResizeMode(QHeaderView::Stretch);
55     QAction *refreshAction = new QAction(tr("Refresh"), tree);
56     metaDataAction = new QAction(tr("Show Schema"), tree);
57     connect(refreshAction, SIGNAL(triggered()), SLOT(refresh()));
58     connect(metaDataAction, SIGNAL(triggered()), SLOT(showMetaData()));
59     tree->addAction(refreshAction);
60     tree->addAction(metaDataAction);
61     tree->setContextMenuPolicy(Qt::ActionsContextMenu);
62
63     layout->addWidget(tree);
64
65     QMetaObject::connectSlotsByName(this);
66 }
67
68 ConnectionWidget::~ConnectionWidget()
69 {
70 }
71
72 static QString qDBCaption(const QSqlDatabase &db)
73 {
74     QString nm = db.driverName();
75     nm.append(QLatin1Char(':'));
76     if (!db.userName().isEmpty())
77         nm.append(db.userName()).append(QLatin1Char('@'));
78     nm.append(db.databaseName());
79     return nm;
80 }
81
82 void ConnectionWidget::refresh()
83 {
84     tree->clear();
85     QStringList connectionNames = QSqlDatabase::connectionNames();
86
87     bool gotActiveDb = false;
88     for (int i = 0; i < connectionNames.count(); ++i) {
89         QTreeWidgetItem *root = new QTreeWidgetItem(tree);
90         QSqlDatabase db = QSqlDatabase::database(connectionNames.at(i), false);
91         root->setText(0, qDBCaption(db));
92         if (connectionNames.at(i) == activeDb) {
93             gotActiveDb = true;
94             setActive(root);
95         }
96         if (db.isOpen()) {
97             QStringList tables = db.tables();
98             for (int t = 0; t < tables.count(); ++t) {
99                 QTreeWidgetItem *table = new QTreeWidgetItem(root);
100                 table->setText(0, tables.at(t));
101             }
102         }
103     }
104     if (!gotActiveDb) {
105         activeDb = connectionNames.value(0);
106         setActive(tree->topLevelItem(0));
107     }
108
109     tree->doItemsLayout(); // HACK
110 }
111
112 QSqlDatabase ConnectionWidget::currentDatabase() const
113 {
114     return QSqlDatabase::database(activeDb);
115 }
116
117 static void qSetBold(QTreeWidgetItem *item, bool bold)
118 {
119     QFont font = item->font(0);
120     font.setBold(bold);
121     item->setFont(0, font);
122 }
123
124 void ConnectionWidget::setActive(QTreeWidgetItem *item)
125 {
126     for (int i = 0; i < tree->topLevelItemCount(); ++i) {
127         if (tree->topLevelItem(i)->font(0).bold())
128             qSetBold(tree->topLevelItem(i), false);
129     }
130
131     if (!item)
132         return;
133
134     qSetBold(item, true);
135     activeDb = QSqlDatabase::connectionNames().value(tree->indexOfTopLevelItem(item));
136 }
137
138 void ConnectionWidget::on_tree_itemActivated(QTreeWidgetItem *item, int /* column */)
139 {
140
141     if (!item)
142         return;
143
144     if (!item->parent()) {
145         setActive(item);
146     } else {
147         setActive(item->parent());
148         emit tableActivated(item->text(0));
149     }
150 }
151
152 void ConnectionWidget::showMetaData()
153 {
154     QTreeWidgetItem *cItem = tree->currentItem();
155     if (!cItem || !cItem->parent())
156         return;
157     setActive(cItem->parent());
158     emit metaDataRequested(cItem->text(0));
159 }
160
161 void ConnectionWidget::on_tree_currentItemChanged(QTreeWidgetItem *current, QTreeWidgetItem *)
162 {
163     metaDataAction->setEnabled(current && current->parent());
164 }
165