SQLiteStudio3/coreSQLiteStudio/parser/ast/sqlitequery.cpp


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

#include "sqlitequery.h"
#include "parser/statementtokenbuilder.h"

SqliteQuery::SqliteQuery()
{
}

SqliteQuery::SqliteQuery(const SqliteQuery& other) :
    SqliteStatement(other), queryType(other.queryType), explain(other.explain), queryPlan(other.queryPlan)
{
}

bool SqliteQuery::isReadOnly()
{
    bool readOnly = true;
    switch (queryType)
    {
        case SqliteQueryType::EMPTY:
        case SqliteQueryType::Analyze:
        case SqliteQueryType::Pragma:
        case SqliteQueryType::Select:
            readOnly = true;
            break;
        case SqliteQueryType::UNDEFINED:
        case SqliteQueryType::AlterTable:
        case SqliteQueryType::Attach:
        case SqliteQueryType::BeginTrans:
        case SqliteQueryType::CommitTrans:
        case SqliteQueryType::Copy:
        case SqliteQueryType::CreateIndex:
        case SqliteQueryType::CreateTable:
        case SqliteQueryType::CreateTrigger:
        case SqliteQueryType::CreateView:
        case SqliteQueryType::CreateVirtualTable:
        case SqliteQueryType::Delete:
        case SqliteQueryType::Detach:
        case SqliteQueryType::DropIndex:
        case SqliteQueryType::DropTable:
        case SqliteQueryType::DropTrigger:
        case SqliteQueryType::DropView:
        case SqliteQueryType::Insert:
        case SqliteQueryType::Reindex:
        case SqliteQueryType::Release:
        case SqliteQueryType::Rollback:
        case SqliteQueryType::Savepoint:
        case SqliteQueryType::Update:
        case SqliteQueryType::Vacuum:
            readOnly = false;
            break;
    }
    return readOnly;
}

TokenList SqliteQuery::rebuildTokensFromContents()
{
    StatementTokenBuilder builder;
    if (explain)
    {
        builder.withKeyword("EXPLAIN").withSpace();
        if (queryPlan)
            builder.withKeyword("QUERY").withSpace().withKeyword("PLAN").withSpace();
    }
    return builder.build();
}