aboutsummaryrefslogtreecommitdiffstats
path: root/SQLiteStudio3/guiSQLiteStudio/dialogs/newconstraintdialog.h
blob: daa226cd0d784d6b7f6de16427d5fba0d729dc7d (plain) (blame)
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
#ifndef NEWCONSTRAINTDIALOG_H
#define NEWCONSTRAINTDIALOG_H

#include "parser/ast/sqlitecreatetable.h"
#include "db/db.h"
#include "dialogs/constraintdialog.h"
#include "iconmanager.h"
#include "guiSQLiteStudio_global.h"
#include <QDialog>
#include <QPointer>

namespace Ui {
    class NewConstraintDialog;
}
class QCommandLinkButton;

class GUI_API_EXPORT NewConstraintDialog : public QDialog
{
        Q_OBJECT

    public:
        explicit NewConstraintDialog(SqliteCreateTable* createTable, Db* db, QWidget *parent = 0);
        explicit NewConstraintDialog(SqliteCreateTable::Column* column, Db* db, QWidget *parent = 0);
        explicit NewConstraintDialog(ConstraintDialog::Constraint constraintType, SqliteCreateTable* createTable, Db* db, QWidget *parent = 0);
        explicit NewConstraintDialog(ConstraintDialog::Constraint constraintType, SqliteCreateTable::Column* column, Db* db, QWidget *parent = 0);
        ~NewConstraintDialog();

        SqliteStatement* getConstraint();
        void disableMode(ConstraintDialog::Constraint constraintType);
        int exec();

    protected:
        void changeEvent(QEvent *e);

    private:
        void init();
        void initTable();
        void initColumn();
        QCommandLinkButton* addButton(const Icon& icon, const QString text, const char* slot);
        int createColumnConstraint(ConstraintDialog::Constraint constraintType);
        int createTableConstraint(ConstraintDialog::Constraint constraintType);
        int editConstraint();

        Ui::NewConstraintDialog *ui = nullptr;
        ConstraintDialog::Type type;
        Db* db = nullptr;
        ConstraintDialog::Constraint predefinedConstraintType = ConstraintDialog::UNKNOWN;
        SqliteStatement* constrStatement = nullptr;
        QPointer<SqliteCreateTable> createTable;
        QPointer<SqliteCreateTable::Column> columnStmt;
        ConstraintDialog* constraintDialog = nullptr;
        QHash<ConstraintDialog::Constraint, QCommandLinkButton*> modeToButton;

    private slots:
        void createTablePk();
        void createTableFk();
        void createTableUnique();
        void createTableCheck();
        void createColumnPk();
        void createColumnFk();
        void createColumnUnique();
        void createColumnCheck();
        void createColumnNotNull();
        void createColumnDefault();
        void createColumnCollate();
        void createColumnGenerated();
};

#endif // NEWCONSTRAINTDIALOG_H