From 3565aad630864ecdbe53fdaa501ea708555b3c7c Mon Sep 17 00:00:00 2001
From: Unit 193 <unit193@unit193.net>
Date: Sun, 30 Apr 2023 18:30:36 -0400
Subject: New upstream version 3.4.4+dfsg.

---
 Plugins/SqlEnterpriseFormatter/formatwith.cpp | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'Plugins/SqlEnterpriseFormatter/formatwith.cpp')

diff --git a/Plugins/SqlEnterpriseFormatter/formatwith.cpp b/Plugins/SqlEnterpriseFormatter/formatwith.cpp
index d3d99d5..c992d33 100644
--- a/Plugins/SqlEnterpriseFormatter/formatwith.cpp
+++ b/Plugins/SqlEnterpriseFormatter/formatwith.cpp
@@ -34,5 +34,16 @@ void FormatWithCommonTableExpression::formatInternal()
     if (cte->indexedColumns.size() > 0)
         withParDefLeft().withStatementList(cte->indexedColumns, "idxCols").withParDefRight();
 
-    withKeyword("AS").withParDefLeft().withStatement(cte->select).withParDefRight();
+    withKeyword("AS");
+    switch (cte->asMode) {
+        case SqliteWith::CommonTableExpression::ANY:
+            break;
+        case SqliteWith::CommonTableExpression::MATERIALIZED:
+            withKeyword("MATERIALIZED");
+            break;
+        case SqliteWith::CommonTableExpression::NOT_MATERIALIZED:
+            withKeyword("NOT").withKeyword("MATERIALIZED");
+            break;
+    }
+    withParDefLeft().withStatement(cte->select).withParDefRight();
 }
-- 
cgit v1.2.3