@zodmon/core 0.8.0 → 0.9.0

package/dist/index.js

@@ -1,3 +1,652 @@
+// src/aggregate/expressions.ts
+var $count = () => ({
+  __accum: true,
+  expr: { $sum: 1 }
+});
+var $sum = (field) => ({
+  __accum: true,
+  expr: { $sum: field }
+});
+var $avg = (field) => ({
+  __accum: true,
+  expr: { $avg: field }
+});
+var $min = (field) => ({
+  __accum: true,
+  expr: { $min: field }
+});
+var $max = (field) => ({
+  __accum: true,
+  expr: { $max: field }
+});
+var $first = (field) => ({
+  __accum: true,
+  expr: { $first: field }
+});
+var $last = (field) => ({
+  __accum: true,
+  expr: { $last: field }
+});
+var $push = (field) => ({
+  __accum: true,
+  expr: { $push: field }
+});
+var $addToSet = (field) => ({
+  __accum: true,
+  expr: { $addToSet: field }
+});
+function createAccumulatorBuilder() {
+  return {
+    count: () => ({ __accum: true, expr: { $sum: 1 } }),
+    sum: (field) => ({
+      __accum: true,
+      expr: { $sum: typeof field === "number" ? field : `$${field}` }
+    }),
+    avg: (field) => ({ __accum: true, expr: { $avg: `$${field}` } }),
+    min: (field) => ({ __accum: true, expr: { $min: `$${field}` } }),
+    max: (field) => ({ __accum: true, expr: { $max: `$${field}` } }),
+    first: (field) => ({ __accum: true, expr: { $first: `$${field}` } }),
+    last: (field) => ({ __accum: true, expr: { $last: `$${field}` } }),
+    push: (field) => ({ __accum: true, expr: { $push: `$${field}` } }),
+    addToSet: (field) => ({ __accum: true, expr: { $addToSet: `$${field}` } })
+    // biome-ignore lint/suspicious/noExplicitAny: Runtime implementation uses string field names and returns plain objects — TypeScript cannot verify that the runtime Accumulator objects match the generic AccumulatorBuilder<T> return types. Safe because type resolution happens at compile time via AccumulatorBuilder<T>, and runtime values are identical to what the standalone $min/$max/etc. produce.
+  };
+}
+function createExpressionBuilder() {
+  const $2 = (field) => `$${field}`;
+  const val = (v) => typeof v === "number" ? v : `$${v}`;
+  const expr = (value) => ({ __expr: true, value });
+  return {
+    // Arithmetic
+    add: (field, value) => expr({ $add: [$2(field), val(value)] }),
+    subtract: (field, value) => expr({ $subtract: [$2(field), val(value)] }),
+    multiply: (field, value) => expr({ $multiply: [$2(field), val(value)] }),
+    divide: (field, value) => expr({ $divide: [$2(field), val(value)] }),
+    mod: (field, value) => expr({ $mod: [$2(field), val(value)] }),
+    abs: (field) => expr({ $abs: $2(field) }),
+    ceil: (field) => expr({ $ceil: $2(field) }),
+    floor: (field) => expr({ $floor: $2(field) }),
+    round: (field, place = 0) => expr({ $round: [$2(field), place] }),
+    // String
+    concat: (...parts) => {
+      const resolved = parts.map((p) => {
+        if (/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(p)) return $2(p);
+        return p;
+      });
+      return expr({ $concat: resolved });
+    },
+    toLower: (field) => expr({ $toLower: $2(field) }),
+    toUpper: (field) => expr({ $toUpper: $2(field) }),
+    trim: (field) => expr({ $trim: { input: $2(field) } }),
+    substr: (field, start, length) => expr({ $substrBytes: [$2(field), start, length] }),
+    // Comparison
+    eq: (field, value) => expr({ $eq: [$2(field), value] }),
+    gt: (field, value) => expr({ $gt: [$2(field), value] }),
+    gte: (field, value) => expr({ $gte: [$2(field), value] }),
+    lt: (field, value) => expr({ $lt: [$2(field), value] }),
+    lte: (field, value) => expr({ $lte: [$2(field), value] }),
+    ne: (field, value) => expr({ $ne: [$2(field), value] }),
+    // Date
+    year: (field) => expr({ $year: $2(field) }),
+    month: (field) => expr({ $month: $2(field) }),
+    dayOfMonth: (field) => expr({ $dayOfMonth: $2(field) }),
+    // Array
+    size: (field) => expr({ $size: $2(field) }),
+    // Conditional
+    cond: (condition, thenValue, elseValue) => expr({ $cond: [condition.value, thenValue, elseValue] }),
+    ifNull: (field, fallback) => expr({ $ifNull: [$2(field), fallback] })
+    // biome-ignore lint/suspicious/noExplicitAny: Runtime implementation uses string field names — TypeScript cannot verify generic ExpressionBuilder<T> return types match. Safe because type resolution happens at compile time.
+  };
+}
+
+// src/schema/ref.ts
+import { z } from "zod";
+var refMetadata = /* @__PURE__ */ new WeakMap();
+function getRefMetadata(schema) {
+  if (typeof schema !== "object" || schema === null) return void 0;
+  return refMetadata.get(schema);
+}
+var REF_GUARD = /* @__PURE__ */ Symbol.for("zodmon_ref");
+function installRefExtension() {
+  const proto = z.ZodType.prototype;
+  if (REF_GUARD in proto) return;
+  Object.defineProperty(proto, "ref", {
+    value(collection2) {
+      refMetadata.set(this, { collection: collection2 });
+      return this;
+    },
+    enumerable: true,
+    configurable: true,
+    writable: true
+  });
+  Object.defineProperty(proto, REF_GUARD, {
+    value: true,
+    enumerable: false,
+    configurable: false,
+    writable: false
+  });
+}
+
+// src/aggregate/pipeline.ts
+var AggregatePipeline = class _AggregatePipeline {
+  definition;
+  nativeCollection;
+  stages;
+  constructor(definition, nativeCollection, stages) {
+    this.definition = definition;
+    this.nativeCollection = nativeCollection;
+    this.stages = stages;
+  }
+  /**
+   * Append an arbitrary aggregation stage to the pipeline (escape hatch).
+   *
+   * Returns a new pipeline instance with the stage appended — the
+   * original pipeline is not modified.
+   *
+   * Optionally accepts a type parameter `TNew` to change the output
+   * type when the stage transforms the document shape.
+   *
+   * @typeParam TNew - The output type after this stage. Defaults to the current output type.
+   * @param stage - A raw MongoDB aggregation stage document (e.g. `{ $match: { ... } }`).
+   * @returns A new pipeline with the stage appended.
+   *
+   * @example
+   * ```ts
+   * const admins = aggregate(users)
+   *   .raw({ $match: { role: 'admin' } })
+   *   .toArray()
+   * ```
+   *
+   * @example
+   * ```ts
+   * // Change output type with a $project stage
+   * const names = aggregate(users)
+   *   .raw<{ name: string }>({ $project: { name: 1, _id: 0 } })
+   *   .toArray()
+   * ```
+   */
+  raw(stage) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      stage
+    ]);
+  }
+  /**
+   * Execute the pipeline and return all results as an array.
+   *
+   * @returns A promise resolving to the array of output documents.
+   *
+   * @example
+   * ```ts
+   * const results = await aggregate(users)
+   *   .raw({ $match: { age: { $gte: 18 } } })
+   *   .toArray()
+   * ```
+   */
+  async toArray() {
+    const cursor = this.nativeCollection.aggregate(this.stages);
+    return await cursor.toArray();
+  }
+  /**
+   * Stream pipeline results one document at a time via `for await...of`.
+   *
+   * @returns An async generator yielding output documents.
+   *
+   * @example
+   * ```ts
+   * for await (const user of aggregate(users).raw({ $match: { role: 'admin' } })) {
+   *   console.log(user.name)
+   * }
+   * ```
+   */
+  async *[Symbol.asyncIterator]() {
+    const cursor = this.nativeCollection.aggregate(this.stages);
+    for await (const doc of cursor) {
+      yield doc;
+    }
+  }
+  /**
+   * Return the query execution plan without running the pipeline.
+   *
+   * Useful for debugging and understanding how MongoDB will process
+   * the pipeline stages.
+   *
+   * @returns A promise resolving to the explain output document.
+   *
+   * @example
+   * ```ts
+   * const plan = await aggregate(users)
+   *   .raw({ $match: { role: 'admin' } })
+   *   .explain()
+   * console.log(plan)
+   * ```
+   */
+  async explain() {
+    const cursor = this.nativeCollection.aggregate(this.stages);
+    return await cursor.explain();
+  }
+  // ── Shape-preserving stages ──────────────────────────────────────
+  /**
+   * Filter documents using a type-safe match expression.
+   *
+   * Appends a `$match` stage to the pipeline. The filter is constrained
+   * to the current output type, so only valid fields and operators are accepted.
+   *
+   * Supports two forms of type narrowing:
+   *
+   * **Tier 1 — Explicit type parameter:**
+   * ```ts
+   * .match<{ role: 'engineer' | 'designer' }>({ role: { $in: ['engineer', 'designer'] } })
+   * // role narrows to 'engineer' | 'designer'
+   * ```
+   *
+   * **Tier 2 — Automatic inference from filter literals:**
+   * ```ts
+   * .match({ role: 'engineer' })           // role narrows to 'engineer'
+   * .match({ role: { $ne: 'intern' } })    // role narrows to Exclude<Role, 'intern'>
+   * .match({ role: { $in: ['engineer', 'designer'] as const } })  // needs as const
+   * ```
+   *
+   * When no type parameter is provided and the filter doesn't contain
+   * inferrable literals, the output type is unchanged (backward compatible).
+   *
+   * @typeParam TNarrow - Optional object mapping field names to narrowed types. Must be a subtype of the corresponding fields in TOutput.
+   * @typeParam F - Inferred from the filter argument. Do not provide explicitly.
+   * @param filter - A type-safe filter for the current output type.
+   * @returns A new pipeline with the `$match` stage appended and output type narrowed.
+   *
+   * @example
+   * ```ts
+   * // Explicit narrowing
+   * const filtered = await users.aggregate()
+   *   .match<{ role: 'engineer' }>({ role: 'engineer' })
+   *   .toArray()
+   * // filtered[0].role → 'engineer'
+   *
+   * // Automatic narrowing with $in (requires as const)
+   * const subset = await users.aggregate()
+   *   .match({ role: { $in: ['engineer', 'designer'] as const } })
+   *   .toArray()
+   * // subset[0].role → 'engineer' | 'designer'
+   * ```
+   */
+  match(filter) {
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $match: filter }
+    ]);
+    return pipeline;
+  }
+  /**
+   * Sort documents by one or more fields.
+   *
+   * Appends a `$sort` stage. Keys are constrained to `keyof TOutput & string`
+   * and values must be `1` (ascending) or `-1` (descending).
+   *
+   * @param spec - A sort specification mapping field names to sort direction.
+   * @returns A new pipeline with the `$sort` stage appended.
+   *
+   * @example
+   * ```ts
+   * const sorted = await aggregate(users)
+   *   .sort({ age: -1, name: 1 })
+   *   .toArray()
+   * ```
+   */
+  sort(spec) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $sort: spec }
+    ]);
+  }
+  /**
+   * Skip a number of documents in the pipeline.
+   *
+   * Appends a `$skip` stage. Commonly used with {@link limit} for pagination.
+   *
+   * @param n - The number of documents to skip.
+   * @returns A new pipeline with the `$skip` stage appended.
+   *
+   * @example
+   * ```ts
+   * // Page 2 (10 items per page)
+   * const page2 = await aggregate(users)
+   *   .sort({ name: 1 })
+   *   .skip(10)
+   *   .limit(10)
+   *   .toArray()
+   * ```
+   */
+  skip(n) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $skip: n }
+    ]);
+  }
+  /**
+   * Limit the number of documents passing through the pipeline.
+   *
+   * Appends a `$limit` stage. Commonly used with {@link skip} for pagination,
+   * or after {@link sort} to get top/bottom N results.
+   *
+   * @param n - The maximum number of documents to pass through.
+   * @returns A new pipeline with the `$limit` stage appended.
+   *
+   * @example
+   * ```ts
+   * const top5 = await aggregate(users)
+   *   .sort({ score: -1 })
+   *   .limit(5)
+   *   .toArray()
+   * ```
+   */
+  limit(n) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $limit: n }
+    ]);
+  }
+  // ── Shape-transforming projection stages ─────────────────────────
+  /**
+   * Include only specified fields in the output.
+   *
+   * Appends a `$project` stage with inclusion (`1`) for each key.
+   * The `_id` field is always included. The output type narrows to
+   * `Pick<TOutput, K | '_id'>`.
+   *
+   * @param spec - An object mapping field names to `1` for inclusion.
+   * @returns A new pipeline with the `$project` stage appended.
+   *
+   * @example
+   * ```ts
+   * const namesOnly = await aggregate(users)
+   *   .project({ name: 1 })
+   *   .toArray()
+   * // [{ _id: ..., name: 'Ada' }, ...]
+   * ```
+   */
+  project(spec) {
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $project: spec }
+    ]);
+    return pipeline;
+  }
+  /**
+   * Variadic shorthand for {@link project} — pick fields to include.
+   *
+   * Generates a `$project` stage that includes only the listed fields
+   * (plus `_id`). Equivalent to `.project({ field1: 1, field2: 1 })`.
+   *
+   * @param fields - Field names to include in the output.
+   * @returns A new pipeline with the `$project` stage appended.
+   *
+   * @example
+   * ```ts
+   * const namesAndRoles = await aggregate(users)
+   *   .pick('name', 'role')
+   *   .toArray()
+   * ```
+   */
+  pick(...fields) {
+    const spec = Object.fromEntries(fields.map((f) => [f, 1]));
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $project: spec }
+    ]);
+    return pipeline;
+  }
+  /**
+   * Exclude specified fields from the output.
+   *
+   * Appends a `$project` stage with exclusion (`0`) for each key.
+   * All other fields pass through. The output type becomes `Omit<TOutput, K>`.
+   *
+   * @param fields - Field names to exclude from the output.
+   * @returns A new pipeline with the `$project` stage appended.
+   *
+   * @example
+   * ```ts
+   * const noAge = await aggregate(users)
+   *   .omit('age')
+   *   .toArray()
+   * ```
+   */
+  omit(...fields) {
+    const spec = Object.fromEntries(fields.map((f) => [f, 0]));
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $project: spec }
+    ]);
+    return pipeline;
+  }
+  groupBy(field, accumulators) {
+    const resolved = typeof accumulators === "function" ? accumulators(createAccumulatorBuilder()) : accumulators;
+    const _id = Array.isArray(field) ? Object.fromEntries(field.map((f) => [f, `$${f}`])) : `$${field}`;
+    const accumExprs = Object.fromEntries(
+      Object.entries(resolved).map(([key, acc]) => [key, acc.expr])
+    );
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $group: { _id, ...accumExprs } }
+    ]);
+    return pipeline;
+  }
+  // Implementation
+  addFields(fields) {
+    const resolved = typeof fields === "function" ? fields(createExpressionBuilder()) : fields;
+    const stage = Object.fromEntries(
+      Object.entries(resolved).map(([k, v]) => [
+        k,
+        v && typeof v === "object" && "__expr" in v ? v.value : v
+      ])
+    );
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $addFields: stage }
+    ]);
+    return pipeline;
+  }
+  // ── unwind stage ─────────────────────────────────────────────────
+  /**
+   * Deconstruct an array field, outputting one document per array element.
+   *
+   * Appends an `$unwind` stage. The unwound field's type changes from
+   * `T[]` to `T` in the output type. Documents with empty or missing
+   * arrays are dropped unless `preserveEmpty` is `true`.
+   *
+   * @param field - The name of the array field to unwind.
+   * @param options - Optional settings for the unwind stage.
+   * @param options.preserveEmpty - If `true`, documents with null, missing, or empty arrays are preserved.
+   * @returns A new pipeline with the `$unwind` stage appended.
+   *
+   * @example
+   * ```ts
+   * const flat = await aggregate(orders)
+   *   .unwind('items')
+   *   .toArray()
+   * // Each result has a single `items` value instead of an array
+   * ```
+   */
+  unwind(field, options) {
+    const stage = options?.preserveEmpty ? { $unwind: { path: `$${field}`, preserveNullAndEmptyArrays: true } } : { $unwind: `$${field}` };
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      stage
+    ]);
+    return pipeline;
+  }
+  lookup(fieldOrFrom, options) {
+    const stages = [...this.stages];
+    if (typeof fieldOrFrom === "object") {
+      const foreignName = fieldOrFrom.name;
+      const foreignField = options?.on;
+      if (!foreignField) {
+        throw new Error(
+          `[zodmon] lookup: reverse lookup on '${foreignName}' requires an 'on' option specifying which field on the foreign collection references this collection.`
+        );
+      }
+      const asField = options?.as ?? foreignName;
+      stages.push({
+        $lookup: {
+          from: foreignName,
+          localField: "_id",
+          foreignField,
+          as: asField
+        }
+      });
+      if (options?.unwind) {
+        stages.push({ $unwind: { path: `$${asField}`, preserveNullAndEmptyArrays: true } });
+      }
+    } else {
+      const shape = this.definition.shape;
+      const fieldSchema = shape[fieldOrFrom];
+      const ref = getRefMetadata(fieldSchema);
+      if (!ref) {
+        throw new Error(
+          `[zodmon] lookup: field '${fieldOrFrom}' has no .ref() metadata. Use .lookup(CollectionDef, { on: foreignKey }) for reverse lookups, or add .ref(TargetCollection) to the field schema.`
+        );
+      }
+      const targetName = ref.collection.name;
+      const asField = options?.as ?? targetName;
+      stages.push({
+        $lookup: {
+          from: targetName,
+          localField: fieldOrFrom,
+          foreignField: "_id",
+          as: asField
+        }
+      });
+      if (options?.unwind) {
+        stages.push({ $unwind: { path: `$${asField}`, preserveNullAndEmptyArrays: true } });
+      }
+    }
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, stages);
+    return pipeline;
+  }
+  // ── Convenience shortcuts ────────────────────────────────────────
+  /**
+   * Count documents per group, sorted by count descending.
+   *
+   * Shorthand for `.groupBy(field, { count: $count() }).sort({ count: -1 })`.
+   *
+   * @param field - The field to group and count by.
+   * @returns A new pipeline producing `{ _id: TOutput[K], count: number }` results.
+   *
+   * @example
+   * ```ts
+   * const roleCounts = await aggregate(users)
+   *   .countBy('role')
+   *   .toArray()
+   * // [{ _id: 'user', count: 3 }, { _id: 'admin', count: 2 }]
+   * ```
+   */
+  countBy(field) {
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $group: { _id: `$${field}`, count: { $sum: 1 } } },
+      { $sort: { count: -1 } }
+    ]);
+    return pipeline;
+  }
+  /**
+   * Sum a numeric field per group, sorted by total descending.
+   *
+   * Shorthand for `.groupBy(field, { total: $sum('$sumField') }).sort({ total: -1 })`.
+   *
+   * @param field - The field to group by.
+   * @param sumField - The numeric field to sum.
+   * @returns A new pipeline producing `{ _id: TOutput[K], total: number }` results.
+   *
+   * @example
+   * ```ts
+   * const revenueByCategory = await aggregate(orders)
+   *   .sumBy('category', 'amount')
+   *   .toArray()
+   * // [{ _id: 'electronics', total: 5000 }, ...]
+   * ```
+   */
+  sumBy(field, sumField) {
+    const pipeline = new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $group: { _id: `$${field}`, total: { $sum: `$${sumField}` } } },
+      { $sort: { total: -1 } }
+    ]);
+    return pipeline;
+  }
+  /**
+   * Sort by a single field with a friendly direction name.
+   *
+   * Shorthand for `.sort({ [field]: direction === 'desc' ? -1 : 1 })`.
+   *
+   * @param field - The field to sort by.
+   * @param direction - Sort direction: `'asc'` (default) or `'desc'`.
+   * @returns A new pipeline with the `$sort` stage appended.
+   *
+   * @example
+   * ```ts
+   * const youngest = await aggregate(users)
+   *   .sortBy('age')
+   *   .toArray()
+   * ```
+   */
+  sortBy(field, direction = "asc") {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $sort: { [field]: direction === "desc" ? -1 : 1 } }
+    ]);
+  }
+  /**
+   * Return the top N documents sorted by a field descending.
+   *
+   * Shorthand for `.sort({ [by]: -1 }).limit(n)`.
+   *
+   * @param n - The number of documents to return.
+   * @param options - An object with a `by` field specifying the sort key.
+   * @returns A new pipeline with `$sort` and `$limit` stages appended.
+   *
+   * @example
+   * ```ts
+   * const top3 = await aggregate(users)
+   *   .top(3, { by: 'score' })
+   *   .toArray()
+   * ```
+   */
+  top(n, options) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $sort: { [options.by]: -1 } },
+      { $limit: n }
+    ]);
+  }
+  /**
+   * Return the bottom N documents sorted by a field ascending.
+   *
+   * Shorthand for `.sort({ [by]: 1 }).limit(n)`.
+   *
+   * @param n - The number of documents to return.
+   * @param options - An object with a `by` field specifying the sort key.
+   * @returns A new pipeline with `$sort` and `$limit` stages appended.
+   *
+   * @example
+   * ```ts
+   * const bottom3 = await aggregate(users)
+   *   .bottom(3, { by: 'score' })
+   *   .toArray()
+   * ```
+   */
+  bottom(n, options) {
+    return new _AggregatePipeline(this.definition, this.nativeCollection, [
+      ...this.stages,
+      { $sort: { [options.by]: 1 } },
+      { $limit: n }
+    ]);
+  }
+};
+function aggregate(handle) {
+  return new AggregatePipeline(handle.definition, handle.native, []);
+}
+
 // src/client/client.ts
 import { MongoClient } from "mongodb";
 
@@ -164,7 +813,7 @@ async function syncIndexes(handle, options) {
 }
 
 // src/crud/delete.ts
-import { z } from "zod";
+import { z as z2 } from "zod";
 
 // src/errors/validation.ts
 var ZodmonValidationError = class extends Error {
@@ -205,7 +854,7 @@ async function findOneAndDelete(handle, filter, options) {
   try {
     return handle.definition.schema.parse(result);
   } catch (err) {
-    if (err instanceof z.ZodError) {
+    if (err instanceof z2.ZodError) {
       throw new ZodmonValidationError(handle.definition.name, err);
     }
     throw err;
@@ -213,7 +862,7 @@ async function findOneAndDelete(handle, filter, options) {
 }
 
 // src/crud/find.ts
-import { z as z3 } from "zod";
+import { z as z4 } from "zod";
 
 // src/errors/not-found.ts
 var ZodmonNotFoundError = class extends Error {
@@ -250,7 +899,7 @@ function checkUnindexedFields(definition, filter) {
 }
 
 // src/query/cursor.ts
-import { z as z2 } from "zod";
+import { z as z3 } from "zod";
 
 // src/crud/paginate.ts
 import { ObjectId } from "mongodb";
@@ -510,7 +1159,7 @@ var TypedFindCursor = class {
     try {
       return this.schema.parse(raw2);
     } catch (err) {
-      if (err instanceof z2.ZodError) {
+      if (err instanceof z3.ZodError) {
         throw new ZodmonValidationError(this.collectionName, err);
       }
       throw err;
@@ -531,7 +1180,7 @@ async function findOne(handle, filter, options) {
   try {
     return handle.definition.schema.parse(raw2);
   } catch (err) {
-    if (err instanceof z3.ZodError) {
+    if (err instanceof z4.ZodError) {
       throw new ZodmonValidationError(handle.definition.name, err);
     }
     throw err;
@@ -553,13 +1202,13 @@ function find(handle, filter, options) {
 }
 
 // src/crud/insert.ts
-import { z as z4 } from "zod";
+import { z as z5 } from "zod";
 async function insertOne(handle, doc) {
   let parsed;
   try {
     parsed = handle.definition.schema.parse(doc);
   } catch (err) {
-    if (err instanceof z4.ZodError) {
+    if (err instanceof z5.ZodError) {
       throw new ZodmonValidationError(handle.definition.name, err);
     }
     throw err;
@@ -574,7 +1223,7 @@ async function insertMany(handle, docs) {
     try {
       parsed.push(handle.definition.schema.parse(doc));
     } catch (err) {
-      if (err instanceof z4.ZodError) {
+      if (err instanceof z5.ZodError) {
         throw new ZodmonValidationError(handle.definition.name, err);
       }
       throw err;
@@ -585,7 +1234,7 @@ async function insertMany(handle, docs) {
 }
 
 // src/crud/update.ts
-import { z as z5 } from "zod";
+import { z as z6 } from "zod";
 async function updateOne(handle, filter, update, options) {
   return await handle.native.updateOne(filter, update, options);
 }
@@ -616,7 +1265,7 @@ async function findOneAndUpdate(handle, filter, update, options) {
   try {
     return handle.definition.schema.parse(result);
   } catch (err) {
-    if (err instanceof z5.ZodError) {
+    if (err instanceof z6.ZodError) {
       throw new ZodmonValidationError(handle.definition.name, err);
     }
     throw err;
@@ -907,6 +1556,27 @@ var CollectionHandle = class {
   async syncIndexes(options) {
     return await syncIndexes(this, options);
   }
+  /**
+   * Start a type-safe aggregation pipeline on this collection.
+   *
+   * Returns a fluent pipeline builder that tracks the output document
+   * shape through each stage. The pipeline is lazy — no query executes
+   * until a terminal method (`toArray`, `for await`, `explain`) is called.
+   *
+   * @returns A new pipeline builder starting with this collection's document type.
+   *
+   * @example
+   * ```ts
+   * const users = db.use(Users)
+   * const result = await users.aggregate()
+   *   .match({ role: 'admin' })
+   *   .groupBy('role', { count: $count() })
+   *   .toArray()
+   * ```
+   */
+  aggregate() {
+    return aggregate(this);
+  }
 };
 
 // src/client/client.ts
@@ -932,9 +1602,7 @@ var Database = class {
    */
   use(def) {
     this._collections.set(def.name, def);
-    const native = this._db.collection(
-      def.name
-    );
+    const native = this._db.collection(def.name);
     return new CollectionHandle(
       def,
       native
@@ -1012,36 +1680,6 @@ import { z as z9 } from "zod";
 
 // src/schema/extensions.ts
 import { z as z7 } from "zod";
-
-// src/schema/ref.ts
-import { z as z6 } from "zod";
-var refMetadata = /* @__PURE__ */ new WeakMap();
-function getRefMetadata(schema) {
-  if (typeof schema !== "object" || schema === null) return void 0;
-  return refMetadata.get(schema);
-}
-var REF_GUARD = /* @__PURE__ */ Symbol.for("zodmon_ref");
-function installRefExtension() {
-  const proto = z6.ZodType.prototype;
-  if (REF_GUARD in proto) return;
-  Object.defineProperty(proto, "ref", {
-    value(collection2) {
-      refMetadata.set(this, { collection: collection2 });
-      return this;
-    },
-    enumerable: true,
-    configurable: true,
-    writable: true
-  });
-  Object.defineProperty(proto, REF_GUARD, {
-    value: true,
-    enumerable: false,
-    configurable: false,
-    writable: false
-  });
-}
-
-// src/schema/extensions.ts
 var indexMetadata = /* @__PURE__ */ new WeakMap();
 function getIndexMetadata(schema) {
   if (typeof schema !== "object" || schema === null) return void 0;
@@ -1291,29 +1929,42 @@ var $ = {
 };
 export {
   $,
+  $addToSet,
   $and,
+  $avg,
+  $count,
   $eq,
   $exists,
+  $first,
   $gt,
   $gte,
   $in,
+  $last,
   $lt,
   $lte,
+  $max,
+  $min,
   $ne,
   $nin,
   $nor,
   $not,
   $or,
+  $push,
   $regex,
+  $sum,
+  AggregatePipeline,
   CollectionHandle,
   Database,
   IndexBuilder,
   TypedFindCursor,
   ZodmonNotFoundError,
   ZodmonValidationError,
+  aggregate,
   checkUnindexedFields,
   collection,
+  createAccumulatorBuilder,
   createClient,
+  createExpressionBuilder,
   deleteMany,
   deleteOne,
   extractComparableOptions,