diff options
Diffstat (limited to 'Plugins/SqlEnterpriseFormatter/formatstatement.h')
| -rw-r--r-- | Plugins/SqlEnterpriseFormatter/formatstatement.h | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/Plugins/SqlEnterpriseFormatter/formatstatement.h b/Plugins/SqlEnterpriseFormatter/formatstatement.h new file mode 100644 index 0000000..1702a3d --- /dev/null +++ b/Plugins/SqlEnterpriseFormatter/formatstatement.h @@ -0,0 +1,198 @@ +#ifndef FORMATSTATEMENT_H +#define FORMATSTATEMENT_H + +#include "parser/ast/sqlitestatement.h" +#include "parser/ast/sqlitesortorder.h" +#include "parser/ast/sqliteconflictalgo.h" +#include "common/utils_sql.h" +#include "sqlenterpriseformatter.h" +#include <QString> +#include <QStringList> +#include <QHash> +#include <QStack> +#include <QVariant> +#include <functional> + +class FormatStatement +{ + public: + enum class ListSeparator + { + NONE, + COMMA, + EXPR_COMMA, + NEW_LINE, + SEMICOLON + }; + + typedef std::function<void(FormatStatement*)> FormatStatementEnricher; + + FormatStatement(); + virtual ~FormatStatement(); + + QString format(); + void setSelectedWrapper(NameWrapper wrapper); + void setConfig(Cfg::SqlEnterpriseFormatterConfig* cfg); + + static FormatStatement* forQuery(SqliteStatement *query); + + FormatStatement& withKeyword(const QString& kw); + FormatStatement& withLinedUpKeyword(const QString& kw, const QString& lineUpName = QString()); + FormatStatement& withId(const QString& id); + FormatStatement& withIdList(const QStringList& names, const QString& indentName = QString(), ListSeparator sep = ListSeparator::COMMA); + FormatStatement& withOperator(const QString& oper); + FormatStatement& withStringOrId(const QString& id); + FormatStatement& withIdDot(); + FormatStatement& withStar(); + FormatStatement& withFloat(double value); + FormatStatement& withInteger(qint64 value); + FormatStatement& withString(const QString& value); + FormatStatement& withBlob(const QString& value); + FormatStatement& withBindParam(const QString& name); + FormatStatement& withParDefLeft(); + FormatStatement& withParDefRight(); + FormatStatement& withParExprLeft(); + FormatStatement& withParExprRight(); + FormatStatement& withParFuncLeft(); + FormatStatement& withParFuncRight(); + FormatStatement& withSemicolon(); + FormatStatement& withListComma(); + FormatStatement& withCommaOper(); + FormatStatement& withSortOrder(SqliteSortOrder sortOrder); + FormatStatement& withConflict(SqliteConflictAlgo onConflict); + FormatStatement& withFuncId(const QString& func); + FormatStatement& withDataType(const QString& dataType); + FormatStatement& withNewLine(); + FormatStatement& withLiteral(const QVariant& value); + FormatStatement& withStatement(SqliteStatement* stmt, const QString& indentName = QString(), FormatStatementEnricher enricher = nullptr); + FormatStatement& markIndent(const QString& name); + FormatStatement& markAndKeepIndent(const QString& name); + FormatStatement& withIncrIndent(int newIndent); + FormatStatement& withIncrIndent(const QString& name = QString()); + FormatStatement& withDecrIndent(); + FormatStatement& markKeywordLineUp(const QString& keyword, const QString& lineUpName = QString()); + FormatStatement& withSeparator(ListSeparator sep); + + template <class T> + FormatStatement& withStatementList(QList<T*> stmtList, const QString& indentName = QString(), ListSeparator sep = ListSeparator::COMMA, + FormatStatementEnricher enricher = nullptr) + { + if (!indentName.isNull()) + markAndKeepIndent(indentName); + + bool first = true; + foreach (T* stmt, stmtList) + { + if (!first) + withSeparator(sep); + + withStatement(stmt, QString(), enricher); + first = false; + } + + if (!indentName.isNull()) + withDecrIndent(); + + return *this; + } + + template <class T> + T* getFormatStatement(SqliteStatement* stmt) + { + return dynamic_cast<T*>(forQuery(stmt, dialect, wrapper, cfg)); + } + + protected: + virtual void formatInternal() = 0; + virtual void resetInternal(); + + Dialect dialect = Dialect::Sqlite3; + NameWrapper wrapper = NameWrapper::BRACKET; + Cfg::SqlEnterpriseFormatterConfig* cfg = nullptr; + + private: + struct FormatToken + { + enum Type + { + KEYWORD, + LINED_UP_KEYWORD, + ID, + STRING_OR_ID, + OPERATOR, + STAR, + FLOAT, + STRING, + INTEGER, + BLOB, + BIND_PARAM, + ID_DOT, + PAR_DEF_LEFT, + PAR_DEF_RIGHT, + PAR_EXPR_LEFT, + PAR_EXPR_RIGHT, + PAR_FUNC_LEFT, + PAR_FUNC_RIGHT, + SEMICOLON, + COMMA_LIST, + COMMA_OPER, // for example in LIMIT + FUNC_ID, + DATA_TYPE, + NEW_LINE, + INDENT_MARKER, + INCR_INDENT, + SET_INDENT, + DECR_INDENT, + MARK_KEYWORD_LINEUP + }; + + Type type; + QVariant value; + QVariant additionalValue; + }; + + void withToken(FormatToken::Type type, const QVariant& value, const QVariant& additionalValue = QVariant()); + void cleanup(); + void buildTokens(); + bool applyIndent(); + void applySpace(FormatToken::Type type); + bool isSpaceExpectingType(FormatToken::Type type); + bool isMetaType(FormatToken::Type type); + void newLine(); + void incrIndent(const QString& name = QString()); + void decrIndent(); + void setIndent(int newIndent); + bool endsWithSpace(); + FormatToken* getLastRealToken(bool skipNewLines = false); + QString detokenize(); + void detokenizeLeftPar(FormatToken* token, bool spaceBefore, bool spaceAfter, bool nlBefore, bool nlAfter); + void detokenizeRightPar(FormatToken* token, bool spaceBefore, bool spaceAfter, bool nlBefore, bool nlAfter); + void resetIndents(); + void removeAllSpaces(); + void removeAllSpacesFromLine(); + void updateLastToken(FormatToken* token); + QString getFinalLineUpName(const QString& lineUpName); + int predictCurrentIndent(FormatToken* currentMetaToken); + bool willStartWithNewLine(FormatToken* token); + void formatId(const QString& value); + int getLineUpValue(const QString& lineUpName); + + static FormatStatement* forQuery(SqliteStatement *query, Dialect dialect, NameWrapper wrapper, Cfg::SqlEnterpriseFormatterConfig* cfg); + + QHash<QString,int> kwLineUpPosition; + QHash<QString,int> namedIndents; + QStack<int> indents; + QList<FormatToken*> tokens; + bool deleteTokens = true; + QStringList lines; + QString line; + FormatToken* lastToken = nullptr; + QString statementName; + FormatStatement* parentFormatStatement = nullptr; + + static qint64 nameSeq; + static const QString SPACE; + static const QString NEWLINE; +}; + +#endif // FORMATSTATEMENT_H |