SQLiteStudio3/coreSQLiteStudio/parser/ast/sqlitealtertable.h


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

#ifndef SQLITEALTERTABLE_H
#define SQLITEALTERTABLE_H

#include "sqlitequery.h"
#include "sqlitecreatetable.h"

class API_EXPORT SqliteAlterTable : public SqliteQuery
{
    public:
        enum class Command
        {
            RENAME,
            ADD_COLUMN,
            DROP_COLUMN,
            null
        };

        SqliteAlterTable();
        SqliteAlterTable(const SqliteAlterTable& other);
        SqliteAlterTable(const QString& name1, const QString& name2, const QString& newName);
        SqliteAlterTable(const QString& name1, const QString& name2, bool columnKw, SqliteCreateTable::Column* column);
        SqliteAlterTable(const QString& name1, const QString& name2, bool columnKw, const QString& dropColumn);
        ~SqliteAlterTable();
        SqliteStatement* clone();

    protected:
        QStringList getColumnsInStatement();
        QStringList getTablesInStatement();
        QStringList getDatabasesInStatement();
        TokenList getColumnTokensInStatement();
        TokenList getTableTokensInStatement();
        TokenList getDatabaseTokensInStatement();
        QList<FullObject> getFullObjectsInStatement();
        TokenList rebuildTokensFromContents();

    private:
        void initName(const QString& name1, const QString& name2);

    public:
        Command command = Command::null;
        QString newName = QString();
        QString database = QString();
        QString table = QString();
        QString dropColumnName = QString();
        bool columnKw = false;
        SqliteCreateTable::Column* newColumn = nullptr;
};

typedef QSharedPointer<SqliteAlterTable> SqliteAlterTablePtr;

#endif // SQLITEALTERTABLE_H