convex-ents 0.18.1 → 0.20.0

package/dist/deletion.js

@@ -69,11 +69,7 @@ function scheduledDeleteFactory(entDefinitions, options) {
       inProgress: import_values.v.boolean()
     },
     handler: async (ctx, { origin, stack, inProgress }) => {
-      const originId = ctx.db.normalizeId(origin.table, origin.id);
-      if (originId === null) {
-        throw new Error(`Invalid ID "${origin.id}" for table ${origin.table}`);
-      }
-      const doc = await ctx.db.get(originId);
+      const doc = await ctx.db.get(origin.table, origin.id);
       if (doc.deletionTime !== origin.deletionTime) {
         if (inProgress) {
           console.error(
@@ -91,7 +87,7 @@ function scheduledDeleteFactory(entDefinitions, options) {
         newCounter(),
         inProgress ? stack : [
           {
-            id: originId,
+            id: origin.id,
             table: origin.table,
             edges: getEdgeArgs(entDefinitions, origin.table)
           }

package/dist/deletion.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/deletion.ts","../src/shared.ts"],"sourcesContent":["import {\n  FunctionReference,\n  GenericMutationCtx,\n  IndexRangeBuilder,\n  RegisteredMutation,\n  internalMutationGeneric as internalMutation,\n  makeFunctionReference,\n} from \"convex/server\";\nimport { GenericId, Infer, convexToJson, v } from \"convex/values\";\nimport { GenericEntsDataModel } from \"./schema\";\nimport { getEdgeDefinitions } from \"./shared\";\n\nexport type ScheduledDeleteFuncRef = FunctionReference<\n  \"mutation\",\n  \"internal\",\n  {\n    origin: Origin;\n    stack: Stack;\n    inProgress: boolean;\n  },\n  void\n>;\n\ntype Origin = {\n  id: string;\n  table: string;\n  deletionTime: number;\n};\n\nconst vApproach = v.union(v.literal(\"cascade\"), v.literal(\"paginate\"));\n\ntype Approach = Infer<typeof vApproach>;\n\nexport function scheduledDeleteFactory<\n  EntsDataModel extends GenericEntsDataModel,\n>(\n  entDefinitions: EntsDataModel,\n  options?: {\n    scheduledDelete: ScheduledDeleteFuncRef;\n  },\n): RegisteredMutation<\n  \"internal\",\n  { origin: Origin; stack: Stack; inProgress: boolean },\n  Promise<void>\n> {\n  const selfRef =\n    options?.scheduledDelete ??\n    (makeFunctionReference(\n      \"functions:scheduledDelete\",\n    ) as unknown as ScheduledDeleteFuncRef);\n  return internalMutation({\n    args: {\n      origin: v.object({\n        id: v.string(),\n        table: v.string(),\n        deletionTime: v.number(),\n      }),\n      stack: v.array(\n        v.union(\n          v.object({\n            id: v.string(),\n            table: v.string(),\n            edges: v.array(\n              v.object({\n                approach: vApproach,\n                table: v.string(),\n                indexName: v.string(),\n              }),\n            ),\n          }),\n          v.object({\n            approach: vApproach,\n            cursor: v.union(v.string(), v.null()),\n            table: v.string(),\n            indexName: v.string(),\n            fieldValue: v.any(),\n          }),\n        ),\n      ),\n      inProgress: v.boolean(),\n    },\n    handler: async (ctx, { origin, stack, inProgress }) => {\n      const originId = ctx.db.normalizeId(origin.table, origin.id);\n      if (originId === null) {\n        throw new Error(`Invalid ID \"${origin.id}\" for table ${origin.table}`);\n      }\n      // Check that we still want to delete\n      // Note: Doesn't support scheduled deletion starting with system table\n      const doc = await ctx.db.get(originId);\n      if (doc.deletionTime !== origin.deletionTime) {\n        if (inProgress) {\n          console.error(\n            `[Ents] Already in-progress scheduled deletion for \"${origin.id}\" was canceled!`,\n          );\n        } else {\n          console.log(\n            `[Ents] Scheduled deletion for \"${origin.id}\" was canceled`,\n          );\n        }\n        return;\n      }\n      await progressScheduledDeletion(\n        { ctx, entDefinitions, selfRef, origin },\n        newCounter(),\n        inProgress\n          ? stack\n          : [\n              {\n                id: originId,\n                table: origin.table,\n                edges: getEdgeArgs(entDefinitions, origin.table),\n              },\n            ],\n      );\n    },\n  });\n}\n\n// Heuristic:\n// Ent at the end of an edge\n//  has soft or scheduled deletion behavior && has cascading edges: schedule individually\n//  has cascading edges: paginate by 1\n//  else: paginate by decent number\nfunction getEdgeArgs(entDefinitions: GenericEntsDataModel, table: string) {\n  const edges = getEdgeDefinitions(entDefinitions, table);\n  return Object.values(edges).flatMap((edgeDefinition) => {\n    if (\n      (edgeDefinition.cardinality === \"single\" &&\n        edgeDefinition.type === \"ref\") ||\n      (edgeDefinition.cardinality === \"multiple\" &&\n        edgeDefinition.type === \"field\")\n    ) {\n      const table = edgeDefinition.to;\n      const targetEdges = getEdgeDefinitions(entDefinitions, table);\n      const hasCascadingEdges = Object.values(targetEdges).some(\n        (edgeDefinition) =>\n          (edgeDefinition.cardinality === \"single\" &&\n            edgeDefinition.type === \"ref\") ||\n          edgeDefinition.cardinality === \"multiple\",\n      );\n      const approach = hasCascadingEdges ? \"cascade\" : \"paginate\";\n\n      const indexName = edgeDefinition.ref;\n      return [{ table, indexName, approach } as const];\n    } else if (edgeDefinition.cardinality === \"multiple\") {\n      const table = edgeDefinition.table;\n      return [\n        {\n          table,\n          indexName: edgeDefinition.field,\n          approach: \"paginate\",\n        } as const,\n        ...(edgeDefinition.symmetric\n          ? [\n              {\n                table,\n                indexName: edgeDefinition.ref,\n                approach: \"paginate\",\n              } as const,\n            ]\n          : []),\n      ];\n    } else {\n      return [];\n    }\n  });\n}\n\ntype PaginationArgs = {\n  approach: Approach;\n  table: string;\n  cursor: string | null;\n  indexName: string;\n  fieldValue: any;\n};\n\ntype EdgeArgs = {\n  approach: Approach;\n  table: string;\n  indexName: string;\n};\n\ntype Stack = (\n  | { id: string; table: string; edges: EdgeArgs[] }\n  | PaginationArgs\n)[];\n\ntype CascadeCtx = {\n  ctx: GenericMutationCtx<any>;\n  entDefinitions: GenericEntsDataModel;\n  selfRef: ScheduledDeleteFuncRef;\n  origin: Origin;\n};\n\nasync function progressScheduledDeletion(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n) {\n  const { ctx } = cascade;\n  const last = stack[stack.length - 1];\n\n  if (\"id\" in last) {\n    const edgeArgs = last.edges[0];\n    if (edgeArgs === undefined) {\n      await ctx.db.delete(last.table, last.id as GenericId<any>);\n      if (stack.length > 1) {\n        await continueOrSchedule(cascade, counter, stack.slice(0, -1));\n      }\n    } else {\n      const updated = { ...last, edges: last.edges.slice(1) };\n      await paginateOrCascade(\n        cascade,\n        counter,\n        stack.slice(0, -1).concat(updated),\n        {\n          cursor: null,\n          fieldValue: last.id,\n          ...edgeArgs,\n        },\n      );\n    }\n  } else {\n    await paginateOrCascade(cascade, counter, stack, last);\n  }\n}\n\nconst MAXIMUM_DOCUMENTS_READ = 8192 / 4;\nconst MAXIMUM_BYTES_READ = 2 ** 18;\n\nasync function paginateOrCascade(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n  { table, approach, indexName, fieldValue, cursor }: PaginationArgs,\n) {\n  const { ctx, entDefinitions } = cascade;\n  const { page, continueCursor, isDone, bytesRead } = await paginate(\n    ctx,\n    { table, indexName, fieldValue },\n    {\n      cursor,\n      ...limitsBasedOnCounter(\n        counter,\n        approach === \"paginate\"\n          ? { numItems: MAXIMUM_DOCUMENTS_READ }\n          : { numItems: 1 },\n      ),\n    },\n  );\n\n  const updatedCounter = incrementCounter(counter, page.length, bytesRead);\n  const updated = {\n    approach,\n    table,\n    cursor: continueCursor,\n    indexName,\n    fieldValue,\n  };\n  const relevantStack = cursor === null ? stack : stack.slice(0, -1);\n  const updatedStack =\n    isDone && (approach === \"paginate\" || page.length === 0)\n      ? relevantStack\n      : relevantStack.concat(\n          approach === \"cascade\"\n            ? [\n                updated,\n                {\n                  id: page[0]._id,\n                  table,\n                  edges: getEdgeArgs(entDefinitions, table),\n                },\n              ]\n            : [updated],\n        );\n  if (approach === \"paginate\") {\n    await Promise.all(page.map((doc) => ctx.db.delete(table, doc._id)));\n  }\n  await continueOrSchedule(cascade, updatedCounter, updatedStack);\n}\n\nasync function continueOrSchedule(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n) {\n  if (shouldSchedule(counter)) {\n    const { ctx, selfRef, origin } = cascade;\n    await ctx.scheduler.runAfter(0, selfRef, {\n      origin,\n      stack,\n      inProgress: true,\n    });\n  } else {\n    await progressScheduledDeletion(cascade, counter, stack);\n  }\n}\n\ntype Counter = {\n  numDocuments: number;\n  numBytesRead: number;\n};\n\nfunction newCounter() {\n  return {\n    numDocuments: 0,\n    numBytesRead: 0,\n  };\n}\n\nfunction incrementCounter(\n  counter: Counter,\n  numDocuments: number,\n  numBytesRead: number,\n) {\n  return {\n    numDocuments: counter.numDocuments + numDocuments,\n    numBytesRead: counter.numBytesRead + numBytesRead,\n  };\n}\n\nfunction limitsBasedOnCounter(\n  counter: Counter,\n  { numItems }: { numItems: number },\n) {\n  return {\n    numItems: Math.max(1, numItems - counter.numDocuments),\n    maximumBytesRead: Math.max(1, MAXIMUM_BYTES_READ - counter.numBytesRead),\n  };\n}\n\nfunction shouldSchedule(counter: Counter) {\n  return (\n    counter.numDocuments >= MAXIMUM_DOCUMENTS_READ ||\n    counter.numBytesRead >= MAXIMUM_BYTES_READ\n  );\n}\n\nasync function paginate(\n  ctx: GenericMutationCtx<any>,\n  {\n    table,\n    indexName,\n    fieldValue,\n  }: { table: string; indexName: string; fieldValue: any },\n  {\n    cursor,\n    numItems,\n    maximumBytesRead,\n  }: {\n    cursor: string | null;\n    numItems: number;\n    maximumBytesRead: number;\n  },\n) {\n  const query = ctx.db\n    .query(table)\n    .withIndex(indexName, (q) =>\n      (q.eq(indexName, fieldValue) as IndexRangeBuilder<any, any, any>).gt(\n        \"_creationTime\",\n        cursor === null ? cursor : +cursor,\n      ),\n    );\n\n  let bytesRead = 0;\n  const results = [];\n  let isDone = true;\n\n  for await (const doc of query) {\n    if (results.length >= numItems) {\n      isDone = false;\n      break;\n    }\n    const size = JSON.stringify(convexToJson(doc)).length * 8;\n\n    results.push(doc);\n    bytesRead += size;\n\n    // Check this after we read the doc, since reading it already\n    // happened anyway, and to make sure we return at least one\n    // result.\n    if (bytesRead > maximumBytesRead) {\n      isDone = false;\n      break;\n    }\n  }\n  return {\n    page: results,\n    continueCursor:\n      results.length === 0\n        ? cursor\n        : \"\" + results[results.length - 1]._creationTime,\n    isDone,\n    bytesRead,\n  };\n}\n","import {\n  DocumentByName,\n  FieldTypeFromFieldPath,\n  SystemDataModel,\n  TableNamesInDataModel,\n} from \"convex/server\";\nimport { EdgeConfig, GenericEdgeConfig, GenericEntsDataModel } from \"./schema\";\n\nexport type EntsSystemDataModel = {\n  [key in keyof SystemDataModel]: SystemDataModel[key] & {\n    edges: Record<string, never>;\n  };\n};\n\nexport type PromiseEdgeResult<\n  EdgeConfig extends GenericEdgeConfig,\n  MultipleRef,\n  MultipleField,\n  SingleOptional,\n  Single,\n> = EdgeConfig[\"cardinality\"] extends \"multiple\"\n  ? EdgeConfig[\"type\"] extends \"ref\"\n    ? MultipleRef\n    : MultipleField\n  : EdgeConfig[\"type\"] extends \"ref\"\n    ? SingleOptional\n    : EdgeConfig[\"optional\"] extends true\n      ? SingleOptional\n      : Single;\n\nexport type IndexFieldTypesForEq<\n  EntsDataModel extends GenericEntsDataModel,\n  Table extends TableNamesInDataModel<EntsDataModel>,\n  T extends string[],\n> = Pop<{\n  [K in keyof T]: FieldTypeFromFieldPath<\n    DocumentByName<EntsDataModel, Table>,\n    T[K]\n  >;\n}>;\n\ntype Pop<T extends any[]> = T extends [...infer Rest, infer _Last]\n  ? Rest\n  : never;\n\nexport function getEdgeDefinitions<\n  EntsDataModel extends GenericEntsDataModel,\n  Table extends TableNamesInDataModel<EntsDataModel>,\n>(entDefinitions: EntsDataModel, table: Table) {\n  return entDefinitions[table].edges as Record<\n    keyof EntsDataModel[Table][\"edges\"],\n    EdgeConfig\n  >;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAOO;AACP,oBAAkD;;;ACqC3C,SAAS,mBAGd,gBAA+B,OAAc;AAC7C,SAAO,eAAe,KAAK,EAAE;AAI/B;;;ADxBA,IAAM,YAAY,gBAAE,MAAM,gBAAE,QAAQ,SAAS,GAAG,gBAAE,QAAQ,UAAU,CAAC;AAI9D,SAAS,uBAGd,gBACA,SAOA;AACA,QAAM,UACJ,SAAS,uBACR;AAAA,IACC;AAAA,EACF;AACF,aAAO,cAAAA,yBAAiB;AAAA,IACtB,MAAM;AAAA,MACJ,QAAQ,gBAAE,OAAO;AAAA,QACf,IAAI,gBAAE,OAAO;AAAA,QACb,OAAO,gBAAE,OAAO;AAAA,QAChB,cAAc,gBAAE,OAAO;AAAA,MACzB,CAAC;AAAA,MACD,OAAO,gBAAE;AAAA,QACP,gBAAE;AAAA,UACA,gBAAE,OAAO;AAAA,YACP,IAAI,gBAAE,OAAO;AAAA,YACb,OAAO,gBAAE,OAAO;AAAA,YAChB,OAAO,gBAAE;AAAA,cACP,gBAAE,OAAO;AAAA,gBACP,UAAU;AAAA,gBACV,OAAO,gBAAE,OAAO;AAAA,gBAChB,WAAW,gBAAE,OAAO;AAAA,cACtB,CAAC;AAAA,YACH;AAAA,UACF,CAAC;AAAA,UACD,gBAAE,OAAO;AAAA,YACP,UAAU;AAAA,YACV,QAAQ,gBAAE,MAAM,gBAAE,OAAO,GAAG,gBAAE,KAAK,CAAC;AAAA,YACpC,OAAO,gBAAE,OAAO;AAAA,YAChB,WAAW,gBAAE,OAAO;AAAA,YACpB,YAAY,gBAAE,IAAI;AAAA,UACpB,CAAC;AAAA,QACH;AAAA,MACF;AAAA,MACA,YAAY,gBAAE,QAAQ;AAAA,IACxB;AAAA,IACA,SAAS,OAAO,KAAK,EAAE,QAAQ,OAAO,WAAW,MAAM;AACrD,YAAM,WAAW,IAAI,GAAG,YAAY,OAAO,OAAO,OAAO,EAAE;AAC3D,UAAI,aAAa,MAAM;AACrB,cAAM,IAAI,MAAM,eAAe,OAAO,EAAE,eAAe,OAAO,KAAK,EAAE;AAAA,MACvE;AAGA,YAAM,MAAM,MAAM,IAAI,GAAG,IAAI,QAAQ;AACrC,UAAI,IAAI,iBAAiB,OAAO,cAAc;AAC5C,YAAI,YAAY;AACd,kBAAQ;AAAA,YACN,sDAAsD,OAAO,EAAE;AAAA,UACjE;AAAA,QACF,OAAO;AACL,kBAAQ;AAAA,YACN,kCAAkC,OAAO,EAAE;AAAA,UAC7C;AAAA,QACF;AACA;AAAA,MACF;AACA,YAAM;AAAA,QACJ,EAAE,KAAK,gBAAgB,SAAS,OAAO;AAAA,QACvC,WAAW;AAAA,QACX,aACI,QACA;AAAA,UACE;AAAA,YACE,IAAI;AAAA,YACJ,OAAO,OAAO;AAAA,YACd,OAAO,YAAY,gBAAgB,OAAO,KAAK;AAAA,UACjD;AAAA,QACF;AAAA,MACN;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAOA,SAAS,YAAY,gBAAsC,OAAe;AACxE,QAAM,QAAQ,mBAAmB,gBAAgB,KAAK;AACtD,SAAO,OAAO,OAAO,KAAK,EAAE,QAAQ,CAAC,mBAAmB;AACtD,QACG,eAAe,gBAAgB,YAC9B,eAAe,SAAS,SACzB,eAAe,gBAAgB,cAC9B,eAAe,SAAS,SAC1B;AACA,YAAMC,SAAQ,eAAe;AAC7B,YAAM,cAAc,mBAAmB,gBAAgBA,MAAK;AAC5D,YAAM,oBAAoB,OAAO,OAAO,WAAW,EAAE;AAAA,QACnD,CAACC,oBACEA,gBAAe,gBAAgB,YAC9BA,gBAAe,SAAS,SAC1BA,gBAAe,gBAAgB;AAAA,MACnC;AACA,YAAM,WAAW,oBAAoB,YAAY;AAEjD,YAAM,YAAY,eAAe;AACjC,aAAO,CAAC,EAAE,OAAAD,QAAO,WAAW,SAAS,CAAU;AAAA,IACjD,WAAW,eAAe,gBAAgB,YAAY;AACpD,YAAMA,SAAQ,eAAe;AAC7B,aAAO;AAAA,QACL;AAAA,UACE,OAAAA;AAAA,UACA,WAAW,eAAe;AAAA,UAC1B,UAAU;AAAA,QACZ;AAAA,QACA,GAAI,eAAe,YACf;AAAA,UACE;AAAA,YACE,OAAAA;AAAA,YACA,WAAW,eAAe;AAAA,YAC1B,UAAU;AAAA,UACZ;AAAA,QACF,IACA,CAAC;AAAA,MACP;AAAA,IACF,OAAO;AACL,aAAO,CAAC;AAAA,IACV;AAAA,EACF,CAAC;AACH;AA4BA,eAAe,0BACb,SACA,SACA,OACA;AACA,QAAM,EAAE,IAAI,IAAI;AAChB,QAAM,OAAO,MAAM,MAAM,SAAS,CAAC;AAEnC,MAAI,QAAQ,MAAM;AAChB,UAAM,WAAW,KAAK,MAAM,CAAC;AAC7B,QAAI,aAAa,QAAW;AAC1B,YAAM,IAAI,GAAG,OAAO,KAAK,OAAO,KAAK,EAAoB;AACzD,UAAI,MAAM,SAAS,GAAG;AACpB,cAAM,mBAAmB,SAAS,SAAS,MAAM,MAAM,GAAG,EAAE,CAAC;AAAA,MAC/D;AAAA,IACF,OAAO;AACL,YAAM,UAAU,EAAE,GAAG,MAAM,OAAO,KAAK,MAAM,MAAM,CAAC,EAAE;AACtD,YAAM;AAAA,QACJ;AAAA,QACA;AAAA,QACA,MAAM,MAAM,GAAG,EAAE,EAAE,OAAO,OAAO;AAAA,QACjC;AAAA,UACE,QAAQ;AAAA,UACR,YAAY,KAAK;AAAA,UACjB,GAAG;AAAA,QACL;AAAA,MACF;AAAA,IACF;AAAA,EACF,OAAO;AACL,UAAM,kBAAkB,SAAS,SAAS,OAAO,IAAI;AAAA,EACvD;AACF;AAEA,IAAM,yBAAyB,OAAO;AACtC,IAAM,qBAAqB,KAAK;AAEhC,eAAe,kBACb,SACA,SACA,OACA,EAAE,OAAO,UAAU,WAAW,YAAY,OAAO,GACjD;AACA,QAAM,EAAE,KAAK,eAAe,IAAI;AAChC,QAAM,EAAE,MAAM,gBAAgB,QAAQ,UAAU,IAAI,MAAM;AAAA,IACxD;AAAA,IACA,EAAE,OAAO,WAAW,WAAW;AAAA,IAC/B;AAAA,MACE;AAAA,MACA,GAAG;AAAA,QACD;AAAA,QACA,aAAa,aACT,EAAE,UAAU,uBAAuB,IACnC,EAAE,UAAU,EAAE;AAAA,MACpB;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,iBAAiB,SAAS,KAAK,QAAQ,SAAS;AACvE,QAAM,UAAU;AAAA,IACd;AAAA,IACA;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,EACF;AACA,QAAM,gBAAgB,WAAW,OAAO,QAAQ,MAAM,MAAM,GAAG,EAAE;AACjE,QAAM,eACJ,WAAW,aAAa,cAAc,KAAK,WAAW,KAClD,gBACA,cAAc;AAAA,IACZ,aAAa,YACT;AAAA,MACE;AAAA,MACA;AAAA,QACE,IAAI,KAAK,CAAC,EAAE;AAAA,QACZ;AAAA,QACA,OAAO,YAAY,gBAAgB,KAAK;AAAA,MAC1C;AAAA,IACF,IACA,CAAC,OAAO;AAAA,EACd;AACN,MAAI,aAAa,YAAY;AAC3B,UAAM,QAAQ,IAAI,KAAK,IAAI,CAAC,QAAQ,IAAI,GAAG,OAAO,OAAO,IAAI,GAAG,CAAC,CAAC;AAAA,EACpE;AACA,QAAM,mBAAmB,SAAS,gBAAgB,YAAY;AAChE;AAEA,eAAe,mBACb,SACA,SACA,OACA;AACA,MAAI,eAAe,OAAO,GAAG;AAC3B,UAAM,EAAE,KAAK,SAAS,OAAO,IAAI;AACjC,UAAM,IAAI,UAAU,SAAS,GAAG,SAAS;AAAA,MACvC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAAA,EACH,OAAO;AACL,UAAM,0BAA0B,SAAS,SAAS,KAAK;AAAA,EACzD;AACF;AAOA,SAAS,aAAa;AACpB,SAAO;AAAA,IACL,cAAc;AAAA,IACd,cAAc;AAAA,EAChB;AACF;AAEA,SAAS,iBACP,SACA,cACA,cACA;AACA,SAAO;AAAA,IACL,cAAc,QAAQ,eAAe;AAAA,IACrC,cAAc,QAAQ,eAAe;AAAA,EACvC;AACF;AAEA,SAAS,qBACP,SACA,EAAE,SAAS,GACX;AACA,SAAO;AAAA,IACL,UAAU,KAAK,IAAI,GAAG,WAAW,QAAQ,YAAY;AAAA,IACrD,kBAAkB,KAAK,IAAI,GAAG,qBAAqB,QAAQ,YAAY;AAAA,EACzE;AACF;AAEA,SAAS,eAAe,SAAkB;AACxC,SACE,QAAQ,gBAAgB,0BACxB,QAAQ,gBAAgB;AAE5B;AAEA,eAAe,SACb,KACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF,GACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF,GAKA;AACA,QAAM,QAAQ,IAAI,GACf,MAAM,KAAK,EACX;AAAA,IAAU;AAAA,IAAW,CAAC,MACpB,EAAE,GAAG,WAAW,UAAU,EAAuC;AAAA,MAChE;AAAA,MACA,WAAW,OAAO,SAAS,CAAC;AAAA,IAC9B;AAAA,EACF;AAEF,MAAI,YAAY;AAChB,QAAM,UAAU,CAAC;AACjB,MAAI,SAAS;AAEb,mBAAiB,OAAO,OAAO;AAC7B,QAAI,QAAQ,UAAU,UAAU;AAC9B,eAAS;AACT;AAAA,IACF;AACA,UAAM,OAAO,KAAK,cAAU,4BAAa,GAAG,CAAC,EAAE,SAAS;AAExD,YAAQ,KAAK,GAAG;AAChB,iBAAa;AAKb,QAAI,YAAY,kBAAkB;AAChC,eAAS;AACT;AAAA,IACF;AAAA,EACF;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,gBACE,QAAQ,WAAW,IACf,SACA,KAAK,QAAQ,QAAQ,SAAS,CAAC,EAAE;AAAA,IACvC;AAAA,IACA;AAAA,EACF;AACF;","names":["internalMutation","table","edgeDefinition"]}
+{"version":3,"sources":["../src/deletion.ts","../src/shared.ts"],"sourcesContent":["import {\n  FunctionReference,\n  GenericMutationCtx,\n  IndexRangeBuilder,\n  RegisteredMutation,\n  internalMutationGeneric as internalMutation,\n  makeFunctionReference,\n} from \"convex/server\";\nimport { GenericId, Infer, convexToJson, v } from \"convex/values\";\nimport { GenericEntsDataModel } from \"./schema\";\nimport { getEdgeDefinitions } from \"./shared\";\n\nexport type ScheduledDeleteFuncRef = FunctionReference<\n  \"mutation\",\n  \"internal\",\n  {\n    origin: Origin;\n    stack: Stack;\n    inProgress: boolean;\n  },\n  void\n>;\n\ntype Origin = {\n  id: string;\n  table: string;\n  deletionTime: number;\n};\n\nconst vApproach = v.union(v.literal(\"cascade\"), v.literal(\"paginate\"));\n\ntype Approach = Infer<typeof vApproach>;\n\nexport function scheduledDeleteFactory<\n  EntsDataModel extends GenericEntsDataModel,\n>(\n  entDefinitions: EntsDataModel,\n  options?: {\n    scheduledDelete: ScheduledDeleteFuncRef;\n  },\n): RegisteredMutation<\n  \"internal\",\n  { origin: Origin; stack: Stack; inProgress: boolean },\n  Promise<void>\n> {\n  const selfRef =\n    options?.scheduledDelete ??\n    (makeFunctionReference(\n      \"functions:scheduledDelete\",\n    ) as unknown as ScheduledDeleteFuncRef);\n  return internalMutation({\n    args: {\n      origin: v.object({\n        id: v.string(),\n        table: v.string(),\n        deletionTime: v.number(),\n      }),\n      stack: v.array(\n        v.union(\n          v.object({\n            id: v.string(),\n            table: v.string(),\n            edges: v.array(\n              v.object({\n                approach: vApproach,\n                table: v.string(),\n                indexName: v.string(),\n              }),\n            ),\n          }),\n          v.object({\n            approach: vApproach,\n            cursor: v.union(v.string(), v.null()),\n            table: v.string(),\n            indexName: v.string(),\n            fieldValue: v.any(),\n          }),\n        ),\n      ),\n      inProgress: v.boolean(),\n    },\n    handler: async (ctx, { origin, stack, inProgress }) => {\n      // Check that we still want to delete\n      // Note: Doesn't support scheduled deletion starting with system table\n      const doc = await ctx.db.get(origin.table, origin.id as GenericId<any>);\n      if (doc.deletionTime !== origin.deletionTime) {\n        if (inProgress) {\n          console.error(\n            `[Ents] Already in-progress scheduled deletion for \"${origin.id}\" was canceled!`,\n          );\n        } else {\n          console.log(\n            `[Ents] Scheduled deletion for \"${origin.id}\" was canceled`,\n          );\n        }\n        return;\n      }\n      await progressScheduledDeletion(\n        { ctx, entDefinitions, selfRef, origin },\n        newCounter(),\n        inProgress\n          ? stack\n          : [\n              {\n                id: origin.id,\n                table: origin.table,\n                edges: getEdgeArgs(entDefinitions, origin.table),\n              },\n            ],\n      );\n    },\n  });\n}\n\n// Heuristic:\n// Ent at the end of an edge\n//  has soft or scheduled deletion behavior && has cascading edges: schedule individually\n//  has cascading edges: paginate by 1\n//  else: paginate by decent number\nfunction getEdgeArgs(entDefinitions: GenericEntsDataModel, table: string) {\n  const edges = getEdgeDefinitions(entDefinitions, table);\n  return Object.values(edges).flatMap((edgeDefinition) => {\n    if (\n      (edgeDefinition.cardinality === \"single\" &&\n        edgeDefinition.type === \"ref\") ||\n      (edgeDefinition.cardinality === \"multiple\" &&\n        edgeDefinition.type === \"field\")\n    ) {\n      const table = edgeDefinition.to;\n      const targetEdges = getEdgeDefinitions(entDefinitions, table);\n      const hasCascadingEdges = Object.values(targetEdges).some(\n        (edgeDefinition) =>\n          (edgeDefinition.cardinality === \"single\" &&\n            edgeDefinition.type === \"ref\") ||\n          edgeDefinition.cardinality === \"multiple\",\n      );\n      const approach = hasCascadingEdges ? \"cascade\" : \"paginate\";\n\n      const indexName = edgeDefinition.ref;\n      return [{ table, indexName, approach } as const];\n    } else if (edgeDefinition.cardinality === \"multiple\") {\n      const table = edgeDefinition.table;\n      return [\n        {\n          table,\n          indexName: edgeDefinition.field,\n          approach: \"paginate\",\n        } as const,\n        ...(edgeDefinition.symmetric\n          ? [\n              {\n                table,\n                indexName: edgeDefinition.ref,\n                approach: \"paginate\",\n              } as const,\n            ]\n          : []),\n      ];\n    } else {\n      return [];\n    }\n  });\n}\n\ntype PaginationArgs = {\n  approach: Approach;\n  table: string;\n  cursor: string | null;\n  indexName: string;\n  fieldValue: any;\n};\n\ntype EdgeArgs = {\n  approach: Approach;\n  table: string;\n  indexName: string;\n};\n\ntype Stack = (\n  | { id: string; table: string; edges: EdgeArgs[] }\n  | PaginationArgs\n)[];\n\ntype CascadeCtx = {\n  ctx: GenericMutationCtx<any>;\n  entDefinitions: GenericEntsDataModel;\n  selfRef: ScheduledDeleteFuncRef;\n  origin: Origin;\n};\n\nasync function progressScheduledDeletion(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n) {\n  const { ctx } = cascade;\n  const last = stack[stack.length - 1];\n\n  if (\"id\" in last) {\n    const edgeArgs = last.edges[0];\n    if (edgeArgs === undefined) {\n      await ctx.db.delete(last.table, last.id as GenericId<any>);\n      if (stack.length > 1) {\n        await continueOrSchedule(cascade, counter, stack.slice(0, -1));\n      }\n    } else {\n      const updated = { ...last, edges: last.edges.slice(1) };\n      await paginateOrCascade(\n        cascade,\n        counter,\n        stack.slice(0, -1).concat(updated),\n        {\n          cursor: null,\n          fieldValue: last.id,\n          ...edgeArgs,\n        },\n      );\n    }\n  } else {\n    await paginateOrCascade(cascade, counter, stack, last);\n  }\n}\n\nconst MAXIMUM_DOCUMENTS_READ = 8192 / 4;\nconst MAXIMUM_BYTES_READ = 2 ** 18;\n\nasync function paginateOrCascade(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n  { table, approach, indexName, fieldValue, cursor }: PaginationArgs,\n) {\n  const { ctx, entDefinitions } = cascade;\n  const { page, continueCursor, isDone, bytesRead } = await paginate(\n    ctx,\n    { table, indexName, fieldValue },\n    {\n      cursor,\n      ...limitsBasedOnCounter(\n        counter,\n        approach === \"paginate\"\n          ? { numItems: MAXIMUM_DOCUMENTS_READ }\n          : { numItems: 1 },\n      ),\n    },\n  );\n\n  const updatedCounter = incrementCounter(counter, page.length, bytesRead);\n  const updated = {\n    approach,\n    table,\n    cursor: continueCursor,\n    indexName,\n    fieldValue,\n  };\n  const relevantStack = cursor === null ? stack : stack.slice(0, -1);\n  const updatedStack =\n    isDone && (approach === \"paginate\" || page.length === 0)\n      ? relevantStack\n      : relevantStack.concat(\n          approach === \"cascade\"\n            ? [\n                updated,\n                {\n                  id: page[0]._id,\n                  table,\n                  edges: getEdgeArgs(entDefinitions, table),\n                },\n              ]\n            : [updated],\n        );\n  if (approach === \"paginate\") {\n    await Promise.all(page.map((doc) => ctx.db.delete(table, doc._id)));\n  }\n  await continueOrSchedule(cascade, updatedCounter, updatedStack);\n}\n\nasync function continueOrSchedule(\n  cascade: CascadeCtx,\n  counter: Counter,\n  stack: Stack,\n) {\n  if (shouldSchedule(counter)) {\n    const { ctx, selfRef, origin } = cascade;\n    await ctx.scheduler.runAfter(0, selfRef, {\n      origin,\n      stack,\n      inProgress: true,\n    });\n  } else {\n    await progressScheduledDeletion(cascade, counter, stack);\n  }\n}\n\ntype Counter = {\n  numDocuments: number;\n  numBytesRead: number;\n};\n\nfunction newCounter() {\n  return {\n    numDocuments: 0,\n    numBytesRead: 0,\n  };\n}\n\nfunction incrementCounter(\n  counter: Counter,\n  numDocuments: number,\n  numBytesRead: number,\n) {\n  return {\n    numDocuments: counter.numDocuments + numDocuments,\n    numBytesRead: counter.numBytesRead + numBytesRead,\n  };\n}\n\nfunction limitsBasedOnCounter(\n  counter: Counter,\n  { numItems }: { numItems: number },\n) {\n  return {\n    numItems: Math.max(1, numItems - counter.numDocuments),\n    maximumBytesRead: Math.max(1, MAXIMUM_BYTES_READ - counter.numBytesRead),\n  };\n}\n\nfunction shouldSchedule(counter: Counter) {\n  return (\n    counter.numDocuments >= MAXIMUM_DOCUMENTS_READ ||\n    counter.numBytesRead >= MAXIMUM_BYTES_READ\n  );\n}\n\nasync function paginate(\n  ctx: GenericMutationCtx<any>,\n  {\n    table,\n    indexName,\n    fieldValue,\n  }: { table: string; indexName: string; fieldValue: any },\n  {\n    cursor,\n    numItems,\n    maximumBytesRead,\n  }: {\n    cursor: string | null;\n    numItems: number;\n    maximumBytesRead: number;\n  },\n) {\n  const query = ctx.db\n    .query(table)\n    .withIndex(indexName, (q) =>\n      (q.eq(indexName, fieldValue) as IndexRangeBuilder<any, any, any>).gt(\n        \"_creationTime\",\n        cursor === null ? cursor : +cursor,\n      ),\n    );\n\n  let bytesRead = 0;\n  const results = [];\n  let isDone = true;\n\n  for await (const doc of query) {\n    if (results.length >= numItems) {\n      isDone = false;\n      break;\n    }\n    const size = JSON.stringify(convexToJson(doc)).length * 8;\n\n    results.push(doc);\n    bytesRead += size;\n\n    // Check this after we read the doc, since reading it already\n    // happened anyway, and to make sure we return at least one\n    // result.\n    if (bytesRead > maximumBytesRead) {\n      isDone = false;\n      break;\n    }\n  }\n  return {\n    page: results,\n    continueCursor:\n      results.length === 0\n        ? cursor\n        : \"\" + results[results.length - 1]._creationTime,\n    isDone,\n    bytesRead,\n  };\n}\n","import {\n  DocumentByName,\n  FieldTypeFromFieldPath,\n  GenericDatabaseReader,\n  GenericDataModel,\n  NamedTableInfo,\n  QueryInitializer,\n  SystemDataModel,\n  SystemTableNames,\n  TableNamesInDataModel,\n} from \"convex/server\";\nimport { EdgeConfig, GenericEdgeConfig, GenericEntsDataModel } from \"./schema\";\nimport { GenericId } from \"convex/values\";\n\nexport type EntsSystemDataModel = {\n  [key in keyof SystemDataModel]: SystemDataModel[key] & {\n    edges: Record<string, never>;\n  };\n};\n\nexport type PromiseEdgeResult<\n  EdgeConfig extends GenericEdgeConfig,\n  MultipleRef,\n  MultipleField,\n  SingleOptional,\n  Single,\n> = EdgeConfig[\"cardinality\"] extends \"multiple\"\n  ? EdgeConfig[\"type\"] extends \"ref\"\n    ? MultipleRef\n    : MultipleField\n  : EdgeConfig[\"type\"] extends \"ref\"\n    ? SingleOptional\n    : EdgeConfig[\"optional\"] extends true\n      ? SingleOptional\n      : Single;\n\nexport type IndexFieldTypesForEq<\n  EntsDataModel extends GenericEntsDataModel,\n  Table extends TableNamesInDataModel<EntsDataModel>,\n  T extends string[],\n> = PopIfSeveral<{\n  [K in keyof T]: FieldTypeFromFieldPath<\n    DocumentByName<EntsDataModel, Table>,\n    T[K]\n  >;\n}>;\n\n// System indexes have only a single field, so we won't to perform\n// equality check on that field. Normal indexes always have _creationTime as the last field.\ntype PopIfSeveral<T extends any[]> = T extends [infer Only]\n  ? [Only]\n  : T extends [...infer Rest, infer _Last]\n    ? Rest\n    : never;\n\nexport function getEdgeDefinitions<\n  EntsDataModel extends GenericEntsDataModel,\n  Table extends TableNamesInDataModel<EntsDataModel>,\n>(entDefinitions: EntsDataModel, table: Table) {\n  return entDefinitions[table].edges as Record<\n    keyof EntsDataModel[Table][\"edges\"],\n    EdgeConfig\n  >;\n}\n\nexport type UniqueIndexFieldName<T extends string[]> = T extends [infer Only]\n  ? Only\n  : T extends [infer Single, \"_creationTime\"]\n    ? Single\n    : never;\n\nexport function systemAwareGet<\n  DataModel extends GenericDataModel,\n  Table extends TableNamesInDataModel<DataModel>,\n>(db: GenericDatabaseReader<DataModel>, table: Table, id: GenericId<Table>) {\n  return isSystemTable(table)\n    ? db.system.get(table, id as any)\n    : db.get(table, id);\n}\n\nexport function systemAwareQuery<\n  DataModel extends GenericDataModel,\n  Table extends TableNamesInDataModel<DataModel>,\n>(\n  db: GenericDatabaseReader<DataModel>,\n  table: Table,\n): QueryInitializer<NamedTableInfo<DataModel, Table>> {\n  return isSystemTable(table)\n    ? (db.system.query(table) as any)\n    : db.query(table);\n}\n\nexport function isSystemTable(table: string): table is SystemTableNames {\n  return table.startsWith(\"_\");\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oBAOO;AACP,oBAAkD;;;AC+C3C,SAAS,mBAGd,gBAA+B,OAAc;AAC7C,SAAO,eAAe,KAAK,EAAE;AAI/B;;;ADlCA,IAAM,YAAY,gBAAE,MAAM,gBAAE,QAAQ,SAAS,GAAG,gBAAE,QAAQ,UAAU,CAAC;AAI9D,SAAS,uBAGd,gBACA,SAOA;AACA,QAAM,UACJ,SAAS,uBACR;AAAA,IACC;AAAA,EACF;AACF,aAAO,cAAAA,yBAAiB;AAAA,IACtB,MAAM;AAAA,MACJ,QAAQ,gBAAE,OAAO;AAAA,QACf,IAAI,gBAAE,OAAO;AAAA,QACb,OAAO,gBAAE,OAAO;AAAA,QAChB,cAAc,gBAAE,OAAO;AAAA,MACzB,CAAC;AAAA,MACD,OAAO,gBAAE;AAAA,QACP,gBAAE;AAAA,UACA,gBAAE,OAAO;AAAA,YACP,IAAI,gBAAE,OAAO;AAAA,YACb,OAAO,gBAAE,OAAO;AAAA,YAChB,OAAO,gBAAE;AAAA,cACP,gBAAE,OAAO;AAAA,gBACP,UAAU;AAAA,gBACV,OAAO,gBAAE,OAAO;AAAA,gBAChB,WAAW,gBAAE,OAAO;AAAA,cACtB,CAAC;AAAA,YACH;AAAA,UACF,CAAC;AAAA,UACD,gBAAE,OAAO;AAAA,YACP,UAAU;AAAA,YACV,QAAQ,gBAAE,MAAM,gBAAE,OAAO,GAAG,gBAAE,KAAK,CAAC;AAAA,YACpC,OAAO,gBAAE,OAAO;AAAA,YAChB,WAAW,gBAAE,OAAO;AAAA,YACpB,YAAY,gBAAE,IAAI;AAAA,UACpB,CAAC;AAAA,QACH;AAAA,MACF;AAAA,MACA,YAAY,gBAAE,QAAQ;AAAA,IACxB;AAAA,IACA,SAAS,OAAO,KAAK,EAAE,QAAQ,OAAO,WAAW,MAAM;AAGrD,YAAM,MAAM,MAAM,IAAI,GAAG,IAAI,OAAO,OAAO,OAAO,EAAoB;AACtE,UAAI,IAAI,iBAAiB,OAAO,cAAc;AAC5C,YAAI,YAAY;AACd,kBAAQ;AAAA,YACN,sDAAsD,OAAO,EAAE;AAAA,UACjE;AAAA,QACF,OAAO;AACL,kBAAQ;AAAA,YACN,kCAAkC,OAAO,EAAE;AAAA,UAC7C;AAAA,QACF;AACA;AAAA,MACF;AACA,YAAM;AAAA,QACJ,EAAE,KAAK,gBAAgB,SAAS,OAAO;AAAA,QACvC,WAAW;AAAA,QACX,aACI,QACA;AAAA,UACE;AAAA,YACE,IAAI,OAAO;AAAA,YACX,OAAO,OAAO;AAAA,YACd,OAAO,YAAY,gBAAgB,OAAO,KAAK;AAAA,UACjD;AAAA,QACF;AAAA,MACN;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAOA,SAAS,YAAY,gBAAsC,OAAe;AACxE,QAAM,QAAQ,mBAAmB,gBAAgB,KAAK;AACtD,SAAO,OAAO,OAAO,KAAK,EAAE,QAAQ,CAAC,mBAAmB;AACtD,QACG,eAAe,gBAAgB,YAC9B,eAAe,SAAS,SACzB,eAAe,gBAAgB,cAC9B,eAAe,SAAS,SAC1B;AACA,YAAMC,SAAQ,eAAe;AAC7B,YAAM,cAAc,mBAAmB,gBAAgBA,MAAK;AAC5D,YAAM,oBAAoB,OAAO,OAAO,WAAW,EAAE;AAAA,QACnD,CAACC,oBACEA,gBAAe,gBAAgB,YAC9BA,gBAAe,SAAS,SAC1BA,gBAAe,gBAAgB;AAAA,MACnC;AACA,YAAM,WAAW,oBAAoB,YAAY;AAEjD,YAAM,YAAY,eAAe;AACjC,aAAO,CAAC,EAAE,OAAAD,QAAO,WAAW,SAAS,CAAU;AAAA,IACjD,WAAW,eAAe,gBAAgB,YAAY;AACpD,YAAMA,SAAQ,eAAe;AAC7B,aAAO;AAAA,QACL;AAAA,UACE,OAAAA;AAAA,UACA,WAAW,eAAe;AAAA,UAC1B,UAAU;AAAA,QACZ;AAAA,QACA,GAAI,eAAe,YACf;AAAA,UACE;AAAA,YACE,OAAAA;AAAA,YACA,WAAW,eAAe;AAAA,YAC1B,UAAU;AAAA,UACZ;AAAA,QACF,IACA,CAAC;AAAA,MACP;AAAA,IACF,OAAO;AACL,aAAO,CAAC;AAAA,IACV;AAAA,EACF,CAAC;AACH;AA4BA,eAAe,0BACb,SACA,SACA,OACA;AACA,QAAM,EAAE,IAAI,IAAI;AAChB,QAAM,OAAO,MAAM,MAAM,SAAS,CAAC;AAEnC,MAAI,QAAQ,MAAM;AAChB,UAAM,WAAW,KAAK,MAAM,CAAC;AAC7B,QAAI,aAAa,QAAW;AAC1B,YAAM,IAAI,GAAG,OAAO,KAAK,OAAO,KAAK,EAAoB;AACzD,UAAI,MAAM,SAAS,GAAG;AACpB,cAAM,mBAAmB,SAAS,SAAS,MAAM,MAAM,GAAG,EAAE,CAAC;AAAA,MAC/D;AAAA,IACF,OAAO;AACL,YAAM,UAAU,EAAE,GAAG,MAAM,OAAO,KAAK,MAAM,MAAM,CAAC,EAAE;AACtD,YAAM;AAAA,QACJ;AAAA,QACA;AAAA,QACA,MAAM,MAAM,GAAG,EAAE,EAAE,OAAO,OAAO;AAAA,QACjC;AAAA,UACE,QAAQ;AAAA,UACR,YAAY,KAAK;AAAA,UACjB,GAAG;AAAA,QACL;AAAA,MACF;AAAA,IACF;AAAA,EACF,OAAO;AACL,UAAM,kBAAkB,SAAS,SAAS,OAAO,IAAI;AAAA,EACvD;AACF;AAEA,IAAM,yBAAyB,OAAO;AACtC,IAAM,qBAAqB,KAAK;AAEhC,eAAe,kBACb,SACA,SACA,OACA,EAAE,OAAO,UAAU,WAAW,YAAY,OAAO,GACjD;AACA,QAAM,EAAE,KAAK,eAAe,IAAI;AAChC,QAAM,EAAE,MAAM,gBAAgB,QAAQ,UAAU,IAAI,MAAM;AAAA,IACxD;AAAA,IACA,EAAE,OAAO,WAAW,WAAW;AAAA,IAC/B;AAAA,MACE;AAAA,MACA,GAAG;AAAA,QACD;AAAA,QACA,aAAa,aACT,EAAE,UAAU,uBAAuB,IACnC,EAAE,UAAU,EAAE;AAAA,MACpB;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,iBAAiB,SAAS,KAAK,QAAQ,SAAS;AACvE,QAAM,UAAU;AAAA,IACd;AAAA,IACA;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,EACF;AACA,QAAM,gBAAgB,WAAW,OAAO,QAAQ,MAAM,MAAM,GAAG,EAAE;AACjE,QAAM,eACJ,WAAW,aAAa,cAAc,KAAK,WAAW,KAClD,gBACA,cAAc;AAAA,IACZ,aAAa,YACT;AAAA,MACE;AAAA,MACA;AAAA,QACE,IAAI,KAAK,CAAC,EAAE;AAAA,QACZ;AAAA,QACA,OAAO,YAAY,gBAAgB,KAAK;AAAA,MAC1C;AAAA,IACF,IACA,CAAC,OAAO;AAAA,EACd;AACN,MAAI,aAAa,YAAY;AAC3B,UAAM,QAAQ,IAAI,KAAK,IAAI,CAAC,QAAQ,IAAI,GAAG,OAAO,OAAO,IAAI,GAAG,CAAC,CAAC;AAAA,EACpE;AACA,QAAM,mBAAmB,SAAS,gBAAgB,YAAY;AAChE;AAEA,eAAe,mBACb,SACA,SACA,OACA;AACA,MAAI,eAAe,OAAO,GAAG;AAC3B,UAAM,EAAE,KAAK,SAAS,OAAO,IAAI;AACjC,UAAM,IAAI,UAAU,SAAS,GAAG,SAAS;AAAA,MACvC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,IACd,CAAC;AAAA,EACH,OAAO;AACL,UAAM,0BAA0B,SAAS,SAAS,KAAK;AAAA,EACzD;AACF;AAOA,SAAS,aAAa;AACpB,SAAO;AAAA,IACL,cAAc;AAAA,IACd,cAAc;AAAA,EAChB;AACF;AAEA,SAAS,iBACP,SACA,cACA,cACA;AACA,SAAO;AAAA,IACL,cAAc,QAAQ,eAAe;AAAA,IACrC,cAAc,QAAQ,eAAe;AAAA,EACvC;AACF;AAEA,SAAS,qBACP,SACA,EAAE,SAAS,GACX;AACA,SAAO;AAAA,IACL,UAAU,KAAK,IAAI,GAAG,WAAW,QAAQ,YAAY;AAAA,IACrD,kBAAkB,KAAK,IAAI,GAAG,qBAAqB,QAAQ,YAAY;AAAA,EACzE;AACF;AAEA,SAAS,eAAe,SAAkB;AACxC,SACE,QAAQ,gBAAgB,0BACxB,QAAQ,gBAAgB;AAE5B;AAEA,eAAe,SACb,KACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF,GACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF,GAKA;AACA,QAAM,QAAQ,IAAI,GACf,MAAM,KAAK,EACX;AAAA,IAAU;AAAA,IAAW,CAAC,MACpB,EAAE,GAAG,WAAW,UAAU,EAAuC;AAAA,MAChE;AAAA,MACA,WAAW,OAAO,SAAS,CAAC;AAAA,IAC9B;AAAA,EACF;AAEF,MAAI,YAAY;AAChB,QAAM,UAAU,CAAC;AACjB,MAAI,SAAS;AAEb,mBAAiB,OAAO,OAAO;AAC7B,QAAI,QAAQ,UAAU,UAAU;AAC9B,eAAS;AACT;AAAA,IACF;AACA,UAAM,OAAO,KAAK,cAAU,4BAAa,GAAG,CAAC,EAAE,SAAS;AAExD,YAAQ,KAAK,GAAG;AAChB,iBAAa;AAKb,QAAI,YAAY,kBAAkB;AAChC,eAAS;AACT;AAAA,IACF;AAAA,EACF;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,gBACE,QAAQ,WAAW,IACf,SACA,KAAK,QAAQ,QAAQ,SAAS,CAAC,EAAE;AAAA,IACvC;AAAA,IACA;AAAA,EACF;AACF;","names":["internalMutation","table","edgeDefinition"]}

package/dist/functions.d.ts

@@ -3,4 +3,4 @@ import 'convex/values';
 import './deletion.js';
 import './schema.js';
 import './shared.js';
-export { Y as DocRetriever, D as Ent, X as EntMutationCtx, W as EntQueryCtx, E as EntsTable, C as EntsTableWriter, G as GenericEnt, U as GenericEntWriter, z as PromiseArray, y as PromiseArrayOrNull, F as PromiseEdge, t as PromiseEdgeEnts, p as PromiseEdgeEntsOrNull, v as PromiseEdgeEntsWriter, r as PromiseEdgeEntsWriterOrNull, H as PromiseEdgeOrThrow, s as PromiseEdgeOrderedEnts, o as PromiseEdgeOrderedEntsOrNull, u as PromiseEdgeOrderedEntsWriter, q as PromiseEdgeOrderedEntsWriterOrNull, I as PromiseEdgeWriter, K as PromiseEdgeWriterOrNull, J as PromiseEdgeWriterOrThrow, x as PromiseEnt, V as PromiseEntId, w as PromiseEntOrNull, T as PromiseEntWriter, S as PromiseEntWriterOrNull, m as PromiseEnts, k as PromiseEntsOrNull, n as PromiseEntsOrNulls, N as PromiseEntsWriter, l as PromiseEntsWriterOrNull, g as PromiseOrderedQuery, f as PromiseOrderedQueryBase, P as PromiseOrderedQueryOrNull, L as PromiseOrderedQueryWriter, a as PromiseOrderedQueryWriterOrNull, j as PromisePaginationResult, i as PromisePaginationResultOrNull, Q as PromisePaginationResultWriter, O as PromisePaginationResultWriterOrNull, h as PromiseQuery, b as PromiseQueryOrNull, M as PromiseQueryWriter, c as PromiseQueryWriterOrNull, e as PromiseTable, d as PromiseTableBase, R as PromiseTableWriter, Z as addEntRules, A as entWrapper, B as entsTableFactory, a0 as getDeletionConfig, _ as getReadRule, $ as getWriteRule } from './writer.js';
+export { a2 as DocRetriever, K as Ent, a1 as EntMutationCtx, a0 as EntQueryCtx, I as EntsTable, J as EntsTableWriter, L as GenericEnt, _ as GenericEntWriter, F as PromiseArray, E as PromiseArrayOrNull, M as PromiseEdge, z as PromiseEdgeEnts, v as PromiseEdgeEntsOrNull, B as PromiseEdgeEntsWriter, x as PromiseEdgeEntsWriterOrNull, N as PromiseEdgeOrThrow, y as PromiseEdgeOrderedEnts, u as PromiseEdgeOrderedEntsOrNull, A as PromiseEdgeOrderedEntsWriter, w as PromiseEdgeOrderedEntsWriterOrNull, O as PromiseEdgeWriter, R as PromiseEdgeWriterOrNull, Q as PromiseEdgeWriterOrThrow, D as PromiseEnt, $ as PromiseEntId, C as PromiseEntOrNull, Z as PromiseEntWriter, Y as PromiseEntWriterOrNull, q as PromiseEnts, o as PromiseEntsOrNull, r as PromiseEntsOrNulls, U as PromiseEntsWriter, p as PromiseEntsWriterOrNull, t as PromiseIds, s as PromiseIdsOrNull, n as PromiseIdsPaginationResult, m as PromiseIdsPaginationResultOrNull, i as PromiseOrderedIdsQuery, a as PromiseOrderedIdsQueryOrNull, h as PromiseOrderedQuery, g as PromiseOrderedQueryBase, P as PromiseOrderedQueryOrNull, S as PromiseOrderedQueryWriter, b as PromiseOrderedQueryWriterOrNull, l as PromisePaginationResult, k as PromisePaginationResultOrNull, W as PromisePaginationResultWriter, V as PromisePaginationResultWriterOrNull, j as PromiseQuery, c as PromiseQueryOrNull, T as PromiseQueryWriter, d as PromiseQueryWriterOrNull, f as PromiseTable, e as PromiseTableBase, X as PromiseTableWriter, a3 as addEntRules, G as entWrapper, H as entsTableFactory, a6 as getDeletionConfig, a4 as getReadRule, a5 as getWriteRule } from './writer.js';

package/dist/functions.js

@@ -43,6 +43,15 @@ function edgeCompoundIndexNameRaw(idA, idB) {
 function getEdgeDefinitions(entDefinitions, table) {
   return entDefinitions[table].edges;
 }
+function systemAwareGet(db, table, id) {
+  return isSystemTable(table) ? db.system.get(table, id) : db.get(table, id);
+}
+function systemAwareQuery(db, table) {
+  return isSystemTable(table) ? db.system.query(table) : db.query(table);
+}
+function isSystemTable(table) {
+  return table.startsWith("_");
+}
 
 // src/writer.ts
 var import_server2 = require("convex/server");
@@ -76,7 +85,7 @@ var WriterImplBase = class _WriterImplBase {
             }
           } else if (edgeDefinition.cardinality === "single") {
             if (edgeDefinition.deletion !== void 0 && (!isDeletingSoftly || edgeDefinition.deletion === "soft")) {
-              const doc = await this.ctx.db.get(id);
+              const doc = await this.ctx.db.get(this.table, id);
               if (doc !== null) {
                 const otherId = doc[edgeDefinition.field];
                 edges[key] = {
@@ -282,7 +291,7 @@ var WriterImplBase = class _WriterImplBase {
     if (id !== void 0) {
       const readPolicy = getReadRule(this.entDefinitions, this.table);
       if (readPolicy !== void 0) {
-        const doc = await this.ctx.db.get(id);
+        const doc = await this.ctx.db.get(this.table, id);
         if (doc === null) {
           throw new Error(
             `Cannot update document with ID "${id}" in table "${this.table} because it does not exist"`
@@ -301,7 +310,7 @@ var WriterImplBase = class _WriterImplBase {
       return;
     }
     const ent = id === void 0 ? void 0 : entWrapper(
-      await this.ctx.db.get(id),
+      await this.ctx.db.get(this.table, id),
       this.ctx,
       this.entDefinitions,
       this.table
@@ -331,9 +340,6 @@ var WriterImplBase = class _WriterImplBase {
     }
   }
 };
-function isSystemTable(table) {
-  return table.startsWith("_");
-}
 
 // src/functions.ts
 var PromiseQueryOrNullImpl = class _PromiseQueryOrNullImpl extends Promise {
@@ -360,7 +366,7 @@ var PromiseQueryOrNullImpl = class _PromiseQueryOrNullImpl extends Promise {
     );
   }
   map(callbackFn) {
-    return new PromiseArrayImpl(async () => {
+    return new PromiseArrayOrNullImpl(async () => {
       const array = await this;
       if (array === null) {
         return null;
@@ -521,6 +527,90 @@ var PromiseQueryOrNullImpl = class _PromiseQueryOrNullImpl extends Promise {
     );
   }
 };
+var PromiseIdsQueryOrNullImpl = class extends Promise {
+  constructor(retrieve, field) {
+    super(() => {
+    });
+    this.retrieve = retrieve;
+    this.field = field;
+  }
+  map(callbackFn) {
+    return new PromiseArrayOrNullImpl(async () => {
+      const array = await this;
+      if (array === null) {
+        return null;
+      }
+      return await Promise.all(array.map(callbackFn));
+    });
+  }
+  paginate(paginationOpts) {
+    return new PromiseIdsPaginationResultOrNullImpl(async () => {
+      const query = await this.retrieve();
+      if (query === null) {
+        return null;
+      }
+      const result = await query.paginate(paginationOpts);
+      return {
+        ...result,
+        page: result.page.map((id) => this._getId(id))
+      };
+    });
+  }
+  take(n) {
+    return new PromiseIdsOrNullImpl(async () => {
+      const query = await this.retrieve();
+      if (query === null) {
+        return null;
+      }
+      const result = await query.take(n);
+      return result.map((id) => this._getId(id));
+    });
+  }
+  async first() {
+    const query = await this.retrieve();
+    if (query === null) {
+      return null;
+    }
+    const doc = await query.first();
+    if (doc === null) {
+      return null;
+    }
+    return this._getId(doc);
+  }
+  async firstX() {
+    const id = await this.first();
+    if (id === null) {
+      throw new Error("Expected at least one ID, but got none");
+    }
+    return id;
+  }
+  async unique() {
+    const query = await this.retrieve();
+    if (query === null) {
+      return null;
+    }
+    const result = await query.unique();
+    if (result === null) {
+      return null;
+    }
+    return this._getId(result);
+  }
+  async uniqueX() {
+    const id = await this.unique();
+    if (id === null) {
+      throw new Error("Expected one unique ID, but got none");
+    }
+    return id;
+  }
+  then(onfulfilled, onrejected) {
+    return this.retrieve().then((query) => query === null ? null : query.collect()).then(
+      (docs) => docs === null ? null : docs.map((doc) => this._getId(doc))
+    ).then(onfulfilled, onrejected);
+  }
+  _getId(doc) {
+    return doc[this.field];
+  }
+};
 var PromisePaginationResultOrNullImpl = class extends Promise {
   constructor(ctx, entDefinitions, table, retrieve) {
     super(() => {
@@ -567,13 +657,33 @@ var PromisePaginationResultOrNullImpl = class extends Promise {
     ).then(onfulfilled, onrejected);
   }
 };
+var PromiseIdsPaginationResultOrNullImpl = class extends Promise {
+  constructor(retrieve) {
+    super(() => {
+    });
+    this.retrieve = retrieve;
+  }
+  async map(callbackFn) {
+    const result = await this;
+    if (result === null) {
+      return null;
+    }
+    return {
+      ...result,
+      page: await Promise.all(result.page.map(callbackFn))
+    };
+  }
+  then(onfulfilled, onrejected) {
+    return this.retrieve().then(onfulfilled, onrejected);
+  }
+};
 var PromiseTableImpl = class extends PromiseQueryOrNullImpl {
   constructor(ctx, entDefinitions, table) {
     super(
       ctx,
       entDefinitions,
       table,
-      async () => isSystemTable(table) ? ctx.db.system.query(table) : ctx.db.query(table)
+      async () => systemAwareQuery(ctx.db, table)
     );
   }
   get(...args) {
@@ -601,7 +711,7 @@ var PromiseTableImpl = class extends PromiseQueryOrNullImpl {
         return {
           id,
           doc: async () => {
-            const doc = await (isSystemTable(this.table) ? this.ctx.db.system.get(id) : this.ctx.db.get(id));
+            const doc = await systemAwareGet(this.ctx.db, this.table, id);
             if (throwIfNull && doc === null) {
               throw new Error(
                 `Document not found with id \`${id}\` in table "${this.table}"`
@@ -617,7 +727,7 @@ var PromiseTableImpl = class extends PromiseQueryOrNullImpl {
           this.table,
           indexName
         );
-        const doc = await this.ctx.db.query(this.table).withIndex(
+        const doc = await systemAwareQuery(this.ctx.db, this.table).withIndex(
           indexName,
           (q) => values.reduce((q2, value, i) => q2.eq(fieldNames[i], value), q)
         ).unique();
@@ -650,7 +760,7 @@ var PromiseTableImpl = class extends PromiseQueryOrNullImpl {
         });
         return await Promise.all(
           ids.map(async (id) => {
-            const doc = await (isSystemTable(this.table) ? this.ctx.db.system.get(id) : this.ctx.db.get(id));
+            const doc = await systemAwareGet(this.ctx.db, this.table, id);
             if (throwIfNull && doc === null) {
               throw new Error(
                 `Document not found with id \`${id}\` in table "${this.table}"`
@@ -661,12 +771,23 @@ var PromiseTableImpl = class extends PromiseQueryOrNullImpl {
         );
       } : async () => {
         const [indexName, values] = args;
+        const fieldNames = getIndexFields(
+          this.entDefinitions,
+          this.table,
+          indexName
+        );
+        if (fieldNames.length > 1) {
+          throw new Error(
+            `Index "${indexName}" has ${fieldNames.length} fields, but getMany() supports only single field indexes`
+          );
+        }
+        const fieldName = fieldNames[0];
         return await Promise.all(
           values.map(async (value) => {
-            const doc = await this.ctx.db.query(this.table).withIndex(indexName, (q) => q.eq(indexName, value)).unique();
+            const doc = await systemAwareQuery(this.ctx.db, this.table).withIndex(indexName, (q) => q.eq(fieldName, value)).unique();
             if (throwIfNull && doc === null) {
               throw new Error(
-                `Table "${this.table}" does not contain document with field "${indexName}" = \`${value}\``
+                `Table "${this.table}" does not contain document with field "${fieldName}" = \`${value}\``
               );
             }
             return doc;
@@ -721,7 +842,7 @@ var PromiseEntsOrNullImpl = class extends Promise {
     this.throwIfNull = throwIfNull;
   }
   map(callbackFn) {
-    return new PromiseArrayImpl(async () => {
+    return new PromiseArrayOrNullImpl(async () => {
       const array = await this;
       if (array === null) {
         return null;
@@ -823,11 +944,34 @@ var PromiseEntsOrNullImpl = class extends Promise {
     ).then(onfulfilled, onrejected);
   }
 };
+var PromiseIdsOrNullImpl = class extends Promise {
+  constructor(retrieve) {
+    super(() => {
+    });
+    this.retrieve = retrieve;
+  }
+  map(callbackFn) {
+    return new PromiseArrayOrNullImpl(async () => {
+      const array = await this;
+      if (array === null) {
+        return null;
+      }
+      return await Promise.all(array.map(callbackFn));
+    });
+  }
+  then(onfulfilled, onrejected) {
+    return this.retrieve().then(onfulfilled, onrejected);
+  }
+};
 var PromiseEdgeOrNullImpl = class _PromiseEdgeOrNullImpl extends PromiseEntsOrNullImpl {
   constructor(ctx, entDefinitions, table, edgeDefinition, retrieveSourceId, retrieveQuery, retrieveDoc = async (edgeDoc) => {
     const sourceId = edgeDoc[edgeDefinition.field];
     const targetId = edgeDoc[edgeDefinition.ref];
-    const doc = await this.ctx.db.get(targetId);
+    const doc = await systemAwareGet(
+      this.ctx.db,
+      edgeDefinition.to,
+      targetId
+    );
     if (doc === null) {
       throw new Error(
         `Dangling reference for edge "${edgeDefinition.name}" in table "${this.table}" for document with ID "${sourceId}": Could not find a document with ID "${targetId}" in table "${edgeDefinition.to}" (edge document ID is "${edgeDoc._id}").`
@@ -854,6 +998,12 @@ var PromiseEdgeOrNullImpl = class _PromiseEdgeOrNullImpl extends PromiseEntsOrNu
     this.retrieveQuery = retrieveQuery;
     this.retrieveDoc = retrieveDoc;
   }
+  ids() {
+    return new PromiseIdsQueryOrNullImpl(
+      () => this.retrieveQuery(),
+      this.edgeDefinition.ref
+    );
+  }
   async has(targetId) {
     const sourceId = await this.retrieveSourceId();
     if (sourceId === null) {
@@ -1123,7 +1273,11 @@ var PromiseEntOrNullImpl = class extends Promise {
                 `Unexpected null reference for edge "${edgeDefinition.name}" in table "${this.table}" on document with ID "${id}": Expected an ID for a document in table "${edgeDefinition.to}".`
               );
             }
-            const otherDoc = await this.ctx.db.get(otherId);
+            const otherDoc = await systemAwareGet(
+              this.ctx.db,
+              edgeDefinition.to,
+              otherId
+            );
             if (otherDoc === null && edgeDefinition.to !== "_scheduled_functions") {
               throw new Error(
                 `Dangling reference for edge "${edgeDefinition.name}" in table "${this.table}" on document with ID "${id}": Could not find a document with ID "${otherId}" in table "${edgeDefinition.to}".`
@@ -1137,7 +1291,7 @@ var PromiseEntOrNullImpl = class extends Promise {
     );
   }
 };
-var PromiseArrayImpl = class extends Promise {
+var PromiseArrayOrNullImpl = class extends Promise {
   constructor(retrieve) {
     super(() => {
     });
@@ -1396,7 +1550,7 @@ var PromiseEntWriterImpl = class extends PromiseEntOrNullImpl {
             }
             if (edgeDefinition.cardinality === "single") {
               if (edgeDefinition.type === "ref") {
-                const oldDoc = await this.ctx.db.get(docId);
+                const oldDoc = await this.ctx.db.get(this.table, docId);
                 if (oldDoc[key] !== void 0 && oldDoc[key] !== idOrIds) {
                   throw new Error("Cannot set 1:1 edge from ref end.");
                 }
@@ -1458,7 +1612,10 @@ var PromiseEntIdImpl = class extends Promise {
       this.table,
       async () => {
         const id = await this.retrieve();
-        return { id, doc: async () => this.ctx.db.get(id) };
+        return {
+          id,
+          doc: async () => systemAwareGet(this.ctx.db, this.table, id)
+        };
       },
       true
     );
@@ -1543,6 +1700,12 @@ async function filterByReadRule(ctx, entDefinitions, table, docs, throwIfNull) {
   return docs.filter((_, i) => decisions[i]);
 }
 function getIndexFields(entDefinitions, table, index) {
+  if (index === "by_id") {
+    return ["_id"];
+  }
+  if (index === "by_creation_time") {
+    return ["_creationTime"];
+  }
   return entDefinitions[table].indexes[index];
 }
 function getReadRule(entDefinitions, table) {