@tanstack/db 0.0.7 → 0.0.9

package/dist/esm/query/compiled-query.js.map

@@ -1 +1 @@
-{"version":3,"file":"compiled-query.js","sources":["../../../src/query/compiled-query.ts"],"sourcesContent":["import { D2, MessageType, MultiSet, output } from \"@electric-sql/d2ts\"\nimport { Effect, batch } from \"@tanstack/store\"\nimport { createCollection } from \"../collection.js\"\nimport { compileQueryPipeline } from \"./pipeline-compiler.js\"\nimport type { Collection } from \"../collection.js\"\nimport type { ChangeMessage, SyncConfig } from \"../types.js\"\nimport type {\n  IStreamBuilder,\n  MultiSetArray,\n  RootStreamBuilder,\n} from \"@electric-sql/d2ts\"\nimport type { QueryBuilder, ResultsFromContext } from \"./query-builder.js\"\nimport type { Context, Schema } from \"./types.js\"\n\nexport function compileQuery<TContext extends Context<Schema>>(\n  queryBuilder: QueryBuilder<TContext>\n) {\n  return new CompiledQuery<\n    ResultsFromContext<TContext> & { _key?: string | number }\n  >(queryBuilder)\n}\n\nexport class CompiledQuery<TResults extends object = Record<string, unknown>> {\n  private graph: D2\n  private inputs: Record<string, RootStreamBuilder<any>>\n  private inputCollections: Record<string, Collection<any>>\n  private resultCollection: Collection<TResults>\n  public state: `compiled` | `running` | `stopped` = `compiled`\n  private version = 0\n  private unsubscribeEffect?: () => void\n\n  constructor(queryBuilder: QueryBuilder<Context<Schema>>) {\n    const query = queryBuilder._query\n    const collections = query.collections\n\n    if (!collections) {\n      throw new Error(`No collections provided`)\n    }\n\n    this.inputCollections = collections\n\n    const graph = new D2({ initialFrontier: this.version })\n    const inputs = Object.fromEntries(\n      Object.entries(collections).map(([key]) => [key, graph.newInput<any>()])\n    )\n\n    const sync: SyncConfig<TResults>[`sync`] = ({ begin, write, commit }) => {\n      compileQueryPipeline<IStreamBuilder<[unknown, TResults]>>(\n        query,\n        inputs\n      ).pipe(\n        output(({ type, data }) => {\n          if (type === MessageType.DATA) {\n            begin()\n            data.collection\n              .getInner()\n              .reduce((acc, [[key, value], multiplicity]) => {\n                const changes = acc.get(key) || {\n                  deletes: 0,\n                  inserts: 0,\n                  value,\n                }\n                if (multiplicity < 0) {\n                  changes.deletes += Math.abs(multiplicity)\n                } else if (multiplicity > 0) {\n                  changes.inserts += multiplicity\n                  changes.value = value\n                }\n                acc.set(key, changes)\n                return acc\n              }, new Map<unknown, { deletes: number; inserts: number; value: TResults }>())\n              .forEach((changes, rawKey) => {\n                const { deletes, inserts, value } = changes\n                const valueWithKey = { ...value, _key: rawKey }\n                if (inserts && !deletes) {\n                  write({\n                    value: valueWithKey,\n                    type: `insert`,\n                  })\n                } else if (inserts >= deletes) {\n                  write({\n                    value: valueWithKey,\n                    type: `update`,\n                  })\n                } else if (deletes > 0) {\n                  write({\n                    value: valueWithKey,\n                    type: `delete`,\n                  })\n                }\n              })\n            commit()\n          }\n        })\n      )\n      graph.finalize()\n    }\n\n    this.graph = graph\n    this.inputs = inputs\n    this.resultCollection = createCollection<TResults>({\n      id: crypto.randomUUID(), // TODO: remove when we don't require any more\n      getId: (val: unknown) => {\n        return (val as any)._key\n      },\n      sync: {\n        sync,\n      },\n    })\n  }\n\n  get results() {\n    return this.resultCollection\n  }\n\n  private sendChangesToInput(\n    inputKey: string,\n    changes: Array<ChangeMessage>,\n    getId: (item: ChangeMessage[`value`]) => any\n  ) {\n    const input = this.inputs[inputKey]!\n    const multiSetArray: MultiSetArray<unknown> = []\n    for (const change of changes) {\n      const key = getId(change.value)\n      if (change.type === `insert`) {\n        multiSetArray.push([[key, change.value], 1])\n      } else if (change.type === `update`) {\n        multiSetArray.push([[key, change.previousValue], -1])\n        multiSetArray.push([[key, change.value], 1])\n      } else {\n        // change.type === `delete`\n        multiSetArray.push([[key, change.value], -1])\n      }\n    }\n    input.sendData(this.version, new MultiSet(multiSetArray))\n  }\n\n  private sendFrontierToInput(inputKey: string) {\n    const input = this.inputs[inputKey]!\n    input.sendFrontier(this.version)\n  }\n\n  private sendFrontierToAllInputs() {\n    Object.entries(this.inputs).forEach(([key]) => {\n      this.sendFrontierToInput(key)\n    })\n  }\n\n  private incrementVersion() {\n    this.version++\n  }\n\n  private runGraph() {\n    this.graph.run()\n  }\n\n  start() {\n    if (this.state === `running`) {\n      throw new Error(`Query is already running`)\n    } else if (this.state === `stopped`) {\n      throw new Error(`Query is stopped`)\n    }\n\n    batch(() => {\n      Object.entries(this.inputCollections).forEach(([key, collection]) => {\n        this.sendChangesToInput(\n          key,\n          collection.currentStateAsChanges(),\n          collection.config.getId\n        )\n      })\n      this.incrementVersion()\n      this.sendFrontierToAllInputs()\n      this.runGraph()\n    })\n\n    const changeEffect = new Effect({\n      fn: () => {\n        batch(() => {\n          Object.entries(this.inputCollections).forEach(([key, collection]) => {\n            this.sendChangesToInput(\n              key,\n              collection.derivedChanges.state,\n              collection.config.getId\n            )\n          })\n          this.incrementVersion()\n          this.sendFrontierToAllInputs()\n          this.runGraph()\n        })\n      },\n      deps: Object.values(this.inputCollections).map(\n        (collection) => collection.derivedChanges\n      ),\n    })\n    this.unsubscribeEffect = changeEffect.mount()\n\n    this.state = `running`\n    return () => {\n      this.stop()\n    }\n  }\n\n  stop() {\n    this.unsubscribeEffect?.()\n    this.unsubscribeEffect = undefined\n    this.state = `stopped`\n  }\n}\n"],"names":[],"mappings":";;;;AAcO,SAAS,aACd,cACA;AACO,SAAA,IAAI,cAET,YAAY;AAChB;AAEO,MAAM,cAAiE;AAAA,EAS5E,YAAY,cAA6C;AAJzD,SAAO,QAA4C;AACnD,SAAQ,UAAU;AAIhB,UAAM,QAAQ,aAAa;AAC3B,UAAM,cAAc,MAAM;AAE1B,QAAI,CAAC,aAAa;AACV,YAAA,IAAI,MAAM,yBAAyB;AAAA,IAAA;AAG3C,SAAK,mBAAmB;AAExB,UAAM,QAAQ,IAAI,GAAG,EAAE,iBAAiB,KAAK,SAAS;AACtD,UAAM,SAAS,OAAO;AAAA,MACpB,OAAO,QAAQ,WAAW,EAAE,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,MAAM,SAAA,CAAe,CAAC;AAAA,IACzE;AAEA,UAAM,OAAqC,CAAC,EAAE,OAAO,OAAO,aAAa;AACvE;AAAA,QACE;AAAA,QACA;AAAA,MAAA,EACA;AAAA,QACA,OAAO,CAAC,EAAE,MAAM,WAAW;AACrB,cAAA,SAAS,YAAY,MAAM;AACvB,kBAAA;AACN,iBAAK,WACF,SACA,EAAA,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,KAAK,GAAG,YAAY,MAAM;AAC7C,oBAAM,UAAU,IAAI,IAAI,GAAG,KAAK;AAAA,gBAC9B,SAAS;AAAA,gBACT,SAAS;AAAA,gBACT;AAAA,cACF;AACA,kBAAI,eAAe,GAAG;AACZ,wBAAA,WAAW,KAAK,IAAI,YAAY;AAAA,cAAA,WAC/B,eAAe,GAAG;AAC3B,wBAAQ,WAAW;AACnB,wBAAQ,QAAQ;AAAA,cAAA;AAEd,kBAAA,IAAI,KAAK,OAAO;AACb,qBAAA;AAAA,YAAA,uBACF,IAAoE,CAAC,EAC3E,QAAQ,CAAC,SAAS,WAAW;AAC5B,oBAAM,EAAE,SAAS,SAAS,MAAU,IAAA;AACpC,oBAAM,eAAe,EAAE,GAAG,OAAO,MAAM,OAAO;AAC1C,kBAAA,WAAW,CAAC,SAAS;AACjB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA,WACQ,WAAW,SAAS;AACvB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA,WACQ,UAAU,GAAG;AAChB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA;AAAA,YACH,CACD;AACI,mBAAA;AAAA,UAAA;AAAA,QAEV,CAAA;AAAA,MACH;AACA,YAAM,SAAS;AAAA,IACjB;AAEA,SAAK,QAAQ;AACb,SAAK,SAAS;AACd,SAAK,mBAAmB,iBAA2B;AAAA,MACjD,IAAI,OAAO,WAAW;AAAA;AAAA,MACtB,OAAO,CAAC,QAAiB;AACvB,eAAQ,IAAY;AAAA,MACtB;AAAA,MACA,MAAM;AAAA,QACJ;AAAA,MAAA;AAAA,IACF,CACD;AAAA,EAAA;AAAA,EAGH,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EAAA;AAAA,EAGN,mBACN,UACA,SACA,OACA;AACM,UAAA,QAAQ,KAAK,OAAO,QAAQ;AAClC,UAAM,gBAAwC,CAAC;AAC/C,eAAW,UAAU,SAAS;AACtB,YAAA,MAAM,MAAM,OAAO,KAAK;AAC1B,UAAA,OAAO,SAAS,UAAU;AACd,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,CAAC,CAAC;AAAA,MAC7C,WAAW,OAAO,SAAS,UAAU;AACrB,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,aAAa,GAAG,EAAE,CAAC;AACtC,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,CAAC,CAAC;AAAA,MAAA,OACtC;AAES,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,EAAE,CAAC;AAAA,MAAA;AAAA,IAC9C;AAEF,UAAM,SAAS,KAAK,SAAS,IAAI,SAAS,aAAa,CAAC;AAAA,EAAA;AAAA,EAGlD,oBAAoB,UAAkB;AACtC,UAAA,QAAQ,KAAK,OAAO,QAAQ;AAC5B,UAAA,aAAa,KAAK,OAAO;AAAA,EAAA;AAAA,EAGzB,0BAA0B;AACzB,WAAA,QAAQ,KAAK,MAAM,EAAE,QAAQ,CAAC,CAAC,GAAG,MAAM;AAC7C,WAAK,oBAAoB,GAAG;AAAA,IAAA,CAC7B;AAAA,EAAA;AAAA,EAGK,mBAAmB;AACpB,SAAA;AAAA,EAAA;AAAA,EAGC,WAAW;AACjB,SAAK,MAAM,IAAI;AAAA,EAAA;AAAA,EAGjB,QAAQ;AACF,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA,IAAI,MAAM,0BAA0B;AAAA,IAC5C,WAAW,KAAK,UAAU,WAAW;AAC7B,YAAA,IAAI,MAAM,kBAAkB;AAAA,IAAA;AAGpC,UAAM,MAAM;AACH,aAAA,QAAQ,KAAK,gBAAgB,EAAE,QAAQ,CAAC,CAAC,KAAK,UAAU,MAAM;AAC9D,aAAA;AAAA,UACH;AAAA,UACA,WAAW,sBAAsB;AAAA,UACjC,WAAW,OAAO;AAAA,QACpB;AAAA,MAAA,CACD;AACD,WAAK,iBAAiB;AACtB,WAAK,wBAAwB;AAC7B,WAAK,SAAS;AAAA,IAAA,CACf;AAEK,UAAA,eAAe,IAAI,OAAO;AAAA,MAC9B,IAAI,MAAM;AACR,cAAM,MAAM;AACH,iBAAA,QAAQ,KAAK,gBAAgB,EAAE,QAAQ,CAAC,CAAC,KAAK,UAAU,MAAM;AAC9D,iBAAA;AAAA,cACH;AAAA,cACA,WAAW,eAAe;AAAA,cAC1B,WAAW,OAAO;AAAA,YACpB;AAAA,UAAA,CACD;AACD,eAAK,iBAAiB;AACtB,eAAK,wBAAwB;AAC7B,eAAK,SAAS;AAAA,QAAA,CACf;AAAA,MACH;AAAA,MACA,MAAM,OAAO,OAAO,KAAK,gBAAgB,EAAE;AAAA,QACzC,CAAC,eAAe,WAAW;AAAA,MAAA;AAAA,IAC7B,CACD;AACI,SAAA,oBAAoB,aAAa,MAAM;AAE5C,SAAK,QAAQ;AACb,WAAO,MAAM;AACX,WAAK,KAAK;AAAA,IACZ;AAAA,EAAA;AAAA,EAGF,OAAO;;AACL,eAAK,sBAAL;AACA,SAAK,oBAAoB;AACzB,SAAK,QAAQ;AAAA,EAAA;AAEjB;"}
+{"version":3,"file":"compiled-query.js","sources":["../../../src/query/compiled-query.ts"],"sourcesContent":["import { D2, MessageType, MultiSet, output } from \"@electric-sql/d2ts\"\nimport { createCollection } from \"../collection.js\"\nimport { compileQueryPipeline } from \"./pipeline-compiler.js\"\nimport type { Collection } from \"../collection.js\"\nimport type { ChangeMessage, SyncConfig } from \"../types.js\"\nimport type {\n  IStreamBuilder,\n  MultiSetArray,\n  RootStreamBuilder,\n} from \"@electric-sql/d2ts\"\nimport type { QueryBuilder, ResultsFromContext } from \"./query-builder.js\"\nimport type { Context, Schema } from \"./types.js\"\n\nexport function compileQuery<TContext extends Context<Schema>>(\n  queryBuilder: QueryBuilder<TContext>\n) {\n  return new CompiledQuery<\n    ResultsFromContext<TContext> & { _key?: string | number }\n  >(queryBuilder)\n}\n\nexport class CompiledQuery<TResults extends object = Record<string, unknown>> {\n  private graph: D2\n  private inputs: Record<string, RootStreamBuilder<any>>\n  private inputCollections: Record<string, Collection<any>>\n  private resultCollection: Collection<TResults>\n  public state: `compiled` | `running` | `stopped` = `compiled`\n  private version = 0\n  private unsubscribeCallbacks: Array<() => void> = []\n\n  constructor(queryBuilder: QueryBuilder<Context<Schema>>) {\n    const query = queryBuilder._query\n    const collections = query.collections\n\n    if (!collections) {\n      throw new Error(`No collections provided`)\n    }\n\n    this.inputCollections = collections\n\n    const graph = new D2({ initialFrontier: this.version })\n    const inputs = Object.fromEntries(\n      Object.entries(collections).map(([key]) => [key, graph.newInput<any>()])\n    )\n\n    const sync: SyncConfig<TResults>[`sync`] = ({ begin, write, commit }) => {\n      compileQueryPipeline<IStreamBuilder<[unknown, TResults]>>(\n        query,\n        inputs\n      ).pipe(\n        output(({ type, data }) => {\n          if (type === MessageType.DATA) {\n            begin()\n            data.collection\n              .getInner()\n              .reduce((acc, [[key, value], multiplicity]) => {\n                const changes = acc.get(key) || {\n                  deletes: 0,\n                  inserts: 0,\n                  value,\n                }\n                if (multiplicity < 0) {\n                  changes.deletes += Math.abs(multiplicity)\n                } else if (multiplicity > 0) {\n                  changes.inserts += multiplicity\n                  changes.value = value\n                }\n                acc.set(key, changes)\n                return acc\n              }, new Map<unknown, { deletes: number; inserts: number; value: TResults }>())\n              .forEach((changes, rawKey) => {\n                const { deletes, inserts, value } = changes\n                const valueWithKey = { ...value, _key: rawKey }\n                if (inserts && !deletes) {\n                  write({\n                    value: valueWithKey,\n                    type: `insert`,\n                  })\n                } else if (inserts >= deletes) {\n                  write({\n                    value: valueWithKey,\n                    type: `update`,\n                  })\n                } else if (deletes > 0) {\n                  write({\n                    value: valueWithKey,\n                    type: `delete`,\n                  })\n                }\n              })\n            commit()\n          }\n        })\n      )\n      graph.finalize()\n    }\n\n    this.graph = graph\n    this.inputs = inputs\n    this.resultCollection = createCollection<TResults>({\n      id: crypto.randomUUID(), // TODO: remove when we don't require any more\n      getKey: (val: unknown) => {\n        return (val as any)._key\n      },\n      sync: {\n        sync,\n      },\n    })\n  }\n\n  get results() {\n    return this.resultCollection\n  }\n\n  private sendChangesToInput(\n    inputKey: string,\n    changes: Array<ChangeMessage>,\n    getKey: (item: ChangeMessage[`value`]) => any\n  ) {\n    const input = this.inputs[inputKey]!\n    const multiSetArray: MultiSetArray<unknown> = []\n    for (const change of changes) {\n      const key = getKey(change.value)\n      if (change.type === `insert`) {\n        multiSetArray.push([[key, change.value], 1])\n      } else if (change.type === `update`) {\n        multiSetArray.push([[key, change.previousValue], -1])\n        multiSetArray.push([[key, change.value], 1])\n      } else {\n        // change.type === `delete`\n        multiSetArray.push([[key, change.value], -1])\n      }\n    }\n    input.sendData(this.version, new MultiSet(multiSetArray))\n  }\n\n  private sendFrontierToInput(inputKey: string) {\n    const input = this.inputs[inputKey]!\n    input.sendFrontier(this.version)\n  }\n\n  private sendFrontierToAllInputs() {\n    Object.entries(this.inputs).forEach(([key]) => {\n      this.sendFrontierToInput(key)\n    })\n  }\n\n  private incrementVersion() {\n    this.version++\n  }\n\n  private runGraph() {\n    this.graph.run()\n  }\n\n  start() {\n    if (this.state === `running`) {\n      throw new Error(`Query is already running`)\n    } else if (this.state === `stopped`) {\n      throw new Error(`Query is stopped`)\n    }\n\n    // Send initial state\n    Object.entries(this.inputCollections).forEach(([key, collection]) => {\n      this.sendChangesToInput(\n        key,\n        collection.currentStateAsChanges(),\n        collection.config.getKey\n      )\n    })\n    this.incrementVersion()\n    this.sendFrontierToAllInputs()\n    this.runGraph()\n\n    // Subscribe to changes\n    Object.entries(this.inputCollections).forEach(([key, collection]) => {\n      const unsubscribe = collection.subscribeChanges((changes) => {\n        this.sendChangesToInput(key, changes, collection.config.getKey)\n        this.incrementVersion()\n        this.sendFrontierToAllInputs()\n        this.runGraph()\n      })\n\n      this.unsubscribeCallbacks.push(unsubscribe)\n    })\n\n    this.state = `running`\n    return () => {\n      this.stop()\n    }\n  }\n\n  stop() {\n    this.unsubscribeCallbacks.forEach((unsubscribe) => unsubscribe())\n    this.unsubscribeCallbacks = []\n    this.state = `stopped`\n  }\n}\n"],"names":[],"mappings":";;;AAaO,SAAS,aACd,cACA;AACO,SAAA,IAAI,cAET,YAAY;AAChB;AAEO,MAAM,cAAiE;AAAA,EAS5E,YAAY,cAA6C;AAJzD,SAAO,QAA4C;AACnD,SAAQ,UAAU;AAClB,SAAQ,uBAA0C,CAAC;AAGjD,UAAM,QAAQ,aAAa;AAC3B,UAAM,cAAc,MAAM;AAE1B,QAAI,CAAC,aAAa;AACV,YAAA,IAAI,MAAM,yBAAyB;AAAA,IAAA;AAG3C,SAAK,mBAAmB;AAExB,UAAM,QAAQ,IAAI,GAAG,EAAE,iBAAiB,KAAK,SAAS;AACtD,UAAM,SAAS,OAAO;AAAA,MACpB,OAAO,QAAQ,WAAW,EAAE,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,MAAM,SAAA,CAAe,CAAC;AAAA,IACzE;AAEA,UAAM,OAAqC,CAAC,EAAE,OAAO,OAAO,aAAa;AACvE;AAAA,QACE;AAAA,QACA;AAAA,MAAA,EACA;AAAA,QACA,OAAO,CAAC,EAAE,MAAM,WAAW;AACrB,cAAA,SAAS,YAAY,MAAM;AACvB,kBAAA;AACN,iBAAK,WACF,SACA,EAAA,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,KAAK,GAAG,YAAY,MAAM;AAC7C,oBAAM,UAAU,IAAI,IAAI,GAAG,KAAK;AAAA,gBAC9B,SAAS;AAAA,gBACT,SAAS;AAAA,gBACT;AAAA,cACF;AACA,kBAAI,eAAe,GAAG;AACZ,wBAAA,WAAW,KAAK,IAAI,YAAY;AAAA,cAAA,WAC/B,eAAe,GAAG;AAC3B,wBAAQ,WAAW;AACnB,wBAAQ,QAAQ;AAAA,cAAA;AAEd,kBAAA,IAAI,KAAK,OAAO;AACb,qBAAA;AAAA,YAAA,uBACF,IAAoE,CAAC,EAC3E,QAAQ,CAAC,SAAS,WAAW;AAC5B,oBAAM,EAAE,SAAS,SAAS,MAAU,IAAA;AACpC,oBAAM,eAAe,EAAE,GAAG,OAAO,MAAM,OAAO;AAC1C,kBAAA,WAAW,CAAC,SAAS;AACjB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA,WACQ,WAAW,SAAS;AACvB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA,WACQ,UAAU,GAAG;AAChB,sBAAA;AAAA,kBACJ,OAAO;AAAA,kBACP,MAAM;AAAA,gBAAA,CACP;AAAA,cAAA;AAAA,YACH,CACD;AACI,mBAAA;AAAA,UAAA;AAAA,QAEV,CAAA;AAAA,MACH;AACA,YAAM,SAAS;AAAA,IACjB;AAEA,SAAK,QAAQ;AACb,SAAK,SAAS;AACd,SAAK,mBAAmB,iBAA2B;AAAA,MACjD,IAAI,OAAO,WAAW;AAAA;AAAA,MACtB,QAAQ,CAAC,QAAiB;AACxB,eAAQ,IAAY;AAAA,MACtB;AAAA,MACA,MAAM;AAAA,QACJ;AAAA,MAAA;AAAA,IACF,CACD;AAAA,EAAA;AAAA,EAGH,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EAAA;AAAA,EAGN,mBACN,UACA,SACA,QACA;AACM,UAAA,QAAQ,KAAK,OAAO,QAAQ;AAClC,UAAM,gBAAwC,CAAC;AAC/C,eAAW,UAAU,SAAS;AACtB,YAAA,MAAM,OAAO,OAAO,KAAK;AAC3B,UAAA,OAAO,SAAS,UAAU;AACd,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,CAAC,CAAC;AAAA,MAC7C,WAAW,OAAO,SAAS,UAAU;AACrB,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,aAAa,GAAG,EAAE,CAAC;AACtC,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,CAAC,CAAC;AAAA,MAAA,OACtC;AAES,sBAAA,KAAK,CAAC,CAAC,KAAK,OAAO,KAAK,GAAG,EAAE,CAAC;AAAA,MAAA;AAAA,IAC9C;AAEF,UAAM,SAAS,KAAK,SAAS,IAAI,SAAS,aAAa,CAAC;AAAA,EAAA;AAAA,EAGlD,oBAAoB,UAAkB;AACtC,UAAA,QAAQ,KAAK,OAAO,QAAQ;AAC5B,UAAA,aAAa,KAAK,OAAO;AAAA,EAAA;AAAA,EAGzB,0BAA0B;AACzB,WAAA,QAAQ,KAAK,MAAM,EAAE,QAAQ,CAAC,CAAC,GAAG,MAAM;AAC7C,WAAK,oBAAoB,GAAG;AAAA,IAAA,CAC7B;AAAA,EAAA;AAAA,EAGK,mBAAmB;AACpB,SAAA;AAAA,EAAA;AAAA,EAGC,WAAW;AACjB,SAAK,MAAM,IAAI;AAAA,EAAA;AAAA,EAGjB,QAAQ;AACF,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA,IAAI,MAAM,0BAA0B;AAAA,IAC5C,WAAW,KAAK,UAAU,WAAW;AAC7B,YAAA,IAAI,MAAM,kBAAkB;AAAA,IAAA;AAI7B,WAAA,QAAQ,KAAK,gBAAgB,EAAE,QAAQ,CAAC,CAAC,KAAK,UAAU,MAAM;AAC9D,WAAA;AAAA,QACH;AAAA,QACA,WAAW,sBAAsB;AAAA,QACjC,WAAW,OAAO;AAAA,MACpB;AAAA,IAAA,CACD;AACD,SAAK,iBAAiB;AACtB,SAAK,wBAAwB;AAC7B,SAAK,SAAS;AAGP,WAAA,QAAQ,KAAK,gBAAgB,EAAE,QAAQ,CAAC,CAAC,KAAK,UAAU,MAAM;AACnE,YAAM,cAAc,WAAW,iBAAiB,CAAC,YAAY;AAC3D,aAAK,mBAAmB,KAAK,SAAS,WAAW,OAAO,MAAM;AAC9D,aAAK,iBAAiB;AACtB,aAAK,wBAAwB;AAC7B,aAAK,SAAS;AAAA,MAAA,CACf;AAEI,WAAA,qBAAqB,KAAK,WAAW;AAAA,IAAA,CAC3C;AAED,SAAK,QAAQ;AACb,WAAO,MAAM;AACX,WAAK,KAAK;AAAA,IACZ;AAAA,EAAA;AAAA,EAGF,OAAO;AACL,SAAK,qBAAqB,QAAQ,CAAC,gBAAgB,aAAa;AAChE,SAAK,uBAAuB,CAAC;AAC7B,SAAK,QAAQ;AAAA,EAAA;AAEjB;"}

package/dist/esm/query/order-by.js

@@ -50,16 +50,13 @@ function processOrderBy(resultPipeline, query, mainTableAlias) {
   }
   const valueExtractor = (namespacedRow) => {
     if (orderByItems.length > 1) {
-      return orderByItems.map((item) => {
-        const val = evaluateOperandOnNamespacedRow(
+      return orderByItems.map(
+        (item) => evaluateOperandOnNamespacedRow(
           namespacedRow,
           item.operand,
           mainTableAlias
-        );
-        return item.direction === `desc` && typeof val === `number` ? -val : item.direction === `desc` && typeof val === `string` ? String.fromCharCode(
-          ...[...val].map((c) => 65535 - c.charCodeAt(0))
-        ) : val;
-      });
+        )
+      );
     } else if (orderByItems.length === 1) {
       const item = orderByItems[0];
       const val = evaluateOperandOnNamespacedRow(
@@ -67,53 +64,59 @@ function processOrderBy(resultPipeline, query, mainTableAlias) {
         item.operand,
         mainTableAlias
       );
-      return item.direction === `desc` && typeof val === `number` ? -val : item.direction === `desc` && typeof val === `string` ? String.fromCharCode(
-        ...[...val].map((c) => 65535 - c.charCodeAt(0))
-      ) : val;
+      return val;
     }
     return null;
   };
-  const comparator = (a, b) => {
-    if (typeof a === `number` && typeof b === `number`) {
-      return a - b;
-    }
+  const ascComparator = (a, b) => {
     if (typeof a === `string` && typeof b === `string`) {
       return a.localeCompare(b);
     }
-    if (typeof a === `boolean` && typeof b === `boolean`) {
-      return a === b ? 0 : a ? 1 : -1;
-    }
-    if (a instanceof Date && b instanceof Date) {
-      return a.getTime() - b.getTime();
-    }
-    if (a === null || b === null) {
-      return 0;
-    }
     if (Array.isArray(a) && Array.isArray(b)) {
       for (let i = 0; i < Math.min(a.length, b.length); i++) {
-        const aVal = a[i];
-        const bVal = b[i];
-        let result;
-        if (typeof aVal === `boolean` && typeof bVal === `boolean`) {
-          result = aVal === bVal ? 0 : aVal ? 1 : -1;
-        } else if (typeof aVal === `number` && typeof bVal === `number`) {
-          result = aVal - bVal;
-        } else if (typeof aVal === `string` && typeof bVal === `string`) {
-          result = aVal.localeCompare(bVal);
-        } else {
-          result = comparator(aVal, bVal);
-        }
+        const result = ascComparator(a[i], b[i]);
         if (result !== 0) {
           return result;
         }
       }
       return a.length - b.length;
     }
-    if (a == null && b == null) {
-      return 0;
+    const bothObjects = typeof a === `object` && typeof b === `object`;
+    const bothDates = a instanceof Date && b instanceof Date;
+    const notNull = a !== null && b !== null;
+    if (bothObjects && !bothDates && notNull) {
+      return a.toString().localeCompare(b.toString());
     }
-    return a.toString().localeCompare(b.toString());
+    if (a < b) return -1;
+    if (a > b) return 1;
+    return 0;
+  };
+  const descComparator = (a, b) => {
+    return ascComparator(b, a);
+  };
+  const makeComparator = (orderByProps) => {
+    return (a, b) => {
+      if (orderByProps.length > 1) {
+        const arrayA = a;
+        const arrayB = b;
+        for (let i = 0; i < orderByProps.length; i++) {
+          const direction = orderByProps[i].direction;
+          const compareFn = direction === `desc` ? descComparator : ascComparator;
+          const result = compareFn(arrayA[i], arrayB[i]);
+          if (result !== 0) {
+            return result;
+          }
+        }
+        return arrayA.length - arrayB.length;
+      }
+      if (orderByProps.length === 1) {
+        const direction = orderByProps[0].direction;
+        return direction === `desc` ? descComparator(a, b) : ascComparator(a, b);
+      }
+      return ascComparator(a, b);
+    };
   };
+  const comparator = makeComparator(orderByItems);
   if (hasOrderIndexColumn) {
     if (orderIndexType === `numeric`) {
       resultPipeline = resultPipeline.pipe(

package/dist/esm/query/order-by.js.map

@@ -1 +1 @@
-{"version":3,"file":"order-by.js","sources":["../../../src/query/order-by.ts"],"sourcesContent":["import {\n  map,\n  orderBy,\n  orderByWithFractionalIndex,\n  orderByWithIndex,\n} from \"@electric-sql/d2ts\"\nimport { evaluateOperandOnNamespacedRow } from \"./extractors\"\nimport { isOrderIndexFunctionCall } from \"./utils\"\nimport type { ConditionOperand, Query } from \"./schema\"\nimport type {\n  KeyedNamespacedRow,\n  NamespacedAndKeyedStream,\n  NamespacedRow,\n} from \"../types\"\n\nexport function processOrderBy(\n  resultPipeline: NamespacedAndKeyedStream,\n  query: Query,\n  mainTableAlias: string\n) {\n  // Check if any column in the SELECT clause is an ORDER_INDEX function call\n  let hasOrderIndexColumn = false\n  let orderIndexType: `numeric` | `fractional` = `numeric`\n  let orderIndexAlias = ``\n\n  // Scan the SELECT clause for ORDER_INDEX functions\n  // TODO: Select is going to be optional in future - we will automatically add an\n  // attribute for the index column\n  for (const item of query.select!) {\n    if (typeof item === `object`) {\n      for (const [alias, expr] of Object.entries(item)) {\n        if (typeof expr === `object` && isOrderIndexFunctionCall(expr)) {\n          hasOrderIndexColumn = true\n          orderIndexAlias = alias\n          orderIndexType = getOrderIndexType(expr)\n          break\n        }\n      }\n    }\n    if (hasOrderIndexColumn) break\n  }\n\n  // Normalize orderBy to an array of objects\n  const orderByItems: Array<{\n    operand: ConditionOperand\n    direction: `asc` | `desc`\n  }> = []\n\n  if (typeof query.orderBy === `string`) {\n    // Handle string format: '@column'\n    orderByItems.push({\n      operand: query.orderBy,\n      direction: `asc`,\n    })\n  } else if (Array.isArray(query.orderBy)) {\n    // Handle array format: ['@column1', { '@column2': 'desc' }]\n    for (const item of query.orderBy) {\n      if (typeof item === `string`) {\n        orderByItems.push({\n          operand: item,\n          direction: `asc`,\n        })\n      } else if (typeof item === `object`) {\n        for (const [column, direction] of Object.entries(item)) {\n          orderByItems.push({\n            operand: column,\n            direction: direction as `asc` | `desc`,\n          })\n        }\n      }\n    }\n  } else if (typeof query.orderBy === `object`) {\n    // Handle object format: { '@column': 'desc' }\n    for (const [column, direction] of Object.entries(query.orderBy)) {\n      orderByItems.push({\n        operand: column,\n        direction: direction as `asc` | `desc`,\n      })\n    }\n  }\n\n  // Create a value extractor function for the orderBy operator\n  // const valueExtractor = ([key, namespacedRow]: [\n  const valueExtractor = (namespacedRow: NamespacedRow) => {\n    // For multiple orderBy columns, create a composite key\n    if (orderByItems.length > 1) {\n      return orderByItems.map((item) => {\n        const val = evaluateOperandOnNamespacedRow(\n          namespacedRow,\n          item.operand,\n          mainTableAlias\n        )\n\n        // Reverse the value for 'desc' ordering\n        return item.direction === `desc` && typeof val === `number`\n          ? -val\n          : item.direction === `desc` && typeof val === `string`\n            ? String.fromCharCode(\n                ...[...val].map((c) => 0xffff - c.charCodeAt(0))\n              )\n            : val\n      })\n    } else if (orderByItems.length === 1) {\n      // For a single orderBy column, use the value directly\n      const item = orderByItems[0]\n      const val = evaluateOperandOnNamespacedRow(\n        namespacedRow,\n        item!.operand,\n        mainTableAlias\n      )\n\n      // Reverse the value for 'desc' ordering\n      return item!.direction === `desc` && typeof val === `number`\n        ? -val\n        : item!.direction === `desc` && typeof val === `string`\n          ? String.fromCharCode(\n              ...[...val].map((c) => 0xffff - c.charCodeAt(0))\n            )\n          : val\n    }\n\n    // Default case - no ordering\n    return null\n  }\n\n  const comparator = (a: unknown, b: unknown): number => {\n    // if a and b are both numbers compare them directly\n    if (typeof a === `number` && typeof b === `number`) {\n      return a - b\n    }\n    // if a and b are both strings, compare them lexicographically\n    if (typeof a === `string` && typeof b === `string`) {\n      return a.localeCompare(b)\n    }\n    // if a and b are both booleans, compare them\n    if (typeof a === `boolean` && typeof b === `boolean`) {\n      return a === b ? 0 : a ? 1 : -1\n    }\n    // if a and b are both dates, compare them\n    if (a instanceof Date && b instanceof Date) {\n      return a.getTime() - b.getTime()\n    }\n    // if a and b are both null, return 0\n    if (a === null || b === null) {\n      return 0\n    }\n\n    // if a and b are both arrays, compare them element by element\n    if (Array.isArray(a) && Array.isArray(b)) {\n      for (let i = 0; i < Math.min(a.length, b.length); i++) {\n        // Get the values from the array\n        const aVal = a[i]\n        const bVal = b[i]\n\n        // Compare the values\n        let result: number\n\n        if (typeof aVal === `boolean` && typeof bVal === `boolean`) {\n          // Special handling for booleans - false comes before true\n          result = aVal === bVal ? 0 : aVal ? 1 : -1\n        } else if (typeof aVal === `number` && typeof bVal === `number`) {\n          // Numeric comparison\n          result = aVal - bVal\n        } else if (typeof aVal === `string` && typeof bVal === `string`) {\n          // String comparison\n          result = aVal.localeCompare(bVal)\n        } else {\n          // Default comparison using the general comparator\n          result = comparator(aVal, bVal)\n        }\n\n        if (result !== 0) {\n          return result\n        }\n      }\n      // All elements are equal up to the minimum length\n      return a.length - b.length\n    }\n    // if a and b are both null/undefined, return 0\n    if (a == null && b == null) {\n      return 0\n    }\n    // Fallback to string comparison for all other cases\n    return (a as any).toString().localeCompare((b as any).toString())\n  }\n\n  // Apply the appropriate orderBy operator based on whether an ORDER_INDEX column is requested\n  if (hasOrderIndexColumn) {\n    if (orderIndexType === `numeric`) {\n      // Use orderByWithIndex for numeric indices\n      resultPipeline = resultPipeline.pipe(\n        orderByWithIndex(valueExtractor, {\n          limit: query.limit,\n          offset: query.offset,\n          comparator,\n        }),\n        map(([key, [value, index]]) => {\n          // Add the index to the result\n          // We add this to the main table alias for now\n          // TODO: re are going to need to refactor the whole order by pipeline\n          const result = {\n            ...(value as Record<string, unknown>),\n            [mainTableAlias]: {\n              ...value[mainTableAlias],\n              [orderIndexAlias]: index,\n            },\n          }\n          return [key, result] as KeyedNamespacedRow\n        })\n      )\n    } else {\n      // Use orderByWithFractionalIndex for fractional indices\n      resultPipeline = resultPipeline.pipe(\n        orderByWithFractionalIndex(valueExtractor, {\n          limit: query.limit,\n          offset: query.offset,\n          comparator,\n        }),\n        map(([key, [value, index]]) => {\n          // Add the index to the result\n          // We add this to the main table alias for now\n          // TODO: re are going to need to refactor the whole order by pipeline\n          const result = {\n            ...(value as Record<string, unknown>),\n            [mainTableAlias]: {\n              ...value[mainTableAlias],\n              [orderIndexAlias]: index,\n            },\n          }\n          return [key, result] as KeyedNamespacedRow\n        })\n      )\n    }\n  } else {\n    // Use regular orderBy if no index column is requested\n    resultPipeline = resultPipeline.pipe(\n      orderBy(valueExtractor, {\n        limit: query.limit,\n        offset: query.offset,\n        comparator,\n      })\n    )\n  }\n\n  return resultPipeline\n}\n\n// Helper function to extract the ORDER_INDEX type from a function call\nfunction getOrderIndexType(obj: any): `numeric` | `fractional` {\n  if (!isOrderIndexFunctionCall(obj)) {\n    throw new Error(`Not an ORDER_INDEX function call`)\n  }\n\n  const arg = obj[`ORDER_INDEX`]\n  if (arg === `numeric` || arg === true || arg === `default`) {\n    return `numeric`\n  } else if (arg === `fractional`) {\n    return `fractional`\n  } else {\n    throw new Error(`Invalid ORDER_INDEX type: ` + arg)\n  }\n}\n"],"names":[],"mappings":";;;AAegB,SAAA,eACd,gBACA,OACA,gBACA;AAEA,MAAI,sBAAsB;AAC1B,MAAI,iBAA2C;AAC/C,MAAI,kBAAkB;AAKX,aAAA,QAAQ,MAAM,QAAS;AAC5B,QAAA,OAAO,SAAS,UAAU;AAC5B,iBAAW,CAAC,OAAO,IAAI,KAAK,OAAO,QAAQ,IAAI,GAAG;AAChD,YAAI,OAAO,SAAS,YAAY,yBAAyB,IAAI,GAAG;AACxC,gCAAA;AACJ,4BAAA;AAClB,2BAAiB,kBAAkB,IAAI;AACvC;AAAA,QAAA;AAAA,MACF;AAAA,IACF;AAEF,QAAI,oBAAqB;AAAA,EAAA;AAI3B,QAAM,eAGD,CAAC;AAEF,MAAA,OAAO,MAAM,YAAY,UAAU;AAErC,iBAAa,KAAK;AAAA,MAChB,SAAS,MAAM;AAAA,MACf,WAAW;AAAA,IAAA,CACZ;AAAA,EACQ,WAAA,MAAM,QAAQ,MAAM,OAAO,GAAG;AAE5B,eAAA,QAAQ,MAAM,SAAS;AAC5B,UAAA,OAAO,SAAS,UAAU;AAC5B,qBAAa,KAAK;AAAA,UAChB,SAAS;AAAA,UACT,WAAW;AAAA,QAAA,CACZ;AAAA,MACH,WAAW,OAAO,SAAS,UAAU;AACnC,mBAAW,CAAC,QAAQ,SAAS,KAAK,OAAO,QAAQ,IAAI,GAAG;AACtD,uBAAa,KAAK;AAAA,YAChB,SAAS;AAAA,YACT;AAAA,UAAA,CACD;AAAA,QAAA;AAAA,MACH;AAAA,IACF;AAAA,EAEO,WAAA,OAAO,MAAM,YAAY,UAAU;AAEjC,eAAA,CAAC,QAAQ,SAAS,KAAK,OAAO,QAAQ,MAAM,OAAO,GAAG;AAC/D,mBAAa,KAAK;AAAA,QAChB,SAAS;AAAA,QACT;AAAA,MAAA,CACD;AAAA,IAAA;AAAA,EACH;AAKI,QAAA,iBAAiB,CAAC,kBAAiC;AAEnD,QAAA,aAAa,SAAS,GAAG;AACpB,aAAA,aAAa,IAAI,CAAC,SAAS;AAChC,cAAM,MAAM;AAAA,UACV;AAAA,UACA,KAAK;AAAA,UACL;AAAA,QACF;AAGA,eAAO,KAAK,cAAc,UAAU,OAAO,QAAQ,WAC/C,CAAC,MACD,KAAK,cAAc,UAAU,OAAO,QAAQ,WAC1C,OAAO;AAAA,UACL,GAAG,CAAC,GAAG,GAAG,EAAE,IAAI,CAAC,MAAM,QAAS,EAAE,WAAW,CAAC,CAAC;AAAA,QAAA,IAEjD;AAAA,MAAA,CACP;AAAA,IAAA,WACQ,aAAa,WAAW,GAAG;AAE9B,YAAA,OAAO,aAAa,CAAC;AAC3B,YAAM,MAAM;AAAA,QACV;AAAA,QACA,KAAM;AAAA,QACN;AAAA,MACF;AAGA,aAAO,KAAM,cAAc,UAAU,OAAO,QAAQ,WAChD,CAAC,MACD,KAAM,cAAc,UAAU,OAAO,QAAQ,WAC3C,OAAO;AAAA,QACL,GAAG,CAAC,GAAG,GAAG,EAAE,IAAI,CAAC,MAAM,QAAS,EAAE,WAAW,CAAC,CAAC;AAAA,MAAA,IAEjD;AAAA,IAAA;AAID,WAAA;AAAA,EACT;AAEM,QAAA,aAAa,CAAC,GAAY,MAAuB;AAErD,QAAI,OAAO,MAAM,YAAY,OAAO,MAAM,UAAU;AAClD,aAAO,IAAI;AAAA,IAAA;AAGb,QAAI,OAAO,MAAM,YAAY,OAAO,MAAM,UAAU;AAC3C,aAAA,EAAE,cAAc,CAAC;AAAA,IAAA;AAG1B,QAAI,OAAO,MAAM,aAAa,OAAO,MAAM,WAAW;AACpD,aAAO,MAAM,IAAI,IAAI,IAAI,IAAI;AAAA,IAAA;AAG3B,QAAA,aAAa,QAAQ,aAAa,MAAM;AAC1C,aAAO,EAAE,YAAY,EAAE,QAAQ;AAAA,IAAA;AAG7B,QAAA,MAAM,QAAQ,MAAM,MAAM;AACrB,aAAA;AAAA,IAAA;AAIT,QAAI,MAAM,QAAQ,CAAC,KAAK,MAAM,QAAQ,CAAC,GAAG;AAC/B,eAAA,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK;AAE/C,cAAA,OAAO,EAAE,CAAC;AACV,cAAA,OAAO,EAAE,CAAC;AAGZ,YAAA;AAEJ,YAAI,OAAO,SAAS,aAAa,OAAO,SAAS,WAAW;AAE1D,mBAAS,SAAS,OAAO,IAAI,OAAO,IAAI;AAAA,QAAA,WAC/B,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU;AAE/D,mBAAS,OAAO;AAAA,QAAA,WACP,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU;AAEtD,mBAAA,KAAK,cAAc,IAAI;AAAA,QAAA,OAC3B;AAEI,mBAAA,WAAW,MAAM,IAAI;AAAA,QAAA;AAGhC,YAAI,WAAW,GAAG;AACT,iBAAA;AAAA,QAAA;AAAA,MACT;AAGK,aAAA,EAAE,SAAS,EAAE;AAAA,IAAA;AAGlB,QAAA,KAAK,QAAQ,KAAK,MAAM;AACnB,aAAA;AAAA,IAAA;AAGT,WAAQ,EAAU,SAAS,EAAE,cAAe,EAAU,UAAU;AAAA,EAClE;AAGA,MAAI,qBAAqB;AACvB,QAAI,mBAAmB,WAAW;AAEhC,uBAAiB,eAAe;AAAA,QAC9B,iBAAiB,gBAAgB;AAAA,UAC/B,OAAO,MAAM;AAAA,UACb,QAAQ,MAAM;AAAA,UACd;AAAA,QAAA,CACD;AAAA,QACD,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,MAAM;AAI7B,gBAAM,SAAS;AAAA,YACb,GAAI;AAAA,YACJ,CAAC,cAAc,GAAG;AAAA,cAChB,GAAG,MAAM,cAAc;AAAA,cACvB,CAAC,eAAe,GAAG;AAAA,YAAA;AAAA,UAEvB;AACO,iBAAA,CAAC,KAAK,MAAM;AAAA,QACpB,CAAA;AAAA,MACH;AAAA,IAAA,OACK;AAEL,uBAAiB,eAAe;AAAA,QAC9B,2BAA2B,gBAAgB;AAAA,UACzC,OAAO,MAAM;AAAA,UACb,QAAQ,MAAM;AAAA,UACd;AAAA,QAAA,CACD;AAAA,QACD,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,MAAM;AAI7B,gBAAM,SAAS;AAAA,YACb,GAAI;AAAA,YACJ,CAAC,cAAc,GAAG;AAAA,cAChB,GAAG,MAAM,cAAc;AAAA,cACvB,CAAC,eAAe,GAAG;AAAA,YAAA;AAAA,UAEvB;AACO,iBAAA,CAAC,KAAK,MAAM;AAAA,QACpB,CAAA;AAAA,MACH;AAAA,IAAA;AAAA,EACF,OACK;AAEL,qBAAiB,eAAe;AAAA,MAC9B,QAAQ,gBAAgB;AAAA,QACtB,OAAO,MAAM;AAAA,QACb,QAAQ,MAAM;AAAA,QACd;AAAA,MACD,CAAA;AAAA,IACH;AAAA,EAAA;AAGK,SAAA;AACT;AAGA,SAAS,kBAAkB,KAAoC;AACzD,MAAA,CAAC,yBAAyB,GAAG,GAAG;AAC5B,UAAA,IAAI,MAAM,kCAAkC;AAAA,EAAA;AAG9C,QAAA,MAAM,IAAI,aAAa;AAC7B,MAAI,QAAQ,aAAa,QAAQ,QAAQ,QAAQ,WAAW;AACnD,WAAA;AAAA,EAAA,WACE,QAAQ,cAAc;AACxB,WAAA;AAAA,EAAA,OACF;AACC,UAAA,IAAI,MAAM,+BAA+B,GAAG;AAAA,EAAA;AAEtD;"}
+{"version":3,"file":"order-by.js","sources":["../../../src/query/order-by.ts"],"sourcesContent":["import {\n  map,\n  orderBy,\n  orderByWithFractionalIndex,\n  orderByWithIndex,\n} from \"@electric-sql/d2ts\"\nimport { evaluateOperandOnNamespacedRow } from \"./extractors\"\nimport { isOrderIndexFunctionCall } from \"./utils\"\nimport type { ConditionOperand, Query } from \"./schema\"\nimport type {\n  KeyedNamespacedRow,\n  NamespacedAndKeyedStream,\n  NamespacedRow,\n} from \"../types\"\n\ntype OrderByItem = {\n  operand: ConditionOperand\n  direction: `asc` | `desc`\n}\n\ntype OrderByItems = Array<OrderByItem>\n\nexport function processOrderBy(\n  resultPipeline: NamespacedAndKeyedStream,\n  query: Query,\n  mainTableAlias: string\n) {\n  // Check if any column in the SELECT clause is an ORDER_INDEX function call\n  let hasOrderIndexColumn = false\n  let orderIndexType: `numeric` | `fractional` = `numeric`\n  let orderIndexAlias = ``\n\n  // Scan the SELECT clause for ORDER_INDEX functions\n  // TODO: Select is going to be optional in future - we will automatically add an\n  // attribute for the index column\n  for (const item of query.select!) {\n    if (typeof item === `object`) {\n      for (const [alias, expr] of Object.entries(item)) {\n        if (typeof expr === `object` && isOrderIndexFunctionCall(expr)) {\n          hasOrderIndexColumn = true\n          orderIndexAlias = alias\n          orderIndexType = getOrderIndexType(expr)\n          break\n        }\n      }\n    }\n    if (hasOrderIndexColumn) break\n  }\n\n  // Normalize orderBy to an array of objects\n  const orderByItems: OrderByItems = []\n\n  if (typeof query.orderBy === `string`) {\n    // Handle string format: '@column'\n    orderByItems.push({\n      operand: query.orderBy,\n      direction: `asc`,\n    })\n  } else if (Array.isArray(query.orderBy)) {\n    // Handle array format: ['@column1', { '@column2': 'desc' }]\n    for (const item of query.orderBy) {\n      if (typeof item === `string`) {\n        orderByItems.push({\n          operand: item,\n          direction: `asc`,\n        })\n      } else if (typeof item === `object`) {\n        for (const [column, direction] of Object.entries(item)) {\n          orderByItems.push({\n            operand: column,\n            direction: direction as `asc` | `desc`,\n          })\n        }\n      }\n    }\n  } else if (typeof query.orderBy === `object`) {\n    // Handle object format: { '@column': 'desc' }\n    for (const [column, direction] of Object.entries(query.orderBy)) {\n      orderByItems.push({\n        operand: column,\n        direction: direction as `asc` | `desc`,\n      })\n    }\n  }\n\n  // Create a value extractor function for the orderBy operator\n  // const valueExtractor = ([key, namespacedRow]: [\n  const valueExtractor = (namespacedRow: NamespacedRow) => {\n    // For multiple orderBy columns, create a composite key\n    if (orderByItems.length > 1) {\n      return orderByItems.map((item) =>\n        evaluateOperandOnNamespacedRow(\n          namespacedRow,\n          item.operand,\n          mainTableAlias\n        )\n      )\n    } else if (orderByItems.length === 1) {\n      // For a single orderBy column, use the value directly\n      const item = orderByItems[0]\n      const val = evaluateOperandOnNamespacedRow(\n        namespacedRow,\n        item!.operand,\n        mainTableAlias\n      )\n      return val\n    }\n\n    // Default case - no ordering\n    return null\n  }\n\n  const ascComparator = (a: any, b: any): number => {\n    // if a and b are both strings, compare them based on locale\n    if (typeof a === `string` && typeof b === `string`) {\n      return a.localeCompare(b)\n    }\n\n    // if a and b are both arrays, compare them element by element\n    if (Array.isArray(a) && Array.isArray(b)) {\n      for (let i = 0; i < Math.min(a.length, b.length); i++) {\n        // Compare the values\n        const result = ascComparator(a[i], b[i])\n\n        if (result !== 0) {\n          return result\n        }\n      }\n      // All elements are equal up to the minimum length\n      return a.length - b.length\n    }\n\n    // If at least one of the values is an object then we don't really know how to meaningfully compare them\n    // therefore we turn them into strings and compare those\n    // There are 2 exceptions:\n    // 1) if both objects are dates then we can compare them\n    // 2) if either object is nullish then we can't call toString on it\n    const bothObjects = typeof a === `object` && typeof b === `object`\n    const bothDates = a instanceof Date && b instanceof Date\n    const notNull = a !== null && b !== null\n    if (bothObjects && !bothDates && notNull) {\n      // Every object should support `toString`\n      return a.toString().localeCompare(b.toString())\n    }\n\n    if (a < b) return -1\n    if (a > b) return 1\n    return 0\n  }\n\n  const descComparator = (a: unknown, b: unknown): number => {\n    return ascComparator(b, a)\n  }\n\n  // Create a multi-property comparator that respects the order and direction of each property\n  const makeComparator = (orderByProps: OrderByItems) => {\n    return (a: unknown, b: unknown) => {\n      // If we're comparing arrays (multiple properties), compare each property in order\n      if (orderByProps.length > 1) {\n        // `a` and `b` must be arrays since `orderByItems.length > 1`\n        // hence the extracted values must be arrays\n        const arrayA = a as Array<unknown>\n        const arrayB = b as Array<unknown>\n        for (let i = 0; i < orderByProps.length; i++) {\n          const direction = orderByProps[i]!.direction\n          const compareFn =\n            direction === `desc` ? descComparator : ascComparator\n          const result = compareFn(arrayA[i], arrayB[i])\n          if (result !== 0) {\n            return result\n          }\n        }\n        // should normally always be 0 because\n        // both values are extracted based on orderByItems\n        return arrayA.length - arrayB.length\n      }\n\n      // Single property comparison\n      if (orderByProps.length === 1) {\n        const direction = orderByProps[0]!.direction\n        return direction === `desc` ? descComparator(a, b) : ascComparator(a, b)\n      }\n\n      return ascComparator(a, b)\n    }\n  }\n  const comparator = makeComparator(orderByItems)\n\n  // Apply the appropriate orderBy operator based on whether an ORDER_INDEX column is requested\n  if (hasOrderIndexColumn) {\n    if (orderIndexType === `numeric`) {\n      // Use orderByWithIndex for numeric indices\n      resultPipeline = resultPipeline.pipe(\n        orderByWithIndex(valueExtractor, {\n          limit: query.limit,\n          offset: query.offset,\n          comparator,\n        }),\n        map(([key, [value, index]]) => {\n          // Add the index to the result\n          // We add this to the main table alias for now\n          // TODO: re are going to need to refactor the whole order by pipeline\n          const result = {\n            ...(value as Record<string, unknown>),\n            [mainTableAlias]: {\n              ...value[mainTableAlias],\n              [orderIndexAlias]: index,\n            },\n          }\n          return [key, result] as KeyedNamespacedRow\n        })\n      )\n    } else {\n      // Use orderByWithFractionalIndex for fractional indices\n      resultPipeline = resultPipeline.pipe(\n        orderByWithFractionalIndex(valueExtractor, {\n          limit: query.limit,\n          offset: query.offset,\n          comparator,\n        }),\n        map(([key, [value, index]]) => {\n          // Add the index to the result\n          // We add this to the main table alias for now\n          // TODO: re are going to need to refactor the whole order by pipeline\n          const result = {\n            ...(value as Record<string, unknown>),\n            [mainTableAlias]: {\n              ...value[mainTableAlias],\n              [orderIndexAlias]: index,\n            },\n          }\n          return [key, result] as KeyedNamespacedRow\n        })\n      )\n    }\n  } else {\n    // Use regular orderBy if no index column is requested\n    resultPipeline = resultPipeline.pipe(\n      orderBy(valueExtractor, {\n        limit: query.limit,\n        offset: query.offset,\n        comparator,\n      })\n    )\n  }\n\n  return resultPipeline\n}\n\n// Helper function to extract the ORDER_INDEX type from a function call\nfunction getOrderIndexType(obj: any): `numeric` | `fractional` {\n  if (!isOrderIndexFunctionCall(obj)) {\n    throw new Error(`Not an ORDER_INDEX function call`)\n  }\n\n  const arg = obj[`ORDER_INDEX`]\n  if (arg === `numeric` || arg === true || arg === `default`) {\n    return `numeric`\n  } else if (arg === `fractional`) {\n    return `fractional`\n  } else {\n    throw new Error(`Invalid ORDER_INDEX type: ` + arg)\n  }\n}\n"],"names":[],"mappings":";;;AAsBgB,SAAA,eACd,gBACA,OACA,gBACA;AAEA,MAAI,sBAAsB;AAC1B,MAAI,iBAA2C;AAC/C,MAAI,kBAAkB;AAKX,aAAA,QAAQ,MAAM,QAAS;AAC5B,QAAA,OAAO,SAAS,UAAU;AAC5B,iBAAW,CAAC,OAAO,IAAI,KAAK,OAAO,QAAQ,IAAI,GAAG;AAChD,YAAI,OAAO,SAAS,YAAY,yBAAyB,IAAI,GAAG;AACxC,gCAAA;AACJ,4BAAA;AAClB,2BAAiB,kBAAkB,IAAI;AACvC;AAAA,QAAA;AAAA,MACF;AAAA,IACF;AAEF,QAAI,oBAAqB;AAAA,EAAA;AAI3B,QAAM,eAA6B,CAAC;AAEhC,MAAA,OAAO,MAAM,YAAY,UAAU;AAErC,iBAAa,KAAK;AAAA,MAChB,SAAS,MAAM;AAAA,MACf,WAAW;AAAA,IAAA,CACZ;AAAA,EACQ,WAAA,MAAM,QAAQ,MAAM,OAAO,GAAG;AAE5B,eAAA,QAAQ,MAAM,SAAS;AAC5B,UAAA,OAAO,SAAS,UAAU;AAC5B,qBAAa,KAAK;AAAA,UAChB,SAAS;AAAA,UACT,WAAW;AAAA,QAAA,CACZ;AAAA,MACH,WAAW,OAAO,SAAS,UAAU;AACnC,mBAAW,CAAC,QAAQ,SAAS,KAAK,OAAO,QAAQ,IAAI,GAAG;AACtD,uBAAa,KAAK;AAAA,YAChB,SAAS;AAAA,YACT;AAAA,UAAA,CACD;AAAA,QAAA;AAAA,MACH;AAAA,IACF;AAAA,EAEO,WAAA,OAAO,MAAM,YAAY,UAAU;AAEjC,eAAA,CAAC,QAAQ,SAAS,KAAK,OAAO,QAAQ,MAAM,OAAO,GAAG;AAC/D,mBAAa,KAAK;AAAA,QAChB,SAAS;AAAA,QACT;AAAA,MAAA,CACD;AAAA,IAAA;AAAA,EACH;AAKI,QAAA,iBAAiB,CAAC,kBAAiC;AAEnD,QAAA,aAAa,SAAS,GAAG;AAC3B,aAAO,aAAa;AAAA,QAAI,CAAC,SACvB;AAAA,UACE;AAAA,UACA,KAAK;AAAA,UACL;AAAA,QAAA;AAAA,MAEJ;AAAA,IAAA,WACS,aAAa,WAAW,GAAG;AAE9B,YAAA,OAAO,aAAa,CAAC;AAC3B,YAAM,MAAM;AAAA,QACV;AAAA,QACA,KAAM;AAAA,QACN;AAAA,MACF;AACO,aAAA;AAAA,IAAA;AAIF,WAAA;AAAA,EACT;AAEM,QAAA,gBAAgB,CAAC,GAAQ,MAAmB;AAEhD,QAAI,OAAO,MAAM,YAAY,OAAO,MAAM,UAAU;AAC3C,aAAA,EAAE,cAAc,CAAC;AAAA,IAAA;AAI1B,QAAI,MAAM,QAAQ,CAAC,KAAK,MAAM,QAAQ,CAAC,GAAG;AAC/B,eAAA,IAAI,GAAG,IAAI,KAAK,IAAI,EAAE,QAAQ,EAAE,MAAM,GAAG,KAAK;AAErD,cAAM,SAAS,cAAc,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;AAEvC,YAAI,WAAW,GAAG;AACT,iBAAA;AAAA,QAAA;AAAA,MACT;AAGK,aAAA,EAAE,SAAS,EAAE;AAAA,IAAA;AAQtB,UAAM,cAAc,OAAO,MAAM,YAAY,OAAO,MAAM;AACpD,UAAA,YAAY,aAAa,QAAQ,aAAa;AAC9C,UAAA,UAAU,MAAM,QAAQ,MAAM;AAChC,QAAA,eAAe,CAAC,aAAa,SAAS;AAExC,aAAO,EAAE,SAAS,EAAE,cAAc,EAAE,UAAU;AAAA,IAAA;AAG5C,QAAA,IAAI,EAAU,QAAA;AACd,QAAA,IAAI,EAAU,QAAA;AACX,WAAA;AAAA,EACT;AAEM,QAAA,iBAAiB,CAAC,GAAY,MAAuB;AAClD,WAAA,cAAc,GAAG,CAAC;AAAA,EAC3B;AAGM,QAAA,iBAAiB,CAAC,iBAA+B;AAC9C,WAAA,CAAC,GAAY,MAAe;AAE7B,UAAA,aAAa,SAAS,GAAG;AAG3B,cAAM,SAAS;AACf,cAAM,SAAS;AACf,iBAAS,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;AACtC,gBAAA,YAAY,aAAa,CAAC,EAAG;AAC7B,gBAAA,YACJ,cAAc,SAAS,iBAAiB;AAC1C,gBAAM,SAAS,UAAU,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;AAC7C,cAAI,WAAW,GAAG;AACT,mBAAA;AAAA,UAAA;AAAA,QACT;AAIK,eAAA,OAAO,SAAS,OAAO;AAAA,MAAA;AAI5B,UAAA,aAAa,WAAW,GAAG;AACvB,cAAA,YAAY,aAAa,CAAC,EAAG;AAC5B,eAAA,cAAc,SAAS,eAAe,GAAG,CAAC,IAAI,cAAc,GAAG,CAAC;AAAA,MAAA;AAGlE,aAAA,cAAc,GAAG,CAAC;AAAA,IAC3B;AAAA,EACF;AACM,QAAA,aAAa,eAAe,YAAY;AAG9C,MAAI,qBAAqB;AACvB,QAAI,mBAAmB,WAAW;AAEhC,uBAAiB,eAAe;AAAA,QAC9B,iBAAiB,gBAAgB;AAAA,UAC/B,OAAO,MAAM;AAAA,UACb,QAAQ,MAAM;AAAA,UACd;AAAA,QAAA,CACD;AAAA,QACD,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,MAAM;AAI7B,gBAAM,SAAS;AAAA,YACb,GAAI;AAAA,YACJ,CAAC,cAAc,GAAG;AAAA,cAChB,GAAG,MAAM,cAAc;AAAA,cACvB,CAAC,eAAe,GAAG;AAAA,YAAA;AAAA,UAEvB;AACO,iBAAA,CAAC,KAAK,MAAM;AAAA,QACpB,CAAA;AAAA,MACH;AAAA,IAAA,OACK;AAEL,uBAAiB,eAAe;AAAA,QAC9B,2BAA2B,gBAAgB;AAAA,UACzC,OAAO,MAAM;AAAA,UACb,QAAQ,MAAM;AAAA,UACd;AAAA,QAAA,CACD;AAAA,QACD,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,MAAM;AAI7B,gBAAM,SAAS;AAAA,YACb,GAAI;AAAA,YACJ,CAAC,cAAc,GAAG;AAAA,cAChB,GAAG,MAAM,cAAc;AAAA,cACvB,CAAC,eAAe,GAAG;AAAA,YAAA;AAAA,UAEvB;AACO,iBAAA,CAAC,KAAK,MAAM;AAAA,QACpB,CAAA;AAAA,MACH;AAAA,IAAA;AAAA,EACF,OACK;AAEL,qBAAiB,eAAe;AAAA,MAC9B,QAAQ,gBAAgB;AAAA,QACtB,OAAO,MAAM;AAAA,QACb,QAAQ,MAAM;AAAA,QACd;AAAA,MACD,CAAA;AAAA,IACH;AAAA,EAAA;AAGK,SAAA;AACT;AAGA,SAAS,kBAAkB,KAAoC;AACzD,MAAA,CAAC,yBAAyB,GAAG,GAAG;AAC5B,UAAA,IAAI,MAAM,kCAAkC;AAAA,EAAA;AAG9C,QAAA,MAAM,IAAI,aAAa;AAC7B,MAAI,QAAQ,aAAa,QAAQ,QAAQ,QAAQ,WAAW;AACnD,WAAA;AAAA,EAAA,WACE,QAAQ,cAAc;AACxB,WAAA;AAAA,EAAA,OACF;AACC,UAAA,IAAI,MAAM,+BAA+B,GAAG;AAAA,EAAA;AAEtD;"}

package/dist/esm/query/query-builder.d.ts

@@ -1,5 +1,5 @@
 import { Collection } from '../collection.js';
-import { Comparator, Condition, Limit, LiteralValue, Offset, OrderBy, Query, Select, WhereCallback } from './schema.js';
+import { Comparator, ComparatorValue, Condition, Limit, LiteralValue, Offset, OrderBy, Query, Select, WhereCallback } from './schema.js';
 import { Context, Flatten, InferResultTypeFromSelectTuple, Input, InputReference, PropertyReference, PropertyReferenceString, RemoveIndexSignature, Schema } from './types.js';
 type CollectionRef = {
     [K: string]: Collection<any>;
@@ -55,7 +55,7 @@ export declare class BaseQueryBuilder<TContext extends Context<Schema>> {
     /**
      * Add a where clause comparing two values.
      */
-    where(left: PropertyReferenceString<TContext> | LiteralValue, operator: Comparator, right: PropertyReferenceString<TContext> | LiteralValue): QueryBuilder<TContext>;
+    where<T extends Comparator>(left: PropertyReferenceString<TContext> | LiteralValue, operator: T, right: ComparatorValue<T, TContext>): QueryBuilder<TContext>;
     /**
      * Add a where clause with a complete condition object.
      */

package/dist/esm/query/query-builder.js.map

@@ -1 +1 @@
-{"version":3,"file":"query-builder.js","sources":["../../../src/query/query-builder.ts"],"sourcesContent":["import type { Collection } from \"../collection\"\nimport type {\n  Comparator,\n  Condition,\n  From,\n  JoinClause,\n  Limit,\n  LiteralValue,\n  Offset,\n  OrderBy,\n  Query,\n  Select,\n  WhereCallback,\n  WithQuery,\n} from \"./schema.js\"\nimport type {\n  Context,\n  Flatten,\n  InferResultTypeFromSelectTuple,\n  Input,\n  InputReference,\n  PropertyReference,\n  PropertyReferenceString,\n  RemoveIndexSignature,\n  Schema,\n} from \"./types.js\"\n\ntype CollectionRef = { [K: string]: Collection<any> }\n\nexport class BaseQueryBuilder<TContext extends Context<Schema>> {\n  private readonly query: Partial<Query<TContext>> = {}\n\n  /**\n   * Create a new QueryBuilder instance.\n   */\n  constructor(query: Partial<Query<TContext>> = {}) {\n    this.query = query\n  }\n\n  from<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ): QueryBuilder<{\n    baseSchema: Flatten<\n      TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n          (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n            ? T\n            : never) &\n            Input\n        >\n      }\n    >\n    schema: Flatten<{\n      [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n        (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n          ? T\n          : never) &\n          Input\n      >\n    }>\n    default: keyof TCollectionRef & string\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(\n    collection: T\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: T\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(\n    collection: T,\n    as: TAs\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: TAs\n  }>\n\n  /**\n   * Specify the collection to query from.\n   * This is the first method that must be called in the chain.\n   *\n   * @param collection The collection name to query from\n   * @param as Optional alias for the collection\n   * @returns A new QueryBuilder with the from clause set\n   */\n  from<\n    T extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof collection === `object` && collection !== null) {\n      return this.fromCollectionRef(collection)\n    } else if (typeof collection === `string`) {\n      return this.fromInputReference(\n        collection as InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>,\n        as\n      )\n    } else {\n      throw new Error(`Invalid collection type`)\n    }\n  }\n\n  private fromCollectionRef<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ) {\n    const keys = Object.keys(collectionRef)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key`)\n    }\n\n    const key = keys[0]!\n    const collection = collectionRef[key]!\n\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = key as From<TContext>\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      schema: {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      default: keyof TCollectionRef & string\n    }>\n  }\n\n  private fromInputReference<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = collection as From<TContext>\n    if (as) {\n      newBuilder.query.as = as\n    }\n\n    // Calculate the result type without deep nesting\n    type ResultSchema = TAs extends undefined\n      ? { [K in T]: TContext[`baseSchema`][T] }\n      : { [K in string & TAs]: TContext[`baseSchema`][T] }\n\n    type ResultDefault = TAs extends undefined ? T : string & TAs\n\n    // Use simpler type assertion to avoid excessive depth\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: ResultSchema\n      default: ResultDefault\n    }>\n  }\n\n  /**\n   * Specify what columns to select.\n   * Overwrites any previous select clause.\n   * Also supports callback functions that receive the row context and return selected data.\n   *\n   * @param selects The columns to select (can include callbacks)\n   * @returns A new QueryBuilder with the select clause set\n   */\n  select<TSelects extends Array<Select<TContext>>>(\n    this: QueryBuilder<TContext>,\n    ...selects: TSelects\n  ) {\n    // Validate function calls in the selects\n    // Need to use a type assertion to bypass deep recursive type checking\n    const validatedSelects = selects.map((select) => {\n      // If the select is an object with aliases, validate each value\n      if (\n        typeof select === `object` &&\n        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n        select !== null &&\n        !Array.isArray(select)\n      ) {\n        const result: Record<string, any> = {}\n\n        for (const [key, value] of Object.entries(select)) {\n          // If it's a function call (object with a single key that is an allowed function name)\n          if (\n            typeof value === `object` &&\n            value !== null &&\n            !Array.isArray(value)\n          ) {\n            const keys = Object.keys(value)\n            if (keys.length === 1) {\n              const funcName = keys[0]!\n              // List of allowed function names from AllowedFunctionName\n              const allowedFunctions = [\n                `SUM`,\n                `COUNT`,\n                `AVG`,\n                `MIN`,\n                `MAX`,\n                `DATE`,\n                `JSON_EXTRACT`,\n                `JSON_EXTRACT_PATH`,\n                `UPPER`,\n                `LOWER`,\n                `COALESCE`,\n                `CONCAT`,\n                `LENGTH`,\n                `ORDER_INDEX`,\n              ]\n\n              if (!allowedFunctions.includes(funcName)) {\n                console.warn(\n                  `Unsupported function: ${funcName}. Expected one of: ${allowedFunctions.join(`, `)}`\n                )\n              }\n            }\n          }\n\n          result[key] = value\n        }\n\n        return result\n      }\n\n      return select\n    })\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called after orderBy\n    if (this._query.orderBy) {\n      validatedSelects.push({ _orderByIndex: { ORDER_INDEX: `numeric` } })\n    }\n\n    const newBuilder = new BaseQueryBuilder<TContext>(\n      (this as BaseQueryBuilder<TContext>).query\n    )\n    newBuilder.query.select = validatedSelects as Array<Select<TContext>>\n\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `result`> & {\n          result: InferResultTypeFromSelectTuple<TContext, TSelects>\n        }\n      >\n    >\n  }\n\n  /**\n   * Add a where clause comparing two values.\n   */\n  where(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: Comparator,\n    right: PropertyReferenceString<TContext> | LiteralValue\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a complete condition object.\n   */\n  where(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a callback function.\n   */\n  where(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause to filter the results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the where clause added\n   */\n  where(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    // Use simplistic approach to avoid deep type errors\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Where is always an array, so initialize or append\n    if (!newBuilder.query.where) {\n      newBuilder.query.where = [condition]\n    } else {\n      newBuilder.query.where = [...newBuilder.query.where, condition]\n    }\n\n    return newBuilder as unknown as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a having clause comparing two values.\n   * For filtering results after they have been grouped.\n   */\n  having(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: Comparator,\n    right: PropertyReferenceString<TContext> | LiteralValue\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a complete condition object.\n   * For filtering results after they have been grouped.\n   */\n  having(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a callback function.\n   * For filtering results after they have been grouped.\n   */\n  having(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause to filter the grouped results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the having clause added\n   */\n  having(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Having is always an array, so initialize or append\n    if (!newBuilder.query.having) {\n      newBuilder.query.having = [condition]\n    } else {\n      newBuilder.query.having = [...newBuilder.query.having, condition]\n    }\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a join clause to the query using a CollectionRef.\n   */\n  join<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query without specifying an alias.\n   * The collection name will be used as the default alias.\n   */\n  join<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: T\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: { [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]> }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query with a specified alias.\n   */\n  join<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  join<\n    TFrom extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n  }): QueryBuilder<any> {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof joinClause.from === `object` && joinClause.from !== null) {\n      return this.joinCollectionRef(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: CollectionRef\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    } else {\n      return this.joinInputReference(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: InputReference<{\n            baseSchema: TContext[`baseSchema`]\n            schema: TContext[`baseSchema`]\n          }>\n          as?: TAs\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    }\n  }\n\n  private joinCollectionRef<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Get the collection key\n    const keys = Object.keys(joinClause.from)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key in CollectionRef`)\n    }\n    const key = keys[0]!\n    const collection = joinClause.from[key]\n    if (!collection) {\n      throw new Error(`Collection not found for key: ${key}`)\n    }\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = {\n      type: joinClause.type,\n      from: key,\n      on: joinClause.on,\n      where: joinClause.where,\n    } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Add the collection to the collections map\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n              (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n                ? T\n                : never) &\n                Input\n            >\n          }\n        }\n      >\n    >\n  }\n\n  private joinInputReference<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = { ...joinClause } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Determine the alias or use the collection name as default\n    const _effectiveAlias = joinClause.as ?? joinClause.from\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in typeof _effectiveAlias]: TContext[`baseSchema`][TFrom]\n          }\n        }\n      >\n    >\n  }\n\n  /**\n   * Add an orderBy clause to sort the results.\n   * Overwrites any previous orderBy clause.\n   *\n   * @param orderBy The order specification\n   * @returns A new QueryBuilder with the orderBy clause set\n   */\n  orderBy(orderBy: OrderBy<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the orderBy clause\n    newBuilder.query.orderBy = orderBy\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called before orderBy\n    newBuilder.query.select = [\n      ...(newBuilder.query.select ?? []),\n      { _orderByIndex: { ORDER_INDEX: `numeric` } },\n    ]\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set a limit on the number of results returned.\n   *\n   * @param limit Maximum number of results to return\n   * @returns A new QueryBuilder with the limit set\n   */\n  limit(limit: Limit<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the limit\n    newBuilder.query.limit = limit\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set an offset to skip a number of results.\n   *\n   * @param offset Number of results to skip\n   * @returns A new QueryBuilder with the offset set\n   */\n  offset(offset: Offset<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the offset\n    newBuilder.query.offset = offset\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a groupBy clause to group the results by one or more columns.\n   *\n   * @param groupBy The column(s) to group by\n   * @returns A new QueryBuilder with the groupBy clause set\n   */\n  groupBy(\n    groupBy: PropertyReference<TContext> | Array<PropertyReference<TContext>>\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the groupBy clause\n    newBuilder.query.groupBy = groupBy\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Define a Common Table Expression (CTE) that can be referenced in the main query.\n   * This allows referencing the CTE by name in subsequent from/join clauses.\n   *\n   * @param name The name of the CTE\n   * @param queryBuilderCallback A function that builds the CTE query\n   * @returns A new QueryBuilder with the CTE added\n   */\n  with<TName extends string, TResult = Record<string, unknown>>(\n    name: TName,\n    queryBuilderCallback: (\n      builder: InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    ) => QueryBuilder<any>\n  ): InitialQueryBuilder<{\n    baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n    schema: TContext[`schema`]\n  }> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a new builder for the CTE\n    const cteBuilder = new BaseQueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: {}\n    }>()\n\n    // Get the CTE query from the callback\n    const cteQueryBuilder = queryBuilderCallback(\n      cteBuilder as InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    )\n\n    // Get the query from the builder\n    const cteQuery = cteQueryBuilder._query\n\n    // Add an 'as' property to the CTE\n    const withQuery: WithQuery<any> = {\n      ...cteQuery,\n      as: name,\n    }\n\n    // Add the CTE to the with array\n    if (!newBuilder.query.with) {\n      newBuilder.query.with = [withQuery]\n    } else {\n      newBuilder.query.with = [...newBuilder.query.with, withQuery]\n    }\n\n    // Use a type cast that simplifies the type structure to avoid recursion\n    return newBuilder as unknown as InitialQueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n      schema: TContext[`schema`]\n    }>\n  }\n\n  get _query(): Query<TContext> {\n    return this.query as Query<TContext>\n  }\n}\n\nexport type InitialQueryBuilder<TContext extends Context<Schema>> = Pick<\n  BaseQueryBuilder<TContext>,\n  `from` | `with`\n>\n\nexport type QueryBuilder<TContext extends Context<Schema>> = Omit<\n  BaseQueryBuilder<TContext>,\n  `from`\n>\n\n/**\n * Create a new query builder with the given schema\n */\nexport function queryBuilder<TBaseSchema extends Schema = {}>() {\n  return new BaseQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>() as InitialQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>\n}\n\nexport type ResultsFromContext<TContext extends Context<Schema>> = Flatten<\n  TContext[`result`] extends object\n    ? TContext[`result`] // If there is a select we will have a result type\n    : TContext[`hasJoin`] extends true\n      ? TContext[`schema`] // If there is a join, the query returns the namespaced schema\n      : TContext[`default`] extends keyof TContext[`schema`]\n        ? TContext[`schema`][TContext[`default`]] // If there is no join we return the flat default schema\n        : never // Should never happen\n>\n\nexport type ResultFromQueryBuilder<TQueryBuilder> = Flatten<\n  TQueryBuilder extends QueryBuilder<infer C>\n    ? C extends { result: infer R }\n      ? R\n      : never\n    : never\n>\n"],"names":[],"mappings":"AA6BO,MAAM,iBAAmD;AAAA;AAAA;AAAA;AAAA,EAM9D,YAAY,QAAkC,IAAI;AALlD,SAAiB,QAAkC,CAAC;AAMlD,SAAK,QAAQ;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmEf,KAQE,YAAe,IAAU;AAEzB,QAAI,OAAO,eAAe,YAAY,eAAe,MAAM;AAClD,aAAA,KAAK,kBAAkB,UAAU;AAAA,IAC1C,WAAW,OAAO,eAAe,UAAU;AACzC,aAAO,KAAK;AAAA,QACV;AAAA,QAIA;AAAA,MACF;AAAA,IAAA,OACK;AACC,YAAA,IAAI,MAAM,yBAAyB;AAAA,IAAA;AAAA,EAC3C;AAAA,EAGM,kBACN,eACA;AArGG;AAsGG,UAAA,OAAO,OAAO,KAAK,aAAa;AAClC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,0BAA0B;AAAA,IAAA;AAGtC,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,cAAc,GAAG;AAE9B,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACb,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAE7B,WAAA;AAAA,EAAA;AAAA,EAuBD,mBAMN,YAAe,IAAU;AACnB,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACxB,QAAI,IAAI;AACN,iBAAW,MAAM,KAAK;AAAA,IAAA;AAWjB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeT,UAEK,SACH;AAGA,UAAM,mBAAmB,QAAQ,IAAI,CAAC,WAAW;AAE/C,UACE,OAAO,WAAW;AAAA,MAElB,WAAW,QACX,CAAC,MAAM,QAAQ,MAAM,GACrB;AACA,cAAM,SAA8B,CAAC;AAErC,mBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,MAAM,GAAG;AAG/C,cAAA,OAAO,UAAU,YACjB,UAAU,QACV,CAAC,MAAM,QAAQ,KAAK,GACpB;AACM,kBAAA,OAAO,OAAO,KAAK,KAAK;AAC1B,gBAAA,KAAK,WAAW,GAAG;AACf,oBAAA,WAAW,KAAK,CAAC;AAEvB,oBAAM,mBAAmB;AAAA,gBACvB;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACF;AAEA,kBAAI,CAAC,iBAAiB,SAAS,QAAQ,GAAG;AAChC,wBAAA;AAAA,kBACN,yBAAyB,QAAQ,sBAAsB,iBAAiB,KAAK,IAAI,CAAC;AAAA,gBACpF;AAAA,cAAA;AAAA,YACF;AAAA,UACF;AAGF,iBAAO,GAAG,IAAI;AAAA,QAAA;AAGT,eAAA;AAAA,MAAA;AAGF,aAAA;AAAA,IAAA,CACR;AAIG,QAAA,KAAK,OAAO,SAAS;AACvB,uBAAiB,KAAK,EAAE,eAAe,EAAE,aAAa,UAAA,GAAa;AAAA,IAAA;AAGrE,UAAM,aAAa,IAAI;AAAA,MACpB,KAAoC;AAAA,IACvC;AACA,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAsCT,MACE,2BACA,UACA,OACwB;AAGlB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,OAAO;AAChB,iBAAA,MAAM,QAAQ,CAAC,SAAS;AAAA,IAAA,OAC9B;AACL,iBAAW,MAAM,QAAQ,CAAC,GAAG,WAAW,MAAM,OAAO,SAAS;AAAA,IAAA;AAGzD,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmCT,OACE,2BACA,UACA,OACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,QAAQ;AACjB,iBAAA,MAAM,SAAS,CAAC,SAAS;AAAA,IAAA,OAC/B;AACL,iBAAW,MAAM,SAAS,CAAC,GAAG,WAAW,MAAM,QAAQ,SAAS;AAAA,IAAA;AAG3D,WAAA;AAAA,EAAA;AAAA,EAgIT,KAQE,YA4CoB;AAEpB,QAAI,OAAO,WAAW,SAAS,YAAY,WAAW,SAAS,MAAM;AACnE,aAAO,KAAK;AAAA,QACV;AAAA,MAMF;AAAA,IAAA,OACK;AACL,aAAO,KAAK;AAAA,QACV;AAAA,MAUF;AAAA,IAAA;AAAA,EACF;AAAA,EAGM,kBAAwD,YAK1C;AAllBjB;AAolBG,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,UAAM,OAAO,OAAO,KAAK,WAAW,IAAI;AACpC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,2CAA2C;AAAA,IAAA;AAEvD,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,WAAW,KAAK,GAAG;AACtC,QAAI,CAAC,YAAY;AACf,YAAM,IAAI,MAAM,iCAAiC,GAAG,EAAE;AAAA,IAAA;AAIxD,UAAM,iBAAiB;AAAA,MACrB,MAAM,WAAW;AAAA,MACjB,MAAM;AAAA,MACN,IAAI,WAAW;AAAA,MACf,OAAO,WAAW;AAAA,IACpB;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAIxD,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAG7B,WAAA;AAAA,EAAA;AAAA,EAgBD,mBAMN,YAMoB;AAEd,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,iBAAiB,EAAE,GAAG,WAAW;AAGnC,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAI3C,eAAW,MAAM,WAAW;AAG7C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBT,QAAQ,SAAoD;AAEpD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAI3B,eAAW,MAAM,SAAS;AAAA,MACxB,GAAI,WAAW,MAAM,UAAU,CAAC;AAAA,MAChC,EAAE,eAAe,EAAE,aAAa,UAAY,EAAA;AAAA,IAC9C;AAEO,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,MAAM,OAAgD;AAE9C,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,QAAQ;AAElB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,OAAO,QAAkD;AAEjD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,QACE,SACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAEpB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWT,KACE,MACA,sBASC;AAEK,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,aAAa,IAAI,iBAGpB;AAGH,UAAM,kBAAkB;AAAA,MACtB;AAAA,IAIF;AAGA,UAAM,WAAW,gBAAgB;AAGjC,UAAM,YAA4B;AAAA,MAChC,GAAG;AAAA,MACH,IAAI;AAAA,IACN;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,SAAS;AAAA,IAAA,OAC7B;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,SAAS;AAAA,IAAA;AAIvD,WAAA;AAAA,EAAA;AAAA,EAMT,IAAI,SAA0B;AAC5B,WAAO,KAAK;AAAA,EAAA;AAEhB;AAeO,SAAS,eAAgD;AAC9D,SAAO,IAAI,iBAGR;AAIL;"}
+{"version":3,"file":"query-builder.js","sources":["../../../src/query/query-builder.ts"],"sourcesContent":["import type { Collection } from \"../collection\"\nimport type {\n  Comparator,\n  ComparatorValue,\n  Condition,\n  From,\n  JoinClause,\n  Limit,\n  LiteralValue,\n  Offset,\n  OrderBy,\n  Query,\n  Select,\n  WhereCallback,\n  WithQuery,\n} from \"./schema.js\"\nimport type {\n  Context,\n  Flatten,\n  InferResultTypeFromSelectTuple,\n  Input,\n  InputReference,\n  PropertyReference,\n  PropertyReferenceString,\n  RemoveIndexSignature,\n  Schema,\n} from \"./types.js\"\n\ntype CollectionRef = { [K: string]: Collection<any> }\n\nexport class BaseQueryBuilder<TContext extends Context<Schema>> {\n  private readonly query: Partial<Query<TContext>> = {}\n\n  /**\n   * Create a new QueryBuilder instance.\n   */\n  constructor(query: Partial<Query<TContext>> = {}) {\n    this.query = query\n  }\n\n  from<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ): QueryBuilder<{\n    baseSchema: Flatten<\n      TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n          (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n            ? T\n            : never) &\n            Input\n        >\n      }\n    >\n    schema: Flatten<{\n      [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n        (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n          ? T\n          : never) &\n          Input\n      >\n    }>\n    default: keyof TCollectionRef & string\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(\n    collection: T\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: T\n  }>\n\n  from<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(\n    collection: T,\n    as: TAs\n  ): QueryBuilder<{\n    baseSchema: TContext[`baseSchema`]\n    schema: {\n      [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n    }\n    default: TAs\n  }>\n\n  /**\n   * Specify the collection to query from.\n   * This is the first method that must be called in the chain.\n   *\n   * @param collection The collection name to query from\n   * @param as Optional alias for the collection\n   * @returns A new QueryBuilder with the from clause set\n   */\n  from<\n    T extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof collection === `object` && collection !== null) {\n      return this.fromCollectionRef(collection)\n    } else if (typeof collection === `string`) {\n      return this.fromInputReference(\n        collection as InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>,\n        as\n      )\n    } else {\n      throw new Error(`Invalid collection type`)\n    }\n  }\n\n  private fromCollectionRef<TCollectionRef extends CollectionRef>(\n    collectionRef: TCollectionRef\n  ) {\n    const keys = Object.keys(collectionRef)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key`)\n    }\n\n    const key = keys[0]!\n    const collection = collectionRef[key]!\n\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = key as From<TContext>\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      schema: {\n        [K in keyof TCollectionRef &\n          string]: (TCollectionRef[keyof TCollectionRef] extends Collection<\n          infer T\n        >\n          ? T\n          : never) &\n          Input\n      }\n      default: keyof TCollectionRef & string\n    }>\n  }\n\n  private fromInputReference<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined,\n  >(collection: T, as?: TAs) {\n    const newBuilder = new BaseQueryBuilder()\n    Object.assign(newBuilder.query, this.query)\n    newBuilder.query.from = collection as From<TContext>\n    if (as) {\n      newBuilder.query.as = as\n    }\n\n    // Calculate the result type without deep nesting\n    type ResultSchema = TAs extends undefined\n      ? { [K in T]: TContext[`baseSchema`][T] }\n      : { [K in string & TAs]: TContext[`baseSchema`][T] }\n\n    type ResultDefault = TAs extends undefined ? T : string & TAs\n\n    // Use simpler type assertion to avoid excessive depth\n    return newBuilder as unknown as QueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: ResultSchema\n      default: ResultDefault\n    }>\n  }\n\n  /**\n   * Specify what columns to select.\n   * Overwrites any previous select clause.\n   * Also supports callback functions that receive the row context and return selected data.\n   *\n   * @param selects The columns to select (can include callbacks)\n   * @returns A new QueryBuilder with the select clause set\n   */\n  select<TSelects extends Array<Select<TContext>>>(\n    this: QueryBuilder<TContext>,\n    ...selects: TSelects\n  ) {\n    // Validate function calls in the selects\n    // Need to use a type assertion to bypass deep recursive type checking\n    const validatedSelects = selects.map((select) => {\n      // If the select is an object with aliases, validate each value\n      if (\n        typeof select === `object` &&\n        // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n        select !== null &&\n        !Array.isArray(select)\n      ) {\n        const result: Record<string, any> = {}\n\n        for (const [key, value] of Object.entries(select)) {\n          // If it's a function call (object with a single key that is an allowed function name)\n          if (\n            typeof value === `object` &&\n            value !== null &&\n            !Array.isArray(value)\n          ) {\n            const keys = Object.keys(value)\n            if (keys.length === 1) {\n              const funcName = keys[0]!\n              // List of allowed function names from AllowedFunctionName\n              const allowedFunctions = [\n                `SUM`,\n                `COUNT`,\n                `AVG`,\n                `MIN`,\n                `MAX`,\n                `DATE`,\n                `JSON_EXTRACT`,\n                `JSON_EXTRACT_PATH`,\n                `UPPER`,\n                `LOWER`,\n                `COALESCE`,\n                `CONCAT`,\n                `LENGTH`,\n                `ORDER_INDEX`,\n              ]\n\n              if (!allowedFunctions.includes(funcName)) {\n                console.warn(\n                  `Unsupported function: ${funcName}. Expected one of: ${allowedFunctions.join(`, `)}`\n                )\n              }\n            }\n          }\n\n          result[key] = value\n        }\n\n        return result\n      }\n\n      return select\n    })\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called after orderBy\n    if (this._query.orderBy) {\n      validatedSelects.push({ _orderByIndex: { ORDER_INDEX: `numeric` } })\n    }\n\n    const newBuilder = new BaseQueryBuilder<TContext>(\n      (this as BaseQueryBuilder<TContext>).query\n    )\n    newBuilder.query.select = validatedSelects as Array<Select<TContext>>\n\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `result`> & {\n          result: InferResultTypeFromSelectTuple<TContext, TSelects>\n        }\n      >\n    >\n  }\n\n  /**\n   * Add a where clause comparing two values.\n   */\n  where<T extends Comparator>(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: T,\n    right: ComparatorValue<T, TContext>\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a complete condition object.\n   */\n  where(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause with a callback function.\n   */\n  where(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a where clause to filter the results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the where clause added\n   */\n  where(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    // Use simplistic approach to avoid deep type errors\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Where is always an array, so initialize or append\n    if (!newBuilder.query.where) {\n      newBuilder.query.where = [condition]\n    } else {\n      newBuilder.query.where = [...newBuilder.query.where, condition]\n    }\n\n    return newBuilder as unknown as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a having clause comparing two values.\n   * For filtering results after they have been grouped.\n   */\n  having(\n    left: PropertyReferenceString<TContext> | LiteralValue,\n    operator: Comparator,\n    right: PropertyReferenceString<TContext> | LiteralValue\n  ): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a complete condition object.\n   * For filtering results after they have been grouped.\n   */\n  having(condition: Condition<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause with a callback function.\n   * For filtering results after they have been grouped.\n   */\n  having(callback: WhereCallback<TContext>): QueryBuilder<TContext>\n\n  /**\n   * Add a having clause to filter the grouped results.\n   * Can be called multiple times to add AND conditions.\n   * Also supports callback functions that receive the row context.\n   *\n   * @param leftOrConditionOrCallback The left operand, complete condition, or callback function\n   * @param operator Optional comparison operator\n   * @param right Optional right operand\n   * @returns A new QueryBuilder with the having clause added\n   */\n  having(\n    leftOrConditionOrCallback: any,\n    operator?: any,\n    right?: any\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    let condition: any\n\n    // Determine if this is a callback, complete condition, or individual parts\n    if (typeof leftOrConditionOrCallback === `function`) {\n      // It's a callback function\n      condition = leftOrConditionOrCallback\n    } else if (operator !== undefined && right !== undefined) {\n      // Create a condition from parts\n      condition = [leftOrConditionOrCallback, operator, right]\n    } else {\n      // Use the provided condition directly\n      condition = leftOrConditionOrCallback\n    }\n\n    // Having is always an array, so initialize or append\n    if (!newBuilder.query.having) {\n      newBuilder.query.having = [condition]\n    } else {\n      newBuilder.query.having = [...newBuilder.query.having, condition]\n    }\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a join clause to the query using a CollectionRef.\n   */\n  join<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n            (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n              ? T\n              : never) &\n              Input\n          >\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query without specifying an alias.\n   * The collection name will be used as the default alias.\n   */\n  join<\n    T extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: T\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: { [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]> }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in T]: RemoveIndexSignature<TContext[`baseSchema`][T]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  /**\n   * Add a join clause to the query with a specified alias.\n   */\n  join<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n      }>\n    >\n  }): QueryBuilder<\n    Flatten<\n      Omit<TContext, `schema`> & {\n        schema: TContext[`schema`] & {\n          [K in TAs]: RemoveIndexSignature<TContext[`baseSchema`][TFrom]>\n        }\n        hasJoin: true\n      }\n    >\n  >\n\n  join<\n    TFrom extends\n      | InputReference<{\n          baseSchema: TContext[`baseSchema`]\n          schema: TContext[`baseSchema`]\n        }>\n      | CollectionRef,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n    where?: Condition<\n      Flatten<{\n        baseSchema: TContext[`baseSchema`]\n        schema: TContext[`schema`] &\n          (TFrom extends CollectionRef\n            ? {\n                [K in keyof TFrom & string]: RemoveIndexSignature<\n                  (TFrom[keyof TFrom] extends Collection<infer T> ? T : never) &\n                    Input\n                >\n              }\n            : TFrom extends InputReference<infer TRef>\n              ? {\n                  [K in keyof TRef & string]: RemoveIndexSignature<\n                    TRef[keyof TRef]\n                  >\n                }\n              : never)\n      }>\n    >\n  }): QueryBuilder<any> {\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    if (typeof joinClause.from === `object` && joinClause.from !== null) {\n      return this.joinCollectionRef(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: CollectionRef\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    } else {\n      return this.joinInputReference(\n        joinClause as {\n          type: `inner` | `left` | `right` | `full` | `cross`\n          from: InputReference<{\n            baseSchema: TContext[`baseSchema`]\n            schema: TContext[`baseSchema`]\n          }>\n          as?: TAs\n          on: Condition<any>\n          where?: Condition<any>\n        }\n      )\n    }\n  }\n\n  private joinCollectionRef<TCollectionRef extends CollectionRef>(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TCollectionRef\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Get the collection key\n    const keys = Object.keys(joinClause.from)\n    if (keys.length !== 1) {\n      throw new Error(`Expected exactly one key in CollectionRef`)\n    }\n    const key = keys[0]!\n    const collection = joinClause.from[key]\n    if (!collection) {\n      throw new Error(`Collection not found for key: ${key}`)\n    }\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = {\n      type: joinClause.type,\n      from: key,\n      on: joinClause.on,\n      where: joinClause.where,\n    } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Add the collection to the collections map\n    newBuilder.query.collections ??= {}\n    newBuilder.query.collections[key] = collection\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in keyof TCollectionRef & string]: RemoveIndexSignature<\n              (TCollectionRef[keyof TCollectionRef] extends Collection<infer T>\n                ? T\n                : never) &\n                Input\n            >\n          }\n        }\n      >\n    >\n  }\n\n  private joinInputReference<\n    TFrom extends InputReference<{\n      baseSchema: TContext[`baseSchema`]\n      schema: TContext[`baseSchema`]\n    }>,\n    TAs extends string | undefined = undefined,\n  >(joinClause: {\n    type: `inner` | `left` | `right` | `full` | `cross`\n    from: TFrom\n    as?: TAs\n    on: Condition<any>\n    where?: Condition<any>\n  }): QueryBuilder<any> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a copy of the join clause for the query\n    const joinClauseCopy = { ...joinClause } as JoinClause<TContext>\n\n    // Add the join clause to the query\n    if (!newBuilder.query.join) {\n      newBuilder.query.join = [joinClauseCopy]\n    } else {\n      newBuilder.query.join = [...newBuilder.query.join, joinClauseCopy]\n    }\n\n    // Determine the alias or use the collection name as default\n    const _effectiveAlias = joinClause.as ?? joinClause.from\n\n    // Return the new builder with updated schema type\n    return newBuilder as QueryBuilder<\n      Flatten<\n        Omit<TContext, `schema`> & {\n          schema: TContext[`schema`] & {\n            [K in typeof _effectiveAlias]: TContext[`baseSchema`][TFrom]\n          }\n        }\n      >\n    >\n  }\n\n  /**\n   * Add an orderBy clause to sort the results.\n   * Overwrites any previous orderBy clause.\n   *\n   * @param orderBy The order specification\n   * @returns A new QueryBuilder with the orderBy clause set\n   */\n  orderBy(orderBy: OrderBy<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the orderBy clause\n    newBuilder.query.orderBy = orderBy\n\n    // Ensure we have an orderByIndex in the select if we have an orderBy\n    // This is required if select is called before orderBy\n    newBuilder.query.select = [\n      ...(newBuilder.query.select ?? []),\n      { _orderByIndex: { ORDER_INDEX: `numeric` } },\n    ]\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set a limit on the number of results returned.\n   *\n   * @param limit Maximum number of results to return\n   * @returns A new QueryBuilder with the limit set\n   */\n  limit(limit: Limit<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the limit\n    newBuilder.query.limit = limit\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Set an offset to skip a number of results.\n   *\n   * @param offset Number of results to skip\n   * @returns A new QueryBuilder with the offset set\n   */\n  offset(offset: Offset<TContext>): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the offset\n    newBuilder.query.offset = offset\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Add a groupBy clause to group the results by one or more columns.\n   *\n   * @param groupBy The column(s) to group by\n   * @returns A new QueryBuilder with the groupBy clause set\n   */\n  groupBy(\n    groupBy: PropertyReference<TContext> | Array<PropertyReference<TContext>>\n  ): QueryBuilder<TContext> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Set the groupBy clause\n    newBuilder.query.groupBy = groupBy\n\n    return newBuilder as QueryBuilder<TContext>\n  }\n\n  /**\n   * Define a Common Table Expression (CTE) that can be referenced in the main query.\n   * This allows referencing the CTE by name in subsequent from/join clauses.\n   *\n   * @param name The name of the CTE\n   * @param queryBuilderCallback A function that builds the CTE query\n   * @returns A new QueryBuilder with the CTE added\n   */\n  with<TName extends string, TResult = Record<string, unknown>>(\n    name: TName,\n    queryBuilderCallback: (\n      builder: InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    ) => QueryBuilder<any>\n  ): InitialQueryBuilder<{\n    baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n    schema: TContext[`schema`]\n  }> {\n    // Create a new builder with a copy of the current query\n    const newBuilder = new BaseQueryBuilder<TContext>()\n    Object.assign(newBuilder.query, this.query)\n\n    // Create a new builder for the CTE\n    const cteBuilder = new BaseQueryBuilder<{\n      baseSchema: TContext[`baseSchema`]\n      schema: {}\n    }>()\n\n    // Get the CTE query from the callback\n    const cteQueryBuilder = queryBuilderCallback(\n      cteBuilder as InitialQueryBuilder<{\n        baseSchema: TContext[`baseSchema`]\n        schema: {}\n      }>\n    )\n\n    // Get the query from the builder\n    const cteQuery = cteQueryBuilder._query\n\n    // Add an 'as' property to the CTE\n    const withQuery: WithQuery<any> = {\n      ...cteQuery,\n      as: name,\n    }\n\n    // Add the CTE to the with array\n    if (!newBuilder.query.with) {\n      newBuilder.query.with = [withQuery]\n    } else {\n      newBuilder.query.with = [...newBuilder.query.with, withQuery]\n    }\n\n    // Use a type cast that simplifies the type structure to avoid recursion\n    return newBuilder as unknown as InitialQueryBuilder<{\n      baseSchema: TContext[`baseSchema`] & { [K in TName]: TResult }\n      schema: TContext[`schema`]\n    }>\n  }\n\n  get _query(): Query<TContext> {\n    return this.query as Query<TContext>\n  }\n}\n\nexport type InitialQueryBuilder<TContext extends Context<Schema>> = Pick<\n  BaseQueryBuilder<TContext>,\n  `from` | `with`\n>\n\nexport type QueryBuilder<TContext extends Context<Schema>> = Omit<\n  BaseQueryBuilder<TContext>,\n  `from`\n>\n\n/**\n * Create a new query builder with the given schema\n */\nexport function queryBuilder<TBaseSchema extends Schema = {}>() {\n  return new BaseQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>() as InitialQueryBuilder<{\n    baseSchema: TBaseSchema\n    schema: {}\n  }>\n}\n\nexport type ResultsFromContext<TContext extends Context<Schema>> = Flatten<\n  TContext[`result`] extends object\n    ? TContext[`result`] // If there is a select we will have a result type\n    : TContext[`hasJoin`] extends true\n      ? TContext[`schema`] // If there is a join, the query returns the namespaced schema\n      : TContext[`default`] extends keyof TContext[`schema`]\n        ? TContext[`schema`][TContext[`default`]] // If there is no join we return the flat default schema\n        : never // Should never happen\n>\n\nexport type ResultFromQueryBuilder<TQueryBuilder> = Flatten<\n  TQueryBuilder extends QueryBuilder<infer C>\n    ? C extends { result: infer R }\n      ? R\n      : never\n    : never\n>\n"],"names":[],"mappings":"AA8BO,MAAM,iBAAmD;AAAA;AAAA;AAAA;AAAA,EAM9D,YAAY,QAAkC,IAAI;AALlD,SAAiB,QAAkC,CAAC;AAMlD,SAAK,QAAQ;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmEf,KAQE,YAAe,IAAU;AAEzB,QAAI,OAAO,eAAe,YAAY,eAAe,MAAM;AAClD,aAAA,KAAK,kBAAkB,UAAU;AAAA,IAC1C,WAAW,OAAO,eAAe,UAAU;AACzC,aAAO,KAAK;AAAA,QACV;AAAA,QAIA;AAAA,MACF;AAAA,IAAA,OACK;AACC,YAAA,IAAI,MAAM,yBAAyB;AAAA,IAAA;AAAA,EAC3C;AAAA,EAGM,kBACN,eACA;AArGG;AAsGG,UAAA,OAAO,OAAO,KAAK,aAAa;AAClC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,0BAA0B;AAAA,IAAA;AAGtC,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,cAAc,GAAG;AAE9B,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACb,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAE7B,WAAA;AAAA,EAAA;AAAA,EAuBD,mBAMN,YAAe,IAAU;AACnB,UAAA,aAAa,IAAI,iBAAiB;AACxC,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAC1C,eAAW,MAAM,OAAO;AACxB,QAAI,IAAI;AACN,iBAAW,MAAM,KAAK;AAAA,IAAA;AAWjB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeT,UAEK,SACH;AAGA,UAAM,mBAAmB,QAAQ,IAAI,CAAC,WAAW;AAE/C,UACE,OAAO,WAAW;AAAA,MAElB,WAAW,QACX,CAAC,MAAM,QAAQ,MAAM,GACrB;AACA,cAAM,SAA8B,CAAC;AAErC,mBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,MAAM,GAAG;AAG/C,cAAA,OAAO,UAAU,YACjB,UAAU,QACV,CAAC,MAAM,QAAQ,KAAK,GACpB;AACM,kBAAA,OAAO,OAAO,KAAK,KAAK;AAC1B,gBAAA,KAAK,WAAW,GAAG;AACf,oBAAA,WAAW,KAAK,CAAC;AAEvB,oBAAM,mBAAmB;AAAA,gBACvB;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACF;AAEA,kBAAI,CAAC,iBAAiB,SAAS,QAAQ,GAAG;AAChC,wBAAA;AAAA,kBACN,yBAAyB,QAAQ,sBAAsB,iBAAiB,KAAK,IAAI,CAAC;AAAA,gBACpF;AAAA,cAAA;AAAA,YACF;AAAA,UACF;AAGF,iBAAO,GAAG,IAAI;AAAA,QAAA;AAGT,eAAA;AAAA,MAAA;AAGF,aAAA;AAAA,IAAA,CACR;AAIG,QAAA,KAAK,OAAO,SAAS;AACvB,uBAAiB,KAAK,EAAE,eAAe,EAAE,aAAa,UAAA,GAAa;AAAA,IAAA;AAGrE,UAAM,aAAa,IAAI;AAAA,MACpB,KAAoC;AAAA,IACvC;AACA,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAsCT,MACE,2BACA,UACA,OACwB;AAGlB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,OAAO;AAChB,iBAAA,MAAM,QAAQ,CAAC,SAAS;AAAA,IAAA,OAC9B;AACL,iBAAW,MAAM,QAAQ,CAAC,GAAG,WAAW,MAAM,OAAO,SAAS;AAAA,IAAA;AAGzD,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAmCT,OACE,2BACA,UACA,OACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAEtC,QAAA;AAGA,QAAA,OAAO,8BAA8B,YAAY;AAEvC,kBAAA;AAAA,IACH,WAAA,aAAa,UAAa,UAAU,QAAW;AAE5C,kBAAA,CAAC,2BAA2B,UAAU,KAAK;AAAA,IAAA,OAClD;AAEO,kBAAA;AAAA,IAAA;AAIV,QAAA,CAAC,WAAW,MAAM,QAAQ;AACjB,iBAAA,MAAM,SAAS,CAAC,SAAS;AAAA,IAAA,OAC/B;AACL,iBAAW,MAAM,SAAS,CAAC,GAAG,WAAW,MAAM,QAAQ,SAAS;AAAA,IAAA;AAG3D,WAAA;AAAA,EAAA;AAAA,EAgIT,KAQE,YA4CoB;AAEpB,QAAI,OAAO,WAAW,SAAS,YAAY,WAAW,SAAS,MAAM;AACnE,aAAO,KAAK;AAAA,QACV;AAAA,MAMF;AAAA,IAAA,OACK;AACL,aAAO,KAAK;AAAA,QACV;AAAA,MAUF;AAAA,IAAA;AAAA,EACF;AAAA,EAGM,kBAAwD,YAK1C;AAllBjB;AAolBG,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,UAAM,OAAO,OAAO,KAAK,WAAW,IAAI;AACpC,QAAA,KAAK,WAAW,GAAG;AACf,YAAA,IAAI,MAAM,2CAA2C;AAAA,IAAA;AAEvD,UAAA,MAAM,KAAK,CAAC;AACZ,UAAA,aAAa,WAAW,KAAK,GAAG;AACtC,QAAI,CAAC,YAAY;AACf,YAAM,IAAI,MAAM,iCAAiC,GAAG,EAAE;AAAA,IAAA;AAIxD,UAAM,iBAAiB;AAAA,MACrB,MAAM,WAAW;AAAA,MACjB,MAAM;AAAA,MACN,IAAI,WAAW;AAAA,MACf,OAAO,WAAW;AAAA,IACpB;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAIxD,qBAAA,OAAM,gBAAN,GAAM,cAAgB,CAAC;AACvB,eAAA,MAAM,YAAY,GAAG,IAAI;AAG7B,WAAA;AAAA,EAAA;AAAA,EAgBD,mBAMN,YAMoB;AAEd,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,iBAAiB,EAAE,GAAG,WAAW;AAGnC,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,cAAc;AAAA,IAAA,OAClC;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,cAAc;AAAA,IAAA;AAI3C,eAAW,MAAM,WAAW;AAG7C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAkBT,QAAQ,SAAoD;AAEpD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAI3B,eAAW,MAAM,SAAS;AAAA,MACxB,GAAI,WAAW,MAAM,UAAU,CAAC;AAAA,MAChC,EAAE,eAAe,EAAE,aAAa,UAAY,EAAA;AAAA,IAC9C;AAEO,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,MAAM,OAAgD;AAE9C,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,QAAQ;AAElB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,OAAO,QAAkD;AAEjD,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,SAAS;AAEnB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAST,QACE,SACwB;AAElB,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAG1C,eAAW,MAAM,UAAU;AAEpB,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWT,KACE,MACA,sBASC;AAEK,UAAA,aAAa,IAAI,iBAA2B;AAClD,WAAO,OAAO,WAAW,OAAO,KAAK,KAAK;AAGpC,UAAA,aAAa,IAAI,iBAGpB;AAGH,UAAM,kBAAkB;AAAA,MACtB;AAAA,IAIF;AAGA,UAAM,WAAW,gBAAgB;AAGjC,UAAM,YAA4B;AAAA,MAChC,GAAG;AAAA,MACH,IAAI;AAAA,IACN;AAGI,QAAA,CAAC,WAAW,MAAM,MAAM;AACf,iBAAA,MAAM,OAAO,CAAC,SAAS;AAAA,IAAA,OAC7B;AACL,iBAAW,MAAM,OAAO,CAAC,GAAG,WAAW,MAAM,MAAM,SAAS;AAAA,IAAA;AAIvD,WAAA;AAAA,EAAA;AAAA,EAMT,IAAI,SAA0B;AAC5B,WAAO,KAAK;AAAA,EAAA;AAEhB;AAeO,SAAS,eAAgD;AAC9D,SAAO,IAAI,iBAGR;AAIL;"}

package/dist/esm/query/schema.d.ts

@@ -1,10 +1,11 @@
-import { Context, InputReference, PropertyReference, PropertyReferenceString, WildcardReferenceString } from './types.js';
+import { Context, InputReference, PropertyReference, PropertyReferenceString, Schema, WildcardReferenceString } from './types.js';
 import { Collection } from '../collection.js';
 export type ColumnName<TColumnNames extends string> = TColumnNames;
 export type JSONLike = string | number | boolean | Date | null | Array<JSONLike> | {
     [key: string]: JSONLike;
 };
 export type LiteralValue = (string & {}) | number | boolean | Date | null | undefined;
+export type ComparatorValue<T extends Comparator, TContext extends Context<Schema>> = T extends `in` | `not in` ? Array<LiteralValue> : PropertyReferenceString<TContext> | LiteralValue;
 export type SafeString<T extends string> = T extends `@${string}` ? never : T;
 export type OptionalSafeString<T extends any> = T extends string ? SafeString<T> : never;
 export type LiteralValueWithSafeString<T extends any> = (OptionalSafeString<T> & {}) | number | boolean | Date | null | undefined;
@@ -63,7 +64,7 @@ export type Select<TContext extends Context = Context> = PropertyReferenceString
 export type SelectCallback<TContext extends Context = Context> = (context: TContext extends {
     schema: infer S;
 } ? S : any) => any;
-export type As<TContext extends Context = Context> = string;
+export type As<_TContext extends Context = Context> = string;
 export type From<TContext extends Context = Context> = InputReference<{
     baseSchema: TContext[`baseSchema`];
     schema: TContext[`baseSchema`];
@@ -74,8 +75,8 @@ export type WhereCallback<TContext extends Context = Context> = (context: TConte
 export type Where<TContext extends Context = Context> = Array<Condition<TContext> | WhereCallback<TContext>>;
 export type Having<TContext extends Context = Context> = Where<TContext>;
 export type GroupBy<TContext extends Context = Context> = PropertyReference<TContext> | Array<PropertyReference<TContext>>;
-export type Limit<TContext extends Context = Context> = number;
-export type Offset<TContext extends Context = Context> = number;
+export type Limit<_TContext extends Context = Context> = number;
+export type Offset<_TContext extends Context = Context> = number;
 export interface BaseQuery<TContext extends Context = Context> {
     select?: Array<Select<TContext>>;
     as?: As<TContext>;

package/dist/esm/transactions.d.ts

@@ -1,13 +1,13 @@
 import { Deferred } from './deferred.js';
-import { PendingMutation, TransactionConfig, TransactionState } from './types.js';
+import { MutationFn, PendingMutation, TransactionConfig, TransactionState } from './types.js';
 export declare function createTransaction(config: TransactionConfig): Transaction;
 export declare function getActiveTransaction(): Transaction | undefined;
-export declare class Transaction {
+export declare class Transaction<T extends object = Record<string, unknown>> {
     id: string;
     state: TransactionState;
-    mutationFn: import('./types.js').MutationFn;
-    mutations: Array<PendingMutation<any>>;
-    isPersisted: Deferred<Transaction>;
+    mutationFn: MutationFn<T>;
+    mutations: Array<PendingMutation<T>>;
+    isPersisted: Deferred<Transaction<T>>;
     autoCommit: boolean;
     createdAt: Date;
     metadata: Record<string, unknown>;
@@ -15,13 +15,13 @@ export declare class Transaction {
         message: string;
         error: Error;
     };
-    constructor(config: TransactionConfig);
+    constructor(config: TransactionConfig<T>);
     setState(newState: TransactionState): void;
-    mutate(callback: () => void): Transaction;
+    mutate(callback: () => void): Transaction<T>;
     applyMutations(mutations: Array<PendingMutation<any>>): void;
     rollback(config?: {
         isSecondaryRollback?: boolean;
-    }): Transaction;
+    }): Transaction<T>;
     touchCollection(): void;
-    commit(): Promise<Transaction>;
+    commit(): Promise<Transaction<T>>;
 }

package/dist/esm/transactions.js

@@ -77,7 +77,7 @@ class Transaction {
   applyMutations(mutations) {
     for (const newMutation of mutations) {
       const existingIndex = this.mutations.findIndex(
-        (m) => m.key === newMutation.key
+        (m) => m.globalKey === newMutation.globalKey
       );
       if (existingIndex >= 0) {
         this.mutations[existingIndex] = newMutation;
@@ -95,9 +95,9 @@ class Transaction {
     this.setState(`failed`);
     if (!isSecondaryRollback) {
       const mutationIds = /* @__PURE__ */ new Set();
-      this.mutations.forEach((m) => mutationIds.add(m.key));
+      this.mutations.forEach((m) => mutationIds.add(m.globalKey));
       for (const t of transactions) {
-        t.state === `pending` && t.mutations.some((m) => mutationIds.has(m.key)) && t.rollback({ isSecondaryRollback: true });
+        t.state === `pending` && t.mutations.some((m) => mutationIds.has(m.globalKey)) && t.rollback({ isSecondaryRollback: true });
       }
     }
     this.isPersisted.reject((_a = this.error) == null ? void 0 : _a.error);
@@ -109,7 +109,7 @@ class Transaction {
     const hasCalled = /* @__PURE__ */ new Set();
     for (const mutation of this.mutations) {
       if (!hasCalled.has(mutation.collection.id)) {
-        mutation.collection.transactions.setState((state) => state);
+        mutation.collection.onTransactionStateChange();
         mutation.collection.commitPendingTransactions();
         hasCalled.add(mutation.collection.id);
       }
@@ -125,8 +125,9 @@ class Transaction {
       return this;
     }
     try {
-      const transactionWithMutations = this;
-      await this.mutationFn({ transaction: transactionWithMutations });
+      await this.mutationFn({
+        transaction: this
+      });
       this.setState(`completed`);
       this.touchCollection();
       this.isPersisted.resolve(this);

package/dist/esm/transactions.js.map

@@ -1 +1 @@
-{"version":3,"file":"transactions.js","sources":["../../src/transactions.ts"],"sourcesContent":["import { createDeferred } from \"./deferred\"\nimport type { Deferred } from \"./deferred\"\nimport type {\n  PendingMutation,\n  TransactionConfig,\n  TransactionState,\n  TransactionWithMutations,\n} from \"./types\"\n\nfunction generateUUID() {\n  // Check if crypto.randomUUID is available (modern browsers and Node.js 15+)\n  if (\n    typeof crypto !== `undefined` &&\n    typeof crypto.randomUUID === `function`\n  ) {\n    return crypto.randomUUID()\n  }\n\n  // Fallback implementation for older environments\n  return `xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx`.replace(/[xy]/g, function (c) {\n    const r = (Math.random() * 16) | 0\n    const v = c === `x` ? r : (r & 0x3) | 0x8\n    return v.toString(16)\n  })\n}\n\nconst transactions: Array<Transaction> = []\nlet transactionStack: Array<Transaction> = []\n\nexport function createTransaction(config: TransactionConfig): Transaction {\n  if (typeof config.mutationFn === `undefined`) {\n    throw `mutationFn is required when creating a transaction`\n  }\n\n  let transactionId = config.id\n  if (!transactionId) {\n    transactionId = generateUUID()\n  }\n  const newTransaction = new Transaction({ ...config, id: transactionId })\n\n  transactions.push(newTransaction)\n\n  return newTransaction\n}\n\nexport function getActiveTransaction(): Transaction | undefined {\n  if (transactionStack.length > 0) {\n    return transactionStack.slice(-1)[0]\n  } else {\n    return undefined\n  }\n}\n\nfunction registerTransaction(tx: Transaction) {\n  transactionStack.push(tx)\n}\n\nfunction unregisterTransaction(tx: Transaction) {\n  transactionStack = transactionStack.filter((t) => t.id !== tx.id)\n}\n\nfunction removeFromPendingList(tx: Transaction) {\n  const index = transactions.findIndex((t) => t.id === tx.id)\n  if (index !== -1) {\n    transactions.splice(index, 1)\n  }\n}\n\nexport class Transaction {\n  public id: string\n  public state: TransactionState\n  public mutationFn\n  public mutations: Array<PendingMutation<any>>\n  public isPersisted: Deferred<Transaction>\n  public autoCommit: boolean\n  public createdAt: Date\n  public metadata: Record<string, unknown>\n  public error?: {\n    message: string\n    error: Error\n  }\n\n  constructor(config: TransactionConfig) {\n    this.id = config.id!\n    this.mutationFn = config.mutationFn\n    this.state = `pending`\n    this.mutations = []\n    this.isPersisted = createDeferred()\n    this.autoCommit = config.autoCommit ?? true\n    this.createdAt = new Date()\n    this.metadata = config.metadata ?? {}\n  }\n\n  setState(newState: TransactionState) {\n    this.state = newState\n\n    if (newState === `completed` || newState === `failed`) {\n      removeFromPendingList(this)\n    }\n  }\n\n  mutate(callback: () => void): Transaction {\n    if (this.state !== `pending`) {\n      throw `You can no longer call .mutate() as the transaction is no longer pending`\n    }\n\n    registerTransaction(this)\n    try {\n      callback()\n    } finally {\n      unregisterTransaction(this)\n    }\n\n    if (this.autoCommit) {\n      this.commit()\n    }\n\n    return this\n  }\n\n  applyMutations(mutations: Array<PendingMutation<any>>): void {\n    for (const newMutation of mutations) {\n      const existingIndex = this.mutations.findIndex(\n        (m) => m.key === newMutation.key\n      )\n\n      if (existingIndex >= 0) {\n        // Replace existing mutation\n        this.mutations[existingIndex] = newMutation\n      } else {\n        // Insert new mutation\n        this.mutations.push(newMutation)\n      }\n    }\n  }\n\n  rollback(config?: { isSecondaryRollback?: boolean }): Transaction {\n    const isSecondaryRollback = config?.isSecondaryRollback ?? false\n    if (this.state === `completed`) {\n      throw `You can no longer call .rollback() as the transaction is already completed`\n    }\n\n    this.setState(`failed`)\n\n    // See if there's any other transactions w/ mutations on the same ids\n    // and roll them back as well.\n    if (!isSecondaryRollback) {\n      const mutationIds = new Set()\n      this.mutations.forEach((m) => mutationIds.add(m.key))\n      for (const t of transactions) {\n        t.state === `pending` &&\n          t.mutations.some((m) => mutationIds.has(m.key)) &&\n          t.rollback({ isSecondaryRollback: true })\n      }\n    }\n\n    // Reject the promise\n    this.isPersisted.reject(this.error?.error)\n    this.touchCollection()\n\n    return this\n  }\n\n  // Tell collection that something has changed with the transaction\n  touchCollection(): void {\n    const hasCalled = new Set()\n    for (const mutation of this.mutations) {\n      if (!hasCalled.has(mutation.collection.id)) {\n        mutation.collection.transactions.setState((state) => state)\n        mutation.collection.commitPendingTransactions()\n        hasCalled.add(mutation.collection.id)\n      }\n    }\n  }\n\n  async commit(): Promise<Transaction> {\n    if (this.state !== `pending`) {\n      throw `You can no longer call .commit() as the transaction is no longer pending`\n    }\n\n    this.setState(`persisting`)\n\n    if (this.mutations.length === 0) {\n      this.setState(`completed`)\n\n      return this\n    }\n\n    // Run mutationFn\n    try {\n      // At this point we know there's at least one mutation\n      // Use type assertion to tell TypeScript about this guarantee\n      const transactionWithMutations =\n        this as unknown as TransactionWithMutations\n      await this.mutationFn({ transaction: transactionWithMutations })\n\n      this.setState(`completed`)\n      this.touchCollection()\n\n      this.isPersisted.resolve(this)\n    } catch (error) {\n      // Update transaction with error information\n      this.error = {\n        message: error instanceof Error ? error.message : String(error),\n        error: error instanceof Error ? error : new Error(String(error)),\n      }\n\n      // rollback the transaction\n      return this.rollback()\n    }\n\n    return this\n  }\n}\n"],"names":[],"mappings":";AASA,SAAS,eAAe;AAEtB,MACE,OAAO,WAAW,eAClB,OAAO,OAAO,eAAe,YAC7B;AACA,WAAO,OAAO,WAAW;AAAA,EAAA;AAI3B,SAAO,uCAAuC,QAAQ,SAAS,SAAU,GAAG;AAC1E,UAAM,IAAK,KAAK,OAAO,IAAI,KAAM;AACjC,UAAM,IAAI,MAAM,MAAM,IAAK,IAAI,IAAO;AAC/B,WAAA,EAAE,SAAS,EAAE;AAAA,EAAA,CACrB;AACH;AAEA,MAAM,eAAmC,CAAC;AAC1C,IAAI,mBAAuC,CAAC;AAErC,SAAS,kBAAkB,QAAwC;AACpE,MAAA,OAAO,OAAO,eAAe,aAAa;AACtC,UAAA;AAAA,EAAA;AAGR,MAAI,gBAAgB,OAAO;AAC3B,MAAI,CAAC,eAAe;AAClB,oBAAgB,aAAa;AAAA,EAAA;AAEzB,QAAA,iBAAiB,IAAI,YAAY,EAAE,GAAG,QAAQ,IAAI,eAAe;AAEvE,eAAa,KAAK,cAAc;AAEzB,SAAA;AACT;AAEO,SAAS,uBAAgD;AAC1D,MAAA,iBAAiB,SAAS,GAAG;AAC/B,WAAO,iBAAiB,MAAM,EAAE,EAAE,CAAC;AAAA,EAAA,OAC9B;AACE,WAAA;AAAA,EAAA;AAEX;AAEA,SAAS,oBAAoB,IAAiB;AAC5C,mBAAiB,KAAK,EAAE;AAC1B;AAEA,SAAS,sBAAsB,IAAiB;AAC9C,qBAAmB,iBAAiB,OAAO,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;AAClE;AAEA,SAAS,sBAAsB,IAAiB;AACxC,QAAA,QAAQ,aAAa,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;AAC1D,MAAI,UAAU,IAAI;AACH,iBAAA,OAAO,OAAO,CAAC;AAAA,EAAA;AAEhC;AAEO,MAAM,YAAY;AAAA,EAcvB,YAAY,QAA2B;AACrC,SAAK,KAAK,OAAO;AACjB,SAAK,aAAa,OAAO;AACzB,SAAK,QAAQ;AACb,SAAK,YAAY,CAAC;AAClB,SAAK,cAAc,eAAe;AAC7B,SAAA,aAAa,OAAO,cAAc;AAClC,SAAA,gCAAgB,KAAK;AACrB,SAAA,WAAW,OAAO,YAAY,CAAC;AAAA,EAAA;AAAA,EAGtC,SAAS,UAA4B;AACnC,SAAK,QAAQ;AAET,QAAA,aAAa,eAAe,aAAa,UAAU;AACrD,4BAAsB,IAAI;AAAA,IAAA;AAAA,EAC5B;AAAA,EAGF,OAAO,UAAmC;AACpC,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA;AAAA,IAAA;AAGR,wBAAoB,IAAI;AACpB,QAAA;AACO,eAAA;AAAA,IAAA,UACT;AACA,4BAAsB,IAAI;AAAA,IAAA;AAG5B,QAAI,KAAK,YAAY;AACnB,WAAK,OAAO;AAAA,IAAA;AAGP,WAAA;AAAA,EAAA;AAAA,EAGT,eAAe,WAA8C;AAC3D,eAAW,eAAe,WAAW;AAC7B,YAAA,gBAAgB,KAAK,UAAU;AAAA,QACnC,CAAC,MAAM,EAAE,QAAQ,YAAY;AAAA,MAC/B;AAEA,UAAI,iBAAiB,GAAG;AAEjB,aAAA,UAAU,aAAa,IAAI;AAAA,MAAA,OAC3B;AAEA,aAAA,UAAU,KAAK,WAAW;AAAA,MAAA;AAAA,IACjC;AAAA,EACF;AAAA,EAGF,SAAS,QAAyD;;AAC1D,UAAA,uBAAsB,iCAAQ,wBAAuB;AACvD,QAAA,KAAK,UAAU,aAAa;AACxB,YAAA;AAAA,IAAA;AAGR,SAAK,SAAS,QAAQ;AAItB,QAAI,CAAC,qBAAqB;AAClB,YAAA,kCAAkB,IAAI;AACvB,WAAA,UAAU,QAAQ,CAAC,MAAM,YAAY,IAAI,EAAE,GAAG,CAAC;AACpD,iBAAW,KAAK,cAAc;AAC5B,UAAE,UAAU,aACV,EAAE,UAAU,KAAK,CAAC,MAAM,YAAY,IAAI,EAAE,GAAG,CAAC,KAC9C,EAAE,SAAS,EAAE,qBAAqB,MAAM;AAAA,MAAA;AAAA,IAC5C;AAIF,SAAK,YAAY,QAAO,UAAK,UAAL,mBAAY,KAAK;AACzC,SAAK,gBAAgB;AAEd,WAAA;AAAA,EAAA;AAAA;AAAA,EAIT,kBAAwB;AAChB,UAAA,gCAAgB,IAAI;AACf,eAAA,YAAY,KAAK,WAAW;AACrC,UAAI,CAAC,UAAU,IAAI,SAAS,WAAW,EAAE,GAAG;AAC1C,iBAAS,WAAW,aAAa,SAAS,CAAC,UAAU,KAAK;AAC1D,iBAAS,WAAW,0BAA0B;AACpC,kBAAA,IAAI,SAAS,WAAW,EAAE;AAAA,MAAA;AAAA,IACtC;AAAA,EACF;AAAA,EAGF,MAAM,SAA+B;AAC/B,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA;AAAA,IAAA;AAGR,SAAK,SAAS,YAAY;AAEtB,QAAA,KAAK,UAAU,WAAW,GAAG;AAC/B,WAAK,SAAS,WAAW;AAElB,aAAA;AAAA,IAAA;AAIL,QAAA;AAGF,YAAM,2BACJ;AACF,YAAM,KAAK,WAAW,EAAE,aAAa,0BAA0B;AAE/D,WAAK,SAAS,WAAW;AACzB,WAAK,gBAAgB;AAEhB,WAAA,YAAY,QAAQ,IAAI;AAAA,aACtB,OAAO;AAEd,WAAK,QAAQ;AAAA,QACX,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,QAC9D,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAAA,MACjE;AAGA,aAAO,KAAK,SAAS;AAAA,IAAA;AAGhB,WAAA;AAAA,EAAA;AAEX;"}
+{"version":3,"file":"transactions.js","sources":["../../src/transactions.ts"],"sourcesContent":["import { createDeferred } from \"./deferred\"\nimport type { Deferred } from \"./deferred\"\nimport type {\n  MutationFn,\n  PendingMutation,\n  TransactionConfig,\n  TransactionState,\n  TransactionWithMutations,\n} from \"./types\"\n\nfunction generateUUID() {\n  // Check if crypto.randomUUID is available (modern browsers and Node.js 15+)\n  if (\n    typeof crypto !== `undefined` &&\n    typeof crypto.randomUUID === `function`\n  ) {\n    return crypto.randomUUID()\n  }\n\n  // Fallback implementation for older environments\n  return `xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx`.replace(/[xy]/g, function (c) {\n    const r = (Math.random() * 16) | 0\n    const v = c === `x` ? r : (r & 0x3) | 0x8\n    return v.toString(16)\n  })\n}\n\nconst transactions: Array<Transaction<any>> = []\nlet transactionStack: Array<Transaction<any>> = []\n\nexport function createTransaction(config: TransactionConfig): Transaction {\n  if (typeof config.mutationFn === `undefined`) {\n    throw `mutationFn is required when creating a transaction`\n  }\n\n  let transactionId = config.id\n  if (!transactionId) {\n    transactionId = generateUUID()\n  }\n  const newTransaction = new Transaction({ ...config, id: transactionId })\n\n  transactions.push(newTransaction)\n\n  return newTransaction\n}\n\nexport function getActiveTransaction(): Transaction | undefined {\n  if (transactionStack.length > 0) {\n    return transactionStack.slice(-1)[0]\n  } else {\n    return undefined\n  }\n}\n\nfunction registerTransaction(tx: Transaction<any>) {\n  transactionStack.push(tx)\n}\n\nfunction unregisterTransaction(tx: Transaction<any>) {\n  transactionStack = transactionStack.filter((t) => t.id !== tx.id)\n}\n\nfunction removeFromPendingList(tx: Transaction<any>) {\n  const index = transactions.findIndex((t) => t.id === tx.id)\n  if (index !== -1) {\n    transactions.splice(index, 1)\n  }\n}\n\nexport class Transaction<T extends object = Record<string, unknown>> {\n  public id: string\n  public state: TransactionState\n  public mutationFn: MutationFn<T>\n  public mutations: Array<PendingMutation<T>>\n  public isPersisted: Deferred<Transaction<T>>\n  public autoCommit: boolean\n  public createdAt: Date\n  public metadata: Record<string, unknown>\n  public error?: {\n    message: string\n    error: Error\n  }\n\n  constructor(config: TransactionConfig<T>) {\n    this.id = config.id!\n    this.mutationFn = config.mutationFn\n    this.state = `pending`\n    this.mutations = []\n    this.isPersisted = createDeferred<Transaction<T>>()\n    this.autoCommit = config.autoCommit ?? true\n    this.createdAt = new Date()\n    this.metadata = config.metadata ?? {}\n  }\n\n  setState(newState: TransactionState) {\n    this.state = newState\n\n    if (newState === `completed` || newState === `failed`) {\n      removeFromPendingList(this)\n    }\n  }\n\n  mutate(callback: () => void): Transaction<T> {\n    if (this.state !== `pending`) {\n      throw `You can no longer call .mutate() as the transaction is no longer pending`\n    }\n\n    registerTransaction(this)\n    try {\n      callback()\n    } finally {\n      unregisterTransaction(this)\n    }\n\n    if (this.autoCommit) {\n      this.commit()\n    }\n\n    return this\n  }\n\n  applyMutations(mutations: Array<PendingMutation<any>>): void {\n    for (const newMutation of mutations) {\n      const existingIndex = this.mutations.findIndex(\n        (m) => m.globalKey === newMutation.globalKey\n      )\n\n      if (existingIndex >= 0) {\n        // Replace existing mutation\n        this.mutations[existingIndex] = newMutation\n      } else {\n        // Insert new mutation\n        this.mutations.push(newMutation)\n      }\n    }\n  }\n\n  rollback(config?: { isSecondaryRollback?: boolean }): Transaction<T> {\n    const isSecondaryRollback = config?.isSecondaryRollback ?? false\n    if (this.state === `completed`) {\n      throw `You can no longer call .rollback() as the transaction is already completed`\n    }\n\n    this.setState(`failed`)\n\n    // See if there's any other transactions w/ mutations on the same ids\n    // and roll them back as well.\n    if (!isSecondaryRollback) {\n      const mutationIds = new Set()\n      this.mutations.forEach((m) => mutationIds.add(m.globalKey))\n      for (const t of transactions) {\n        t.state === `pending` &&\n          t.mutations.some((m) => mutationIds.has(m.globalKey)) &&\n          t.rollback({ isSecondaryRollback: true })\n      }\n    }\n\n    // Reject the promise\n    this.isPersisted.reject(this.error?.error)\n    this.touchCollection()\n\n    return this\n  }\n\n  // Tell collection that something has changed with the transaction\n  touchCollection(): void {\n    const hasCalled = new Set()\n    for (const mutation of this.mutations) {\n      if (!hasCalled.has(mutation.collection.id)) {\n        mutation.collection.onTransactionStateChange()\n        mutation.collection.commitPendingTransactions()\n        hasCalled.add(mutation.collection.id)\n      }\n    }\n  }\n\n  async commit(): Promise<Transaction<T>> {\n    if (this.state !== `pending`) {\n      throw `You can no longer call .commit() as the transaction is no longer pending`\n    }\n\n    this.setState(`persisting`)\n\n    if (this.mutations.length === 0) {\n      this.setState(`completed`)\n\n      return this\n    }\n\n    // Run mutationFn\n    try {\n      // At this point we know there's at least one mutation\n      // We've already verified mutations is non-empty, so this cast is safe\n      // Use a direct type assertion instead of object spreading to preserve the original type\n      await this.mutationFn({\n        transaction: this as unknown as TransactionWithMutations<T>,\n      })\n\n      this.setState(`completed`)\n      this.touchCollection()\n\n      this.isPersisted.resolve(this)\n    } catch (error) {\n      // Update transaction with error information\n      this.error = {\n        message: error instanceof Error ? error.message : String(error),\n        error: error instanceof Error ? error : new Error(String(error)),\n      }\n\n      // rollback the transaction\n      return this.rollback()\n    }\n\n    return this\n  }\n}\n"],"names":[],"mappings":";AAUA,SAAS,eAAe;AAEtB,MACE,OAAO,WAAW,eAClB,OAAO,OAAO,eAAe,YAC7B;AACA,WAAO,OAAO,WAAW;AAAA,EAAA;AAI3B,SAAO,uCAAuC,QAAQ,SAAS,SAAU,GAAG;AAC1E,UAAM,IAAK,KAAK,OAAO,IAAI,KAAM;AACjC,UAAM,IAAI,MAAM,MAAM,IAAK,IAAI,IAAO;AAC/B,WAAA,EAAE,SAAS,EAAE;AAAA,EAAA,CACrB;AACH;AAEA,MAAM,eAAwC,CAAC;AAC/C,IAAI,mBAA4C,CAAC;AAE1C,SAAS,kBAAkB,QAAwC;AACpE,MAAA,OAAO,OAAO,eAAe,aAAa;AACtC,UAAA;AAAA,EAAA;AAGR,MAAI,gBAAgB,OAAO;AAC3B,MAAI,CAAC,eAAe;AAClB,oBAAgB,aAAa;AAAA,EAAA;AAEzB,QAAA,iBAAiB,IAAI,YAAY,EAAE,GAAG,QAAQ,IAAI,eAAe;AAEvE,eAAa,KAAK,cAAc;AAEzB,SAAA;AACT;AAEO,SAAS,uBAAgD;AAC1D,MAAA,iBAAiB,SAAS,GAAG;AAC/B,WAAO,iBAAiB,MAAM,EAAE,EAAE,CAAC;AAAA,EAAA,OAC9B;AACE,WAAA;AAAA,EAAA;AAEX;AAEA,SAAS,oBAAoB,IAAsB;AACjD,mBAAiB,KAAK,EAAE;AAC1B;AAEA,SAAS,sBAAsB,IAAsB;AACnD,qBAAmB,iBAAiB,OAAO,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;AAClE;AAEA,SAAS,sBAAsB,IAAsB;AAC7C,QAAA,QAAQ,aAAa,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;AAC1D,MAAI,UAAU,IAAI;AACH,iBAAA,OAAO,OAAO,CAAC;AAAA,EAAA;AAEhC;AAEO,MAAM,YAAwD;AAAA,EAcnE,YAAY,QAA8B;AACxC,SAAK,KAAK,OAAO;AACjB,SAAK,aAAa,OAAO;AACzB,SAAK,QAAQ;AACb,SAAK,YAAY,CAAC;AAClB,SAAK,cAAc,eAA+B;AAC7C,SAAA,aAAa,OAAO,cAAc;AAClC,SAAA,gCAAgB,KAAK;AACrB,SAAA,WAAW,OAAO,YAAY,CAAC;AAAA,EAAA;AAAA,EAGtC,SAAS,UAA4B;AACnC,SAAK,QAAQ;AAET,QAAA,aAAa,eAAe,aAAa,UAAU;AACrD,4BAAsB,IAAI;AAAA,IAAA;AAAA,EAC5B;AAAA,EAGF,OAAO,UAAsC;AACvC,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA;AAAA,IAAA;AAGR,wBAAoB,IAAI;AACpB,QAAA;AACO,eAAA;AAAA,IAAA,UACT;AACA,4BAAsB,IAAI;AAAA,IAAA;AAG5B,QAAI,KAAK,YAAY;AACnB,WAAK,OAAO;AAAA,IAAA;AAGP,WAAA;AAAA,EAAA;AAAA,EAGT,eAAe,WAA8C;AAC3D,eAAW,eAAe,WAAW;AAC7B,YAAA,gBAAgB,KAAK,UAAU;AAAA,QACnC,CAAC,MAAM,EAAE,cAAc,YAAY;AAAA,MACrC;AAEA,UAAI,iBAAiB,GAAG;AAEjB,aAAA,UAAU,aAAa,IAAI;AAAA,MAAA,OAC3B;AAEA,aAAA,UAAU,KAAK,WAAW;AAAA,MAAA;AAAA,IACjC;AAAA,EACF;AAAA,EAGF,SAAS,QAA4D;;AAC7D,UAAA,uBAAsB,iCAAQ,wBAAuB;AACvD,QAAA,KAAK,UAAU,aAAa;AACxB,YAAA;AAAA,IAAA;AAGR,SAAK,SAAS,QAAQ;AAItB,QAAI,CAAC,qBAAqB;AAClB,YAAA,kCAAkB,IAAI;AACvB,WAAA,UAAU,QAAQ,CAAC,MAAM,YAAY,IAAI,EAAE,SAAS,CAAC;AAC1D,iBAAW,KAAK,cAAc;AAC5B,UAAE,UAAU,aACV,EAAE,UAAU,KAAK,CAAC,MAAM,YAAY,IAAI,EAAE,SAAS,CAAC,KACpD,EAAE,SAAS,EAAE,qBAAqB,MAAM;AAAA,MAAA;AAAA,IAC5C;AAIF,SAAK,YAAY,QAAO,UAAK,UAAL,mBAAY,KAAK;AACzC,SAAK,gBAAgB;AAEd,WAAA;AAAA,EAAA;AAAA;AAAA,EAIT,kBAAwB;AAChB,UAAA,gCAAgB,IAAI;AACf,eAAA,YAAY,KAAK,WAAW;AACrC,UAAI,CAAC,UAAU,IAAI,SAAS,WAAW,EAAE,GAAG;AAC1C,iBAAS,WAAW,yBAAyB;AAC7C,iBAAS,WAAW,0BAA0B;AACpC,kBAAA,IAAI,SAAS,WAAW,EAAE;AAAA,MAAA;AAAA,IACtC;AAAA,EACF;AAAA,EAGF,MAAM,SAAkC;AAClC,QAAA,KAAK,UAAU,WAAW;AACtB,YAAA;AAAA,IAAA;AAGR,SAAK,SAAS,YAAY;AAEtB,QAAA,KAAK,UAAU,WAAW,GAAG;AAC/B,WAAK,SAAS,WAAW;AAElB,aAAA;AAAA,IAAA;AAIL,QAAA;AAIF,YAAM,KAAK,WAAW;AAAA,QACpB,aAAa;AAAA,MAAA,CACd;AAED,WAAK,SAAS,WAAW;AACzB,WAAK,gBAAgB;AAEhB,WAAA,YAAY,QAAQ,IAAI;AAAA,aACtB,OAAO;AAEd,WAAK,QAAQ;AAAA,QACX,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,QAC9D,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AAAA,MACjE;AAGA,aAAO,KAAK,SAAS;AAAA,IAAA;AAGhB,WAAA;AAAA,EAAA;AAEX;"}