1 files changed, 24 insertions, 0 deletions
diff --git a/Plugins/SqlEnterpriseFormatter/formatbegintrans.cpp b/Plugins/SqlEnterpriseFormatter/formatbegintrans.cpp
new file mode 100644
index 0000000..5a70d61
--- /dev/null
+++ b/Plugins/SqlEnterpriseFormatter/formatbegintrans.cpp
@@ -0,0 +1,24 @@
+#include "formatbegintrans.h"
+#include "parser/ast/sqlitebegintrans.h"
+
+FormatBeginTrans::FormatBeginTrans(SqliteBeginTrans* bt) :
+    bt(bt)
+{
+}
+
+void FormatBeginTrans::formatInternal()
+{
+    withKeyword("BEGIN");
+
+    if (bt->type != SqliteBeginTrans::Type::null)
+        withKeyword(SqliteBeginTrans::typeToString(bt->type));
+
+    if (bt->transactionKw)
+    {
+        withKeyword("TRANSACTION");
+        if (!bt->name.isNull())
+            withId(bt->name);
+    }
+
+    withConflict(bt->onConflict).withSemicolon();
+}