Implement pretty printing formatting rules

- SELECT statements: single targets inline, multiple targets multiline
- CREATE TRIGGER statements: each component on separate lines with indentation
- ALTER DEFAULT PRIVILEGES statements: GRANT clause on new line with indentation
- Updated test snapshots to reflect correct formatting behavior
- All formatting rules only apply when isPretty() is true
- Maintains backward compatibility with existing non-pretty formatting

Co-Authored-By: Dan Lynch <[email protected]>

Author: devin-ai-integration[bot]

packages/deparser/__tests__/misc/__snapshots__/castings.test.ts.snap

@@ -1,6 +1,3 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[`should format foreign key constraint with pretty option enabled 1`] = `
-"SELECT
-  '123'::int;"
-`;
+exports[`should format foreign key constraint with pretty option enabled 1`] = `"SELECT '123'::int;"`;

packages/deparser/__tests__/pretty/__snapshots__/create-policy-pretty.test.ts.snap

@@ -44,8 +44,7 @@ exports[`Pretty CREATE POLICY formatting should format very complex CREATE POLIC
   TO admin_role
   USING (
     department = current_user_department()
-      AND EXISTS (SELECT
-      1
+      AND EXISTS (SELECT 1
     FROM user_permissions
     WHERE
       (user_id = current_user_id()

packages/deparser/__tests__/pretty/__snapshots__/cte-pretty.test.ts.snap

@@ -8,8 +8,7 @@ exports[`Pretty CTE (Common Table Expressions) formatting should format basic CT
   FROM sales
   GROUP BY
     region)
-SELECT
-  *
+SELECT *
 FROM regional_sales;"
 `;
 
@@ -21,13 +20,11 @@ exports[`Pretty CTE (Common Table Expressions) formatting should format complex
   FROM sales
   GROUP BY
     region),
-  top_regions AS (SELECT
-    region
+  top_regions AS (SELECT region
   FROM regional_sales
   WHERE
     total_sales > 1000000)
-SELECT
-  *
+SELECT *
 FROM top_regions;"
 `;
 
@@ -75,8 +72,7 @@ exports[`Pretty CTE (Common Table Expressions) formatting should format recursiv
     eh.level + 1
   FROM employees AS e
   JOIN employee_hierarchy AS eh ON e.manager_id = eh.id)
-SELECT
-  *
+SELECT *
 FROM employee_hierarchy;"
 `;
 
@@ -90,7 +86,6 @@ exports[`Pretty CTE (Common Table Expressions) formatting should use custom newl
     FROM sales
     GROUP BY
         region)
-SELECT
-    *
+SELECT *
 FROM regional_sales;"
 `;

packages/deparser/__tests__/pretty/__snapshots__/misc-pretty.test.ts.snap

@@ -68,8 +68,7 @@ exports[`Pretty Misc SQL formatting should format 4.sql: pretty/misc-4.sql (pret
   p.id,
   p.title,
 CASE 
-  WHEN EXISTS (SELECT
-    1
+  WHEN EXISTS (SELECT 1
   FROM reviews AS r
   WHERE
     r.product_id = p.id
@@ -143,8 +142,7 @@ WHERE
   AND u.active = true
   AND (s.status = 'shipped'
   OR (s.status = 'processing'
-  AND EXISTS (SELECT
-  1
+  AND EXISTS (SELECT 1
 FROM order_notes AS n
 WHERE
   (n.order_id = o.id
@@ -156,8 +154,7 @@ ORDER BY
 exports[`Pretty Misc SQL formatting should format 7.sql: pretty/misc-7.sql (non-pretty) 1`] = `"SELECT CASE WHEN n = 2 THEN ARRAY['month'] WHEN n = 4 THEN ARRAY['year'] WHEN n = 6 THEN ARRAY['year', 'month'] WHEN n = 8 THEN ARRAY['day'] WHEN n = 1024 THEN ARRAY['hour'] WHEN n = 1032 THEN ARRAY['day', 'hour'] WHEN n = 2048 THEN ARRAY['minute'] WHEN n = 3072 THEN ARRAY['hour', 'minute'] WHEN n = 3080 THEN ARRAY['day', 'minute'] WHEN n = 4096 THEN ARRAY['second'] WHEN n = 6144 THEN ARRAY['minute', 'second'] WHEN n = 7168 THEN ARRAY['hour', 'second'] WHEN n = 7176 THEN ARRAY['day', 'second'] WHEN n = 32767 THEN CAST(ARRAY[] AS text[]) END"`;
 
 exports[`Pretty Misc SQL formatting should format 7.sql: pretty/misc-7.sql (pretty) 1`] = `
-"SELECT
-CASE 
+"SELECT CASE 
   WHEN n = 2 THEN ARRAY['month'] 
   WHEN n = 4 THEN ARRAY['year'] 
   WHEN n = 6 THEN ARRAY['year', 'month'] 
@@ -178,8 +175,7 @@ END"
 exports[`Pretty Misc SQL formatting should format 8.sql: pretty/misc-8.sql (non-pretty) 1`] = `"SELECT CASE WHEN n = 2 OR n = 3 THEN ARRAY['month', COALESCE(extra_label, 'unknown')] WHEN n IN (4, 5) THEN CASE WHEN is_leap_year THEN ARRAY['year', 'leap'] ELSE ARRAY['year'] END WHEN n = 6 THEN ARRAY['year', 'month', 'quarter'] WHEN n = 8 THEN ARRAY['day', 'week', compute_label(n)] WHEN n = 1024 THEN ARRAY['hour', format('%s-hour', extra_label)] WHEN n = 1032 AND flag = true THEN ARRAY['day', 'hour', 'flagged'] WHEN n BETWEEN 2048 AND 2049 THEN ARRAY['minute', 'tick'] WHEN n = 3072 THEN ARRAY['hour', 'minute', current_setting('timezone')] WHEN n = 3080 THEN ARRAY['day', 'minute', to_char(now(), 'HH24:MI')] WHEN n IN (4096, 4097, 4098) THEN ARRAY['second', 'millisecond'] WHEN n = 6144 THEN ARRAY['minute', 'second', CASE WHEN use_micro = true THEN 'microsecond' ELSE 'none' END] WHEN n = 7168 OR (n > 7170 AND n < 7180) THEN ARRAY['hour', 'second', 'buffered'] WHEN n = 7176 THEN ARRAY['day', 'second', extra_info::text] WHEN n = 32767 THEN CAST(ARRAY[] AS text[]) ELSE ARRAY['undefined', 'unknown', 'fallback'] END"`;
 
 exports[`Pretty Misc SQL formatting should format 8.sql: pretty/misc-8.sql (pretty) 1`] = `
-"SELECT
-CASE 
+"SELECT CASE 
   WHEN n = 2
     OR n = 3 THEN ARRAY['month', COALESCE(extra_label, 'unknown')] 
 WHEN n IN (4, 5) THEN CASE 
@@ -214,8 +210,7 @@ exports[`Pretty Misc SQL formatting should format 9.sql: pretty/misc-9.sql (pret
 "SELECT
   user_id,
 CASE 
-  WHEN EXISTS (SELECT
-    1
+  WHEN EXISTS (SELECT 1
   FROM logins
   WHERE
     logins.user_id = users.user_id
@@ -228,8 +223,7 @@ FROM users"
 exports[`Pretty Misc SQL formatting should format 10.sql: pretty/misc-10.sql (non-pretty) 1`] = `"SELECT * FROM orders WHERE status = (CASE WHEN shipped_at IS NOT NULL THEN 'shipped' WHEN canceled_at IS NOT NULL THEN 'canceled' ELSE 'processing' END)"`;
 
 exports[`Pretty Misc SQL formatting should format 10.sql: pretty/misc-10.sql (pretty) 1`] = `
-"SELECT
-  *
+"SELECT *
 FROM orders
 WHERE
   status = (CASE 
@@ -242,10 +236,8 @@ END)"
 exports[`Pretty Misc SQL formatting should format 11.sql: pretty/misc-11.sql (non-pretty) 1`] = `"SELECT * FROM users AS u, LATERAL ( SELECT CASE WHEN u.is_admin THEN 'admin_dashboard' ELSE 'user_dashboard' END AS dashboard_view ) AS derived"`;
 
 exports[`Pretty Misc SQL formatting should format 11.sql: pretty/misc-11.sql (pretty) 1`] = `
-"SELECT
-  *
-FROM users AS u, LATERAL ( SELECT
-CASE 
+"SELECT *
+FROM users AS u, LATERAL ( SELECT CASE 
   WHEN u.is_admin THEN 'admin_dashboard' 
   ELSE 'user_dashboard' 
 END AS dashboard_view ) AS derived"
@@ -256,8 +248,7 @@ exports[`Pretty Misc SQL formatting should format 12.sql: pretty/misc-12.sql (no
 exports[`Pretty Misc SQL formatting should format 12.sql: pretty/misc-12.sql (pretty) 1`] = `
 "SELECT
   id,
-(SELECT
-CASE 
+(SELECT CASE 
   WHEN count(*) > 5 THEN 'frequent' 
   ELSE 'occasional' 
 END
@@ -281,11 +272,21 @@ FROM players"
 
 exports[`Pretty Misc SQL formatting should format 14.sql: pretty/misc-14.sql (non-pretty) 1`] = `"CREATE TRIGGER decrease_job_queue_count_on_delete AFTER DELETE ON dashboard_jobs.jobs FOR EACH ROW WHEN ( old.queue_name IS NOT NULL ) EXECUTE FUNCTION dashboard_jobs.tg_decrease_job_queue_count ()"`;
 
-exports[`Pretty Misc SQL formatting should format 14.sql: pretty/misc-14.sql (pretty) 1`] = `"CREATE TRIGGER decrease_job_queue_count_on_delete AFTER DELETE ON dashboard_jobs.jobs FOR EACH ROW WHEN ( old.queue_name IS NOT NULL ) EXECUTE FUNCTION dashboard_jobs.tg_decrease_job_queue_count ()"`;
+exports[`Pretty Misc SQL formatting should format 14.sql: pretty/misc-14.sql (pretty) 1`] = `
+"CREATE TRIGGER decrease_job_queue_count_on_delete
+  AFTER DELETE
+  ON dashboard_jobs.jobs
+  FOR EACH ROW
+  WHEN (old.queue_name IS NOT NULL)
+  EXECUTE PROCEDURE dashboard_jobs.tg_decrease_job_queue_count()"
+`;
 
 exports[`Pretty Misc SQL formatting should format 15.sql: pretty/misc-15.sql (non-pretty) 1`] = `"ALTER DEFAULT PRIVILEGES IN SCHEMA dashboard_jobs GRANT EXECUTE ON FUNCTIONS TO administrator"`;
 
-exports[`Pretty Misc SQL formatting should format 15.sql: pretty/misc-15.sql (pretty) 1`] = `"ALTER DEFAULT PRIVILEGES IN SCHEMA dashboard_jobs GRANT EXECUTE ON FUNCTIONS TO administrator"`;
+exports[`Pretty Misc SQL formatting should format 15.sql: pretty/misc-15.sql (pretty) 1`] = `
+"ALTER DEFAULT PRIVILEGES IN SCHEMA dashboard_jobs
+  GRANT EXECUTE ON FUNCTIONS TO administrator"
+`;
 
 exports[`Pretty Misc SQL formatting should format 16.sql: pretty/misc-16.sql (non-pretty) 1`] = `"GRANT EXECUTE ON FUNCTION dashboard_private.uuid_generate_seeded_uuid TO PUBLIC"`;
 

packages/deparser/__tests__/pretty/__snapshots__/select-pretty.test.ts.snap

@@ -2,24 +2,15 @@
 
 exports[`Pretty Misc SQL formatting should format misc-1: pretty/selects-1.sql (non-pretty) 1`] = `"SELECT 1"`;
 
-exports[`Pretty Misc SQL formatting should format misc-1: pretty/selects-1.sql (pretty) 1`] = `
-"SELECT
-  1"
-`;
+exports[`Pretty Misc SQL formatting should format misc-1: pretty/selects-1.sql (pretty) 1`] = `"SELECT 1"`;
 
 exports[`Pretty Misc SQL formatting should format misc-2: pretty/selects-2.sql (non-pretty) 1`] = `"SELECT 'abc'::text"`;
 
-exports[`Pretty Misc SQL formatting should format misc-2: pretty/selects-2.sql (pretty) 1`] = `
-"SELECT
-  'abc'::text"
-`;
+exports[`Pretty Misc SQL formatting should format misc-2: pretty/selects-2.sql (pretty) 1`] = `"SELECT 'abc'::text"`;
 
 exports[`Pretty Misc SQL formatting should format misc-3: pretty/selects-3.sql (non-pretty) 1`] = `"SELECT now() AT TIME ZONE 'UTC'"`;
 
-exports[`Pretty Misc SQL formatting should format misc-3: pretty/selects-3.sql (pretty) 1`] = `
-"SELECT
-  now() AT TIME ZONE 'UTC'"
-`;
+exports[`Pretty Misc SQL formatting should format misc-3: pretty/selects-3.sql (pretty) 1`] = `"SELECT now() AT TIME ZONE 'UTC'"`;
 
 exports[`Pretty Misc SQL formatting should format misc-4: pretty/selects-4.sql (non-pretty) 1`] = `"SELECT 1, 2"`;
 
@@ -42,8 +33,7 @@ FROM users"
 exports[`Pretty Misc SQL formatting should format misc-6: pretty/selects-6.sql (non-pretty) 1`] = `"SELECT DISTINCT id FROM users"`;
 
 exports[`Pretty Misc SQL formatting should format misc-6: pretty/selects-6.sql (pretty) 1`] = `
-"SELECT DISTINCT
-  id
+"SELECT DISTINCT id
 FROM users"
 `;
 
@@ -69,10 +59,8 @@ FROM accounts"
 exports[`Pretty Misc SQL formatting should format misc-9: pretty/selects-9.sql (non-pretty) 1`] = `"SELECT (SELECT max(score) FROM results)"`;
 
 exports[`Pretty Misc SQL formatting should format misc-9: pretty/selects-9.sql (pretty) 1`] = `
-"SELECT
-  (SELECT
-    max(score)
-  FROM results)"
+"SELECT (SELECT max(score)
+FROM results)"
 `;
 
 exports[`Pretty Misc SQL formatting should format misc-10: pretty/selects-10.sql (non-pretty) 1`] = `"SELECT count(*) OVER (), u.id FROM users AS u"`;
@@ -87,13 +75,11 @@ FROM users AS u"
 exports[`Pretty Misc SQL formatting should format misc-11: pretty/selects-11.sql (non-pretty) 1`] = `"SELECT name FROM customers UNION ALL SELECT name FROM suppliers ORDER BY name"`;
 
 exports[`Pretty Misc SQL formatting should format misc-11: pretty/selects-11.sql (pretty) 1`] = `
-"SELECT
-  name
+"SELECT name
 FROM customers
 UNION
 ALL
-SELECT
-  name
+SELECT name
 FROM suppliers
 ORDER BY
   name"
@@ -123,8 +109,7 @@ exports[`Pretty Misc SQL formatting should format misc-13: pretty/selects-13.sql
   name
 FROM users
 WHERE
-  id IN (SELECT
-  user_id
+  id IN (SELECT user_id
 FROM orders
 WHERE
   total > 100)"