Format DO statement

Author: nene

src/syntax/do.ts

@@ -0,0 +1,6 @@
+import { DoStmt } from "sql-parser-cst";
+import { CstToDocMap } from "../CstToDocMap";
+
+export const doMap: CstToDocMap<DoStmt> = {
+  do_stmt: (print) => print.spaced(["doKw", "language", "body"]),
+};

src/syntax/transformMap.ts

@@ -10,6 +10,7 @@ import { createTableMap } from "./create_table";
 import { dataTypeMap } from "./data_type";
 import { dclMap } from "./dcl";
 import { deleteMap } from "./delete";
+import { doMap } from "./do";
 import { domainMap } from "./domain";
 import { dropTableMap } from "./drop_table";
 import { explainMap } from "./explain";
@@ -56,6 +57,7 @@ export const transformMap: Partial<CstToDocMap<Node>> = {
   ...dataTypeMap,
   ...dclMap,
   ...deleteMap,
+  ...doMap,
   ...domainMap,
   ...dropTableMap,
   ...explainMap,

test/proc/do.test.ts

@@ -0,0 +1,20 @@
+import dedent from "dedent-js";
+import { testPostgresql } from "../test_utils";
+
+describe("do", () => {
+  it(`formats DO statement`, async () => {
+    await testPostgresql(dedent`
+      DO $$
+        BEGIN
+          PERFORM proc_name(arg1, arg2, arg3);
+        END
+      $$
+    `);
+  });
+
+  it(`formats DO [LANGUAGE <language>]`, async () => {
+    await testPostgresql(dedent`
+      DO LANGUAGE plpgsql 'SELECT 1;'
+    `);
+  });
+});