okmodel 0.5.1 → 0.7.0

package/dist/shared/1est4ezg.js

@@ -0,0 +1,313 @@
+import {
+  queryHandle,
+  readHandle,
+  readCall,
+  attachHttp
+} from "./tbrvgw6z.js";
+import {
+  emitWhere,
+  emitOrder,
+  refuseHidden,
+  rejectKeys,
+  readLimit,
+  projectExpr,
+  cell,
+  indexes,
+  quote,
+  list,
+  isRecord,
+  fail2
+} from "./67txcrcw.js";
+import {
+  OkmError2,
+  throwNamed2
+} from "./52szzakm.js";
+
+// src/runtime/aggregate.ts
+var BUCKET = Symbol.for("okmodel.bucket");
+var bucketChunk;
+function loadBucketGroup() {
+  return bucketChunk ??= import("./3v7t1yrv.js");
+}
+var OPTIONS = [
+  "where",
+  "groupBy",
+  "count",
+  "sum",
+  "avg",
+  "min",
+  "max",
+  "orderBy",
+  "limit",
+  "route",
+  "signal",
+  "timeout"
+];
+var NUMERIC = /^(smallint|integer|bigint|real|double precision|numeric|decimal)\b/;
+var AS_NUMBER = (wire) => Number(wire);
+var AS_TEXT = (wire) => wire;
+var UNORDERED = /^(json|jsonb|bytea|point|line|tsvector)\b|\[\]$/;
+var FUNCTIONS = [
+  { option: "sum", tag: "s", numeric: true },
+  { option: "avg", tag: "a", numeric: true },
+  { option: "min", tag: "n", numeric: false },
+  { option: "max", tag: "x", numeric: false }
+];
+function build(session, table, options, mods, view) {
+  try {
+    if (seesBucket(options))
+      return bucketHandle(session, table, options, mods, view);
+    return plan(session, table, options, mods, view);
+  } catch (error) {
+    throw error instanceof OkmError2 ? attachHttp(session.http, error) : error;
+  }
+}
+function bucketHandle(session, table, options, mods, view) {
+  const open = () => loadBucketGroup().then((mod) => {
+    try {
+      return { it: plan(session, table, options, mods, view, mod) };
+    } catch (error) {
+      throw error instanceof OkmError2 ? attachHttp(session.http, error) : error;
+    }
+  });
+  return queryHandle(() => open().then((box) => box.it), {
+    inspect: () => open().then((box) => box.it.inspect()),
+    sql: () => open().then((box) => box.it.sql()),
+    all: (reason) => build(session, table, options, { ...mods, all: reason }, view)
+  });
+}
+function seesBucket(options) {
+  if (!isRecord(options))
+    return false;
+  const groupBy = options.groupBy;
+  if (!Array.isArray(groupBy))
+    return false;
+  for (const item of groupBy) {
+    if (typeof item === "object" && item !== null && BUCKET in item)
+      return true;
+  }
+  return false;
+}
+function plan(session, table, options, mods, view, buckets) {
+  const record = isRecord(options) ? options : {};
+  rejectKeys(record, OPTIONS, "aggregate");
+  const indexed = indexes(session.schema).get(table);
+  if (indexed === undefined)
+    fail2("OKM1120", `Table ${table} is not in the schema.`);
+  const spec = parse(indexed, record, buckets);
+  if (spec.groups.length > 0 && record.limit === undefined && mods.all === undefined) {
+    fail2("OKM1101", "aggregate() with groupBy needs a limit, or .all(reason) for every group.");
+  }
+  const names = groupNames(spec.groups);
+  const orderBy = record.orderBy;
+  if (orderBy !== undefined) {
+    if (!isRecord(orderBy))
+      fail2("OKM1120", "orderBy must be an object of fields.");
+    for (const field of Object.keys(orderBy)) {
+      if (!names.includes(field)) {
+        const column = indexed.columns.get(field);
+        if (column !== undefined)
+          refuseHidden(column);
+        throwNamed2("OKM1120", field, names, `aggregate() orders by a groupBy field. Accepted names: ${list(names)}.`);
+      }
+    }
+  }
+  if (record.limit !== undefined)
+    readLimit(record.limit);
+  const base = readCall("find", table, {
+    where: record.where,
+    orderBy,
+    limit: record.limit,
+    signal: record.signal,
+    timeout: record.timeout,
+    route: record.route
+  }, mods.all, session.scope, session.schema.model[table], session.schema, view);
+  const call = { ...base, op: "aggregate", build: statement(spec, buckets) };
+  return readHandle(session, table, call, mods, (compiled, rows) => decode(compiled, rows));
+}
+function parse(table, record, buckets) {
+  const grouped = readGroups(table, record.groupBy, buckets);
+  const count = record.count === true;
+  if (record.count !== undefined && typeof record.count !== "boolean") {
+    fail2("OKM1120", "count is true when you want a row count.");
+  }
+  const calls = [];
+  for (const { option, tag, numeric } of FUNCTIONS) {
+    for (const column of columns(table, record[option], option)) {
+      const fits = numeric ? NUMERIC.test(column.dataType) : !UNORDERED.test(column.dataType);
+      if (!fits) {
+        fail2("OKM1124", `${option} does not apply to ${column.dataType} column ${column.field}. ${numeric ? "sum and avg take integer, real, and numeric columns." : "min and max need an ordered type."}`);
+      }
+      calls.push({ option, tag, column, decode: numeric ? sumDecode(column) : column.decode });
+    }
+  }
+  if (!count && calls.length === 0 && grouped.groups.length === 0) {
+    fail2("OKM1120", "aggregate() needs count, sum, avg, min, max, or groupBy.");
+  }
+  return { groups: grouped.groups, bucketed: grouped.bucketed, count, calls };
+}
+function sumDecode(column) {
+  const kind = typeof (column.decode?.("0") ?? "");
+  if (kind === "string")
+    return AS_TEXT;
+  if (kind === "number")
+    return AS_NUMBER;
+  return fail2("OKM1124", `sum and avg need a column whose value is a number or a string. ${column.field} reads as ${kind}.`);
+}
+function columns(table, value, option) {
+  if (value === undefined)
+    return [];
+  if (!Array.isArray(value))
+    fail2("OKM1120", `${option} must be a list of field names.`);
+  const out = [];
+  for (const name of value) {
+    const column = typeof name === "string" ? table.columns.get(name) : undefined;
+    if (typeof name !== "string" || column === undefined) {
+      throwNamed2("OKM1120", String(name), table.names, `Field ${String(name)} is not on ${table.model.name}. Accepted names: ${list(table.names)}.`);
+    }
+    if (column.hidden) {
+      fail2("OKM1120", `Field ${column.field} is hidden, so ${option} cannot read it.`);
+    }
+    out.push(column);
+  }
+  return out;
+}
+function groupNames(groups) {
+  const names = [];
+  for (const group of groups)
+    names.push(group.kind === "field" ? group.column.field : group.as);
+  return names;
+}
+function readGroups(table, value, buckets) {
+  if (value === undefined)
+    return { groups: [], bucketed: false };
+  if (!Array.isArray(value))
+    fail2("OKM1120", "groupBy must be a list of field names.");
+  const fields = new Set;
+  const aliases = new Set;
+  const groups = [];
+  let bucketed = false;
+  for (const item of value) {
+    if (typeof item === "object" && item !== null && BUCKET in item) {
+      if (buckets === undefined)
+        fail2("OKM1120", "bucket group");
+      groups.push(buckets.readBucket(table, item, fields, aliases));
+      bucketed = true;
+      continue;
+    }
+    const column = typeof item === "string" ? table.columns.get(item) : undefined;
+    if (typeof item !== "string" || column === undefined) {
+      throwNamed2("OKM1120", String(item), table.names, `Field ${String(item)} is not on ${table.model.name}. Accepted names: ${list(table.names)}.`);
+    }
+    if (column.hidden) {
+      fail2("OKM1120", `Field ${column.field} is hidden, so groupBy cannot read it.`);
+    }
+    if (aliases.has(column.field)) {
+      fail2("OKM1120", `group field ${column.field} clashes with a bucket as.`);
+    }
+    fields.add(column.field);
+    groups.push({ kind: "field", column });
+  }
+  return { groups, bucketed };
+}
+function statement(spec, buckets) {
+  return (schema, table, sink, outputs, call) => {
+    sink.mark(`aggregate|${table.model.name}|g:${groupMark(spec.groups)}|c:${String(spec.count)}|${spec.calls.map((item) => `${item.tag}:${item.column.field}`).join(",")}|`);
+    sink.text("select ");
+    const wrote = { at: 0 };
+    for (const group of spec.groups) {
+      if (group.kind === "field") {
+        put(sink, outputs, wrote, projectExpr(`t.${quote(group.column.sql)}`, group.column.dataType), {
+          key: `g:${group.column.field}`,
+          decode: group.column.decode
+        });
+      } else {
+        if (buckets === undefined)
+          fail2("OKM1120", "bucket group");
+        buckets.writeBucket(sink, outputs, wrote, group);
+      }
+    }
+    if (spec.count)
+      put(sink, outputs, wrote, "count(*)::text", { key: "c:", decode: undefined });
+    for (const item of spec.calls) {
+      const ref = `t.${quote(item.column.sql)}`;
+      const expr = item.option === "sum" || item.option === "avg" ? `${item.option}(${ref})::text` : projectExpr(`${item.option}(${ref})`, item.column.dataType);
+      put(sink, outputs, wrote, expr, {
+        key: `${item.tag}:${item.column.field}`,
+        decode: item.decode
+      });
+    }
+    sink.text(" from ");
+    sink.text(quote(table.model.sql));
+    sink.text(" t");
+    emitWhere(schema, table, call.where, sink, "t", 0);
+    if (spec.groups.length > 0) {
+      sink.text(" group by ");
+      if (spec.bucketed) {
+        if (buckets === undefined)
+          fail2("OKM1120", "bucket group");
+        buckets.writeGrouped(sink, groupNames(spec.groups), call.orderBy);
+      } else {
+        sink.text(spec.groups.map((group) => fieldRef(group)).join(", "));
+        if (isRecord(call.orderBy) && Object.keys(call.orderBy).length > 0) {
+          emitOrder(table, call.orderBy, sink, "t");
+        } else {
+          sink.text(" order by ");
+          sink.text(spec.groups.map((group) => `${fieldRef(group)} asc nulls last`).join(", "));
+        }
+      }
+    }
+    if (call.limit !== undefined) {
+      const limit = String(call.limit);
+      sink.text(" limit ");
+      sink.param(limit);
+      sink.mark(`l:${limit}`);
+    }
+    if (typeof call.all === "string")
+      sink.mark(`all:${call.all}`);
+  };
+}
+function groupMark(groups) {
+  const parts = [];
+  for (const group of groups) {
+    parts.push(group.kind === "field" ? group.column.field : `b:${group.column.field}:${group.as}:${group.unit}:${String(group.shift)}`);
+  }
+  return parts.join(",");
+}
+function fieldRef(group) {
+  if (group.kind !== "field")
+    return fail2("OKM1120", "bucket group");
+  return `t.${quote(group.column.sql)}`;
+}
+function put(sink, outputs, wrote, expr, out) {
+  if (wrote.at > 0)
+    sink.text(", ");
+  sink.text(expr);
+  outputs?.fields.push({ key: out.key, at: wrote.at, decode: out.decode });
+  wrote.at += 1;
+}
+function decode(compiled, rows) {
+  const items = [];
+  for (const row of rows) {
+    const item = {};
+    for (const field of compiled.outputs.fields) {
+      const value = row[field.at];
+      const tag = field.key.slice(0, 2);
+      const name = field.key.slice(2);
+      if (tag === "g:") {
+        item[name] = cell(value, field.decode);
+      } else if (tag === "c:") {
+        item.count = Number(value ?? 0);
+      } else {
+        const group = tag === "s:" ? "sum" : tag === "a:" ? "avg" : tag === "n:" ? "min" : "max";
+        const into = item[group] ??= {};
+        into[name] = cell(value, field.decode);
+      }
+    }
+    items.push(item);
+  }
+  return items;
+}
+export {
+  build
+};

package/dist/shared/{4az561fn.js → 1qa0qajz.js}

@@ -4,12 +4,9 @@ import {
   nextTransactionDepth,
   EMPTY_NOTICES,
   resultFrom
-} from "./cgbamx3e.js";
+} from "./kb2sdhfb.js";
 
 // src/adapters/pg/session.ts
-function quoteIdent2(name) {
-  return `"${name.replaceAll('"', '""')}"`;
-}
 function noticeBuffer() {
   let notices;
   return {
@@ -63,7 +60,7 @@ class WireSession {
         return resultFrom(result, this.notices.since(start));
       }, (error) => {
         this.onLeave();
-        return driverErrors().then((errors) => Promise.reject(errors.mapDriverError(error)));
+        return driverErrors().then((errors) => Promise.reject(errors.mapDriverError(stampQuery(error, text))));
       });
     }
     return this.watched(pending, watch, start, text);
@@ -110,7 +107,7 @@ class WireSession {
       return resultFrom(result, this.notices.since(start));
     }, (error) => {
       this.finishWatched(pending, watch, onAbort);
-      return classify(error, watch);
+      return classify(stampQuery(error, text), watch);
     });
   }
   finishWatched(pending, watch, onAbort) {
@@ -119,22 +116,10 @@ class WireSession {
     watch.signal.removeEventListener("abort", onAbort);
   }
 }
-async function* cursorStream(run, text, params) {
-  const name = "okm_stream";
-  await run("BEGIN");
-  try {
-    await run(`DECLARE ${name} NO SCROLL CURSOR FOR ${text}`, params);
-    for (;; ) {
-      const chunk = await run(`FETCH 64 FROM ${name}`);
-      if (chunk.rows.length === 0)
-        break;
-      yield chunk.rows;
-    }
-  } finally {
-    await run("ROLLBACK").catch(() => {
-      return;
-    });
-  }
+function stampQuery(error, text) {
+  if (typeof error === "object" && error !== null && !("query" in error))
+    Reflect.set(error, "query", text);
+  return error;
 }
 async function classify(error, watch) {
   const errors = await driverErrors();
@@ -151,4 +136,4 @@ async function classify(error, watch) {
   return Promise.reject(errors.mapDriverError(error));
 }
 
-export { quoteIdent2, noticeBuffer, WireSession, cursorStream };
+export { noticeBuffer, WireSession };