aboutsummaryrefslogtreecommitdiffstats
path: root/Plugins/SqlEnterpriseFormatter/formatcreatetrigger.cpp
blob: 01351e610ecfad751605605a450ba921d9084b15 (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
#include "formatcreatetrigger.h"
#include "parser/ast/sqliteexpr.h"

FormatCreateTrigger::FormatCreateTrigger(SqliteCreateTrigger* createTrig) :
    createTrig(createTrig)
{
}

void FormatCreateTrigger::formatInternal()
{
    handleExplainQuery(createTrig);
    withKeyword("CREATE");
    if (createTrig->tempKw)
        withKeyword("TEMP");
    else if (createTrig->temporaryKw)
        withKeyword("TEMPORARY");

    withKeyword("TRIGGER");
    if (createTrig->ifNotExistsKw)
        withKeyword("IF").withKeyword("NOT").withKeyword("EXISTS");

    if (dialect == Dialect::Sqlite3 && !createTrig->database.isNull())
        withId(createTrig->database).withIdDot();

    withId(createTrig->trigger);
    switch (createTrig->eventTime)
    {
        case SqliteCreateTrigger::Time::BEFORE:
            withKeyword("BEFORE");
            break;
        case SqliteCreateTrigger::Time::AFTER:
            withKeyword("AFTER");
            break;
        case SqliteCreateTrigger::Time::INSTEAD_OF:
            withKeyword("INSTEAD").withKeyword("OF");
            break;
        case SqliteCreateTrigger::Time::null:
            break;
    }

    withStatement(createTrig->event).withKeyword("ON");
    if (dialect == Dialect::Sqlite2 && !createTrig->database.isNull())
        withId(createTrig->database).withIdDot();

    withId(createTrig->table);

    switch (createTrig->scope)
    {
        case SqliteCreateTrigger::Scope::FOR_EACH_ROW:
            withKeyword("FOR").withKeyword("EACH").withKeyword("ROW");
            break;
        case SqliteCreateTrigger::Scope::FOR_EACH_STATEMENT:
            withKeyword("FOR").withKeyword("EACH").withKeyword("STATEMENT");
            break;
        case SqliteCreateTrigger::Scope::null:
            break;
    }

    if (createTrig->precondition)
        withKeyword("WHEN").withStatement(createTrig->precondition);

    withNewLine().withKeyword("BEGIN").withNewLine().withIncrIndent().withStatementList(createTrig->queries, QString(), ListSeparator::SEMICOLON).withSemicolon();
    withDecrIndent().withKeyword("END").withSemicolon();
}


FormatCreateTriggerEvent::FormatCreateTriggerEvent(SqliteCreateTrigger::Event* ev) :
    ev(ev)
{
}

void FormatCreateTriggerEvent::formatInternal()
{
    switch (ev->type)
    {
        case SqliteCreateTrigger::Event::INSERT:
            withKeyword("INSERT");
            break;
        case SqliteCreateTrigger::Event::UPDATE:
            withKeyword("UPDATE");
            break;
        case SqliteCreateTrigger::Event::DELETE:
            withKeyword("DELETE");
            break;
        case SqliteCreateTrigger::Event::UPDATE_OF:
            withKeyword("UPDATE").withKeyword("OF").withIdList(ev->columnNames);
            break;
        case SqliteCreateTrigger::Event::null:
            break;
    }
}