aboutsummaryrefslogtreecommitdiffstats
path: root/Plugins/SqlEnterpriseFormatter/formatcreatevirtualtable.cpp
blob: 6efd8634a7571b0295c9ea725f48baa28daeec21 (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
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
#include "formatcreatevirtualtable.h"
#include "parser/ast/sqlitecreatevirtualtable.h"
#include "parser/lexer.h"

FormatCreateVirtualTable::FormatCreateVirtualTable(SqliteCreateVirtualTable* cvt) :
    cvt(cvt)
{
}

void FormatCreateVirtualTable::formatInternal()
{
    handleExplainQuery(cvt);
    withKeyword("CREATE").withKeyword("VIRTUAL").withKeyword("TABLE");
    if (cvt->ifNotExistsKw)
        withKeyword("IF").withKeyword("NOT").withKeyword("EXISTS");

    if (!cvt->database.isNull())
        withId(cvt->database).withIdDot();

    withId(cvt->table).withKeyword("USING").withId(cvt->module);
    if (!cvt->args.isEmpty())
    {
        withParDefLeft();
        int i = 0;
        for (const QString& arg : cvt->args)
        {
            if (i > 0)
                withListComma();

            for (const TokenPtr& tk : Lexer::tokenize(arg))
                handleToken(tk);

            i++;
        }
        withParDefRight();
    }

    withSemicolon();
}

void FormatCreateVirtualTable::handleToken(const TokenPtr& token)
{
    switch (token->type)
    {
        case Token::OTHER:
            withId(token->value);
            break;
        case Token::STRING:
            withString(token->value);
            break;
        case Token::COMMENT:
            // TODO Format comment here
            break;
        case Token::FLOAT:
            withFloat(token->value.toDouble());
            break;
        case Token::INTEGER:
            withInteger(token->value.toLongLong());
            break;
        case Token::BIND_PARAM:
            withBindParam(token->value);
            break;
        case Token::OPERATOR:
            withOperator(token->value);
            break;
        case Token::PAR_LEFT:
            withParDefLeft();
            break;
        case Token::PAR_RIGHT:
            withParDefRight();
            break;
        case Token::BLOB:
            withBlob(token->value);
            break;
        case Token::KEYWORD:
            withKeyword(token->value);
            break;
        case Token::SPACE:
        case Token::INVALID:
        case Token::CTX_COLUMN:
        case Token::CTX_TABLE:
        case Token::CTX_DATABASE:
        case Token::CTX_FUNCTION:
        case Token::CTX_COLLATION:
        case Token::CTX_INDEX:
        case Token::CTX_TRIGGER:
        case Token::CTX_VIEW:
        case Token::CTX_JOIN_OPTS:
        case Token::CTX_TABLE_NEW:
        case Token::CTX_INDEX_NEW:
        case Token::CTX_VIEW_NEW:
        case Token::CTX_TRIGGER_NEW:
        case Token::CTX_ALIAS:
        case Token::CTX_TRANSACTION:
        case Token::CTX_COLUMN_NEW:
        case Token::CTX_COLUMN_TYPE:
        case Token::CTX_CONSTRAINT:
        case Token::CTX_FK_MATCH:
        case Token::CTX_PRAGMA:
        case Token::CTX_ROWID_KW:
        case Token::CTX_STRICT_KW:
        case Token::CTX_NEW_KW:
        case Token::CTX_OLD_KW:
        case Token::CTX_ERROR_MESSAGE:
            break;
    }
}