Format CREATE/ALTER/DROP POLICY

Author: nene

src/syntax/policy.ts

@@ -0,0 +1,41 @@
+import { AllPolicyNodes } from "sql-parser-cst";
+import { group, join } from "../print_utils";
+import { CstToDocMap } from "../CstToDocMap";
+
+export const policyMap: CstToDocMap<AllPolicyNodes> = {
+  create_policy_stmt: (print, node) =>
+    group([
+      print.spaced(["createPolicyKw", "name", "onKw", "table"]),
+      node.clauses.length > 0 ? print.dynamicLine() : [],
+      join(print.dynamicLine(), print("clauses")),
+    ]),
+
+  policy_permissive_clause: (print) =>
+    group(print.spaced(["asKw", "permissiveKw"])),
+  policy_restrictive_clause: (print) =>
+    group(print.spaced(["asKw", "restrictiveKw"])),
+  policy_command_clause: (print) => group(print.spaced(["forKw", "commandKw"])),
+  policy_roles_clause: (print) => group(print.spaced(["toKw", "roles"])),
+  policy_using_clause: (print) => group(print.spaced(["usingKw", "expr"])),
+  policy_check_clause: (print) =>
+    group(print.spaced(["withKw", "checkKw", "expr"])),
+
+  alter_policy_stmt: (print) =>
+    group([
+      print.spaced(["alterPolicyKw", "name", "onKw", "table"]),
+      print.dynamicLine(),
+      join(print.dynamicLine(), print("actions")),
+    ]),
+
+  drop_policy_stmt: (print) =>
+    group(
+      print.spaced([
+        "dropPolicyKw",
+        "ifExistsKw",
+        "name",
+        "onKw",
+        "table",
+        "behaviorKw",
+      ]),
+    ),
+};

src/syntax/transformMap.ts

@@ -22,6 +22,7 @@ import { mergeMap } from "./merge";
 import { otherClausesMap } from "./other_clauses";
 import { renameTableMap } from "./rename_table";
 import { roleMap } from "./role";
+import { policyMap } from "./policy";
 import { preparedStatementsMap } from "./prepared_statements";
 import { proceduralLanguageMap } from "./procedural_language";
 import { procedureMap } from "./procedure";
@@ -64,6 +65,7 @@ export const transformMap: CstToDocMap<Node> = {
   ...insertMap,
   ...mergeMap,
   ...otherClausesMap,
+  ...policyMap,
   ...preparedStatementsMap,
   ...procClauseMap,
   ...proceduralLanguageMap,

test/ddl/policy.test.ts

@@ -0,0 +1,68 @@
+import dedent from "dedent-js";
+import { testPostgresql } from "../test_utils";
+
+describe("policy", () => {
+  describe("create policy", () => {
+    it(`formats minimal CREATE POLICY`, async () => {
+      await testPostgresql(dedent`
+        CREATE POLICY be_kind_policy ON permissions
+      `);
+    });
+
+    it(`formats single-line short CREATE POLICY`, async () => {
+      await testPostgresql(dedent`
+        CREATE POLICY be_kind_policy ON permissions AS PERMISSIVE FOR SELECT
+      `);
+    });
+
+    it(`formats multi-line short CREATE POLICY (if user prefers)`, async () => {
+      await testPostgresql(dedent`
+        CREATE POLICY be_kind_policy ON permissions
+        AS PERMISSIVE
+        FOR SELECT
+      `);
+    });
+
+    it(`formats CREATE POLICY with all possible clauses`, async () => {
+      await testPostgresql(dedent`
+        CREATE POLICY be_kind_policy ON permissions
+        AS RESTRICTIVE
+        FOR SELECT
+        TO johnny, sally
+        USING (kind = 'public')
+        WITH CHECK (kind = 'public')
+      `);
+    });
+  });
+
+  describe("alter policy", () => {
+    it(`formats ALTER POLICY .. RENAME`, async () => {
+      await testPostgresql(dedent`
+        ALTER POLICY be_kind ON users RENAME TO be_evil
+      `);
+    });
+
+    it(`formats ALTER POLICY .. altering of various clauses`, async () => {
+      await testPostgresql(dedent`
+        ALTER POLICY be_kind ON users
+        TO johnny, sally
+        USING (kind = 'public')
+        WITH CHECK (kind = 'public')
+      `);
+    });
+  });
+
+  describe("drop policy", () => {
+    it(`formats basic DROP POLICY`, async () => {
+      await testPostgresql(dedent`
+        DROP POLICY be_kind ON admin
+      `);
+    });
+
+    it(`formats IF EXISTS and CASCADE/RESTRICT`, async () => {
+      await testPostgresql(dedent`
+        DROP POLICY IF EXISTS be_kind ON admin CASCADE
+      `);
+    });
+  });
+});