@crvouga/mockingbird-service-postgres 3.0.1 → 3.1.0

package/dist/wire/cli.js

@@ -927,7 +927,7 @@ var start = async (target, values, config) => {
   });
   const listening = await listen(runtime, { port: config.port, host: config.host });
   target.listening?.(listening);
-  console.log(`${target.name} mock listening on ${listening.url}`);
+  console.log(`${target.name} emulator listening on ${listening.url}`);
   console.log(`${target.name} health: GET ${listening.url}${adminPrefix}/health`);
   console.log(`${target.name} admin: ${listening.url}${adminPrefix} (${config.adminKey ? "x-mockingbird-admin-key required" : "open \u2014 pass --admin-key to lock"})`);
   console.log(`${target.name} admin ui: ${listening.url}${adminPrefix}/ui`);
@@ -947,7 +947,7 @@ var untilSignal = async (servers) => new Promise((resolve2) => {
   process.once("SIGTERM", stop2);
 });
 var serveCommand = (target) => ({
-  summary: `Serve the ${target.name} mock or a configured fleet`,
+  summary: `Serve the ${target.name} emulator or a configured fleet`,
   options: { ...COMMON_SERVE_OPTIONS, ..."options" in target ? target.options : {} },
   async run(values) {
     const log2 = values["log-requests"] === true ? "json" : asString(values.log) ?? "pretty";
@@ -973,7 +973,7 @@ var serveCommand = (target) => ({
               console.log(JSON.stringify(manifest));
             else
               for (const [name, endpoint] of Object.entries(manifest.services))
-                console.log(`${name} mock listening on ${endpoint.url}`);
+                console.log(`${name} emulator listening on ${endpoint.url}`);
           }
         });
         return untilSignal([fleet]);
@@ -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);