Imported Upstream version 2.99.6upstream/2.99.6

1 files changed, 241 insertions, 0 deletions

diff --git a/SQLiteStudio3/guiSQLiteStudio/windows/tablewindow.h b/SQLiteStudio3/guiSQLiteStudio/windows/tablewindow.h
new file mode 100644
index 0000000..b0ee1e3
--- /dev/null
+++ b/SQLiteStudio3/guiSQLiteStudio/windows/tablewindow.h
@@ -0,0 +1,241 @@
+#ifndef TABLEWINDOW_H
+#define TABLEWINDOW_H
+
+#include "db/db.h"
+#include "mdichild.h"
+#include "common/extactioncontainer.h"
+#include "parser/ast/sqlitecreatetable.h"
+#include "parser/ast/sqlitecreateindex.h"
+#include "dialogs/constraintdialog.h"
+#include "db/chainexecutor.h"
+#include "guiSQLiteStudio_global.h"
+#include <QPointer>
+
+class TableModifier;
+class SqlTableModel;
+class ExtLineEdit;
+class IntValidator;
+class QLabel;
+class TableStructureModel;
+class TableConstraintsModel;
+class QProgressBar;
+class WidgetCover;
+class SqliteSyntaxHighlighter;
+class ConstraintTabModel;
+
+namespace Ui {
+    class TableWindow;
+}
+
+CFG_KEY_LIST(TableWindow, QObject::tr("Table window"),
+     CFG_KEY_ENTRY(REFRESH_STRUCTURE,       Qt::Key_F5,              QObject::tr("Refresh table structure"))
+     CFG_KEY_ENTRY(ADD_COLUMN,              Qt::Key_Insert,          QObject::tr("Add new column"))
+     CFG_KEY_ENTRY(EDIT_COLUMN,             Qt::Key_Return,          QObject::tr("Edit selected column"))
+     CFG_KEY_ENTRY(DEL_COLUMN,              Qt::Key_Delete,          QObject::tr("Delete selected column"))
+     CFG_KEY_ENTRY(EXPORT,                  Qt::CTRL + Qt::Key_E,    QObject::tr("Export table data"))
+     CFG_KEY_ENTRY(IMPORT,                  Qt::CTRL + Qt::Key_I,    QObject::tr("Import data to the table"))
+     CFG_KEY_ENTRY(ADD_TABLE_CONSTRAINT,    Qt::Key_Insert,          QObject::tr("Add new table constraint"))
+     CFG_KEY_ENTRY(EDIT_TABLE_CONSTRAINT,   Qt::Key_Return,          QObject::tr("Edit selected table constraint"))
+     CFG_KEY_ENTRY(DEL_TABLE_CONSTRAINT,    Qt::Key_Delete,          QObject::tr("Delete selected table constraint"))
+     CFG_KEY_ENTRY(REFRESH_INDEXES,         Qt::Key_F5,              QObject::tr("Refresh table index list"))
+     CFG_KEY_ENTRY(ADD_INDEX,               Qt::Key_Insert,          QObject::tr("Add new index"))
+     CFG_KEY_ENTRY(EDIT_INDEX,              Qt::Key_Return,          QObject::tr("Edit selected index"))
+     CFG_KEY_ENTRY(DEL_INDEX,               Qt::Key_Delete,          QObject::tr("Delete selected index"))
+     CFG_KEY_ENTRY(REFRESH_TRIGGERS,        Qt::Key_F5,              QObject::tr("Refresh table trigger list"))
+     CFG_KEY_ENTRY(ADD_TRIGGER,             Qt::Key_Insert,          QObject::tr("Add new trigger"))
+     CFG_KEY_ENTRY(EDIT_TRIGGER,            Qt::Key_Return,          QObject::tr("Edit selected trigger"))
+     CFG_KEY_ENTRY(DEL_TRIGGER,             Qt::Key_Delete,          QObject::tr("Delete selected trigger"))
+     CFG_KEY_ENTRY(NEXT_TAB,                Qt::ALT + Qt::Key_Right, QObject::tr("Go to next tab"))
+     CFG_KEY_ENTRY(PREV_TAB,                Qt::ALT + Qt::Key_Left,  QObject::tr("Go to previous tab"))
+)
+
+class GUI_API_EXPORT TableWindow : public MdiChild
+{
+        Q_OBJECT
+        Q_ENUMS(Action)
+
+    public:
+        enum Action
+        {
+            // Structure tab
+            REFRESH_STRUCTURE,
+            COMMIT_STRUCTURE,
+            ROLLBACK_STRUCTURE,
+            ADD_COLUMN,
+            EDIT_COLUMN,
+            DEL_COLUMN,
+            MOVE_COLUMN_UP,
+            MOVE_COLUMN_DOWN,
+            ADD_TABLE_CONSTRAINT,
+            EDIT_TABLE_CONSTRAINT,
+            DEL_TABLE_CONSTRAINT,
+            ADD_TABLE_PK,
+            ADD_TABLE_FK,
+            ADD_TABLE_UNIQUE,
+            ADD_TABLE_CHECK,
+            MOVE_CONSTRAINT_UP,
+            MOVE_CONSTRAINT_DOWN,
+            EXPORT,
+            IMPORT,
+            POPULATE,
+            CREATE_SIMILAR,
+            // Indexes tab
+            REFRESH_INDEXES,
+            ADD_INDEX,
+            EDIT_INDEX,
+            DEL_INDEX,
+            // Triggers tab
+            REFRESH_TRIGGERS,
+            ADD_TRIGGER,
+            EDIT_TRIGGER,
+            DEL_TRIGGER,
+            // All tabs
+            NEXT_TAB,
+            PREV_TAB
+        };
+
+        enum ToolBar
+        {
+            TOOLBAR_STRUCTURE,
+            TOOLBAR_INDEXES,
+            TOOLBAR_TRIGGERS
+        };
+
+        explicit TableWindow(QWidget *parent = 0);
+        TableWindow(Db* db, QWidget *parent = 0);
+        TableWindow(const TableWindow& win);
+        TableWindow(QWidget *parent, Db* db, const QString& database, const QString& table);
+        ~TableWindow();
+
+        static void staticInit();
+        static void insertAction(ExtActionPrototype* action, ToolBar toolbar = TOOLBAR_STRUCTURE);
+        static void insertActionBefore(ExtActionPrototype* action, Action beforeAction, ToolBar toolbar = TOOLBAR_STRUCTURE);
+        static void insertActionAfter(ExtActionPrototype* action, Action afterAction, ToolBar toolbar = TOOLBAR_STRUCTURE);
+        static void removeAction(ExtActionPrototype* action, ToolBar toolbar = TOOLBAR_STRUCTURE);
+
+        QString getTable() const;
+        Db* getDb() const;
+        bool handleInitialFocus();
+        bool isUncommited() const;
+        QString getQuitUncommitedConfirmMessage() const;
+        void useCurrentTableAsBaseForNew();
+        Db* getAssociatedDb() const;
+
+    protected:
+        void changeEvent(QEvent *e);
+        QVariant saveSession();
+        bool restoreSession(const QVariant& sessionValue);
+        Icon* getIconNameForMdiWindow();
+        QString getTitleForMdiWindow();
+        void createActions();
+        void setupDefShortcuts();
+        bool restoreSessionNextTime();
+        QToolBar* getToolBar(int toolbar) const;
+
+    private:
+        void init();
+        void newTable();
+        void parseDdl();
+        void initDbAndTable();
+        void setupCoverWidget();
+        void createStructureActions();
+        void createDataGridActions();
+        void createDataFormActions();
+        void createIndexActions();
+        void createTriggerActions();
+        void editColumn(const QModelIndex& idx);
+        void delColumn(const QModelIndex& idx);
+        void editConstraint(const QModelIndex& idx);
+        void delConstraint(const QModelIndex& idx);
+        void executeStructureChanges();
+        void updateAfterInit();
+        QModelIndex structureCurrentIndex() const;
+        void addConstraint(ConstraintDialog::Constraint mode);
+        bool validate(bool skipWarning = false);
+        bool isModified() const;
+        TokenList indexColumnTokens(SqliteCreateIndexPtr index);
+        QString getCurrentIndex() const;
+        QString getCurrentTrigger() const;
+        void applyInitialTab();
+
+        int newTableWindowNum = 1;
+
+        Db* db = nullptr;
+        QString database;
+        QString table;
+        Ui::TableWindow *ui = nullptr;
+        SqlTableModel* dataModel = nullptr;
+        bool dataLoaded = false;
+        bool existingTable = true;
+        SqliteCreateTablePtr createTable;
+        SqliteCreateTablePtr originalCreateTable;
+        TableStructureModel* structureModel = nullptr;
+        TableConstraintsModel* structureConstraintsModel = nullptr;
+        ConstraintTabModel* constraintTabModel = nullptr;
+        WidgetCover* widgetCover = nullptr;
+        ChainExecutor* structureExecutor = nullptr;
+        TableModifier* tableModifier = nullptr;
+        bool modifyingThisTable = false;
+
+    private slots:
+        void executionSuccessful();
+        void executionFailed(const QString& errorText);
+        void dbClosedFinalCleanup();
+        void checkIfTableDeleted(const QString& database, const QString& object, DbObjectType type);
+        void checkIfIndexDeleted(const QString& object);
+        void checkIfTriggerDeleted(const QString& object);
+        void refreshStructure();
+        void commitStructure(bool skipWarning = false);
+        void changesSuccessfullyCommited();
+        void changesFailedToCommit(int errorCode, const QString& errorText);
+        void rollbackStructure();
+        void editColumn();
+        void delColumn();
+        void moveColumnUp();
+        void moveColumnDown();
+        void addConstraint();
+        void editConstraint();
+        void delConstraint();
+        void moveConstraintUp();
+        void moveConstraintDown();
+        void addPk();
+        void addFk();
+        void addUnique();
+        void addCheck();
+        void exportTable();
+        void importTable();
+        void populateTable();
+        void createSimilarTable();
+        void tabChanged(int newTab);
+        void updateStructureToolbarState();
+        void updateStructureCommitState();
+        void updateTableConstraintsToolbarState();
+        void updateDdlTab();
+        void updateNewTableState();
+        void on_structureView_doubleClicked(const QModelIndex &index);
+        void on_tableConstraintsView_doubleClicked(const QModelIndex &index);
+        void nameChanged();
+        void withOutRowIdChanged();
+        void addIndex();
+        void editIndex();
+        void delIndex();
+        void addTrigger();
+        void editTrigger();
+        void delTrigger();
+        void updateIndexesState();
+        void updateTriggersState();
+        void nextTab();
+        void prevTab();
+
+    public slots:
+        void updateIndexes();
+        void updateTriggers();
+        void addColumn();
+        void editColumn(const QString& columnName);
+        void delColumn(const QString& columnName);
+
+    signals:
+        void modifyStatusChanged();
+};
+
+#endif // TABLEWINDOW_H