Imported Upstream version 2.99.6upstream/2.99.6

1 files changed, 24 insertions, 0 deletions

diff --git a/Plugins/SqlEnterpriseFormatter/formatpragma.cpp b/Plugins/SqlEnterpriseFormatter/formatpragma.cpp
new file mode 100644
index 0000000..0b6491a
--- /dev/null
+++ b/Plugins/SqlEnterpriseFormatter/formatpragma.cpp
@@ -0,0 +1,24 @@
+#include "formatpragma.h"
+#include "parser/ast/sqlitepragma.h"
+
+FormatPragma::FormatPragma(SqlitePragma* pragma) :
+    pragma(pragma)
+{
+}
+
+void FormatPragma::formatInternal()
+{
+    withKeyword("PRAGMA");
+
+    if (!pragma->database.isNull())
+        withId(pragma->database).withIdDot();
+
+    withId(pragma->pragmaName);
+
+    if (pragma->equalsOp)
+        withOperator("=").withLiteral(pragma->value);
+    else if (pragma->parenthesis)
+        withParExprLeft().withLiteral(pragma->value).withParExprRight();
+
+    withSemicolon();
+}