1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
|
#ifndef DBTREE_H
#define DBTREE_H
#include "db/db.h"
#include "common/extactioncontainer.h"
#include "mainwindow.h"
#include "dbtree/dbtreeitem.h"
#include "guiSQLiteStudio_global.h"
#include <QDockWidget>
#include <QSet>
#include <QAtomicInt>
class WidgetCover;
class QAction;
class QMenu;
class DbTreeModel;
class QStandardItem;
class QTimer;
class TableWindow;
class ViewWindow;
class UserInputFilter;
class DbTreeView;
namespace Ui {
class DbTree;
}
CFG_KEY_LIST(DbTree, QObject::tr("Database list"),
CFG_KEY_ENTRY(DEL_SELECTED, Qt::Key_Delete, QObject::tr("Delete selected item"))
CFG_KEY_ENTRY(CLEAR_FILTER, Qt::Key_Escape, QObject::tr("Clear filter contents"))
CFG_KEY_ENTRY(REFRESH_SCHEMA, Qt::Key_F5, QObject::tr("Refresh schema"))
CFG_KEY_ENTRY(REFRESH_SCHEMAS, Qt::SHIFT + Qt::Key_F5, QObject::tr("Refresh all schemas"))
CFG_KEY_ENTRY(ADD_DB, Qt::CTRL + Qt::Key_O, QObject::tr("Add database"))
CFG_KEY_ENTRY(SELECT_ALL, Qt::CTRL + Qt::Key_A, QObject::tr("Select all items"))
CFG_KEY_ENTRY(COPY, Qt::CTRL + Qt::Key_C, QObject::tr("Copy selected item(s)"))
CFG_KEY_ENTRY(PASTE, Qt::CTRL + Qt::Key_V, QObject::tr("Paste from clipboard"))
)
class GUI_API_EXPORT DbTree : public QDockWidget, public ExtActionContainer
{
Q_OBJECT
Q_ENUMS(Action)
public:
friend class DbTreeView;
enum Action
{
COPY,
PASTE,
SELECT_ALL,
DEL_SELECTED,
CREATE_GROUP,
DELETE_GROUP,
RENAME_GROUP,
ADD_DB,
EDIT_DB,
DELETE_DB,
CONNECT_TO_DB,
DISCONNECT_FROM_DB,
IMPORT_INTO_DB,
EXPORT_DB,
VACUUM_DB,
INTEGRITY_CHECK,
ADD_TABLE,
EDIT_TABLE,
DEL_TABLE,
EXPORT_TABLE,
IMPORT_TABLE,
POPULATE_TABLE,
ADD_INDEX,
EDIT_INDEX,
DEL_INDEX,
ADD_TRIGGER,
EDIT_TRIGGER,
DEL_TRIGGER,
ADD_VIEW,
EDIT_VIEW,
DEL_VIEW,
ADD_COLUMN,
EDIT_COLUMN,
DEL_COLUMN,
CLEAR_FILTER,
REFRESH_SCHEMAS,
REFRESH_SCHEMA,
CREATE_SIMILAR_TABLE,
RESET_AUTOINCREMENT,
ERASE_TABLE_DATA,
GENERATE_SELECT,
GENERATE_UPDATE,
GENERATE_INSERT,
GENERATE_DELETE,
OPEN_DB_DIRECTORY,
EXEC_SQL_FROM_FILE,
_separator // Never use it directly, it's just for menu setup
};
enum ToolBar
{
};
explicit DbTree(QWidget *parent = 0);
~DbTree();
static void staticInit();
void init();
void updateActionStates(const QStandardItem* item);
void setupActionsForMenu(DbTreeItem* currItem, QMenu* contextMenu);
QVariant saveSession();
void restoreSession(const QVariant& sessionValue);
DbTreeModel* getModel() const;
DbTreeView* getView() const;
void showRefreshWidgetCover();
void hideRefreshWidgetCover();
void setSelectedItem(DbTreeItem* item);
bool isMimeDataValidForItem(const QMimeData* mimeData, const DbTreeItem* item, bool forPasting = false);
QToolBar* getToolBar(int toolbar) const;
Db* getSelectedDb();
Db* getSelectedOpenDb();
QSet<Db*> getSelectedDatabases();
static bool isItemDraggable(const DbTreeItem* item);
protected:
void createActions();
void setupDefShortcuts();
private:
typedef std::function<bool(DbTreeItem*)> ItemFilterFunc;
void setActionEnabled(int action, bool enabled);
TableWindow* openTable(DbTreeItem* item);
TableWindow* openTable(Db* db, const QString& database, const QString& table);
TableWindow* newTable(Db* db);
ViewWindow* openView(DbTreeItem* item);
ViewWindow* openView(Db* db, const QString& database, const QString& view);
ViewWindow* newView(Db* db);
void editIndex(DbTreeItem* item);
void editTrigger(DbTreeItem* item);
void delSelectedObject();
void filterUndeletableItems(QList<DbTreeItem*>& items);
void filterItemsWithParentInList(QList<DbTreeItem*>& items);
void deleteItem(DbTreeItem* item);
void deleteSelected(DbTreeItem::Type itemType);
QHash<Db*, QList<DbTreeItem*>> groupByDb(const QList<DbTreeItem*> items);
QStringList itemsToNames(const QList<DbTreeItem*>& items);
QString getSelectedTableName() const;
QString getSelectedIndexName() const;
QString getSelectedTriggerName() const;
QString getSelectedViewName() const;
QList<DbTreeItem*> getSelectedItems(DbTreeItem::Type itemType);
QList<DbTreeItem*> getSelectedItems(ItemFilterFunc filterFunc = nullptr);
void execFromFileAsync(const QString& path, Db* db, bool ignoreErrors, const QString& codec);
bool execQueryFromFile(Db* db, const QString& sql);
void handleFileQueryExecution(Db* db, int executed, int attemptedExecutions, bool ok, bool ignoreErrors, int millis);
QList<QPair<QString, QString>> executeFileQueries(Db* db, QTextStream& stream, int& executed, int& attemptedExecutions, bool& ok, bool ignoreErrors, qint64 fileSize);
bool shouldSkipQueryFromFileExecution(const QString& sql);
static bool areDbTreeItemsValidForItem(QList<DbTreeItem*> srcItems, const DbTreeItem* dstItem, bool forPasting = false);
static bool areUrlsValidForItem(const QList<QUrl>& srcUrls, const DbTreeItem* dstItem);
static void initDndTypes();
Ui::DbTree *ui = nullptr;
DbTreeModel* treeModel = nullptr;
WidgetCover* treeRefreshWidgetCover = nullptr;
WidgetCover* fileExecWidgetCover = nullptr;
QAtomicInt executingQueriesFromFile = 0;
Db* executingQueriesFromFileDb = nullptr;
static QHash<DbTreeItem::Type,QList<DbTreeItem::Type>> allowedTypesInside;
static QSet<DbTreeItem::Type> draggableTypes;
static const constexpr int ITEM_TEXT_LIMIT = 300;
public slots:
void refreshSchema(Db* db);
void refreshSchemas();
void interrupt();
void updateActionsForCurrent();
private slots:
void copy();
void paste();
void selectAll();
void createGroup();
void deleteGroup();
void renameGroup();
void addDb();
void editDb();
void removeDb();
void connectToDb();
void disconnectFromDb();
void import();
void exportDb();
void addTable();
void editTable();
void delTable();
void addIndex();
void editIndex();
void delIndex();
void addTrigger();
void editTrigger();
void delTrigger();
void addView();
void editView();
void delView();
void exportTable();
void importTable();
void populateTable();
void addColumn();
void editColumn();
void delColumn();
void vacuumDb();
void integrityCheck();
void createSimilarTable();
void resetAutoincrement();
void eraseTableData();
void addColumn(DbTreeItem* item);
void editColumn(DbTreeItem* item);
void delColumn(DbTreeItem* item);
void currentChanged(const QModelIndex & current, const QModelIndex & previous);
void deleteSelected(ItemFilterFunc filterFunc = nullptr);
void deleteItems(const QList<DbTreeItem*>& itemsToDelete);
void refreshSchema();
void dbConnected(Db* db);
void dbDisconnected(Db* db);
void updateDbIcon(Db* db);
void refreshFont();
void generateSelectForTable();
void generateInsertForTable();
void generateUpdateForTable();
void generateDeleteForTable();
void openDbDirectory();
void execSqlFromFile();
void setFileExecProgress(int newValue);
void hideFileExecCover();
void showFileExecErrors(const QList<QPair<QString, QString>>& errors, bool rolledBack);
signals:
void updateFileExecProgress(int value);
void fileExecCoverToBeClosed();
void fileExecErrors(const QList<QPair<QString, QString>>& errors, bool rolledBack);
void schemaNeedsRefreshing(Db* db);
};
int qHash(DbTree::Action action);
#define DBTREE MainWindow::getInstance()->getDbTree()
#endif // DBTREE_H
|
