Support array literals and accessors in PostgreSQL

Author: nene

src/languages/postgresql.formatter.ts

@@ -1675,6 +1675,8 @@ export default class PostgreSqlFormatter extends Formatter {
       reservedJoins,
       reservedDependentClauses,
       reservedKeywords: dedupe([...Object.values(reservedFunctions).flat(), ...reservedKeywords]),
+      openParens: ['(', '['],
+      closeParens: [')', ']'],
       stringTypes: [{ quote: "''", prefixes: ['U&', 'E', 'X', 'B'] }, '$$'],
       identTypes: [{ quote: '""', prefixes: ['U&'] }],
       identChars: { rest: '$' },

test/postgresql.test.ts

@@ -17,6 +17,7 @@ import supportsDeleteFrom from './features/deleteFrom';
 import supportsComments from './features/comments';
 import supportsIdentifiers from './features/identifiers';
 import supportsParams from './options/param';
+import supportsArrayAndMapAccessors from './features/arrayAndMapAccessors';
 
 describe('PostgreSqlFormatter', () => {
   const language = 'postgresql';
@@ -26,6 +27,7 @@ describe('PostgreSqlFormatter', () => {
   supportsComments(format);
   supportsCreateTable(format);
   supportsConstraints(format);
+  supportsArrayAndMapAccessors(format);
   supportsAlterTable(format);
   supportsDeleteFrom(format);
   supportsStrings(format, ["''", "U&''", "X''"]);
@@ -79,4 +81,11 @@ describe('PostgreSqlFormatter', () => {
         $$update$$
     `);
   });
+
+  it('supports ARRAY literals', () => {
+    expect(format('SELECT ARRAY[1, 2, 3]')).toBe(dedent`
+      SELECT
+        ARRAY[1, 2, 3]
+    `);
+  });
 });