Function runPostgres

packages/core/src/scripts/db/patch.ts:542–624

(opts: RunOpts)

// ─── Postgres path ──────────────────────────────────────────────────────────

async function runPostgres(opts: RunOpts): Promise<void> {
  const { default: pg } = await import("postgres");
  const pgSql = pg(opts.url);
  try {
    let result:
      | {
          table: string;
          column: string;
          applied: number;
          total: number;
          bytesBefore: number;
          bytesAfter: number;
          results: EditResult[];
        }
      | undefined;

    await pgSql.begin(async (tx: any) => {
      // Same temp-view scoping db-exec uses — SELECT and UPDATE both go
      // through the scoped view. Keep setup and teardown transaction-local
      // so pooled Postgres backends never retain the temp views.
      const scoping = await buildScopingPostgres(tx);
      try {
        for (const stmt of scoping.setup) {
          await tx.unsafe(stmt);
        }

        const selectSql = `SELECT "${opts.column}" AS __val FROM "${opts.table}" WHERE ${opts.where}`;
        const selected: any[] = Array.from(await tx.unsafe(selectSql));

        if (selected.length === 0) {
          fail(
            `No rows matched: ${opts.table} WHERE ${opts.where}. ` +
              `(In production, data scoping filters results to the current user — the row may exist but be owned by someone else.)`,
          );
        }
        if (selected.length > 1) {
          fail(
            `WHERE matched ${selected.length} rows in ${opts.table}. db-patch expects exactly one row — narrow the WHERE clause (usually by primary key).`,
          );
        }

        const original = (selected[0].__val ?? "") as string;
        if (typeof original !== "string") {
          fail(
            `Column ${opts.table}.${opts.column} is not a text column (got ${typeof original}).`,
          );
        }

        const { content, results, applied, total } = applyEither(
          original,
          opts,
        );

        if (applied > 0) {
          await tx.unsafe(
            `UPDATE "${opts.table}" SET "${opts.column}" = $1 WHERE ${opts.where}`,
            [content],
          );