@crvouga/mockingbird-service-postgres 1.5.1 → 1.7.0

package/dist/admin.js

@@ -717,12 +717,12 @@ function bootAdmin(config) {
       throw new Error(`admin shell is missing #${id}`);
     return node;
   };
-  const errorMessage = (data, fallback) => {
+  const errorMessage = (data, fallback2) => {
     if (!isRecord(data))
-      return fallback;
+      return fallback2;
     const error = data.error;
     if (!isRecord(error) || typeof error.message !== "string" || error.message === "") {
-      return fallback;
+      return fallback2;
     }
     return error.message;
   };
@@ -981,8 +981,8 @@ function bootAdmin(config) {
     const data = await api("GET", "/namespaces");
     const select = el("namespace", HTMLSelectElement);
     const listed2 = requireList(isRecord(data) ? data.namespaces : void 0, "Namespaces");
-    const fallback = isRecord(data) && typeof data.default === "string" && data.default !== "" ? data.default : "default";
-    const names = (listed2.length > 0 ? listed2.map((name) => String(name)) : [fallback]).slice();
+    const fallback2 = isRecord(data) && typeof data.default === "string" && data.default !== "" ? data.default : "default";
+    const names = (listed2.length > 0 ? listed2.map((name) => String(name)) : [fallback2]).slice();
     if (!names.includes(state.namespace))
       names.unshift(state.namespace);
     select.innerHTML = names.map((name) => `<option value="${esc(name)}">${esc(name)}</option>`).join("");
@@ -1561,29 +1561,30 @@ var createControlPlane = (context) => {
       return json(200, { status: "ok", reset: target === "*" ? context.namespaces() : [target] });
     },
     "GET /namespaces": () => json(200, { default: context.defaultNamespace, namespaces: context.namespaces() }),
-    "GET /clock": () => json(200, context.clock.state()),
-    "POST /clock": ({ body }) => {
+    "GET /clock": ({ namespace }) => json(200, (context.namespaceClock?.(namespace) ?? context.clock).state()),
+    "POST /clock": ({ body, namespace }) => {
+      const clock = context.namespaceClock?.(namespace) ?? context.clock;
       if (!isRecord2(body))
         return adminError(400, "expected a JSON object");
       if (body.reset === true)
-        context.clock.reset();
+        clock.reset();
       if (body.freeze === true)
-        context.clock.freeze();
+        clock.freeze();
       if (body.set !== void 0) {
         const instant = parseInstant(body.set);
         if (instant === void 0)
           return adminError(400, "set: expected epoch ms or ISO-8601");
-        context.clock.set(instant);
+        clock.set(instant);
       }
       if (body.advance !== void 0) {
         const delta = parseDuration(body.advance);
         if (delta === void 0)
           return adminError(400, 'advance: expected ms or "15m"-style');
-        context.clock.advance(delta);
+        clock.advance(delta);
       }
       if (body.freeze === false)
-        context.clock.unfreeze();
-      return json(200, context.clock.state());
+        clock.unfreeze();
+      return json(200, clock.state());
     },
     "GET /faults": () => json(200, { faults: context.faults.list() }),
     "POST /faults": ({ body, namespace }) => {
@@ -2425,7 +2426,33 @@ var faultResponse = (rule) => {
 };
 var createFaultRegistry = (rng = createRng(0), sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms))) => {
   const entries = [];
+  let isolated = false;
+  const scopedRngs = /* @__PURE__ */ new Map();
+  const random = (namespace) => {
+    if (!isolated || namespace === "default")
+      return rng;
+    let scoped = scopedRngs.get(namespace);
+    if (!scoped) {
+      scoped = createRng(rng.seed);
+      scopedRngs.set(namespace, scoped);
+    }
+    return scoped;
+  };
   return {
+    isolateNamespaces() {
+      isolated = true;
+    },
+    snapshot(namespace) {
+      return {
+        rules: structuredClone(entries.map((entry, position) => ({
+          ...entry.rule,
+          remaining: entry.remaining,
+          hits: entry.hits,
+          position
+        })).filter((entry) => entry.namespace === namespace)),
+        rngState: random(namespace).state()
+      };
+    },
     add(rule) {
       const existing = entries.findIndex((e) => e.rule.id === rule.id);
       const entry = { rule, remaining: rule.count ?? null, hits: 0 };
@@ -2445,6 +2472,16 @@ var createFaultRegistry = (rng = createRng(0), sleep = (ms) => new Promise((reso
     },
     clear() {
       entries.length = 0;
+      rng.reset();
+      scopedRngs.clear();
+    },
+    restore(namespace, state) {
+      for (let i = entries.length - 1; i >= 0; i--)
+        if (entries[i]?.rule.namespace === namespace)
+          entries.splice(i, 1);
+      for (const { remaining, hits, position, ...rule } of state.rules)
+        entries.splice(Math.min(position, entries.length), 0, { rule, remaining, hits });
+      random(namespace).setState(state.rngState);
     },
     async take(candidate) {
       const hits = [];
@@ -2454,7 +2491,7 @@ var createFaultRegistry = (rng = createRng(0), sleep = (ms) => new Promise((reso
         if (!matches(entry.rule, candidate))
           continue;
         const rate = entry.rule.rate ?? 1;
-        if (rng.next() >= rate)
+        if (random(candidate.namespace).next() >= rate)
           continue;
         entry.hits++;
         if (entry.remaining !== null)
@@ -2521,7 +2558,8 @@ var createJournal = (size = DEFAULT_JOURNAL_SIZE) => {
 };
 
 // ../core/dist/metrics.js
-var createMetrics = () => {
+var createMetrics = (partition = true) => {
+  const namespaces = /* @__PURE__ */ new Map();
   let requests = 0;
   let faults = 0;
   let totalDurationMs = 0;
@@ -2529,6 +2567,14 @@ var createMetrics = () => {
   const unmatched = /* @__PURE__ */ new Map();
   return {
     record(entry) {
+      if (partition) {
+        let scoped = namespaces.get(entry.namespace);
+        if (!scoped) {
+          scoped = createMetrics(false);
+          namespaces.set(entry.namespace, scoped);
+        }
+        scoped.record(entry);
+      }
       requests++;
       totalDurationMs += entry.durationMs;
       if (entry.faultId !== void 0)
@@ -2540,7 +2586,7 @@ var createMetrics = () => {
         unmatched.set(route, (unmatched.get(route) ?? 0) + 1);
       }
     },
-    report: () => ({
+    report: (namespace) => namespace !== void 0 ? (namespaces.get(namespace) ?? createMetrics(false)).report() : {
       requests,
       byOperation: Object.fromEntries([...byOperation].sort(([a], [b]) => a.localeCompare(b))),
       unmatched: [...unmatched].sort((a, b) => b[1] - a[1] || a[0].localeCompare(b[0])).map(([route, count]) => {
@@ -2549,8 +2595,13 @@ var createMetrics = () => {
       }),
       faults,
       totalDurationMs
-    }),
-    reset() {
+    },
+    reset(namespace) {
+      if (namespace !== void 0) {
+        namespaces.delete(namespace);
+        return;
+      }
+      namespaces.clear();
       requests = 0;
       faults = 0;
       totalDurationMs = 0;
@@ -2629,9 +2680,9 @@ var kindFromType = (type) => {
     return "unknown";
   return "string";
 };
-var countParam = (value, fallback, max) => {
+var countParam = (value, fallback2, max) => {
   if (value === null)
-    return fallback;
+    return fallback2;
   if (!/^\d+$/.test(value))
     return void 0;
   return Math.min(max, Number(value));
@@ -6344,7 +6395,13 @@ var DatabaseState = class _DatabaseState {
     this.settings.set("lc_numeric", "C");
     this.settings.set("lc_time", "C");
     this.settings.set("default_transaction_isolation", "read committed");
+    this.settings.set("lock_timeout", "0");
+    this.settings.set("idle_in_transaction_session_timeout", "0");
+    this.settings.set("transaction_timeout", "0");
     this.settings.set("statement_timeout", "0");
+    this.settings.set("check_function_bodies", "on");
+    this.settings.set("xmloption", "content");
+    this.settings.set("row_security", "on");
     this.settings.set("array_nulls", "on");
     this.settings.set("backslash_quote", "safe_encoding");
   }
@@ -9946,6 +10003,19 @@ function exactDegTan(x) {
   return Math.tan(x * Math.PI / 180);
 }
 
+// src/sql/deparse.ts
+function quoteIdent(name) {
+  if (/^[a-z_][a-z0-9_$]*$/.test(name)) return name;
+  return `"${name.replaceAll('"', '""')}"`;
+}
+
+// src/runtime/database-context.ts
+var contexts = /* @__PURE__ */ new WeakMap();
+var fallback = { name: "postgres", names: () => ["postgres"] };
+function databaseCatalogContext(state) {
+  return contexts.get(state) ?? fallback;
+}
+
 // src/functions/misc-fns.ts
 var PG_VERSION_TEXT = "PostgreSQL 18.3 (postgres-mem) on TypeScript, in-memory engine";
 function findSequenceForCall(ctx, name) {
@@ -10029,8 +10099,8 @@ function sequenceNextval(ctx, seq) {
 function getMiscFunctions() {
   const m = /* @__PURE__ */ new Map();
   m.set("version", () => tv("text", PG_VERSION_TEXT));
-  m.set("current_database", () => tv("name", "postgres"));
-  m.set("current_catalog", () => tv("name", "postgres"));
+  m.set("current_database", (ctx) => tv("name", databaseCatalogContext(ctx.state).name));
+  m.set("current_catalog", (ctx) => tv("name", databaseCatalogContext(ctx.state).name));
   m.set("current_schema", (ctx) => {
     const path = ctx.state.effectiveSearchPath();
     const first = path.find((s) => ctx.state.schemas.has(s));
@@ -10152,7 +10222,7 @@ function getMiscFunctions() {
       if (!col.identity) return tv("text", null);
       return tv(
         "text",
-        col.identity.sequence.split(".").map((p) => p).join(".")
+        col.identity.sequence.split(".").map(quoteIdent).join(".")
       );
     })
   );
@@ -17402,14 +17472,19 @@ var Parser = class {
   parseIdentityOptions() {
     const options = {};
     if (this.eatPunct("(")) {
-      Object.assign(options, this.parseSequenceOptions());
+      Object.assign(options, this.parseSequenceOptions(true));
       this.expectPunct(")");
     }
     return options;
   }
-  parseSequenceOptions() {
+  parseSequenceOptions(identity = false) {
     const options = {};
     for (; ; ) {
+      if (identity && this.eatKw("sequence")) {
+        this.expectKw("name");
+        options.sequenceName = this.qualifiedName();
+        continue;
+      }
       if (this.eatKw("increment")) {
         this.eatKw("by");
         options.increment = this.parseSignedBigInt();
@@ -23411,6 +23486,27 @@ function parseCsvLine(line, delim, quote) {
   }
   return fields;
 }
+function splitCsvRecords(data, quote) {
+  const records = [];
+  let start = 0;
+  let quoted = false;
+  for (let i = 0; i < data.length; i++) {
+    const char = data[i];
+    if (char === quote) {
+      if (quoted && data[i + 1] === quote) i++;
+      else quoted = !quoted;
+      continue;
+    }
+    if (char === "\n" && !quoted) {
+      const end = i > start && data[i - 1] === "\r" ? i - 1 : i;
+      records.push(data.slice(start, end));
+      start = i + 1;
+    }
+  }
+  if (quoted) throw pgError("invalid_text_representation", "unterminated CSV quoted field", "22P04");
+  if (start < data.length) records.push(data.slice(start));
+  return records;
+}
 function executeCopyFromData(env, stmt, data) {
   const format = String(stmt.options.format ?? "text").toLowerCase();
   if (format !== "text" && format !== "csv") throw unsupported(`COPY format ${format}`);
@@ -23421,7 +23517,7 @@ function executeCopyFromData(env, stmt, data) {
   const nullStr = String(stmt.options.null ?? (format === "csv" ? "" : "\\N"));
   const quote = String(stmt.options.quote ?? '"');
   const header = stmt.options.header === true || String(stmt.options.header ?? "").toLowerCase() === "true";
-  let lines = data.split(/\r?\n/);
+  let lines = format === "csv" ? splitCsvRecords(data, quote) : data.split(/\r?\n/);
   const endMarker = lines.indexOf("\\.");
   if (endMarker !== -1) lines = lines.slice(0, endMarker);
   if (lines.length > 0 && lines[lines.length - 1] === "") lines = lines.slice(0, -1);
@@ -24155,7 +24251,8 @@ function pgSettings(ctx) {
     "pg_settings"
   );
 }
-function pgDatabase() {
+function pgDatabase(ctx) {
+  const names = databaseCatalogContext(ctx.state).names();
   return rel(
     [
       ["oid", "oid"],
@@ -24164,7 +24261,7 @@ function pgDatabase() {
       ["datistemplate", "bool"],
       ["datallowconn", "bool"]
     ],
-    [[5, "postgres", 6, false, true]],
+    names.map((name, index) => [5 + index, name, 6, false, true]),
     "pg_database"
   );
 }
@@ -24199,13 +24296,14 @@ function pgTrigger(ctx) {
     "pg_trigger"
   );
 }
-var CATALOG_NAME = "postgres";
+var catalogName = (state) => databaseCatalogContext(state).name;
 function infoSchemata(state) {
+  const catalog = catalogName(state);
   const rows = [
-    [CATALOG_NAME, "pg_catalog", OWNER],
-    [CATALOG_NAME, "information_schema", OWNER]
+    [catalog, "pg_catalog", OWNER],
+    [catalog, "information_schema", OWNER]
   ];
-  for (const s of state.schemas.values()) rows.push([CATALOG_NAME, s.name, OWNER]);
+  for (const s of state.schemas.values()) rows.push([catalog, s.name, OWNER]);
   return rel(
     [
       ["catalog_name", "name"],
@@ -24217,13 +24315,14 @@ function infoSchemata(state) {
   );
 }
 function infoTables(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const schema of ctx.state.schemas.values()) {
     for (const t of schema.tables.values()) {
-      rows.push([CATALOG_NAME, t.schema, t.name, t.temp ? "LOCAL TEMPORARY" : "BASE TABLE", "YES"]);
+      rows.push([catalog, t.schema, t.name, t.temp ? "LOCAL TEMPORARY" : "BASE TABLE", "YES"]);
     }
     for (const v of schema.views.values()) {
-      if (!v.materialized) rows.push([CATALOG_NAME, v.schema, v.name, "VIEW", "NO"]);
+      if (!v.materialized) rows.push([catalog, v.schema, v.name, "VIEW", "NO"]);
     }
   }
   return rel(
@@ -24239,6 +24338,7 @@ function infoTables(ctx) {
   );
 }
 function infoColumns(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const t of allTables(ctx.state)) {
     t.columns.forEach((c, i) => {
@@ -24249,7 +24349,7 @@ function infoColumns(ctx) {
       const dataType = isArrayType(id) ? "ARRAY" : isEnumType(id) ? "USER-DEFINED" : typeDisplayName(id);
       const udt = isArrayType(id) ? `_${arrayUdt(id)}` : isEnumType(id) ? id.slice(5).split(".").pop() : id;
       rows.push([
-        CATALOG_NAME,
+        catalog,
         t.schema,
         t.name,
         c.name,
@@ -24294,10 +24394,11 @@ function arrayUdt(t) {
   return isEnumType(elem) ? elem.slice(5).split(".").pop() : elem;
 }
 function infoViews(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const schema of ctx.state.schemas.values()) {
     for (const v of schema.views.values()) {
-      if (!v.materialized) rows.push([CATALOG_NAME, v.schema, v.name, null]);
+      if (!v.materialized) rows.push([catalog, v.schema, v.name, null]);
     }
   }
   return rel(
@@ -24312,10 +24413,11 @@ function infoViews(ctx) {
   );
 }
 function infoSequences(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const s of allSequences(ctx.state)) {
     rows.push([
-      CATALOG_NAME,
+      catalog,
       s.schema,
       s.name,
       typeDisplayName(s.dataType),
@@ -24343,11 +24445,12 @@ function infoSequences(ctx) {
   );
 }
 function infoTableConstraints(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const t of allTables(ctx.state)) {
     for (const c of t.constraints) {
       const type = c.kind === "primary_key" ? "PRIMARY KEY" : c.kind === "unique" ? "UNIQUE" : c.kind === "check" ? "CHECK" : "FOREIGN KEY";
-      rows.push([CATALOG_NAME, t.schema, c.name, CATALOG_NAME, t.schema, t.name, type]);
+      rows.push([catalog, t.schema, c.name, catalog, t.schema, t.name, type]);
     }
   }
   return rel(
@@ -24365,12 +24468,13 @@ function infoTableConstraints(ctx) {
   );
 }
 function infoKeyColumnUsage(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const t of allTables(ctx.state)) {
     for (const c of t.constraints) {
       if (c.kind !== "primary_key" && c.kind !== "unique" && c.kind !== "foreign_key") continue;
       c.columns.forEach((col, i) => {
-        rows.push([CATALOG_NAME, t.schema, c.name, CATALOG_NAME, t.schema, t.name, col, i + 1]);
+        rows.push([catalog, t.schema, c.name, catalog, t.schema, t.name, col, i + 1]);
       });
     }
   }
@@ -24390,12 +24494,13 @@ function infoKeyColumnUsage(ctx) {
   );
 }
 function infoRoutines(ctx) {
+  const catalog = catalogName(ctx.state);
   const rows = [];
   for (const schema of ctx.state.schemas.values()) {
     for (const fns of schema.functions.values()) {
       for (const f of fns) {
         rows.push([
-          CATALOG_NAME,
+          catalog,
           f.schema,
           f.name,
           "FUNCTION",
@@ -24452,7 +24557,7 @@ function buildCatalogRelation(ctx, schema, name) {
       case "pg_settings":
         return pgSettings(ctx);
       case "pg_database":
-        return pgDatabase();
+        return pgDatabase(ctx);
       case "pg_roles":
       case "pg_user":
         return pgRoles();
@@ -24633,6 +24738,11 @@ var MAX_IDENT = 63;
 function truncIdent(s) {
   return s.length > MAX_IDENT ? s.slice(0, MAX_IDENT) : s;
 }
+function identitySequenceTarget(env, tableSchema, defaultName, options) {
+  const parts = options.sequenceName;
+  if (!parts || parts.length === 0) return { schema: tableSchema, name: defaultName };
+  return targetSchema(env, parts);
+}
 var SEQ_LIMITS = {
   int2: { min: -32768n, max: 32767n },
   int4: { min: -2147483648n, max: 2147483647n },
@@ -24809,8 +24919,8 @@ function buildColumn(env, schema, tableName, def, out, taken) {
         break;
       }
       case "generated_identity": {
-        const seqName = truncIdent(`${tableName}_${def.name}_seq`);
-        const seq = buildSequence(env, schema, seqName, c.options, false);
+        const target = identitySequenceTarget(env, schema, truncIdent(`${tableName}_${def.name}_seq`), c.options);
+        const seq = buildSequence(env, target.schema, target.name, c.options, false);
         if (!c.options.as) {
           seq.dataType = col.type.id === "int2" || col.type.id === "int4" || col.type.id === "int8" ? col.type.id : "int8";
           const limits = SEQ_LIMITS[seq.dataType];
@@ -24819,7 +24929,7 @@ function buildColumn(env, schema, tableName, def, out, taken) {
         }
         seq.ownedBy = { table: tableName, column: def.name };
         out.sequences.push(seq);
-        col.identity = { always: c.always, sequence: `${schema.name}.${seqName}` };
+        col.identity = { always: c.always, sequence: `${target.schema.name}.${target.name}` };
         col.notNull = true;
         break;
       }
@@ -24955,10 +25065,11 @@ function executeCreateTable(env, stmt) {
     throw pgError("invalid_table_definition", `multiple primary keys for table "${name}" are not allowed`, "42P16");
   }
   for (const seq of built.sequences) {
-    if (schema.hasRelation(seq.name)) {
+    const seqSchema = state.getSchema(seq.schema);
+    if (seqSchema.hasRelation(seq.name)) {
       throw pgError("duplicate_table", `relation "${seq.name}" already exists`, "42P07");
     }
-    schema.sequences.set(seq.name, seq);
+    seqSchema.sequences.set(seq.name, seq);
   }
   const table = new TableData(schema.name, name, built.columns, state.nextOid(), stmt.temp);
   table.constraints = built.constraints;
@@ -25310,7 +25421,13 @@ function executeAlterTable(env, stmt) {
         const built = { columns: [], constraints: [], sequences: [] };
         const taken = constraintNames(table);
         const col = buildColumn(env, schema, table.name, action.column, built, taken);
-        for (const seq of built.sequences) schema.sequences.set(seq.name, seq);
+        for (const seq of built.sequences) {
+          const seqSchema = state.getSchema(seq.schema);
+          if (seqSchema.hasRelation(seq.name)) {
+            throw pgError("duplicate_table", `relation "${seq.name}" already exists`, "42P07");
+          }
+          seqSchema.sequences.set(seq.name, seq);
+        }
         table.columns.push(col);
         table.constraints.push(...built.constraints);
         for (let i = 0; i < rows.length; i++) {
@@ -25630,12 +25747,32 @@ function executeAlterTable(env, stmt) {
             "42P16"
           );
         }
-        const seqName = truncIdent(`${table.name}_${action.column}_seq`);
-        const seq = buildSequence(env, schema, seqName, action.options, false);
+        if (!col.notNull) {
+          throw pgError(
+            "object_not_in_prerequisite_state",
+            `column "${action.column}" of relation "${table.name}" must be declared NOT NULL before identity can be added`,
+            "55000"
+          );
+        }
+        const target = identitySequenceTarget(
+          env,
+          schema,
+          truncIdent(`${table.name}_${action.column}_seq`),
+          action.options
+        );
+        if (target.schema.hasRelation(target.name)) {
+          throw pgError("duplicate_table", `relation "${target.name}" already exists`, "42P07");
+        }
+        const seq = buildSequence(env, target.schema, target.name, action.options, false);
+        if (!action.options.as) {
+          seq.dataType = col.type.id === "int2" || col.type.id === "int4" || col.type.id === "int8" ? col.type.id : "int8";
+          const limits = SEQ_LIMITS[seq.dataType];
+          if (action.options.maxValue === void 0) seq.maxValue = seq.increment > 0n ? limits.max : -1n;
+          if (action.options.minValue === void 0) seq.minValue = seq.increment > 0n ? 1n : limits.min;
+        }
         seq.ownedBy = { table: table.name, column: action.column };
-        schema.sequences.set(seqName, seq);
-        col.identity = { always: true, sequence: `${schema.name}.${seqName}` };
-        col.notNull = true;
+        target.schema.sequences.set(target.name, seq);
+        col.identity = { always: action.always, sequence: `${target.schema.name}.${target.name}` };
         break;
       }
       case "drop_identity": {
@@ -26504,7 +26641,7 @@ if (typeof disposeKey === "symbol") {
 
 // src/admin.ts
 var QUERY_CAP = 500;
-var quoteIdent = (name) => `"${name.replaceAll('"', '""')}"`;
+var quoteIdent2 = (name) => `"${name.replaceAll('"', '""')}"`;
 var bounded = (value) => {
   if (!Number.isSafeInteger(value) || value < 0) throw new Error("limit and offset must be non-negative integers");
   return value;
@@ -26577,7 +26714,7 @@ var adapt = (db) => ({
   tables: () => listed(db),
   page(schema, table, limit, offset) {
     const found = requireTable(db, schema, table);
-    const from = `${quoteIdent(schema)}.${quoteIdent(table)}`;
+    const from = `${quoteIdent2(schema)}.${quoteIdent2(table)}`;
     const take = bounded(limit);
     const skip = bounded(offset);
     const totalRow = db.query(`SELECT count(*) AS count FROM ${from}`);