Add pretty printing for ALTER TABLE ADD CONSTRAINT

- Place ADD CONSTRAINT on new line with indentation when pretty printing enabled
- Maintains existing behavior for non-pretty mode
- Follows existing pretty printing patterns in codebase
- Updated constraint test snapshots to reflect new formatting

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

Author: devin-ai-integration[bot]

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

@@ -28,28 +28,56 @@ exports[`non-pretty: pretty/constraints-13.sql 1`] = `"ALTER TABLE products ADD
 
 exports[`non-pretty: pretty/constraints-14.sql 1`] = `"ALTER TABLE users ADD CONSTRAINT unique_email UNIQUE (email)"`;
 
-exports[`pretty: pretty/constraints-1.sql 1`] = `"ALTER TABLE public.users ADD CONSTRAINT users_pkey PRIMARY KEY (id)"`;
+exports[`pretty: pretty/constraints-1.sql 1`] = `
+"ALTER TABLE public.users 
+  ADD CONSTRAINT users_pkey PRIMARY KEY (id)"
+`;
 
-exports[`pretty: pretty/constraints-2.sql 1`] = `"ALTER TABLE "App"."User Data" ADD CONSTRAINT "Unique_Full Name" UNIQUE ("Full Name")"`;
+exports[`pretty: pretty/constraints-2.sql 1`] = `
+"ALTER TABLE "App"."User Data" 
+  ADD CONSTRAINT "Unique_Full Name" UNIQUE ("Full Name")"
+`;
 
-exports[`pretty: pretty/constraints-3.sql 1`] = `"ALTER TABLE school.attendance ADD CONSTRAINT attendance_unique UNIQUE ("Student ID", "Class ID")"`;
+exports[`pretty: pretty/constraints-3.sql 1`] = `
+"ALTER TABLE school.attendance 
+  ADD CONSTRAINT attendance_unique UNIQUE ("Student ID", "Class ID")"
+`;
 
-exports[`pretty: pretty/constraints-4.sql 1`] = `"ALTER TABLE "Orders"."OrderLines" ADD CONSTRAINT "FK_Order_Ref" FOREIGN KEY (order_id) REFERENCES "Orders"."Order" ("OrderID")"`;
+exports[`pretty: pretty/constraints-4.sql 1`] = `
+"ALTER TABLE "Orders"."OrderLines" 
+  ADD CONSTRAINT "FK_Order_Ref" FOREIGN KEY (order_id) REFERENCES "Orders"."Order" ("OrderID")"
+`;
 
-exports[`pretty: pretty/constraints-5.sql 1`] = `"ALTER TABLE "x-Schema"."z-Table" ADD CONSTRAINT "zNameFormatCheck" CHECK ("Z-Name" ~ '^[A-Z]')"`;
+exports[`pretty: pretty/constraints-5.sql 1`] = `
+"ALTER TABLE "x-Schema"."z-Table" 
+  ADD CONSTRAINT "zNameFormatCheck" CHECK ("Z-Name" ~ '^[A-Z]')"
+`;
 
-exports[`pretty: pretty/constraints-6.sql 1`] = `"ALTER TABLE data.snapshots ADD CONSTRAINT metadata_has_key CHECK (metadata ? 'type')"`;
+exports[`pretty: pretty/constraints-6.sql 1`] = `
+"ALTER TABLE data.snapshots 
+  ADD CONSTRAINT metadata_has_key CHECK (metadata ? 'type')"
+`;
 
-exports[`pretty: pretty/constraints-7.sql 1`] = `"ALTER TABLE "Billing"."Invoices" ADD CONSTRAINT "FK_Client_ID" FOREIGN KEY ("Client ID") REFERENCES "Clients"."ClientBase" ("Client ID")"`;
+exports[`pretty: pretty/constraints-7.sql 1`] = `
+"ALTER TABLE "Billing"."Invoices" 
+  ADD CONSTRAINT "FK_Client_ID" FOREIGN KEY ("Client ID") REFERENCES "Clients"."ClientBase" ("Client ID")"
+`;
 
-exports[`pretty: pretty/constraints-8.sql 1`] = `"ALTER TABLE "API Keys" ADD CONSTRAINT "PK_KeyID" PRIMARY KEY ("KeyID")"`;
+exports[`pretty: pretty/constraints-8.sql 1`] = `
+"ALTER TABLE "API Keys" 
+  ADD CONSTRAINT "PK_KeyID" PRIMARY KEY ("KeyID")"
+`;
 
 exports[`pretty: pretty/constraints-9.sql 1`] = `
-"ALTER TABLE finance.transactions ADD CONSTRAINT tax_rate_range CHECK (tax_rate >= 0
-  AND tax_rate <= 1)"
+"ALTER TABLE finance.transactions 
+  ADD CONSTRAINT tax_rate_range CHECK (tax_rate >= 0
+    AND tax_rate <= 1)"
 `;
 
-exports[`pretty: pretty/constraints-10.sql 1`] = `"ALTER TABLE school.enrollments ADD CONSTRAINT fk_student_course FOREIGN KEY (student_id, course_id) REFERENCES school.courses_students (student_id, course_id)"`;
+exports[`pretty: pretty/constraints-10.sql 1`] = `
+"ALTER TABLE school.enrollments 
+  ADD CONSTRAINT fk_student_course FOREIGN KEY (student_id, course_id) REFERENCES school.courses_students (student_id, course_id)"
+`;
 
 exports[`pretty: pretty/constraints-11.sql 1`] = `
 "CREATE TABLE orders (
@@ -66,13 +94,20 @@ exports[`pretty: pretty/constraints-11.sql 1`] = `
 `;
 
 exports[`pretty: pretty/constraints-12.sql 1`] = `
-"ALTER TABLE products ADD CONSTRAINT fk_category FOREIGN KEY (category_id) REFERENCES categories (id)
-  ON UPDATE CASCADE
-  ON DELETE SET NULL
-  DEFERRABLE
-  INITIALLY DEFERRED"
+"ALTER TABLE products 
+  ADD CONSTRAINT fk_category FOREIGN KEY (category_id) REFERENCES categories (id)
+    ON UPDATE CASCADE
+    ON DELETE SET NULL
+    DEFERRABLE
+    INITIALLY DEFERRED"
 `;
 
-exports[`pretty: pretty/constraints-13.sql 1`] = `"ALTER TABLE products ADD CONSTRAINT check_price CHECK (price > 0)"`;
+exports[`pretty: pretty/constraints-13.sql 1`] = `
+"ALTER TABLE products 
+  ADD CONSTRAINT check_price CHECK (price > 0)"
+`;
 
-exports[`pretty: pretty/constraints-14.sql 1`] = `"ALTER TABLE users ADD CONSTRAINT unique_email UNIQUE (email)"`;
+exports[`pretty: pretty/constraints-14.sql 1`] = `
+"ALTER TABLE users 
+  ADD CONSTRAINT unique_email UNIQUE (email)"
+`;

packages/deparser/src/deparser.ts

@@ -4571,10 +4571,24 @@ export class Deparser implements DeparserVisitor {
     }
 
     if (node.cmds && node.cmds.length > 0) {
-      const commandsStr = ListUtils.unwrapList(node.cmds)
-        .map(cmd => this.visit(cmd, alterContext))
-        .join(', ');
-      output.push(commandsStr);
+      const commands = ListUtils.unwrapList(node.cmds);
+      if (this.formatter.isPretty()) {
+        const commandsStr = commands
+          .map(cmd => {
+            const cmdStr = this.visit(cmd, alterContext);
+            if (cmdStr.startsWith('ADD CONSTRAINT') || cmdStr.startsWith('ADD ')) {
+              return this.formatter.newline() + this.formatter.indent(cmdStr);
+            }
+            return cmdStr;
+          })
+          .join(',');
+        output.push(commandsStr);
+      } else {
+        const commandsStr = commands
+          .map(cmd => this.visit(cmd, alterContext))
+          .join(', ');
+        output.push(commandsStr);
+      }
     }
 
     return output.join(' ');