Format TIMESTAMP WITH TIME ZONE in Trino

Fixes #936

Author: nene

src/languages/trino/trino.formatter.ts

@@ -123,6 +123,8 @@ const reservedKeywordPhrases = expandPhrases([
   '{ROWS | RANGE | GROUPS} BETWEEN',
   // comparison operator
   'IS [NOT] DISTINCT FROM',
+  // https://trino.io/docs/current/language/types.html#timestamp-p-with-time-zone
+  '[TIMESTAMP | TIME] {WITH | WITHOUT} TIME ZONE',
 ]);
 
 const reservedDataTypePhrases = expandPhrases([]);

test/trino.test.ts

@@ -123,4 +123,22 @@ describe('TrinoFormatter', () => {
         )
     `);
   });
+
+  // Regression test for issues #936
+  it('formats TIMESTAMP WITH TIME ZONE syntax', () => {
+    expect(
+      format(`
+        CREATE TABLE time_table (id INT,
+          created_at TIMESTAMP WITH TIME ZONE,
+          deleted_at TIME WITH TIME ZONE,
+          modified_at TIMESTAMP(0) WITH TIME ZONE);`)
+    ).toBe(dedent`
+      CREATE TABLE time_table (
+        id INT,
+        created_at TIMESTAMP WITH TIME ZONE,
+        deleted_at TIME WITH TIME ZONE,
+        modified_at TIMESTAMP(0) WITH TIME ZONE
+      );
+    `);
+  });
 });