@rocicorp/zero 0.25.0 → 0.26.0-canary.0

package/out/zqlite/src/query-builder.js.map

@@ -1 +1 @@
-{"version":3,"file":"query-builder.js","sourceRoot":"","sources":["../../../../zqlite/src/query-builder.ts"],"names":[],"mappings":"AAWA,OAAO,EAAC,GAAG,EAAC,MAAM,mBAAmB,CAAC;AAatC,MAAM,UAAU,gBAAgB,CAC9B,SAAiB,EACjB,OAAoC,EACpC,UAAkC,EAClC,OAAwC,EACxC,KAAe,EACf,OAA4B,EAC5B,KAAwB;IAExB,IAAI,KAAK,GAAG,GAAG,CAAA,UAAU,GAAG,CAAC,IAAI,CAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAC3C,GAAG,CAAA,GAAG,CACP,SAAS,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;IACjC,MAAM,WAAW,GAAe,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAEtE,IAAI,KAAK,EAAE,CAAC;QACV,WAAW,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;IAC3E,CAAC;IAED,IAAI,OAAO,EAAE,CAAC;QACZ,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,KAAK,GAAG,GAAG,CAAA,GAAG,KAAK,UAAU,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAA,OAAO,CAAC,EAAE,CAAC;IACnE,CAAC;IAED,OAAO,GAAG,CAAA,GAAG,KAAK,IAAI,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,UAAkC,EAClC,OAAoC;IAEpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,WAAW,GAAe,EAAE,CAAC;IACnC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,WAAW,CAAC,IAAI,CACd,GAAG,CAAA,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CACnE,CAAC;IACJ,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAe,EAAE,OAAgB;IAC5D,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,GAAG,CAAA,YAAY,GAAG,CAAC,IAAI,CAC5B,KAAK,CAAC,GAAG,CACP,CAAC,CAAC,EAAE,CACF,GAAG,CAAA,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,qBAAqB,CAChD,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAChC,EAAE,CACN,EACD,GAAG,CAAA,IAAI,CACR,EAAE,CAAC;IACN,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,CAAA,YAAY,GAAG,CAAC,IAAI,CAC5B,KAAK,CAAC,GAAG,CACP,CAAC,CAAC,EAAE,CAAC,GAAG,CAAA,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAChE,EACD,GAAG,CAAA,IAAI,CACR,EAAE,CAAC;IACN,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,OAA4B;IACvD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,KAAK,KAAK;YACR,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;gBAClC,CAAC,CAAC,GAAG,CAAA,IAAI,GAAG,CAAC,IAAI,CACb,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CACjC,YAAY,CAAC,SAAgC,CAAC,CAC/C,EACD,GAAG,CAAA,OAAO,CACX,GAAG;gBACN,CAAC,CAAC,GAAG,CAAA,MAAM,CAAC;QAChB,KAAK,IAAI;YACP,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;gBAClC,CAAC,CAAC,GAAG,CAAA,IAAI,GAAG,CAAC,IAAI,CACb,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CACjC,YAAY,CAAC,SAAgC,CAAC,CAC/C,EACD,GAAG,CAAA,MAAM,CACV,GAAG;gBACN,CAAC,CAAC,GAAG,CAAA,OAAO,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAuB;IACnD,MAAM,EAAC,EAAE,EAAC,GAAG,MAAM,CAAC;IACpB,IAAI,EAAE,KAAK,IAAI,IAAI,EAAE,KAAK,QAAQ,EAAE,CAAC;QACnC,QAAQ,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAC1B,KAAK,SAAS;gBACZ,OAAO,GAAG,CAAA,GAAG,kBAAkB,CAC7B,MAAM,CAAC,IAAI,CACZ,IAAI,GAAG,CAAC,qBAAqB,CAC5B,MAAM,CAAC,EAAE,CACV,iCAAiC,IAAI,CAAC,SAAS,CAC9C,MAAM,CAAC,KAAK,CAAC,KAAK,CACnB,IAAI,CAAC;YACR,KAAK,QAAQ;gBACX,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QACN,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAA,GAAG,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,qBAAqB;IACvE,+DAA+D;IAC/D,mDAAmD;IACnD,MAAM,CAAC,EAAE,KAAK,OAAO;QACnB,CAAC,CAAC,MAAM;QACR,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,WAAW;YACzB,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,MAAM,CAAC,EAAE,CAChB,IAAI,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;AAC1C,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAoB;IAC9C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,QAAQ;YACX,OAAO,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/B,KAAK,SAAS;YACZ,OAAO,GAAG,CAAA,GAAG,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;QACnE,KAAK,QAAQ;YACX,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;IACN,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,KAAc;IAC/B,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,OAAO,KAAK,KAAK,QAAQ;QAC9B,CAAC,CAAC,QAAQ;QACV,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ;YACzB,CAAC,CAAC,QAAQ;YACV,CAAC,CAAC,OAAO,KAAK,KAAK,SAAS;gBAC1B,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,MAAM,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,CAAU,EAAE,IAAe;IACtD,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,SAAS;YACZ,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,KAAK,QAAQ,CAAC;QACd,KAAK,QAAQ,CAAC;QACd,KAAK,MAAM;YACT,OAAO,CAAC,CAAC;QACX,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,SAAS,sBAAsB,CAC7B,KAAY,EACZ,OAA4B,EAC5B,KAAe,EACf,WAAwC;IAExC,MAAM,WAAW,GAAe,EAAE,CAAC;IACnC,MAAM,EAAC,GAAG,EAAE,IAAI,EAAE,KAAK,EAAC,GAAG,KAAK,CAAC;IAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,MAAM,eAAe,GAAG,YAAY,CAClC,IAAI,CAAC,MAAM,CAAC,EACZ,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CACzB,CAAC;gBACF,IAAI,UAAU,KAAK,KAAK,EAAE,CAAC;oBACzB,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,KAAK,CAAC,IAAI,CACR,GAAG,CAAA,IAAI,eAAe,eAAe,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,eAAe,GAAG,CAC/E,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,OAAsB,CAAC;wBACvB,KAAK,CAAC,IAAI,CACR,GAAG,CAAA,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,eAAe,GAAG,CACjF,CAAC;oBACJ,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,UAA2B,CAAC;oBAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,KAAK,CAAC,IAAI,CACR,GAAG,CAAA,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,eAAe,GAAG,CACjF,CAAC;oBACJ,CAAC;yBAAM,CAAC;wBACN,OAAsB,CAAC;wBACvB,KAAK,CAAC,IAAI,CACR,GAAG,CAAA,IAAI,eAAe,eAAe,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,eAAe,GAAG,CAC/E,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC1B,KAAK,CAAC,IAAI,CACR,GAAG,CAAA,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,YAAY,CACxC,IAAI,CAAC,MAAM,CAAC,EACZ,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CACzB,EAAE,CACJ,CAAC;YACJ,CAAC;QACH,CAAC;QACD,WAAW,CAAC,IAAI,CAAC,GAAG,CAAA,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAA,OAAO,CAAC,GAAG,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,WAAW,CAAC,IAAI,CACd,GAAG,CAAA,IAAI,GAAG,CAAC,IAAI,CACb,KAAK,CAAC,GAAG,CACP,CAAC,CAAC,EAAE,CACF,GAAG,CAAA,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,YAAY,CACtC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACV,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CACvB,EAAE,CACN,EACD,GAAG,CAAA,OAAO,CACX,GAAG,CACL,CAAC;IACJ,CAAC;IAED,OAAO,GAAG,CAAA,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAA,MAAM,CAAC,GAAG,CAAC;AACpD,CAAC"}
+{"version":3,"file":"query-builder.js","sources":["../../../../zqlite/src/query-builder.ts"],"sourcesContent":["import type {SQLQuery} from '@databases/sql';\nimport type {\n  Condition,\n  Ordering,\n  SimpleCondition,\n  ValuePosition,\n} from '../../zero-protocol/src/ast.ts';\nimport type {\n  SchemaValue,\n  ValueType,\n} from '../../zero-schema/src/table-schema.ts';\nimport {sql} from './internal/sql.ts';\nimport type {Constraint} from '../../zql/src/ivm/constraint.ts';\nimport type {Start} from '../../zql/src/ivm/operator.ts';\n\n/**\n * Condition type without correlated subqueries.\n * This matches the output of transformFilters from zql/builder/filter.ts\n */\nexport type NoSubqueryCondition = Exclude<\n  Condition,\n  {type: 'correlatedSubquery'}\n>;\n\nexport function buildSelectQuery(\n  tableName: string,\n  columns: Record<string, SchemaValue>,\n  constraint: Constraint | undefined,\n  filters: NoSubqueryCondition | undefined,\n  order: Ordering,\n  reverse: boolean | undefined,\n  start: Start | undefined,\n) {\n  let query = sql`SELECT ${sql.join(\n    Object.keys(columns).map(c => sql.ident(c)),\n    sql`,`,\n  )} FROM ${sql.ident(tableName)}`;\n  const constraints: SQLQuery[] = constraintsToSQL(constraint, columns);\n\n  if (start) {\n    constraints.push(gatherStartConstraints(start, reverse, order, columns));\n  }\n\n  if (filters) {\n    constraints.push(filtersToSQL(filters));\n  }\n\n  if (constraints.length > 0) {\n    query = sql`${query} WHERE ${sql.join(constraints, sql` AND `)}`;\n  }\n\n  return sql`${query} ${orderByToSQL(order, !!reverse)}`;\n}\n\nexport function constraintsToSQL(\n  constraint: Constraint | undefined,\n  columns: Record<string, SchemaValue>,\n) {\n  if (!constraint) {\n    return [];\n  }\n\n  const constraints: SQLQuery[] = [];\n  for (const [key, value] of Object.entries(constraint)) {\n    constraints.push(\n      sql`${sql.ident(key)} = ${toSQLiteType(value, columns[key].type)}`,\n    );\n  }\n\n  return constraints;\n}\n\nexport function orderByToSQL(order: Ordering, reverse: boolean): SQLQuery {\n  if (reverse) {\n    return sql`ORDER BY ${sql.join(\n      order.map(\n        s =>\n          sql`${sql.ident(s[0])} ${sql.__dangerous__rawValue(\n            s[1] === 'asc' ? 'desc' : 'asc',\n          )}`,\n      ),\n      sql`, `,\n    )}`;\n  } else {\n    return sql`ORDER BY ${sql.join(\n      order.map(\n        s => sql`${sql.ident(s[0])} ${sql.__dangerous__rawValue(s[1])}`,\n      ),\n      sql`, `,\n    )}`;\n  }\n}\n\n/**\n * Converts filters (conditions) to SQL WHERE clause.\n * This applies all filters present in the AST for a query to the source.\n */\nexport function filtersToSQL(filters: NoSubqueryCondition): SQLQuery {\n  switch (filters.type) {\n    case 'simple':\n      return simpleConditionToSQL(filters);\n    case 'and':\n      return filters.conditions.length > 0\n        ? sql`(${sql.join(\n            filters.conditions.map(condition =>\n              filtersToSQL(condition as NoSubqueryCondition),\n            ),\n            sql` AND `,\n          )})`\n        : sql`TRUE`;\n    case 'or':\n      return filters.conditions.length > 0\n        ? sql`(${sql.join(\n            filters.conditions.map(condition =>\n              filtersToSQL(condition as NoSubqueryCondition),\n            ),\n            sql` OR `,\n          )})`\n        : sql`FALSE`;\n  }\n}\n\nfunction simpleConditionToSQL(filter: SimpleCondition): SQLQuery {\n  const {op} = filter;\n  if (op === 'IN' || op === 'NOT IN') {\n    switch (filter.right.type) {\n      case 'literal':\n        return sql`${valuePositionToSQL(\n          filter.left,\n        )} ${sql.__dangerous__rawValue(\n          filter.op,\n        )} (SELECT value FROM json_each(${JSON.stringify(\n          filter.right.value,\n        )}))`;\n      case 'static':\n        throw new Error(\n          'Static parameters must be replaced before conversion to SQL',\n        );\n    }\n  }\n  return sql`${valuePositionToSQL(filter.left)} ${sql.__dangerous__rawValue(\n    // SQLite's LIKE operator is case-insensitive by default, so we\n    // convert ILIKE to LIKE and NOT ILIKE to NOT LIKE.\n    filter.op === 'ILIKE'\n      ? 'LIKE'\n      : filter.op === 'NOT ILIKE'\n        ? 'NOT LIKE'\n        : filter.op,\n  )} ${valuePositionToSQL(filter.right)}`;\n}\n\nfunction valuePositionToSQL(value: ValuePosition): SQLQuery {\n  switch (value.type) {\n    case 'column':\n      return sql.ident(value.name);\n    case 'literal':\n      return sql`${toSQLiteType(value.value, getJsType(value.value))}`;\n    case 'static':\n      throw new Error(\n        'Static parameters must be replaced before conversion to SQL',\n      );\n  }\n}\n\nfunction getJsType(value: unknown): ValueType {\n  if (value === null) {\n    return 'null';\n  }\n  return typeof value === 'string'\n    ? 'string'\n    : typeof value === 'number'\n      ? 'number'\n      : typeof value === 'boolean'\n        ? 'boolean'\n        : 'json';\n}\n\nexport function toSQLiteType(v: unknown, type: ValueType): unknown {\n  switch (type) {\n    case 'boolean':\n      return v === null ? null : v ? 1 : 0;\n    case 'number':\n    case 'string':\n    case 'null':\n      return v;\n    case 'json':\n      return JSON.stringify(v);\n  }\n}\n\n/**\n * The ordering could be complex such as:\n * `ORDER BY a ASC, b DESC, c ASC`\n *\n * In those cases, we need to encode the constraints as various\n * `OR` clauses.\n *\n * E.g.,\n *\n * to get the row after (a = 1, b = 2, c = 3) would be:\n *\n * `WHERE a > 1 OR (a = 1 AND b < 2) OR (a = 1 AND b = 2 AND c > 3)`\n *\n * - after vs before flips the comparison operators.\n * - inclusive adds a final `OR` clause for the exact match.\n */\nfunction gatherStartConstraints(\n  start: Start,\n  reverse: boolean | undefined,\n  order: Ordering,\n  columnTypes: Record<string, SchemaValue>,\n): SQLQuery {\n  const constraints: SQLQuery[] = [];\n  const {row: from, basis} = start;\n\n  for (let i = 0; i < order.length; i++) {\n    const group: SQLQuery[] = [];\n    const [iField, iDirection] = order[i];\n    for (let j = 0; j <= i; j++) {\n      if (j === i) {\n        const constraintValue = toSQLiteType(\n          from[iField],\n          columnTypes[iField].type,\n        );\n        if (iDirection === 'asc') {\n          if (!reverse) {\n            group.push(\n              sql`(${constraintValue} IS NULL OR ${sql.ident(iField)} > ${constraintValue})`,\n            );\n          } else {\n            reverse satisfies true;\n            group.push(\n              sql`(${sql.ident(iField)} IS NULL OR ${sql.ident(iField)} < ${constraintValue})`,\n            );\n          }\n        } else {\n          iDirection satisfies 'desc';\n          if (!reverse) {\n            group.push(\n              sql`(${sql.ident(iField)} IS NULL OR ${sql.ident(iField)} < ${constraintValue})`,\n            );\n          } else {\n            reverse satisfies true;\n            group.push(\n              sql`(${constraintValue} IS NULL OR ${sql.ident(iField)} > ${constraintValue})`,\n            );\n          }\n        }\n      } else {\n        const [jField] = order[j];\n        group.push(\n          sql`${sql.ident(jField)} IS ${toSQLiteType(\n            from[jField],\n            columnTypes[jField].type,\n          )}`,\n        );\n      }\n    }\n    constraints.push(sql`(${sql.join(group, sql` AND `)})`);\n  }\n\n  if (basis === 'at') {\n    constraints.push(\n      sql`(${sql.join(\n        order.map(\n          s =>\n            sql`${sql.ident(s[0])} IS ${toSQLiteType(\n              from[s[0]],\n              columnTypes[s[0]].type,\n            )}`,\n        ),\n        sql` AND `,\n      )})`,\n    );\n  }\n\n  return sql`(${sql.join(constraints, sql` OR `)})`;\n}\n"],"names":[],"mappings":";AAwBO,SAAS,iBACd,WACA,SACA,YACA,SACA,OACA,SACA,OACA;AACA,MAAI,QAAQ,aAAa,IAAI;AAAA,IAC3B,OAAO,KAAK,OAAO,EAAE,IAAI,CAAA,MAAK,IAAI,MAAM,CAAC,CAAC;AAAA,IAC1C;AAAA,EAAA,CACD,SAAS,IAAI,MAAM,SAAS,CAAC;AAC9B,QAAM,cAA0B,iBAAiB,YAAY,OAAO;AAEpE,MAAI,OAAO;AACT,gBAAY,KAAK,uBAAuB,OAAO,SAAS,OAAO,OAAO,CAAC;AAAA,EACzE;AAEA,MAAI,SAAS;AACX,gBAAY,KAAK,aAAa,OAAO,CAAC;AAAA,EACxC;AAEA,MAAI,YAAY,SAAS,GAAG;AAC1B,YAAQ,MAAM,KAAK,UAAU,IAAI,KAAK,aAAa,UAAU,CAAC;AAAA,EAChE;AAEA,SAAO,MAAM,KAAK,IAAI,aAAa,OAAO,CAAC,CAAC,OAAO,CAAC;AACtD;AAEO,SAAS,iBACd,YACA,SACA;AACA,MAAI,CAAC,YAAY;AACf,WAAO,CAAA;AAAA,EACT;AAEA,QAAM,cAA0B,CAAA;AAChC,aAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,gBAAY;AAAA,MACV,MAAM,IAAI,MAAM,GAAG,CAAC,MAAM,aAAa,OAAO,QAAQ,GAAG,EAAE,IAAI,CAAC;AAAA,IAAA;AAAA,EAEpE;AAEA,SAAO;AACT;AAEO,SAAS,aAAa,OAAiB,SAA4B;AACxE,MAAI,SAAS;AACX,WAAO,eAAe,IAAI;AAAA,MACxB,MAAM;AAAA,QACJ,CAAA,MACE,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI;AAAA,UAC3B,EAAE,CAAC,MAAM,QAAQ,SAAS;AAAA,QAAA,CAC3B;AAAA,MAAA;AAAA,MAEL;AAAA,IAAA,CACD;AAAA,EACH,OAAO;AACL,WAAO,eAAe,IAAI;AAAA,MACxB,MAAM;AAAA,QACJ,CAAA,MAAK,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,sBAAsB,EAAE,CAAC,CAAC,CAAC;AAAA,MAAA;AAAA,MAE/D;AAAA,IAAA,CACD;AAAA,EACH;AACF;AAMO,SAAS,aAAa,SAAwC;AACnE,UAAQ,QAAQ,MAAA;AAAA,IACd,KAAK;AACH,aAAO,qBAAqB,OAAO;AAAA,IACrC,KAAK;AACH,aAAO,QAAQ,WAAW,SAAS,IAC/B,OAAO,IAAI;AAAA,QACT,QAAQ,WAAW;AAAA,UAAI,CAAA,cACrB,aAAa,SAAgC;AAAA,QAAA;AAAA,QAE/C;AAAA,MAAA,CACD,MACD;AAAA,IACN,KAAK;AACH,aAAO,QAAQ,WAAW,SAAS,IAC/B,OAAO,IAAI;AAAA,QACT,QAAQ,WAAW;AAAA,UAAI,CAAA,cACrB,aAAa,SAAgC;AAAA,QAAA;AAAA,QAE/C;AAAA,MAAA,CACD,MACD;AAAA,EAAA;AAEV;AAEA,SAAS,qBAAqB,QAAmC;AAC/D,QAAM,EAAC,OAAM;AACb,MAAI,OAAO,QAAQ,OAAO,UAAU;AAClC,YAAQ,OAAO,MAAM,MAAA;AAAA,MACnB,KAAK;AACH,eAAO,MAAM;AAAA,UACX,OAAO;AAAA,QAAA,CACR,IAAI,IAAI;AAAA,UACP,OAAO;AAAA,QAAA,CACR,iCAAiC,KAAK;AAAA,UACrC,OAAO,MAAM;AAAA,QAAA,CACd;AAAA,MACH,KAAK;AACH,cAAM,IAAI;AAAA,UACR;AAAA,QAAA;AAAA,IACF;AAAA,EAEN;AACA,SAAO,MAAM,mBAAmB,OAAO,IAAI,CAAC,IAAI,IAAI;AAAA;AAAA;AAAA,IAGlD,OAAO,OAAO,UACV,SACA,OAAO,OAAO,cACZ,aACA,OAAO;AAAA,EAAA,CACd,IAAI,mBAAmB,OAAO,KAAK,CAAC;AACvC;AAEA,SAAS,mBAAmB,OAAgC;AAC1D,UAAQ,MAAM,MAAA;AAAA,IACZ,KAAK;AACH,aAAO,IAAI,MAAM,MAAM,IAAI;AAAA,IAC7B,KAAK;AACH,aAAO,MAAM,aAAa,MAAM,OAAO,UAAU,MAAM,KAAK,CAAC,CAAC;AAAA,IAChE,KAAK;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MAAA;AAAA,EACF;AAEN;AAEA,SAAS,UAAU,OAA2B;AAC5C,MAAI,UAAU,MAAM;AAClB,WAAO;AAAA,EACT;AACA,SAAO,OAAO,UAAU,WACpB,WACA,OAAO,UAAU,WACf,WACA,OAAO,UAAU,YACf,YACA;AACV;AAEO,SAAS,aAAa,GAAY,MAA0B;AACjE,UAAQ,MAAA;AAAA,IACN,KAAK;AACH,aAAO,MAAM,OAAO,OAAO,IAAI,IAAI;AAAA,IACrC,KAAK;AAAA,IACL,KAAK;AAAA,IACL,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO,KAAK,UAAU,CAAC;AAAA,EAAA;AAE7B;AAkBA,SAAS,uBACP,OACA,SACA,OACA,aACU;AACV,QAAM,cAA0B,CAAA;AAChC,QAAM,EAAC,KAAK,MAAM,MAAA,IAAS;AAE3B,WAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,UAAM,QAAoB,CAAA;AAC1B,UAAM,CAAC,QAAQ,UAAU,IAAI,MAAM,CAAC;AACpC,aAAS,IAAI,GAAG,KAAK,GAAG,KAAK;AAC3B,UAAI,MAAM,GAAG;AACX,cAAM,kBAAkB;AAAA,UACtB,KAAK,MAAM;AAAA,UACX,YAAY,MAAM,EAAE;AAAA,QAAA;AAEtB,YAAI,eAAe,OAAO;AACxB,cAAI,CAAC,SAAS;AACZ,kBAAM;AAAA,cACJ,OAAO,eAAe,eAAe,IAAI,MAAM,MAAM,CAAC,MAAM,eAAe;AAAA,YAAA;AAAA,UAE/E,OAAO;AAEL,kBAAM;AAAA,cACJ,OAAO,IAAI,MAAM,MAAM,CAAC,eAAe,IAAI,MAAM,MAAM,CAAC,MAAM,eAAe;AAAA,YAAA;AAAA,UAEjF;AAAA,QACF,OAAO;AAEL,cAAI,CAAC,SAAS;AACZ,kBAAM;AAAA,cACJ,OAAO,IAAI,MAAM,MAAM,CAAC,eAAe,IAAI,MAAM,MAAM,CAAC,MAAM,eAAe;AAAA,YAAA;AAAA,UAEjF,OAAO;AAEL,kBAAM;AAAA,cACJ,OAAO,eAAe,eAAe,IAAI,MAAM,MAAM,CAAC,MAAM,eAAe;AAAA,YAAA;AAAA,UAE/E;AAAA,QACF;AAAA,MACF,OAAO;AACL,cAAM,CAAC,MAAM,IAAI,MAAM,CAAC;AACxB,cAAM;AAAA,UACJ,MAAM,IAAI,MAAM,MAAM,CAAC,OAAO;AAAA,YAC5B,KAAK,MAAM;AAAA,YACX,YAAY,MAAM,EAAE;AAAA,UAAA,CACrB;AAAA,QAAA;AAAA,MAEL;AAAA,IACF;AACA,gBAAY,KAAK,OAAO,IAAI,KAAK,OAAO,UAAU,CAAC,GAAG;AAAA,EACxD;AAEA,MAAI,UAAU,MAAM;AAClB,gBAAY;AAAA,MACV,OAAO,IAAI;AAAA,QACT,MAAM;AAAA,UACJ,CAAA,MACE,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO;AAAA,YAC1B,KAAK,EAAE,CAAC,CAAC;AAAA,YACT,YAAY,EAAE,CAAC,CAAC,EAAE;AAAA,UAAA,CACnB;AAAA,QAAA;AAAA,QAEL;AAAA,MAAA,CACD;AAAA,IAAA;AAAA,EAEL;AAEA,SAAO,OAAO,IAAI,KAAK,aAAa,SAAS,CAAC;AAChD;"}

package/out/zqlite/src/query-delegate.d.ts

@@ -1,30 +1,16 @@
 import type { LogContext } from '@rocicorp/logger';
 import type { LogConfig } from '../../otel/src/log-options.ts';
-import type { Schema } from '../../zero-schema/src/builder/schema-builder.ts';
-import type { FilterInput } from '../../zql/src/ivm/filter-operators.ts';
-import { MemoryStorage } from '../../zql/src/ivm/memory-storage.ts';
-import type { Input } from '../../zql/src/ivm/operator.ts';
-import type { Source, SourceInput } from '../../zql/src/ivm/source.ts';
-import type { CommitListener, QueryDelegate } from '../../zql/src/query/query-delegate.ts';
+import type { Schema } from '../../zero-types/src/schema.ts';
+import type { Source } from '../../zql/src/ivm/source.ts';
+import { QueryDelegateBase } from '../../zql/src/query/query-delegate-base.ts';
+import type { CommitListener } from '../../zql/src/query/query-delegate.ts';
 import type { Database } from './db.ts';
-export declare class QueryDelegateImpl implements QueryDelegate {
+export declare class QueryDelegateImpl extends QueryDelegateBase {
     #private;
     readonly defaultQueryComplete = true;
     constructor(lc: LogContext, db: Database, schema: Schema, logConfig?: LogConfig);
     getSource(tableName: string): Source;
-    createStorage(): MemoryStorage;
-    decorateSourceInput(input: SourceInput): Input;
-    decorateInput(input: Input): Input;
-    decorateFilterInput(input: FilterInput): FilterInput;
-    addServerQuery(): () => void;
-    addEdge(): void;
-    addCustomQuery(): () => void;
-    updateServerQuery(): void;
-    updateCustomQuery(): void;
-    flushQueryChanges(): void;
     onTransactionCommit(cb: CommitListener): () => void;
     batchViewUpdates<T>(applyViewUpdates: () => T): T;
-    assertValidRunOptions(): void;
-    addMetric(): void;
 }
 //# sourceMappingURL=query-delegate.d.ts.map

package/out/zqlite/src/query-delegate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"query-delegate.d.ts","sourceRoot":"","sources":["../../../../zqlite/src/query-delegate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,+BAA+B,CAAC;AAC7D,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,iDAAiD,CAAC;AAC5E,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,uCAAuC,CAAC;AACvE,OAAO,EAAC,aAAa,EAAC,MAAM,qCAAqC,CAAC;AAClE,OAAO,KAAK,EAAC,KAAK,EAAC,MAAM,+BAA+B,CAAC;AACzD,OAAO,KAAK,EAAC,MAAM,EAAE,WAAW,EAAC,MAAM,6BAA6B,CAAC;AACrE,OAAO,KAAK,EACV,cAAc,EACd,aAAa,EACd,MAAM,uCAAuC,CAAC;AAC/C,OAAO,KAAK,EAAC,QAAQ,EAAC,MAAM,SAAS,CAAC;AAGtC,qBAAa,iBAAkB,YAAW,aAAa;;IAMrD,QAAQ,CAAC,oBAAoB,QAAQ;gBAInC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,QAAQ,EACZ,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,SAAS;IAevB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAqBpC,aAAa;IAIb,mBAAmB,CAAC,KAAK,EAAE,WAAW,GAAG,KAAK;IAI9C,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,KAAK;IAIlC,mBAAmB,CAAC,KAAK,EAAE,WAAW,GAAG,WAAW;IAIpD,cAAc;IAId,OAAO;IAEP,cAAc;IAId,iBAAiB;IACjB,iBAAiB;IACjB,iBAAiB;IACjB,mBAAmB,CAAC,EAAE,EAAE,cAAc;IAMtC,gBAAgB,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC;IAOjD,qBAAqB;IACrB,SAAS;CACV"}
+{"version":3,"file":"query-delegate.d.ts","sourceRoot":"","sources":["../../../../zqlite/src/query-delegate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,UAAU,EAAC,MAAM,kBAAkB,CAAC;AACjD,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,+BAA+B,CAAC;AAC7D,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,gCAAgC,CAAC;AAC3D,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAC,iBAAiB,EAAC,MAAM,4CAA4C,CAAC;AAC7E,OAAO,KAAK,EAAC,cAAc,EAAC,MAAM,uCAAuC,CAAC;AAC1E,OAAO,KAAK,EAAC,QAAQ,EAAC,MAAM,SAAS,CAAC;AAGtC,qBAAa,iBAAkB,SAAQ,iBAAiB;;IAMtD,QAAQ,CAAC,oBAAoB,QAAQ;gBAInC,EAAE,EAAE,UAAU,EACd,EAAE,EAAE,QAAQ,EACZ,MAAM,EAAE,MAAM,EACd,SAAS,CAAC,EAAE,SAAS;IAevB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IAqBpC,mBAAmB,CAAC,EAAE,EAAE,cAAc;IAM7B,gBAAgB,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC;CAO3D"}

package/out/zqlite/src/query-delegate.js

@@ -1,72 +1,58 @@
-import { MemoryStorage } from "../../zql/src/ivm/memory-storage.js";
+import { QueryDelegateBase } from "../../zql/src/query/query-delegate-base.js";
 import { TableSource } from "./table-source.js";
-export class QueryDelegateImpl {
-    #lc;
-    #db;
-    #schema;
-    #sources;
-    #logConfig;
-    defaultQueryComplete = true;
-    #commitObservers = new Set();
-    constructor(lc, db, schema, logConfig) {
-        this.#lc = lc.withContext('class', 'QueryDelegateImpl');
-        this.#db = db;
-        this.#schema = schema;
-        this.#sources = new Map();
-        this.#logConfig = logConfig ?? {
-            format: 'text',
-            ivmSampling: 0,
-            level: 'info',
-            slowHydrateThreshold: 0,
-            slowRowThreshold: 0,
-        };
-    }
-    getSource(tableName) {
-        let source = this.#sources.get(tableName);
-        if (source) {
-            return source;
-        }
-        const tableSchema = this.#schema.tables[tableName];
-        source = new TableSource(this.#lc, this.#logConfig, this.#db, tableName, tableSchema.columns, tableSchema.primaryKey);
-        this.#sources.set(tableName, source);
-        return source;
-    }
-    createStorage() {
-        return new MemoryStorage();
-    }
-    decorateSourceInput(input) {
-        return input;
-    }
-    decorateInput(input) {
-        return input;
-    }
-    decorateFilterInput(input) {
-        return input;
-    }
-    addServerQuery() {
-        return () => { };
-    }
-    addEdge() { }
-    addCustomQuery() {
-        return () => { };
-    }
-    updateServerQuery() { }
-    updateCustomQuery() { }
-    flushQueryChanges() { }
-    onTransactionCommit(cb) {
-        this.#commitObservers.add(cb);
-        return () => {
-            this.#commitObservers.delete(cb);
-        };
-    }
-    batchViewUpdates(applyViewUpdates) {
-        const ret = applyViewUpdates();
-        for (const observer of this.#commitObservers) {
-            observer();
-        }
-        return ret;
-    }
-    assertValidRunOptions() { }
-    addMetric() { }
+class QueryDelegateImpl extends QueryDelegateBase {
+  #lc;
+  #db;
+  #schema;
+  #sources = /* @__PURE__ */ new Map();
+  #logConfig;
+  defaultQueryComplete = true;
+  #commitObservers = /* @__PURE__ */ new Set();
+  constructor(lc, db, schema, logConfig) {
+    super();
+    this.#lc = lc.withContext("class", "QueryDelegateImpl");
+    this.#db = db;
+    this.#schema = schema;
+    this.#logConfig = logConfig ?? {
+      format: "text",
+      ivmSampling: 0,
+      level: "info",
+      slowHydrateThreshold: 0,
+      slowRowThreshold: 0
+    };
+  }
+  getSource(tableName) {
+    let source = this.#sources.get(tableName);
+    if (source) {
+      return source;
+    }
+    const tableSchema = this.#schema.tables[tableName];
+    source = new TableSource(
+      this.#lc,
+      this.#logConfig,
+      this.#db,
+      tableName,
+      tableSchema.columns,
+      tableSchema.primaryKey
+    );
+    this.#sources.set(tableName, source);
+    return source;
+  }
+  onTransactionCommit(cb) {
+    this.#commitObservers.add(cb);
+    return () => {
+      this.#commitObservers.delete(cb);
+    };
+  }
+  batchViewUpdates(applyViewUpdates) {
+    const ret = applyViewUpdates();
+    for (const observer of this.#commitObservers) {
+      observer();
+    }
+    return ret;
+  }
 }
-//# sourceMappingURL=query-delegate.js.map
+export {
+  QueryDelegateImpl
+};
+//# sourceMappingURL=query-delegate.js.map

package/out/zqlite/src/query-delegate.js.map

@@ -1 +1 @@
-{"version":3,"file":"query-delegate.js","sourceRoot":"","sources":["../../../../zqlite/src/query-delegate.ts"],"names":[],"mappings":"AAIA,OAAO,EAAC,aAAa,EAAC,MAAM,qCAAqC,CAAC;AAQlE,OAAO,EAAC,WAAW,EAAC,MAAM,mBAAmB,CAAC;AAE9C,MAAM,OAAO,iBAAiB;IACnB,GAAG,CAAa;IAChB,GAAG,CAAW;IACd,OAAO,CAAS;IAChB,QAAQ,CAAsB;IAC9B,UAAU,CAAY;IACtB,oBAAoB,GAAG,IAAI,CAAC;IAC5B,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAElD,YACE,EAAc,EACd,EAAY,EACZ,MAAc,EACd,SAAqB;QAErB,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QACxD,IAAI,CAAC,GAAG,GAAG,EAAE,CAAC;QACd,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,SAAS,IAAI;YAC7B,MAAM,EAAE,MAAM;YACd,WAAW,EAAE,CAAC;YACd,KAAK,EAAE,MAAM;YACb,oBAAoB,EAAE,CAAC;YACvB,gBAAgB,EAAE,CAAC;SACpB,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,SAAiB;QACzB,IAAI,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEnD,MAAM,GAAG,IAAI,WAAW,CACtB,IAAI,CAAC,GAAG,EACR,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,GAAG,EACR,SAAS,EACT,WAAW,CAAC,OAAO,EACnB,WAAW,CAAC,UAAU,CACvB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,aAAa,EAAE,CAAC;IAC7B,CAAC;IAED,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,aAAa,CAAC,KAAY;QACxB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,mBAAmB,CAAC,KAAkB;QACpC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc;QACZ,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,KAAI,CAAC;IAEZ,cAAc;QACZ,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;IAClB,CAAC;IAED,iBAAiB,KAAI,CAAC;IACtB,iBAAiB,KAAI,CAAC;IACtB,iBAAiB,KAAI,CAAC;IACtB,mBAAmB,CAAC,EAAkB;QACpC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9B,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC;IACJ,CAAC;IACD,gBAAgB,CAAI,gBAAyB;QAC3C,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC7C,QAAQ,EAAE,CAAC;QACb,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IACD,qBAAqB,KAAI,CAAC;IAC1B,SAAS,KAAI,CAAC;CACf"}
+{"version":3,"file":"query-delegate.js","sources":["../../../../zqlite/src/query-delegate.ts"],"sourcesContent":["import type {LogContext} from '@rocicorp/logger';\nimport type {LogConfig} from '../../otel/src/log-options.ts';\nimport type {Schema} from '../../zero-types/src/schema.ts';\nimport type {Source} from '../../zql/src/ivm/source.ts';\nimport {QueryDelegateBase} from '../../zql/src/query/query-delegate-base.ts';\nimport type {CommitListener} from '../../zql/src/query/query-delegate.ts';\nimport type {Database} from './db.ts';\nimport {TableSource} from './table-source.ts';\n\nexport class QueryDelegateImpl extends QueryDelegateBase {\n  readonly #lc: LogContext;\n  readonly #db: Database;\n  readonly #schema: Schema;\n  readonly #sources: Map<string, Source> = new Map();\n  readonly #logConfig: LogConfig;\n  readonly defaultQueryComplete = true;\n  readonly #commitObservers = new Set<() => void>();\n\n  constructor(\n    lc: LogContext,\n    db: Database,\n    schema: Schema,\n    logConfig?: LogConfig,\n  ) {\n    super();\n    this.#lc = lc.withContext('class', 'QueryDelegateImpl');\n    this.#db = db;\n    this.#schema = schema;\n    this.#logConfig = logConfig ?? {\n      format: 'text',\n      ivmSampling: 0,\n      level: 'info',\n      slowHydrateThreshold: 0,\n      slowRowThreshold: 0,\n    };\n  }\n\n  getSource(tableName: string): Source {\n    let source = this.#sources.get(tableName);\n    if (source) {\n      return source;\n    }\n\n    const tableSchema = this.#schema.tables[tableName];\n\n    source = new TableSource(\n      this.#lc,\n      this.#logConfig,\n      this.#db,\n      tableName,\n      tableSchema.columns,\n      tableSchema.primaryKey,\n    );\n\n    this.#sources.set(tableName, source);\n    return source;\n  }\n\n  onTransactionCommit(cb: CommitListener) {\n    this.#commitObservers.add(cb);\n    return () => {\n      this.#commitObservers.delete(cb);\n    };\n  }\n  override batchViewUpdates<T>(applyViewUpdates: () => T): T {\n    const ret = applyViewUpdates();\n    for (const observer of this.#commitObservers) {\n      observer();\n    }\n    return ret;\n  }\n}\n"],"names":[],"mappings":";;AASO,MAAM,0BAA0B,kBAAkB;AAAA,EAC9C;AAAA,EACA;AAAA,EACA;AAAA,EACA,+BAAoC,IAAA;AAAA,EACpC;AAAA,EACA,uBAAuB;AAAA,EACvB,uCAAuB,IAAA;AAAA,EAEhC,YACE,IACA,IACA,QACA,WACA;AACA,UAAA;AACA,SAAK,MAAM,GAAG,YAAY,SAAS,mBAAmB;AACtD,SAAK,MAAM;AACX,SAAK,UAAU;AACf,SAAK,aAAa,aAAa;AAAA,MAC7B,QAAQ;AAAA,MACR,aAAa;AAAA,MACb,OAAO;AAAA,MACP,sBAAsB;AAAA,MACtB,kBAAkB;AAAA,IAAA;AAAA,EAEtB;AAAA,EAEA,UAAU,WAA2B;AACnC,QAAI,SAAS,KAAK,SAAS,IAAI,SAAS;AACxC,QAAI,QAAQ;AACV,aAAO;AAAA,IACT;AAEA,UAAM,cAAc,KAAK,QAAQ,OAAO,SAAS;AAEjD,aAAS,IAAI;AAAA,MACX,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL;AAAA,MACA,YAAY;AAAA,MACZ,YAAY;AAAA,IAAA;AAGd,SAAK,SAAS,IAAI,WAAW,MAAM;AACnC,WAAO;AAAA,EACT;AAAA,EAEA,oBAAoB,IAAoB;AACtC,SAAK,iBAAiB,IAAI,EAAE;AAC5B,WAAO,MAAM;AACX,WAAK,iBAAiB,OAAO,EAAE;AAAA,IACjC;AAAA,EACF;AAAA,EACS,iBAAoB,kBAA8B;AACzD,UAAM,MAAM,iBAAA;AACZ,eAAW,YAAY,KAAK,kBAAkB;AAC5C,eAAA;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACF;"}

package/out/zqlite/src/sqlite-cost-model.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-cost-model.d.ts","sourceRoot":"","sources":["../../../../zqlite/src/sqlite-cost-model.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,mBAAmB,EAEpB,MAAM,6CAA6C,CAAC;AAIrD,OAAO,KAAK,EAAC,QAAQ,EAAY,MAAM,SAAS,CAAC;AAIjD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,sCAAsC,CAAC;AAgBtE;;;;;;;;GAQG;AACH,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,QAAQ,EACZ,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE;IAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;CAAC,CAAC,GAC9D,mBAAmB,CA2CrB;AAgHD,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK9C"}
+{"version":3,"file":"sqlite-cost-model.d.ts","sourceRoot":"","sources":["../../../../zqlite/src/sqlite-cost-model.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,mBAAmB,EAEpB,MAAM,6CAA6C,CAAC;AAIrD,OAAO,KAAK,EAAC,QAAQ,EAAY,MAAM,SAAS,CAAC;AAIjD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,sCAAsC,CAAC;AAiBtE;;;;;;;;GAQG;AACH,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,QAAQ,EACZ,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE;IAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;CAAC,CAAC,GAC9D,mBAAmB,CAqDrB;AAoHD,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK9C"}

package/out/zqlite/src/sqlite-cost-model.js

@@ -1,134 +1,110 @@
-import SQLite3Database from '@rocicorp/zero-sqlite3';
+import SQLite3Database from "@rocicorp/zero-sqlite3";
 import { buildSelectQuery } from "./query-builder.js";
-import { compile } from "./internal/sql.js";
+import { compileInline } from "./internal/sql-inline.js";
 import { assert } from "../../shared/src/asserts.js";
 import { must } from "../../shared/src/must.js";
-/**
- * Creates a SQLite-based cost model for query planning.
- * Uses SQLite's scanstatus API to estimate query costs based on the actual
- * SQLite query planner's analysis.
- *
- * @param db Database instance for preparing statements
- * @param tableSpecs Map of table names to their table specs with ZQL schemas
- * @returns ConnectionCostModel function for use with the planner
- */
-export function createSQLiteCostModel(db, tableSpecs) {
-    return (tableName, sort, filters, constraint) => {
-        // Transform filters to remove correlated subqueries
-        // The cost model can't handle correlated subqueries, so we estimate cost
-        // without them. This is conservative - actual cost may be higher.
-        const noSubqueryFilters = filters
-            ? removeCorrelatedSubqueries(filters)
-            : undefined;
-        // Build the SQL query using the same logic as actual queries
-        const { zqlSpec } = must(tableSpecs.get(tableName));
-        const query = buildSelectQuery(tableName, zqlSpec, constraint, noSubqueryFilters, sort, undefined, // reverse is undefined here
-        undefined);
-        const sql = compile(query);
-        // Prepare statement to get scanstatus information
-        const stmt = db.prepare(sql);
-        // Get scanstatus loops from the prepared statement
-        const loops = getScanstatusLoops(stmt);
-        // Scanstatus should always be available - if we get no loops, something is wrong
-        assert(loops.length > 0, `Expected scanstatus to return at least one loop for query: ${sql}`);
-        return estimateCost(loops);
-    };
+import { SQLiteStatFanout } from "./sqlite-stat-fanout.js";
+function createSQLiteCostModel(db, tableSpecs) {
+  const fanoutEstimator = new SQLiteStatFanout(db);
+  return (tableName, sort, filters, constraint) => {
+    const noSubqueryFilters = filters ? removeCorrelatedSubqueries(filters) : void 0;
+    const { zqlSpec } = must(tableSpecs.get(tableName));
+    const query = buildSelectQuery(
+      tableName,
+      zqlSpec,
+      constraint,
+      noSubqueryFilters,
+      sort,
+      void 0,
+      // reverse is undefined here
+      void 0
+      // start is undefined here
+    );
+    const sql = compileInline(query);
+    const stmt = db.prepare(sql);
+    const loops = getScanstatusLoops(stmt);
+    assert(
+      loops.length > 0,
+      `Expected scanstatus to return at least one loop for query: ${sql}`
+    );
+    const ret = estimateCost(
+      loops,
+      (columns) => fanoutEstimator.getFanout(tableName, columns)
+    );
+    return ret;
+  };
 }
-/**
- * Removes correlated subqueries from conditions.
- * The cost model estimates cost without correlated subqueries since
- * they can't be included in the scanstatus query.
- */
 function removeCorrelatedSubqueries(condition) {
-    switch (condition.type) {
-        case 'correlatedSubquery':
-            // Remove correlated subqueries - we can't estimate their cost via scanstatus
-            return undefined;
-        case 'simple':
-            return condition;
-        case 'and': {
-            const filtered = condition.conditions
-                .map(c => removeCorrelatedSubqueries(c))
-                .filter((c) => c !== undefined);
-            if (filtered.length === 0)
-                return undefined;
-            if (filtered.length === 1)
-                return filtered[0];
-            return { type: 'and', conditions: filtered };
-        }
-        case 'or': {
-            const filtered = condition.conditions
-                .map(c => removeCorrelatedSubqueries(c))
-                .filter((c) => c !== undefined);
-            if (filtered.length === 0)
-                return undefined;
-            if (filtered.length === 1)
-                return filtered[0];
-            return { type: 'or', conditions: filtered };
-        }
+  switch (condition.type) {
+    case "correlatedSubquery":
+      return void 0;
+    case "simple":
+      return condition;
+    case "and": {
+      const filtered = condition.conditions.map((c) => removeCorrelatedSubqueries(c)).filter((c) => c !== void 0);
+      if (filtered.length === 0) return void 0;
+      if (filtered.length === 1) return filtered[0];
+      return { type: "and", conditions: filtered };
     }
+    case "or": {
+      const filtered = condition.conditions.map((c) => removeCorrelatedSubqueries(c)).filter((c) => c !== void 0);
+      if (filtered.length === 0) return void 0;
+      if (filtered.length === 1) return filtered[0];
+      return { type: "or", conditions: filtered };
+    }
+  }
 }
-/**
- * Gets scanstatus loop information from a prepared statement.
- * Iterates through all query elements and extracts loop statistics.
- *
- * Uses SQLITE_SCANSTAT_COMPLEX flag (1) to get all loops including sorting operations.
- *
- * @param stmt Prepared statement to get scanstatus from
- * @returns Array of loop information, or empty array if scanstatus unavailable
- */
 function getScanstatusLoops(stmt) {
-    const loops = [];
-    // Iterate through query elements by incrementing idx until we get undefined
-    // which indicates we've reached the end
-    for (let idx = 0;; idx++) {
-        const selectId = stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_SELECTID, 1);
-        if (selectId === undefined) {
-            break;
-        }
-        loops.push({
-            selectId: must(selectId),
-            parentId: must(stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_PARENTID, 1)),
-            explain: must(stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EXPLAIN, 1)),
-            est: must(stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EST, 1)),
-        });
+  const loops = [];
+  for (let idx = 0; ; idx++) {
+    const selectId = stmt.scanStatus(
+      idx,
+      SQLite3Database.SQLITE_SCANSTAT_SELECTID,
+      1
+    );
+    if (selectId === void 0) {
+      break;
     }
-    return loops.sort((a, b) => a.selectId - b.selectId);
+    loops.push({
+      selectId: must(selectId),
+      parentId: must(
+        stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_PARENTID, 1)
+      ),
+      explain: must(
+        stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EXPLAIN, 1)
+      ),
+      est: must(stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EST, 1))
+    });
+  }
+  return loops.sort((a, b) => a.selectId - b.selectId);
 }
-/**
- * Estimates the cost of a query based on scanstats from sqlite3_stmt_scanstatus_v2
- */
-function estimateCost(scanstats) {
-    // Sort by selectId to process in execution order
-    const sorted = [...scanstats].sort((a, b) => a.selectId - b.selectId);
-    let totalRows = 0;
-    let totalCost = 0;
-    // Identify if there are multiple top-level (parentId=0) operations
-    // If so, the first is typically the scan, and subsequent ones are sorts
-    const topLevelOps = sorted.filter(s => s.parentId === 0);
-    // We only consider top level ops since ZQL queries are single-table when hitting SQLite.
-    // We do have a nested op in the case of `WHERE x IN (:arg)` but it is negligible
-    // assuming :arg is small.
-    let firstLoop = true;
-    for (const op of topLevelOps) {
-        if (firstLoop) {
-            // First top-level op is the main scan
-            // and determines the total number of rows output.
-            totalRows = op.est;
-            firstLoop = false;
-        }
-        else {
-            if (op.explain.includes('ORDER BY')) {
-                totalCost += btreeCost(totalRows);
-            }
-        }
+function estimateCost(scanstats, fanout) {
+  const sorted = [...scanstats].sort((a, b) => a.selectId - b.selectId);
+  let totalRows = 0;
+  let totalCost = 0;
+  const topLevelOps = sorted.filter((s) => s.parentId === 0);
+  let firstLoop = true;
+  for (const op of topLevelOps) {
+    if (firstLoop) {
+      totalRows = op.est;
+      firstLoop = false;
+    } else {
+      if (op.explain.includes("ORDER BY")) {
+        totalCost += btreeCost(totalRows);
+      }
     }
-    return { rows: totalRows, startupCost: totalCost };
+  }
+  return {
+    rows: totalRows,
+    startupCost: totalCost,
+    fanout
+  };
 }
-export function btreeCost(rows) {
-    // B-Tree construction is ~O(n log n) so we estimate the cost as such.
-    // We divide the cost by 10 because sorting in SQLite is ~10x faster
-    // than bringing the data into JS and sorting there.
-    return (rows * Math.log2(rows)) / 10;
+function btreeCost(rows) {
+  return rows * Math.log2(rows) / 10;
 }
-//# sourceMappingURL=sqlite-cost-model.js.map
+export {
+  btreeCost,
+  createSQLiteCostModel
+};
+//# sourceMappingURL=sqlite-cost-model.js.map

package/out/zqlite/src/sqlite-cost-model.js.map

@@ -1 +1 @@
-{"version":3,"file":"sqlite-cost-model.js","sourceRoot":"","sources":["../../../../zqlite/src/sqlite-cost-model.ts"],"names":[],"mappings":"AAMA,OAAO,eAAe,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAC,gBAAgB,EAA2B,MAAM,oBAAoB,CAAC;AAE9E,OAAO,EAAC,OAAO,EAAC,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAC,MAAM,EAAC,MAAM,6BAA6B,CAAC;AACnD,OAAO,EAAC,IAAI,EAAC,MAAM,0BAA0B,CAAC;AAiB9C;;;;;;;;GAQG;AACH,MAAM,UAAU,qBAAqB,CACnC,EAAY,EACZ,UAA+D;IAE/D,OAAO,CACL,SAAiB,EACjB,IAAc,EACd,OAA8B,EAC9B,UAAyC,EAC1B,EAAE;QACjB,oDAAoD;QACpD,yEAAyE;QACzE,kEAAkE;QAClE,MAAM,iBAAiB,GAAG,OAAO;YAC/B,CAAC,CAAC,0BAA0B,CAAC,OAAO,CAAC;YACrC,CAAC,CAAC,SAAS,CAAC;QAEd,6DAA6D;QAC7D,MAAM,EAAC,OAAO,EAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;QAElD,MAAM,KAAK,GAAG,gBAAgB,CAC5B,SAAS,EACT,OAAO,EACP,UAAU,EACV,iBAAiB,EACjB,IAAI,EACJ,SAAS,EAAE,4BAA4B;QACvC,SAAS,CACV,CAAC;QAEF,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QAE3B,kDAAkD;QAClD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAE7B,mDAAmD;QACnD,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAEvC,iFAAiF;QACjF,MAAM,CACJ,KAAK,CAAC,MAAM,GAAG,CAAC,EAChB,8DAA8D,GAAG,EAAE,CACpE,CAAC;QAEF,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,0BAA0B,CACjC,SAAoB;IAEpB,QAAQ,SAAS,CAAC,IAAI,EAAE,CAAC;QACvB,KAAK,oBAAoB;YACvB,6EAA6E;YAC7E,OAAO,SAAS,CAAC;QACnB,KAAK,QAAQ;YACX,OAAO,SAAS,CAAC;QACnB,KAAK,KAAK,CAAC,CAAC,CAAC;YACX,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC;iBACvC,MAAM,CAAC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;YAC5D,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAC;YAC5C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC9C,OAAO,EAAC,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAC,CAAC;QAC7C,CAAC;QACD,KAAK,IAAI,CAAC,CAAC,CAAC;YACV,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC;iBACvC,MAAM,CAAC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;YAC5D,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAC;YAC5C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC9C,OAAO,EAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAC,CAAC;QAC5C,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,SAAS,kBAAkB,CAAC,IAAe;IACzC,MAAM,KAAK,GAAqB,EAAE,CAAC;IAEnC,4EAA4E;IAC5E,wCAAwC;IACxC,KAAK,IAAI,GAAG,GAAG,CAAC,GAAI,GAAG,EAAE,EAAE,CAAC;QAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAC9B,GAAG,EACH,eAAe,CAAC,wBAAwB,EACxC,CAAC,CACF,CAAC;QAEF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM;QACR,CAAC;QAED,KAAK,CAAC,IAAI,CAAC;YACT,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC;YACxB,QAAQ,EAAE,IAAI,CACZ,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC,CAClE;YACD,OAAO,EAAE,IAAI,CACX,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,eAAe,CAAC,uBAAuB,EAAE,CAAC,CAAC,CACjE;YACD,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,eAAe,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;SACxE,CAAC,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;AACvD,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,SAA2B;IAI/C,iDAAiD;IACjD,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEtE,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,mEAAmE;IACnE,wEAAwE;IACxE,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC;IAEzD,yFAAyF;IACzF,iFAAiF;IACjF,0BAA0B;IAC1B,IAAI,SAAS,GAAG,IAAI,CAAC;IACrB,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC7B,IAAI,SAAS,EAAE,CAAC;YACd,sCAAsC;YACtC,kDAAkD;YAClD,SAAS,GAAG,EAAE,CAAC,GAAG,CAAC;YACnB,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,IAAI,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpC,SAAS,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAC,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAC,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAY;IACpC,sEAAsE;IACtE,oEAAoE;IACpE,oDAAoD;IACpD,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;AACvC,CAAC"}
+{"version":3,"file":"sqlite-cost-model.js","sources":["../../../../zqlite/src/sqlite-cost-model.ts"],"sourcesContent":["import type {Condition, Ordering} from '../../zero-protocol/src/ast.ts';\nimport type {\n  ConnectionCostModel,\n  CostModelCost,\n} from '../../zql/src/planner/planner-connection.ts';\nimport type {PlannerConstraint} from '../../zql/src/planner/planner-constraint.ts';\nimport SQLite3Database from '@rocicorp/zero-sqlite3';\nimport {buildSelectQuery, type NoSubqueryCondition} from './query-builder.ts';\nimport type {Database, Statement} from './db.ts';\nimport {compileInline} from './internal/sql-inline.ts';\nimport {assert} from '../../shared/src/asserts.ts';\nimport {must} from '../../shared/src/must.ts';\nimport type {SchemaValue} from '../../zero-types/src/schema-value.ts';\nimport {SQLiteStatFanout} from './sqlite-stat-fanout.ts';\n\n/**\n * Loop information returned by SQLite's scanstatus API.\n */\ninterface ScanstatusLoop {\n  /** Unique identifier for this loop */\n  selectId: number;\n  /** Parent loop ID, or 0 for root loops */\n  parentId: number;\n  /** Estimated rows emitted per turn of parent loop */\n  est: number;\n  /** EXPLAIN text for this loop to determine: b-tree vs list subquery */\n  explain: string;\n}\n\n/**\n * Creates a SQLite-based cost model for query planning.\n * Uses SQLite's scanstatus API to estimate query costs based on the actual\n * SQLite query planner's analysis.\n *\n * @param db Database instance for preparing statements\n * @param tableSpecs Map of table names to their table specs with ZQL schemas\n * @returns ConnectionCostModel function for use with the planner\n */\nexport function createSQLiteCostModel(\n  db: Database,\n  tableSpecs: Map<string, {zqlSpec: Record<string, SchemaValue>}>,\n): ConnectionCostModel {\n  const fanoutEstimator = new SQLiteStatFanout(db);\n  return (\n    tableName: string,\n    sort: Ordering,\n    filters: Condition | undefined,\n    constraint: PlannerConstraint | undefined,\n  ): CostModelCost => {\n    // Transform filters to remove correlated subqueries\n    // The cost model can't handle correlated subqueries, so we estimate cost\n    // without them. This is conservative - actual cost may be higher.\n    const noSubqueryFilters = filters\n      ? removeCorrelatedSubqueries(filters)\n      : undefined;\n\n    // Build the SQL query using the same logic as actual queries\n    const {zqlSpec} = must(tableSpecs.get(tableName));\n\n    const query = buildSelectQuery(\n      tableName,\n      zqlSpec,\n      constraint,\n      noSubqueryFilters,\n      sort,\n      undefined, // reverse is undefined here\n      undefined, // start is undefined here\n    );\n\n    // Use compileInline to inline actual values into the SQL for cost estimation.\n    // This allows SQLite's query planner to see real values and make better decisions\n    // about index usage and query plans. This is safe here because it's only used for\n    // cost estimation, not for executing user-facing queries (which use parameterized\n    // queries via the standard compile() function).\n    const sql = compileInline(query);\n\n    // Prepare statement to get scanstatus information\n    const stmt = db.prepare(sql);\n\n    // Get scanstatus loops from the prepared statement\n    const loops = getScanstatusLoops(stmt);\n\n    // Scanstatus should always be available - if we get no loops, something is wrong\n    assert(\n      loops.length > 0,\n      `Expected scanstatus to return at least one loop for query: ${sql}`,\n    );\n\n    const ret = estimateCost(loops, (columns: string[]) =>\n      fanoutEstimator.getFanout(tableName, columns),\n    );\n\n    return ret;\n  };\n}\n\n/**\n * Removes correlated subqueries from conditions.\n * The cost model estimates cost without correlated subqueries since\n * they can't be included in the scanstatus query.\n */\nfunction removeCorrelatedSubqueries(\n  condition: Condition,\n): NoSubqueryCondition | undefined {\n  switch (condition.type) {\n    case 'correlatedSubquery':\n      // Remove correlated subqueries - we can't estimate their cost via scanstatus\n      return undefined;\n    case 'simple':\n      return condition;\n    case 'and': {\n      const filtered = condition.conditions\n        .map(c => removeCorrelatedSubqueries(c))\n        .filter((c): c is NoSubqueryCondition => c !== undefined);\n      if (filtered.length === 0) return undefined;\n      if (filtered.length === 1) return filtered[0];\n      return {type: 'and', conditions: filtered};\n    }\n    case 'or': {\n      const filtered = condition.conditions\n        .map(c => removeCorrelatedSubqueries(c))\n        .filter((c): c is NoSubqueryCondition => c !== undefined);\n      if (filtered.length === 0) return undefined;\n      if (filtered.length === 1) return filtered[0];\n      return {type: 'or', conditions: filtered};\n    }\n  }\n}\n\n/**\n * Gets scanstatus loop information from a prepared statement.\n * Iterates through all query elements and extracts loop statistics.\n *\n * Uses SQLITE_SCANSTAT_COMPLEX flag (1) to get all loops including sorting operations.\n *\n * @param stmt Prepared statement to get scanstatus from\n * @returns Array of loop information, or empty array if scanstatus unavailable\n */\nfunction getScanstatusLoops(stmt: Statement): ScanstatusLoop[] {\n  const loops: ScanstatusLoop[] = [];\n\n  // Iterate through query elements by incrementing idx until we get undefined\n  // which indicates we've reached the end\n  for (let idx = 0; ; idx++) {\n    const selectId = stmt.scanStatus(\n      idx,\n      SQLite3Database.SQLITE_SCANSTAT_SELECTID,\n      1,\n    );\n\n    if (selectId === undefined) {\n      break;\n    }\n\n    loops.push({\n      selectId: must(selectId),\n      parentId: must(\n        stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_PARENTID, 1),\n      ),\n      explain: must(\n        stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EXPLAIN, 1),\n      ),\n      est: must(stmt.scanStatus(idx, SQLite3Database.SQLITE_SCANSTAT_EST, 1)),\n    });\n  }\n\n  return loops.sort((a, b) => a.selectId - b.selectId);\n}\n\n/**\n * Estimates the cost of a query based on scanstats from sqlite3_stmt_scanstatus_v2\n */\nfunction estimateCost(\n  scanstats: ScanstatusLoop[],\n  fanout: CostModelCost['fanout'],\n): CostModelCost {\n  // Sort by selectId to process in execution order\n  const sorted = [...scanstats].sort((a, b) => a.selectId - b.selectId);\n\n  let totalRows = 0;\n  let totalCost = 0;\n\n  // Identify if there are multiple top-level (parentId=0) operations\n  // If so, the first is typically the scan, and subsequent ones are sorts\n  const topLevelOps = sorted.filter(s => s.parentId === 0);\n\n  // We only consider top level ops since ZQL queries are single-table when hitting SQLite.\n  // We do have a nested op in the case of `WHERE x IN (:arg)` but it is negligible\n  // assuming :arg is small.\n  let firstLoop = true;\n  for (const op of topLevelOps) {\n    if (firstLoop) {\n      // First top-level op is the main scan\n      // and determines the total number of rows output.\n      totalRows = op.est;\n      firstLoop = false;\n    } else {\n      if (op.explain.includes('ORDER BY')) {\n        totalCost += btreeCost(totalRows);\n      }\n    }\n  }\n\n  return {\n    rows: totalRows,\n    startupCost: totalCost,\n    fanout,\n  };\n}\n\nexport function btreeCost(rows: number): number {\n  // B-Tree construction is ~O(n log n) so we estimate the cost as such.\n  // We divide the cost by 10 because sorting in SQLite is ~10x faster\n  // than bringing the data into JS and sorting there.\n  return (rows * Math.log2(rows)) / 10;\n}\n"],"names":[],"mappings":";;;;;;AAsCO,SAAS,sBACd,IACA,YACqB;AACrB,QAAM,kBAAkB,IAAI,iBAAiB,EAAE;AAC/C,SAAO,CACL,WACA,MACA,SACA,eACkB;AAIlB,UAAM,oBAAoB,UACtB,2BAA2B,OAAO,IAClC;AAGJ,UAAM,EAAC,QAAA,IAAW,KAAK,WAAW,IAAI,SAAS,CAAC;AAEhD,UAAM,QAAQ;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA;AAAA,MACA;AAAA;AAAA,IAAA;AAQF,UAAM,MAAM,cAAc,KAAK;AAG/B,UAAM,OAAO,GAAG,QAAQ,GAAG;AAG3B,UAAM,QAAQ,mBAAmB,IAAI;AAGrC;AAAA,MACE,MAAM,SAAS;AAAA,MACf,8DAA8D,GAAG;AAAA,IAAA;AAGnE,UAAM,MAAM;AAAA,MAAa;AAAA,MAAO,CAAC,YAC/B,gBAAgB,UAAU,WAAW,OAAO;AAAA,IAAA;AAG9C,WAAO;AAAA,EACT;AACF;AAOA,SAAS,2BACP,WACiC;AACjC,UAAQ,UAAU,MAAA;AAAA,IAChB,KAAK;AAEH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK,OAAO;AACV,YAAM,WAAW,UAAU,WACxB,IAAI,CAAA,MAAK,2BAA2B,CAAC,CAAC,EACtC,OAAO,CAAC,MAAgC,MAAM,MAAS;AAC1D,UAAI,SAAS,WAAW,EAAG,QAAO;AAClC,UAAI,SAAS,WAAW,EAAG,QAAO,SAAS,CAAC;AAC5C,aAAO,EAAC,MAAM,OAAO,YAAY,SAAA;AAAA,IACnC;AAAA,IACA,KAAK,MAAM;AACT,YAAM,WAAW,UAAU,WACxB,IAAI,CAAA,MAAK,2BAA2B,CAAC,CAAC,EACtC,OAAO,CAAC,MAAgC,MAAM,MAAS;AAC1D,UAAI,SAAS,WAAW,EAAG,QAAO;AAClC,UAAI,SAAS,WAAW,EAAG,QAAO,SAAS,CAAC;AAC5C,aAAO,EAAC,MAAM,MAAM,YAAY,SAAA;AAAA,IAClC;AAAA,EAAA;AAEJ;AAWA,SAAS,mBAAmB,MAAmC;AAC7D,QAAM,QAA0B,CAAA;AAIhC,WAAS,MAAM,KAAK,OAAO;AACzB,UAAM,WAAW,KAAK;AAAA,MACpB;AAAA,MACA,gBAAgB;AAAA,MAChB;AAAA,IAAA;AAGF,QAAI,aAAa,QAAW;AAC1B;AAAA,IACF;AAEA,UAAM,KAAK;AAAA,MACT,UAAU,KAAK,QAAQ;AAAA,MACvB,UAAU;AAAA,QACR,KAAK,WAAW,KAAK,gBAAgB,0BAA0B,CAAC;AAAA,MAAA;AAAA,MAElE,SAAS;AAAA,QACP,KAAK,WAAW,KAAK,gBAAgB,yBAAyB,CAAC;AAAA,MAAA;AAAA,MAEjE,KAAK,KAAK,KAAK,WAAW,KAAK,gBAAgB,qBAAqB,CAAC,CAAC;AAAA,IAAA,CACvE;AAAA,EACH;AAEA,SAAO,MAAM,KAAK,CAAC,GAAG,MAAM,EAAE,WAAW,EAAE,QAAQ;AACrD;AAKA,SAAS,aACP,WACA,QACe;AAEf,QAAM,SAAS,CAAC,GAAG,SAAS,EAAE,KAAK,CAAC,GAAG,MAAM,EAAE,WAAW,EAAE,QAAQ;AAEpE,MAAI,YAAY;AAChB,MAAI,YAAY;AAIhB,QAAM,cAAc,OAAO,OAAO,CAAA,MAAK,EAAE,aAAa,CAAC;AAKvD,MAAI,YAAY;AAChB,aAAW,MAAM,aAAa;AAC5B,QAAI,WAAW;AAGb,kBAAY,GAAG;AACf,kBAAY;AAAA,IACd,OAAO;AACL,UAAI,GAAG,QAAQ,SAAS,UAAU,GAAG;AACnC,qBAAa,UAAU,SAAS;AAAA,MAClC;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,aAAa;AAAA,IACb;AAAA,EAAA;AAEJ;AAEO,SAAS,UAAU,MAAsB;AAI9C,SAAQ,OAAO,KAAK,KAAK,IAAI,IAAK;AACpC;"}