okmodel 0.6.0 → 0.8.0

package/dist/tooling/shared/yfr8hzhm.js

@@ -0,0 +1,604 @@
+import {
+  throwNamed,
+  orderHasRelevance,
+  emitRelationFrom,
+  needsOperatorSql,
+  decodeResult,
+  emitJoin,
+  emitWhere,
+  emitOrder,
+  emitPrimaryOrder,
+  refuseHidden,
+  selectedColumns,
+  rejectKeys,
+  readLimit,
+  projectExpr,
+  jsonExpr,
+  cell,
+  indexes,
+  quote,
+  list,
+  isRecord,
+  registerFailFix,
+  fail2
+} from "./f9mphxej.js";
+
+// src/runtime/include.ts
+registerFailFix("OKM1105", "Pass limit on the include, or call .all(reason).");
+var INCLUDE_OPTIONS = ["all", "include", "limit", "orderBy", "select", "where"];
+function emitIncludeSelect(include, alias, sink, outputs, column) {
+  const morph = include.morphTo;
+  if (morph !== undefined) {
+    if (column > 0 || outputs?.fields.length)
+      sink.text(", ");
+    sink.text(alias);
+    sink.text(".payload");
+    outputs?.includes.push({
+      name: include.name,
+      kind: "one",
+      at: column,
+      fields: [],
+      nested: [],
+      morph: {
+        typeKey: morph.typeKey,
+        targets: morph.targets.map((target) => ({
+          type: target.type,
+          fields: target.columns.map((item, at) => ({
+            key: item.field,
+            at,
+            decode: item.decode
+          }))
+        }))
+      }
+    });
+    return column + 1;
+  }
+  const selected = include.columns;
+  sink.mark(`i:${include.name}:${include.relation.kind}`);
+  if (include.relation.kind === "many") {
+    if (column > 0 || outputs?.fields.length)
+      sink.text(", ");
+    sink.text(alias);
+    sink.text(".rows");
+    outputs?.includes.push({
+      name: include.name,
+      kind: "many",
+      at: column,
+      fields: selected.map((item, index) => ({
+        key: item.field,
+        at: index,
+        decode: item.decode
+      })),
+      nested: include.nested.map((nested, index) => ({
+        name: nested.name,
+        kind: nested.relation.kind,
+        at: selected.length + index,
+        fields: nested.columns.map((item, at) => ({
+          key: item.field,
+          at,
+          decode: item.decode
+        })),
+        nested: []
+      }))
+    });
+    return column + 1;
+  }
+  const projected = include.presence === undefined ? selected : [...selected, include.presence];
+  const start = column;
+  for (let index = 0;index < projected.length; index += 1) {
+    if (start + index > 0)
+      sink.text(", ");
+    sink.text(alias);
+    sink.text(".c");
+    sink.text(String(index));
+    sink.mark(`s:${projected[index]?.field ?? ""}`);
+  }
+  const nestedStart = start + projected.length;
+  let cursor = nestedStart;
+  const nestedOut = [];
+  for (let index = 0;index < include.nested.length; index += 1) {
+    const nested = include.nested[index];
+    if (nested === undefined)
+      continue;
+    const width = nested.relation.kind === "many" ? 1 : nested.presence === undefined ? nested.columns.length : nested.columns.length + 1;
+    for (let slot = 0;slot < width; slot += 1) {
+      sink.text(", ");
+      sink.text(alias);
+      sink.text(".n");
+      sink.text(String(index));
+      sink.text("_");
+      sink.text(String(slot));
+    }
+    nestedOut.push({
+      name: nested.name,
+      kind: nested.relation.kind,
+      at: nested.relation.kind === "many" ? cursor : nested.presence === undefined ? cursor : cursor + nested.columns.length,
+      fields: nested.columns.map((item, at) => ({
+        key: item.field,
+        at: nested.relation.kind === "many" ? at : cursor + at,
+        decode: item.decode
+      })),
+      nested: []
+    });
+    cursor += width;
+  }
+  outputs?.includes.push({
+    name: include.name,
+    kind: "one",
+    at: start + include.anchor,
+    fields: selected.map((item, index) => ({
+      key: item.field,
+      at: start + index,
+      decode: item.decode
+    })),
+    nested: nestedOut
+  });
+  return cursor;
+}
+function emitIncludeJoin(schema, include, sink, parentAlias, alias, depth) {
+  if (include.morphTo !== undefined) {
+    emitMorphJoin(schema, include, sink, parentAlias, alias, depth);
+    return;
+  }
+  const child = indexes(schema).get(include.relation.table);
+  if (child === undefined) {
+    fail2("OKM1120", `Relation ${include.name} names ${include.relation.table}, which is not in the schema.`);
+  }
+  const childAlias = `r${String(depth)}_${include.name}`;
+  sink.text(" left join lateral (");
+  if (include.relation.kind === "many") {
+    sink.text("select coalesce(json_agg(item order by ord), '[]'::json) as rows from (select item, row_number() over () as ord from (select json_build_array(");
+    const selected = include.columns;
+    for (let index = 0;index < selected.length; index += 1) {
+      const item = selected[index];
+      if (item === undefined)
+        continue;
+      if (index > 0)
+        sink.text(", ");
+      sink.text(jsonExpr(`${childAlias}.${quote(item.sql)}`, item.dataType));
+    }
+    if (selected.length === 0)
+      sink.text("1");
+    for (const nested of include.nested) {
+      sink.text(", ");
+      emitNestedAggregate(schema, nested, sink, childAlias, depth + 1);
+    }
+    sink.text(") as item from ");
+    emitRelationFrom(schema, include.relation, child, sink, parentAlias, childAlias, depth + 1);
+    emitWhere(schema, child, include.where, sink, childAlias, depth + 1, true);
+    emitIncludeOrder(child, include.orderBy, sink, childAlias);
+    if (include.limit !== undefined) {
+      sink.text(" limit ");
+      sink.param(String(include.limit));
+      sink.mark(`l:${String(include.limit)}`);
+    }
+    sink.text(") limited) ordered) ");
+    sink.text(alias);
+    sink.text(" on true");
+    return;
+  }
+  sink.text("select ");
+  const projected = include.presence === undefined ? include.columns : [...include.columns, include.presence];
+  for (let index = 0;index < projected.length; index += 1) {
+    const item = projected[index];
+    if (item === undefined)
+      continue;
+    if (index > 0)
+      sink.text(", ");
+    sink.text(projectExpr(`${childAlias}.${quote(item.sql)}`, item.dataType));
+    sink.text(" as c");
+    sink.text(String(index));
+  }
+  if (projected.length === 0)
+    sink.text("1 as c0");
+  for (let index = 0;index < include.nested.length; index += 1) {
+    const nested = include.nested[index];
+    if (nested === undefined)
+      continue;
+    if (nested.relation.kind === "many") {
+      sink.text(", ");
+      sink.text("n");
+      sink.text(String(index));
+      sink.text(".rows as n");
+      sink.text(String(index));
+      sink.text("_0");
+      continue;
+    }
+    const width = nested.presence === undefined ? nested.columns.length : nested.columns.length + 1;
+    for (let slot = 0;slot < width; slot += 1) {
+      sink.text(", ");
+      sink.text("n");
+      sink.text(String(index));
+      sink.text(".c");
+      sink.text(String(slot));
+      sink.text(" as n");
+      sink.text(String(index));
+      sink.text("_");
+      sink.text(String(slot));
+    }
+  }
+  sink.text(" from ");
+  sink.text(quote(child.model.sql));
+  sink.text(" ");
+  sink.text(childAlias);
+  for (let index = 0;index < include.nested.length; index += 1) {
+    const nested = include.nested[index];
+    if (nested === undefined)
+      continue;
+    emitIncludeJoin(schema, nested, sink, childAlias, `n${String(index)}`, depth + 1);
+  }
+  sink.text(" where ");
+  emitJoin(sink, parentAlias, childAlias, include.relation);
+  emitWhere(schema, child, include.where, sink, childAlias, depth + 1, true);
+  sink.text(" limit 1) ");
+  sink.text(alias);
+  sink.text(" on true");
+}
+function emitNestedAggregate(schema, include, sink, alias, depth) {
+  if (include.relation.morph?.direction === "to") {
+    fail2("OKM1120", `Include ${include.name} is a morph and cannot be nested in this version.`);
+  }
+  const child = indexes(schema).get(include.relation.table);
+  if (child === undefined) {
+    fail2("OKM1120", `Relation ${include.name} names ${include.relation.table}, which is not in the schema.`);
+  }
+  const childAlias = `n${String(depth)}_${include.name}`;
+  sink.text("(select coalesce(json_agg(item order by ord), '[]'::json) from (select item, row_number() over () as ord from (select json_build_array(");
+  for (let index = 0;index < include.columns.length; index += 1) {
+    const item = include.columns[index];
+    if (item === undefined)
+      continue;
+    if (index > 0)
+      sink.text(", ");
+    sink.text(jsonExpr(`${childAlias}.${quote(item.sql)}`, item.dataType));
+  }
+  if (include.columns.length === 0)
+    sink.text("1");
+  sink.text(") as item from ");
+  emitRelationFrom(schema, include.relation, child, sink, alias, childAlias, depth);
+  emitWhere(schema, child, include.where, sink, childAlias, depth, true);
+  emitIncludeOrder(child, include.orderBy, sink, childAlias);
+  if (include.limit !== undefined) {
+    sink.text(" limit ");
+    sink.param(String(include.limit));
+    sink.mark(`l:${String(include.limit)}`);
+  }
+  sink.text(") limited) ordered)");
+}
+function includeColumns(table, select) {
+  const columns = selectedColumns(table, select);
+  if (select === undefined)
+    return omitHidden(columns);
+  for (const column of columns)
+    refuseHidden(column);
+  return columns;
+}
+function omitHidden(columns) {
+  for (const column of columns) {
+    if (!column.hidden)
+      continue;
+    return columns.filter((item) => !item.hidden);
+  }
+  return columns;
+}
+function readIncludes(schema, table, include, all) {
+  if (include === undefined)
+    return [];
+  if (!isRecord(include)) {
+    fail2("OKM1120", "include must be an object of relations.");
+  }
+  const parsed = [];
+  for (const name of Object.keys(include)) {
+    const relation = table.relations.get(name);
+    if (relation === undefined) {
+      throwNamed("OKM1120", name, [...table.relations.keys()], `Include ${name} is not a relation of ${table.model.name}. Accepted names: ${list([...table.relations.keys()])}.`);
+    }
+    parsed.push(parseInclude(schema, name, relation, include[name], all));
+  }
+  return parsed;
+}
+function parseInclude(schema, name, relation, value, all) {
+  if (relation.morph?.direction === "to")
+    return parseMorph(schema, name, relation, value);
+  if (value === true) {
+    if (relation.kind === "many" && all === undefined) {
+      fail2("OKM1105", `Include ${name} is to-many and has no limit. Pass limit, or call .all(reason).`);
+    }
+    const child = childIndex(schema, relation);
+    const columns = omitHidden(selectedColumns(child, undefined));
+    return {
+      name,
+      relation,
+      where: undefined,
+      orderBy: undefined,
+      limit: undefined,
+      columns,
+      ...anchored(child, columns),
+      nested: []
+    };
+  }
+  if (!isRecord(value)) {
+    fail2("OKM1120", `Include ${name} must be true or an object.`);
+  }
+  rejectKeys(value, INCLUDE_OPTIONS, `include ${name}`);
+  const reason = typeof value.all === "string" ? value.all : all;
+  let limit;
+  if (value.limit !== undefined)
+    limit = readLimit(value.limit);
+  if (relation.kind === "many" && limit === undefined && reason === undefined) {
+    fail2("OKM1105", `Include ${name} is to-many and has no limit. Pass limit, or call .all(reason).`);
+  }
+  const child = childIndex(schema, relation);
+  const columns = includeColumns(child, value.select);
+  const nestedValue = value.include;
+  const nested = nestedValue === undefined ? [] : readIncludes(schema, child, nestedValue, reason);
+  return {
+    name,
+    relation,
+    where: value.where,
+    orderBy: value.orderBy,
+    limit,
+    columns,
+    ...anchored(child, columns),
+    nested
+  };
+}
+function anchored(table, columns) {
+  const pk = table.model.primary[0];
+  if (pk === undefined)
+    return { presence: undefined, anchor: 0 };
+  const selected = columns.findIndex((column) => column.field === pk);
+  if (selected >= 0)
+    return { presence: undefined, anchor: selected };
+  return { presence: table.columns.get(pk), anchor: columns.length };
+}
+function childIndex(schema, relation) {
+  const child = indexes(schema).get(relation.table);
+  if (child === undefined) {
+    fail2("OKM1120", `Relation ${relation.name} names ${relation.table}, which is not in the schema.`);
+  }
+  return child;
+}
+function parseMorph(schema, name, relation, value) {
+  if (value !== true) {
+    fail2("OKM1120", `Include ${name} is a morph. Pass true. select, where, orderBy, and limit are not available on it.`);
+  }
+  const morph = relation.morph;
+  if (morph === undefined)
+    fail2("OKM1120", `Include ${name} is not a morph.`);
+  const planned = [];
+  for (const target of morph.targets) {
+    const child = indexes(schema).get(target.table);
+    if (child === undefined) {
+      fail2("OKM1120", `Relation ${name} names ${target.table}, which is not in the schema.`);
+    }
+    planned.push({
+      type: target.type,
+      table: target.table,
+      sql: target.sql,
+      keySql: target.keySql,
+      columns: omitHidden(selectedColumns(child, undefined))
+    });
+  }
+  return {
+    name,
+    relation,
+    where: undefined,
+    orderBy: undefined,
+    limit: undefined,
+    columns: [],
+    presence: undefined,
+    anchor: 0,
+    nested: [],
+    morphTo: {
+      typeKey: morph.typeField,
+      typeSql: morph.typeSql,
+      idSql: morph.idSql,
+      targets: planned
+    }
+  };
+}
+function emitMorphJoin(schema, include, sink, parentAlias, alias, depth) {
+  const morph = include.morphTo;
+  if (morph === undefined)
+    fail2("OKM1120", `Include ${include.name} is not a morph.`);
+  sink.text(" left join lateral (select payload from (");
+  for (let index = 0;index < morph.targets.length; index += 1) {
+    const target = morph.targets[index];
+    if (target === undefined)
+      continue;
+    if (index > 0)
+      sink.text(" union all ");
+    const childAlias = `m${String(depth)}_${include.name}_${String(index)}`;
+    const child = indexes(schema).get(target.table);
+    if (child === undefined) {
+      fail2("OKM1120", `Relation ${include.name} names ${target.table}, which is not in the schema.`);
+    }
+    sink.text("select json_build_array(");
+    sink.param(target.type);
+    sink.text("::text");
+    sink.mark(`mt:${target.type}`);
+    for (const column of target.columns) {
+      sink.text(", ");
+      sink.text(jsonExpr(`${childAlias}.${quote(column.sql)}`, column.dataType));
+    }
+    sink.text(") as payload from ");
+    sink.text(quote(target.sql));
+    sink.text(" ");
+    sink.text(childAlias);
+    sink.text(" where ");
+    sink.text(parentAlias);
+    sink.text(".");
+    sink.text(quote(morph.typeSql));
+    sink.text(" = ");
+    sink.param(target.type);
+    sink.text("::text");
+    sink.text(" and ");
+    sink.text(childAlias);
+    sink.text(".");
+    sink.text(quote(target.keySql));
+    sink.text(" = ");
+    sink.text(parentAlias);
+    sink.text(".");
+    sink.text(quote(morph.idSql));
+    emitWhere(schema, child, undefined, sink, childAlias, depth + 1, true);
+  }
+  sink.text(") matched limit 1) ");
+  sink.text(alias);
+  sink.text(" on true");
+}
+function decodeMorph(include, row) {
+  const morph = include.morph;
+  if (morph === undefined)
+    return null;
+  const raw = row[include.at];
+  if (raw === null || raw === undefined)
+    return null;
+  const parsed = JSON.parse(raw);
+  if (!Array.isArray(parsed))
+    return null;
+  const type = parsed[0];
+  if (typeof type !== "string")
+    return null;
+  let target;
+  for (const item of morph.targets) {
+    if (item.type === type)
+      target = item;
+  }
+  if (target === undefined)
+    return null;
+  const child = {};
+  for (const field of target.fields) {
+    child[field.key] = jsonCell(parsed[field.at + 1], field.decode);
+  }
+  child[morph.typeKey] = type;
+  return child;
+}
+function decodeIncluded(plan, rows, table) {
+  return decodeResult(plan, rows, table, decodeRowIncluded);
+}
+function decodeRowIncluded(outputs, row) {
+  const result = {};
+  for (const field of outputs.fields) {
+    result[field.key] = cell(row[field.at], field.decode);
+  }
+  for (const include of outputs.includes) {
+    result[include.name] = decodeInclude(include, row);
+  }
+  return result;
+}
+function decodeInclude(include, row) {
+  if (include.morph !== undefined)
+    return decodeMorph(include, row);
+  if (include.kind === "many") {
+    const raw = row[include.at];
+    if (raw === null || raw === undefined)
+      return [];
+    const parsed = JSON.parse(raw);
+    if (!Array.isArray(parsed))
+      return [];
+    const rows = [];
+    for (const item of parsed) {
+      if (!Array.isArray(item))
+        continue;
+      const child = {};
+      for (const field of include.fields) {
+        child[field.key] = jsonCell(item[field.at], field.decode);
+      }
+      for (const nested of include.nested) {
+        child[nested.name] = decodeJsonInclude(nested, item[nested.at]);
+      }
+      rows.push(child);
+    }
+    return rows;
+  }
+  const presence = row[include.at];
+  if (presence === null || presence === undefined)
+    return null;
+  const child = {};
+  for (const field of include.fields) {
+    child[field.key] = cell(row[field.at], field.decode);
+  }
+  for (const nested of include.nested) {
+    child[nested.name] = decodeInclude(nested, row);
+  }
+  return child;
+}
+function decodeJsonInclude(include, value) {
+  if (!Array.isArray(value))
+    return include.kind === "many" ? [] : null;
+  const rows = [];
+  for (const item of value) {
+    if (!Array.isArray(item))
+      continue;
+    const child = {};
+    for (const field of include.fields) {
+      child[field.key] = jsonCell(item[field.at], field.decode);
+    }
+    rows.push(child);
+  }
+  return include.kind === "one" ? rows[0] ?? null : rows;
+}
+function jsonCell(value, decode) {
+  if (value === null || value === undefined)
+    return null;
+  const wire = typeof value === "string" ? value : JSON.stringify(value);
+  return decode === undefined ? wire : decode(wire);
+}
+function emitIncludeOrder(child, orderBy, sink, alias) {
+  emitOrder(child, orderBy, sink, alias);
+  const named = isRecord(orderBy) ? Object.keys(orderBy) : [];
+  if (named.length === 0) {
+    if (child.model.primary.length > 0)
+      emitPrimaryOrder(child, sink, alias);
+    return;
+  }
+  for (const field of child.model.primary) {
+    if (named.includes(field))
+      continue;
+    const column = child.columns.get(field);
+    if (column === undefined)
+      continue;
+    sink.text(", ");
+    sink.text(alias);
+    sink.text(".");
+    sink.text(quote(column.sql));
+    sink.text(" asc");
+  }
+}
+function needsOperator(schema, call) {
+  const table = indexes(schema).get(call.table);
+  if (table === undefined)
+    return false;
+  return walkOperators(schema, table, call.include);
+}
+function walkOperators(schema, table, include) {
+  if (!isRecord(include))
+    return false;
+  for (const name of Object.keys(include)) {
+    const value = include[name];
+    if (!isRecord(value))
+      continue;
+    const relation = table.relations.get(name);
+    if (relation === undefined)
+      continue;
+    const child = indexes(schema).get(relation.table);
+    if (child === undefined)
+      continue;
+    if (needsOperatorSql(child, value.where) || orderHasRelevance(value.orderBy))
+      return true;
+    if (walkOperators(schema, child, value.include))
+      return true;
+  }
+  return false;
+}
+var includeHooks = {
+  read: readIncludes,
+  select: emitIncludeSelect,
+  join: emitIncludeJoin,
+  needsOperator
+};
+export {
+  decodeIncluded,
+  includeHooks
+};