snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/sql/ast/returning.go

@@ -0,0 +1,24 @@
+package ast
+
+import "github.com/sqlc-dev/sqlc/internal/sql/format"
+
+// formatReturningOptions writes the PostgreSQL 18 RETURNING WITH (...) option
+// list that renames the OLD and NEW aliases available in a RETURNING clause.
+func formatReturningOptions(buf *TrackedBuffer, d format.Dialect, oldAlias, newAlias string) {
+	if oldAlias == "" && newAlias == "" {
+		return
+	}
+	buf.WriteString("WITH (")
+	if oldAlias != "" {
+		buf.WriteString("OLD AS ")
+		buf.WriteString(d.QuoteIdent(oldAlias))
+	}
+	if newAlias != "" {
+		if oldAlias != "" {
+			buf.WriteString(", ")
+		}
+		buf.WriteString("NEW AS ")
+		buf.WriteString(d.QuoteIdent(newAlias))
+	}
+	buf.WriteString(") ")
+}