1 files changed, 31 insertions, 0 deletions
diff --git a/Plugins/SqlEnterpriseFormatter/formatwith.h b/Plugins/SqlEnterpriseFormatter/formatwith.h
new file mode 100644
index 0000000..618b73b
--- /dev/null
+++ b/Plugins/SqlEnterpriseFormatter/formatwith.h
@@ -0,0 +1,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