aboutsummaryrefslogtreecommitdiffstats
path: root/Plugins/SqlEnterpriseFormatter/formatwith.h
blob: 618b73ba26fda3b6cce8817bd1b607d86d134299 (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
#ifndef FORMATWITH_H
#define FORMATWITH_H

#include "formatstatement.h"
#include "parser/ast/sqlitewith.h"

class FormatWith : public FormatStatement
{
    public:
        FormatWith(SqliteWith* with);

        void setLineUpKeyword(const QString& kw);
        void formatInternal();

    private:
        SqliteWith *with = nullptr;
        QString lineUpKeyword;
};

class FormatWithCommonTableExpression : public FormatStatement
{
    public:
        FormatWithCommonTableExpression(SqliteWith::CommonTableExpression* cte);

        void formatInternal();

    private:
        SqliteWith::CommonTableExpression* cte = nullptr;
};

#endif // FORMATWITH_H