@crvouga/mockingbird-service-postgres 3.0.1 → 3.0.2

package/dist/wire/cli.js

@@ -25620,6 +25620,46 @@ var decodeParam = (oid, format, bytes) => {
       throw pgError2("internal", `binary parameter format is not supported for type oid ${oid}`, "0A000");
   }
 };
+function returningShape(db, sql) {
+  let statements;
+  try {
+    statements = parse(sql);
+  } catch {
+    return null;
+  }
+  const stmt = statements.length === 1 ? statements[0] : void 0;
+  if (!stmt || stmt.type !== "insert" && stmt.type !== "update" && stmt.type !== "delete") return null;
+  const targets = stmt.returning;
+  if (!targets || targets.length === 0) return null;
+  const table = db.state.findTable(stmt.table);
+  if (!table) return null;
+  const label = stmt.alias ?? table.name;
+  const columns = [];
+  const columnTypes = [];
+  for (const target of targets) {
+    if (target.expr.type === "star") {
+      const qualifier = target.expr.table?.at(-1);
+      if (qualifier !== void 0 && qualifier !== label && qualifier !== table.name) return null;
+      for (const column of table.columns) {
+        columns.push(column.name);
+        columnTypes.push(column.type.id);
+      }
+      continue;
+    }
+    if (target.expr.type === "colref") {
+      const name = target.expr.parts.at(-1) ?? "?column?";
+      const column = table.columns.find((item) => item.name === name);
+      columns.push(target.alias ?? name);
+      columnTypes.push(column?.type.id ?? "text");
+      continue;
+    }
+    columns.push(target.alias ?? inferColumnName(target.expr));
+    columnTypes.push("text");
+  }
+  if (columns.length === 0) return null;
+  const command2 = stmt.type === "insert" ? "INSERT" : stmt.type === "update" ? "UPDATE" : "DELETE";
+  return { columns, columnTypes, rows: [], rowCount: 0, command: command2 };
+}
 var Connection = class {
   constructor(socket, cluster, options2) {
     this.socket = socket;
@@ -26500,11 +26540,16 @@ var Connection = class {
     else this.write(backend.rowDescription(this.fields(outcome.result, portal.resultFormats)));
   }
   /**
-   * The row shape of a statement that returns rows, for Describe on a statement: a trial run with
-   * null parameters inside a transaction that is rolled back. Anything else (or a trial that
-   * fails) is described as returning no data.
+   * The row shape of a statement that returns rows, for Describe on a statement.
+   * `INSERT`/`UPDATE`/`DELETE ... RETURNING` is described from the catalog: a trial run binds
+   * nulls and fails NOT NULL, and Describe would answer NoData. Bun then reads one DataRow cell
+   * and treats the rest of the message as a broken protocol frame.
+   * Other row-returning statements are a trial run with null parameters, rolled back. A trial
+   * that fails is described as returning no data.
    */
   async shapeOf(prepared) {
+    const described = returningShape(this.database, prepared.sql);
+    if (described) return described;
     if (!RETURNS_ROWS.test(prepared.sql)) return null;
     const db = this.database;
     await this.cluster.acquireTurn(this);