@atproto/bsky 0.0.272 → 0.0.274

package/dist/data-plane/server/db/pagination.js.map

@@ -1 +1 @@
-{"version":3,"file":"pagination.js","sourceRoot":"","sources":["../../../../src/data-plane/server/db/pagination.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,GAAG,EAAE,MAAM,QAAQ,CAAA;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAA;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AAS1D;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAgB,aAAa;IACjC,YACS,OAAc,EACd,SAAgB;uBADhB,OAAO;yBACP,SAAS;IACf,CAAC;IAIJ,cAAc,CAAC,OAAgB;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;QAChE,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,CAAC,OAAY;QACf,IAAI,CAAC,OAAO;YAAE,OAAM;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;IAChC,CAAC;IACD,MAAM,CAAC,SAAkB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAA;IAC3C,CAAC;IACD,UAAU,CAAC,MAAqB;QAC9B,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,GAAG,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,SAAS,EAAE,CAAA;IACjD,CAAC;IACD,YAAY,CAAC,SAAkB;QAC7B,IAAI,CAAC,SAAS;YAAE,OAAM;QACtB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACpC,MAAM,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,GAAG,MAAM,CAAA;QAC9C,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO;YACP,SAAS;SACV,CAAA;IACH,CAAC;IACD,MAAM,CAAC,OAAY,EAAE,SAA0B,EAAE,QAAkB;QACjE,IAAI,OAAO,KAAK,SAAS;YAAE,OAAM;QACjC,IAAI,QAAQ,EAAE,CAAC;YACb,mIAAmI;YACnI,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;gBACxB,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,QAAQ,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,CAAA;YAC1G,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,QAAQ,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,CAAA;YAC1G,CAAC;QACH,CAAC;aAAM,CAAC;YACN,uHAAuH;YACvH,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;gBACxB,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,SAAS,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,QAAQ,IAAI,CAAC,SAAS,MAAM,OAAO,CAAC,SAAS,IAAI,CAAA;YACtJ,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,SAAS,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,QAAQ,IAAI,CAAC,SAAS,MAAM,OAAO,CAAC,SAAS,IAAI,CAAA;YACtJ,CAAC;QACH,CAAC;IACH,CAAC;IACD,QAAQ,CACN,EAAM,EACN,IAOC;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,GAAG,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,IAAI,CAAA;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;QACvE,OAAO,EAAE;aACN,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,KAAe,CAAC,CAAC;aAC7C,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACrB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CACtE;aACA,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACtB,CAAC;aACE,OAAO,CACN,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,iBAAiB;YACrC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,kBAAkB,CACzC;aACA,OAAO,CACN,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,SAAS,iBAAiB;YACvC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,SAAS,kBAAkB,CAC3C,CACJ;aACA,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAO,CAAA;IAC3E,CAAC;CACF;AAKD,MAAM,OAAO,aAEX,SAAQ,aAAkD;IAE1D,WAAW,CAAwC,MAAqB;QACtE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC1D,CAAC;IACD,qBAAqB,CAAC,OAA6B;QACjD,OAAO;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;YACvD,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAoB;QACxC,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;QAC1D,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,WAAW,EAAE;YAClC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAA;IACH,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,aAGtC;IACA,WAAW,CAAC,MAA0C;QACpD,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC7D,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,aAGtC;IACA,WAAW,CAAC,MAA0C;QACpD,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC7D,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,CAItB,EAAM,EACN,IAQC,EACG,EAAE;IACN,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;AACvC,CAAC,CAAA;AAUD;;GAEG;AACH,MAAM,OAAgB,gBAAgB;IACpC,YAAmB,OAAc;uBAAd,OAAO;IAAU,CAAC;IAIrC,cAAc,CAAC,OAAgB;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;QAChE,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,CAAC,OAAY;QACf,IAAI,CAAC,OAAO;YAAE,OAAM;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;IAChC,CAAC;IACD,MAAM,CAAC,SAAkB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAA;IAC3C,CAAC;IACD,UAAU,CAAC,MAAwB;QACjC,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,MAAM,CAAC,OAAO,CAAA;IACvB,CAAC;IACD,YAAY,CAAC,SAAkB;QAC7B,IAAI,CAAC,SAAS;YAAE,OAAM;QACtB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACpC,MAAM,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,GAAG,MAAM,CAAA;QACnC,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO;SACR,CAAA;IACH,CAAC;IACD,MAAM,CAAC,OAAY,EAAE,SAA0B;QAC7C,IAAI,OAAO,KAAK,SAAS;YAAE,OAAM;QACjC,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YACxB,OAAO,GAAG,CAAS,GAAG,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QAC3D,CAAC;QACD,OAAO,GAAG,CAAS,GAAG,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;IAC3D,CAAC;IACD,QAAQ,CACN,EAAM,EACN,IAMC;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,GAAG,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAA;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAA;QAC1D,OAAO,EAAE;aACN,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,KAAe,CAAC,CAAC;aAC7C,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;aAC1D,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACtB,CAAC,CAAC,OAAO,CACP,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,iBAAiB;YACrC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,kBAAkB,CACzC,CACF;aACA,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAO,CAAA;IAClE,CAAC;CACF;AAKD,MAAM,OAAO,UAAsC,SAAQ,gBAG1D;IAEC,WAAW,CAAkC,MAAkB;QAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAA;IACnC,CAAC;IACD,qBAAqB,CAAC,OAA0B;QAC9C,OAAO;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE;SACjD,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAuB;QAC3C,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC5C,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,WAAW,EAAE;SACnC,CAAA;IACH,CAAC;CACF;AAED,MAAM,OAAO,YAAa,SAAQ,UAEhC;IACA,WAAW,CAAC,MAA0B;QACpC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAA;IACnC,CAAC;CACF;AAED,MAAM,OAAO,eAAgB,SAAQ,UAEnC;IACA,WAAW,CAAC,MAA6B;QACvC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,CAAA;IACtC,CAAC;CACF;AAKD,MAAM,OAAO,OAAgC,SAAQ,gBAGpD;IAEC,WAAW,CAA+B,MAAkB;QAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAA;IAC5B,CAAC;IACD,qBAAqB,CAAC,OAA0B;QAC9C,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAuB;QAC3C,IAAI,CAAC;YACH,oBAAoB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YACpC,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,WAAY,SAAQ,OAE/B;IACA,WAAW,CAAC,MAAuB;QACjC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAChC,CAAC;CACF","sourcesContent":["import { type SqlBool, sql } from 'kysely'\nimport { ensureValidRecordKey } from '@atproto/syntax'\nimport { InvalidRequestError } from '@atproto/xrpc-server'\nimport type { AnyQb, DbRef } from './util.js'\n\ntype KeysetCursor = { primary: string; secondary: string }\ntype KeysetLabeledResult = {\n  primary: string | number\n  secondary: string | number\n}\n\n/**\n * The GenericKeyset is an abstract class that sets-up the interface and partial implementation\n * of a keyset-paginated cursor with two parts. There are three types involved:\n *  - Result: a raw result (i.e. a row from the db) containing data that will make-up a cursor.\n *    - E.g. { createdAt: '2022-01-01T12:00:00Z', cid: 'bafyx' }\n *  - LabeledResult: a Result processed such that the \"primary\" and \"secondary\" parts of the cursor are labeled.\n *    - E.g. { primary: '2022-01-01T12:00:00Z', secondary: 'bafyx' }\n *  - Cursor: the two string parts that make-up the packed/string cursor.\n *    - E.g. packed cursor '1641038400000__bafyx' in parts { primary: '1641038400000', secondary: 'bafyx' }\n *\n * These types relate as such. Implementers define the relations marked with a *:\n *   Result -*-> LabeledResult <-*-> Cursor <--> packed/string cursor\n *                     ↳ SQL Condition\n */\nexport abstract class GenericKeyset<R, LR extends KeysetLabeledResult> {\n  constructor(\n    public primary: DbRef,\n    public secondary: DbRef,\n  ) {}\n  abstract labelResult(result: R): LR\n  abstract labeledResultToCursor(labeled: LR): KeysetCursor\n  abstract cursorToLabeledResult(cursor: KeysetCursor): LR\n  packFromResult(results: R | R[]): string | undefined {\n    const result = Array.isArray(results) ? results.at(-1) : results\n    if (!result) return\n    return this.pack(this.labelResult(result))\n  }\n  pack(labeled?: LR): string | undefined {\n    if (!labeled) return\n    const cursor = this.labeledResultToCursor(labeled)\n    return this.packCursor(cursor)\n  }\n  unpack(cursorStr?: string): LR | undefined {\n    const cursor = this.unpackCursor(cursorStr)\n    if (!cursor) return\n    return this.cursorToLabeledResult(cursor)\n  }\n  packCursor(cursor?: KeysetCursor): string | undefined {\n    if (!cursor) return\n    return `${cursor.primary}__${cursor.secondary}`\n  }\n  unpackCursor(cursorStr?: string): KeysetCursor | undefined {\n    if (!cursorStr) return\n    const result = cursorStr.split('__')\n    const [primary, secondary, ...others] = result\n    if (!primary || !secondary || others.length > 0) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary,\n      secondary,\n    }\n  }\n  getSql(labeled?: LR, direction?: 'asc' | 'desc', tryIndex?: boolean) {\n    if (labeled === undefined) return\n    if (tryIndex) {\n      // The tryIndex param will likely disappear and become the default implementation: here for now for gradual rollout query-by-query.\n      if (direction === 'asc') {\n        return sql<SqlBool>`((${this.primary}, ${this.secondary}) > (${labeled.primary}, ${labeled.secondary}))`\n      } else {\n        return sql<SqlBool>`((${this.primary}, ${this.secondary}) < (${labeled.primary}, ${labeled.secondary}))`\n      }\n    } else {\n      // @NOTE this implementation can struggle to use an index on (primary, secondary) for pagination due to the \"or\" usage.\n      if (direction === 'asc') {\n        return sql<SqlBool>`((${this.primary} > ${labeled.primary}) or (${this.primary} = ${labeled.primary} and ${this.secondary} > ${labeled.secondary}))`\n      } else {\n        return sql<SqlBool>`((${this.primary} < ${labeled.primary}) or (${this.primary} = ${labeled.primary} and ${this.secondary} < ${labeled.secondary}))`\n      }\n    }\n  }\n  paginate<QB extends AnyQb>(\n    qb: QB,\n    opts: {\n      limit?: number\n      cursor?: string\n      direction?: 'asc' | 'desc'\n      tryIndex?: boolean\n      // By default, pg does nullsFirst\n      nullsLast?: boolean\n    },\n  ): QB {\n    const { limit, cursor, direction = 'desc', tryIndex, nullsLast } = opts\n    const keysetSql = this.getSql(this.unpack(cursor), direction, tryIndex)\n    return qb\n      .$if(!!limit, (q) => q.limit(limit as number))\n      .$if(!nullsLast, (q) =>\n        q.orderBy(this.primary, direction).orderBy(this.secondary, direction),\n      )\n      .$if(!!nullsLast, (q) =>\n        q\n          .orderBy(\n            direction === 'asc'\n              ? sql`${this.primary} asc nulls last`\n              : sql`${this.primary} desc nulls last`,\n          )\n          .orderBy(\n            direction === 'asc'\n              ? sql`${this.secondary} asc nulls last`\n              : sql`${this.secondary} desc nulls last`,\n          ),\n      )\n      .$if(!!keysetSql, (qb) => (keysetSql ? qb.where(keysetSql) : qb)) as QB\n  }\n}\n\ntype SortAtCidResult = { sortAt: string; cid: string }\ntype TimeCidLabeledResult = KeysetCursor\n\nexport class TimeCidKeyset<\n  TimeCidResult = SortAtCidResult,\n> extends GenericKeyset<TimeCidResult, TimeCidLabeledResult> {\n  labelResult(result: TimeCidResult): TimeCidLabeledResult\n  labelResult<TimeCidResult extends SortAtCidResult>(result: TimeCidResult) {\n    return { primary: result.sortAt, secondary: result.cid }\n  }\n  labeledResultToCursor(labeled: TimeCidLabeledResult) {\n    return {\n      primary: new Date(labeled.primary).getTime().toString(),\n      secondary: labeled.secondary,\n    }\n  }\n  cursorToLabeledResult(cursor: KeysetCursor) {\n    const primaryDate = new Date(parseInt(cursor.primary, 10))\n    if (isNaN(primaryDate.getTime())) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary: primaryDate.toISOString(),\n      secondary: cursor.secondary,\n    }\n  }\n}\n\nexport class CreatedAtDidKeyset extends TimeCidKeyset<{\n  createdAt: string\n  did: string // dids are treated identically to cids in TimeCidKeyset\n}> {\n  labelResult(result: { createdAt: string; did: string }) {\n    return { primary: result.createdAt, secondary: result.did }\n  }\n}\n\nexport class IndexedAtDidKeyset extends TimeCidKeyset<{\n  indexedAt: string\n  did: string // dids are treated identically to cids in TimeCidKeyset\n}> {\n  labelResult(result: { indexedAt: string; did: string }) {\n    return { primary: result.indexedAt, secondary: result.did }\n  }\n}\n\n/**\n * This is being deprecated. Use {@link GenericKeyset#paginate} instead.\n */\nexport const paginate = <\n  QB extends AnyQb,\n  K extends GenericKeyset<unknown, any>,\n>(\n  qb: QB,\n  opts: {\n    limit?: number\n    cursor?: string\n    direction?: 'asc' | 'desc'\n    keyset: K\n    tryIndex?: boolean\n    // By default, pg does nullsFirst\n    nullsLast?: boolean\n  },\n): QB => {\n  return opts.keyset.paginate(qb, opts)\n}\n\ntype SingleKeyCursor = {\n  primary: string\n}\n\ntype SingleKeyLabeledResult = {\n  primary: string | number\n}\n\n/**\n * GenericSingleKey is similar to {@link GenericKeyset} but for a single key cursor.\n */\nexport abstract class GenericSingleKey<R, LR extends SingleKeyLabeledResult> {\n  constructor(public primary: DbRef) {}\n  abstract labelResult(result: R): LR\n  abstract labeledResultToCursor(labeled: LR): SingleKeyCursor\n  abstract cursorToLabeledResult(cursor: SingleKeyCursor): LR\n  packFromResult(results: R | R[]): string | undefined {\n    const result = Array.isArray(results) ? results.at(-1) : results\n    if (!result) return\n    return this.pack(this.labelResult(result))\n  }\n  pack(labeled?: LR): string | undefined {\n    if (!labeled) return\n    const cursor = this.labeledResultToCursor(labeled)\n    return this.packCursor(cursor)\n  }\n  unpack(cursorStr?: string): LR | undefined {\n    const cursor = this.unpackCursor(cursorStr)\n    if (!cursor) return\n    return this.cursorToLabeledResult(cursor)\n  }\n  packCursor(cursor?: SingleKeyCursor): string | undefined {\n    if (!cursor) return\n    return cursor.primary\n  }\n  unpackCursor(cursorStr?: string): SingleKeyCursor | undefined {\n    if (!cursorStr) return\n    const result = cursorStr.split('__')\n    const [primary, ...others] = result\n    if (!primary || others.length > 0) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary,\n    }\n  }\n  getSql(labeled?: LR, direction?: 'asc' | 'desc') {\n    if (labeled === undefined) return\n    if (direction === 'asc') {\n      return sql<SqlBool>`${this.primary} > ${labeled.primary}`\n    }\n    return sql<SqlBool>`${this.primary} < ${labeled.primary}`\n  }\n  paginate<QB extends AnyQb>(\n    qb: QB,\n    opts: {\n      limit?: number\n      cursor?: string\n      direction?: 'asc' | 'desc'\n      // By default, pg does nullsFirst\n      nullsLast?: boolean\n    },\n  ): QB {\n    const { limit, cursor, direction = 'desc', nullsLast } = opts\n    const keySql = this.getSql(this.unpack(cursor), direction)\n    return qb\n      .$if(!!limit, (q) => q.limit(limit as number))\n      .$if(!nullsLast, (q) => q.orderBy(this.primary, direction))\n      .$if(!!nullsLast, (q) =>\n        q.orderBy(\n          direction === 'asc'\n            ? sql`${this.primary} asc nulls last`\n            : sql`${this.primary} desc nulls last`,\n        ),\n      )\n      .$if(!!keySql, (qb) => (keySql ? qb.where(keySql) : qb)) as QB\n  }\n}\n\ntype SortAtResult = { sortAt: string }\ntype TimeLabeledResult = SingleKeyCursor\n\nexport class IsoTimeKey<TimeResult = SortAtResult> extends GenericSingleKey<\n  TimeResult,\n  TimeLabeledResult\n> {\n  labelResult(result: TimeResult): TimeLabeledResult\n  labelResult<TimeResult extends SortAtResult>(result: TimeResult) {\n    return { primary: result.sortAt }\n  }\n  labeledResultToCursor(labeled: TimeLabeledResult) {\n    return {\n      primary: new Date(labeled.primary).toISOString(),\n    }\n  }\n  cursorToLabeledResult(cursor: SingleKeyCursor) {\n    const primaryDate = new Date(cursor.primary)\n    if (isNaN(primaryDate.getTime())) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary: primaryDate.toISOString(),\n    }\n  }\n}\n\nexport class IsoSortAtKey extends IsoTimeKey<{\n  sortAt: string\n}> {\n  labelResult(result: { sortAt: string }) {\n    return { primary: result.sortAt }\n  }\n}\n\nexport class IsoUpdatedAtKey extends IsoTimeKey<{\n  updatedAt: string\n}> {\n  labelResult(result: { updatedAt: string }) {\n    return { primary: result.updatedAt }\n  }\n}\n\ntype KeyResult = { key: string }\ntype RkeyLabeledResult = SingleKeyCursor\n\nexport class RkeyKey<RkeyResult = KeyResult> extends GenericSingleKey<\n  RkeyResult,\n  RkeyLabeledResult\n> {\n  labelResult(result: RkeyResult): RkeyLabeledResult\n  labelResult<RkeyResult extends KeyResult>(result: RkeyResult) {\n    return { primary: result }\n  }\n  labeledResultToCursor(labeled: RkeyLabeledResult) {\n    return {\n      primary: labeled.primary,\n    }\n  }\n  cursorToLabeledResult(cursor: SingleKeyCursor) {\n    try {\n      ensureValidRecordKey(cursor.primary)\n      return {\n        primary: cursor.primary,\n      }\n    } catch {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n  }\n}\n\nexport class StashKeyKey extends RkeyKey<{\n  key: string\n}> {\n  labelResult(result: { key: string }) {\n    return { primary: result.key }\n  }\n}\n"]}
+{"version":3,"file":"pagination.js","sourceRoot":"","sources":["../../../../src/data-plane/server/db/pagination.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,GAAG,EAAE,MAAM,QAAQ,CAAA;AAC1C,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAA;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AAS1D;;;;;;;;;;;;;GAaG;AACH,MAAM,OAAgB,aAAa;IACjC,YACS,OAAc,EACd,SAAgB;uBADhB,OAAO;yBACP,SAAS;IACf,CAAC;IAIJ,cAAc,CAAC,OAAgB;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;QAChE,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,CACF,OAAiB,EACjB,KAAa;QAEb,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;QACrC,OAAO;YACL,KAAK;YACL,MAAM,EACJ,OAAO,CAAC,MAAM,GAAG,KAAK,IAAI,KAAK,CAAC,MAAM;gBACpC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC9C,CAAC,CAAC,SAAS;SAChB,CAAA;IACH,CAAC;IACD,IAAI,CAAC,OAAY;QACf,IAAI,CAAC,OAAO;YAAE,OAAM;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;IAChC,CAAC;IACD,MAAM,CAAC,SAAkB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAA;IAC3C,CAAC;IACD,UAAU,CAAC,MAAqB;QAC9B,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,GAAG,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,SAAS,EAAE,CAAA;IACjD,CAAC;IACD,YAAY,CAAC,SAAkB;QAC7B,IAAI,CAAC,SAAS;YAAE,OAAM;QACtB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACpC,MAAM,CAAC,OAAO,EAAE,SAAS,EAAE,GAAG,MAAM,CAAC,GAAG,MAAM,CAAA;QAC9C,IAAI,CAAC,OAAO,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO;YACP,SAAS;SACV,CAAA;IACH,CAAC;IACD,MAAM,CAAC,OAAY,EAAE,SAA0B,EAAE,QAAkB;QACjE,IAAI,OAAO,KAAK,SAAS;YAAE,OAAM;QACjC,IAAI,QAAQ,EAAE,CAAC;YACb,mIAAmI;YACnI,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;gBACxB,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,QAAQ,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,CAAA;YAC1G,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,QAAQ,OAAO,CAAC,OAAO,KAAK,OAAO,CAAC,SAAS,IAAI,CAAA;YAC1G,CAAC;QACH,CAAC;aAAM,CAAC;YACN,uHAAuH;YACvH,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;gBACxB,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,SAAS,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,QAAQ,IAAI,CAAC,SAAS,MAAM,OAAO,CAAC,SAAS,IAAI,CAAA;YACtJ,CAAC;iBAAM,CAAC;gBACN,OAAO,GAAG,CAAS,KAAK,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,SAAS,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,QAAQ,IAAI,CAAC,SAAS,MAAM,OAAO,CAAC,SAAS,IAAI,CAAA;YACtJ,CAAC;QACH,CAAC;IACH,CAAC;IACD,QAAQ,CACN,EAAM,EACN,IAOC;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,GAAG,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,IAAI,CAAA;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAA;QACvE,OAAO,EAAE;aACN,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAE,KAAgB,GAAG,CAAC,CAAC,CAAC;aACnD,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACrB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CACtE;aACA,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACtB,CAAC;aACE,OAAO,CACN,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,iBAAiB;YACrC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,kBAAkB,CACzC;aACA,OAAO,CACN,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,SAAS,iBAAiB;YACvC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,SAAS,kBAAkB,CAC3C,CACJ;aACA,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAO,CAAA;IAC3E,CAAC;CACF;AAKD,MAAM,OAAO,aAEX,SAAQ,aAAkD;IAE1D,WAAW,CAAwC,MAAqB;QACtE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC1D,CAAC;IACD,qBAAqB,CAAC,OAA6B;QACjD,OAAO;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;YACvD,SAAS,EAAE,OAAO,CAAC,SAAS;SAC7B,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAoB;QACxC,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;QAC1D,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,WAAW,EAAE;YAClC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAA;IACH,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,aAGtC;IACA,WAAW,CAAC,MAA0C;QACpD,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC7D,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,aAGtC;IACA,WAAW,CAAC,MAA0C;QACpD,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAC7D,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,CAItB,EAAM,EACN,IAQC,EACG,EAAE;IACN,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;AACvC,CAAC,CAAA;AAUD;;GAEG;AACH,MAAM,OAAgB,gBAAgB;IACpC,YAAmB,OAAc;uBAAd,OAAO;IAAU,CAAC;IAIrC,cAAc,CAAC,OAAgB;QAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;QAChE,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,CACF,OAAiB,EACjB,KAAa;QAEb,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;QACrC,OAAO;YACL,KAAK;YACL,MAAM,EACJ,OAAO,CAAC,MAAM,GAAG,KAAK,IAAI,KAAK,CAAC,MAAM;gBACpC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC9C,CAAC,CAAC,SAAS;SAChB,CAAA;IACH,CAAC;IACD,IAAI,CAAC,OAAY;QACf,IAAI,CAAC,OAAO;YAAE,OAAM;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;IAChC,CAAC;IACD,MAAM,CAAC,SAAkB;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAA;IAC3C,CAAC;IACD,UAAU,CAAC,MAAwB;QACjC,IAAI,CAAC,MAAM;YAAE,OAAM;QACnB,OAAO,MAAM,CAAC,OAAO,CAAA;IACvB,CAAC;IACD,YAAY,CAAC,SAAkB;QAC7B,IAAI,CAAC,SAAS;YAAE,OAAM;QACtB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QACpC,MAAM,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,GAAG,MAAM,CAAA;QACnC,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO;SACR,CAAA;IACH,CAAC;IACD,MAAM,CAAC,OAAY,EAAE,SAA0B;QAC7C,IAAI,OAAO,KAAK,SAAS;YAAE,OAAM;QACjC,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YACxB,OAAO,GAAG,CAAS,GAAG,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QAC3D,CAAC;QACD,OAAO,GAAG,CAAS,GAAG,IAAI,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;IAC3D,CAAC;IACD,QAAQ,CACN,EAAM,EACN,IAMC;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,GAAG,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAA;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC,CAAA;QAC1D,OAAO,EAAE;aACN,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAE,KAAgB,GAAG,CAAC,CAAC,CAAC;aACnD,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;aAC1D,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CACtB,CAAC,CAAC,OAAO,CACP,SAAS,KAAK,KAAK;YACjB,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,iBAAiB;YACrC,CAAC,CAAC,GAAG,CAAA,GAAG,IAAI,CAAC,OAAO,kBAAkB,CACzC,CACF;aACA,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAO,CAAA;IAClE,CAAC;CACF;AAKD,MAAM,OAAO,UAAsC,SAAQ,gBAG1D;IAEC,WAAW,CAAkC,MAAkB;QAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAA;IACnC,CAAC;IACD,qBAAqB,CAAC,OAA0B;QAC9C,OAAO;YACL,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE;SACjD,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAuB;QAC3C,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;QAC5C,IAAI,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;QACD,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,WAAW,EAAE;SACnC,CAAA;IACH,CAAC;CACF;AAED,MAAM,OAAO,YAAa,SAAQ,UAEhC;IACA,WAAW,CAAC,MAA0B;QACpC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,CAAA;IACnC,CAAC;CACF;AAED,MAAM,OAAO,eAAgB,SAAQ,UAEnC;IACA,WAAW,CAAC,MAA6B;QACvC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,EAAE,CAAA;IACtC,CAAC;CACF;AAKD,MAAM,OAAO,OAAgC,SAAQ,gBAGpD;IAEC,WAAW,CAA+B,MAAkB;QAC1D,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAA;IAC5B,CAAC;IACD,qBAAqB,CAAC,OAA0B;QAC9C,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAA;IACH,CAAC;IACD,qBAAqB,CAAC,MAAuB;QAC3C,IAAI,CAAC;YACH,oBAAoB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YACpC,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAA;QACH,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,mBAAmB,CAAC,kBAAkB,CAAC,CAAA;QACnD,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,WAAY,SAAQ,OAE/B;IACA,WAAW,CAAC,MAAuB;QACjC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,CAAA;IAChC,CAAC;CACF","sourcesContent":["import { type SqlBool, sql } from 'kysely'\nimport { ensureValidRecordKey } from '@atproto/syntax'\nimport { InvalidRequestError } from '@atproto/xrpc-server'\nimport type { AnyQb, DbRef } from './util.js'\n\ntype KeysetCursor = { primary: string; secondary: string }\ntype KeysetLabeledResult = {\n  primary: string | number\n  secondary: string | number\n}\n\n/**\n * The GenericKeyset is an abstract class that sets-up the interface and partial implementation\n * of a keyset-paginated cursor with two parts. There are three types involved:\n *  - Result: a raw result (i.e. a row from the db) containing data that will make-up a cursor.\n *    - E.g. { createdAt: '2022-01-01T12:00:00Z', cid: 'bafyx' }\n *  - LabeledResult: a Result processed such that the \"primary\" and \"secondary\" parts of the cursor are labeled.\n *    - E.g. { primary: '2022-01-01T12:00:00Z', secondary: 'bafyx' }\n *  - Cursor: the two string parts that make-up the packed/string cursor.\n *    - E.g. packed cursor '1641038400000__bafyx' in parts { primary: '1641038400000', secondary: 'bafyx' }\n *\n * These types relate as such. Implementers define the relations marked with a *:\n *   Result -*-> LabeledResult <-*-> Cursor <--> packed/string cursor\n *                     ↳ SQL Condition\n */\nexport abstract class GenericKeyset<R, LR extends KeysetLabeledResult> {\n  constructor(\n    public primary: DbRef,\n    public secondary: DbRef,\n  ) {}\n  abstract labelResult(result: R): LR\n  abstract labeledResultToCursor(labeled: LR): KeysetCursor\n  abstract cursorToLabeledResult(cursor: KeysetCursor): LR\n  packFromResult(results: R | R[]): string | undefined {\n    const result = Array.isArray(results) ? results.at(-1) : results\n    if (!result) return\n    return this.pack(this.labelResult(result))\n  }\n  page<Result extends R>(\n    results: Result[],\n    limit: number,\n  ): { items: Result[]; cursor?: string } {\n    const items = results.slice(0, limit)\n    return {\n      items,\n      cursor:\n        results.length > limit && items.length\n          ? this.packFromResult(items[items.length - 1])\n          : undefined,\n    }\n  }\n  pack(labeled?: LR): string | undefined {\n    if (!labeled) return\n    const cursor = this.labeledResultToCursor(labeled)\n    return this.packCursor(cursor)\n  }\n  unpack(cursorStr?: string): LR | undefined {\n    const cursor = this.unpackCursor(cursorStr)\n    if (!cursor) return\n    return this.cursorToLabeledResult(cursor)\n  }\n  packCursor(cursor?: KeysetCursor): string | undefined {\n    if (!cursor) return\n    return `${cursor.primary}__${cursor.secondary}`\n  }\n  unpackCursor(cursorStr?: string): KeysetCursor | undefined {\n    if (!cursorStr) return\n    const result = cursorStr.split('__')\n    const [primary, secondary, ...others] = result\n    if (!primary || !secondary || others.length > 0) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary,\n      secondary,\n    }\n  }\n  getSql(labeled?: LR, direction?: 'asc' | 'desc', tryIndex?: boolean) {\n    if (labeled === undefined) return\n    if (tryIndex) {\n      // The tryIndex param will likely disappear and become the default implementation: here for now for gradual rollout query-by-query.\n      if (direction === 'asc') {\n        return sql<SqlBool>`((${this.primary}, ${this.secondary}) > (${labeled.primary}, ${labeled.secondary}))`\n      } else {\n        return sql<SqlBool>`((${this.primary}, ${this.secondary}) < (${labeled.primary}, ${labeled.secondary}))`\n      }\n    } else {\n      // @NOTE this implementation can struggle to use an index on (primary, secondary) for pagination due to the \"or\" usage.\n      if (direction === 'asc') {\n        return sql<SqlBool>`((${this.primary} > ${labeled.primary}) or (${this.primary} = ${labeled.primary} and ${this.secondary} > ${labeled.secondary}))`\n      } else {\n        return sql<SqlBool>`((${this.primary} < ${labeled.primary}) or (${this.primary} = ${labeled.primary} and ${this.secondary} < ${labeled.secondary}))`\n      }\n    }\n  }\n  paginate<QB extends AnyQb>(\n    qb: QB,\n    opts: {\n      limit?: number\n      cursor?: string\n      direction?: 'asc' | 'desc'\n      tryIndex?: boolean\n      // By default, pg does nullsFirst\n      nullsLast?: boolean\n    },\n  ): QB {\n    const { limit, cursor, direction = 'desc', tryIndex, nullsLast } = opts\n    const keysetSql = this.getSql(this.unpack(cursor), direction, tryIndex)\n    return qb\n      .$if(!!limit, (q) => q.limit((limit as number) + 1))\n      .$if(!nullsLast, (q) =>\n        q.orderBy(this.primary, direction).orderBy(this.secondary, direction),\n      )\n      .$if(!!nullsLast, (q) =>\n        q\n          .orderBy(\n            direction === 'asc'\n              ? sql`${this.primary} asc nulls last`\n              : sql`${this.primary} desc nulls last`,\n          )\n          .orderBy(\n            direction === 'asc'\n              ? sql`${this.secondary} asc nulls last`\n              : sql`${this.secondary} desc nulls last`,\n          ),\n      )\n      .$if(!!keysetSql, (qb) => (keysetSql ? qb.where(keysetSql) : qb)) as QB\n  }\n}\n\ntype SortAtCidResult = { sortAt: string; cid: string }\ntype TimeCidLabeledResult = KeysetCursor\n\nexport class TimeCidKeyset<\n  TimeCidResult = SortAtCidResult,\n> extends GenericKeyset<TimeCidResult, TimeCidLabeledResult> {\n  labelResult(result: TimeCidResult): TimeCidLabeledResult\n  labelResult<TimeCidResult extends SortAtCidResult>(result: TimeCidResult) {\n    return { primary: result.sortAt, secondary: result.cid }\n  }\n  labeledResultToCursor(labeled: TimeCidLabeledResult) {\n    return {\n      primary: new Date(labeled.primary).getTime().toString(),\n      secondary: labeled.secondary,\n    }\n  }\n  cursorToLabeledResult(cursor: KeysetCursor) {\n    const primaryDate = new Date(parseInt(cursor.primary, 10))\n    if (isNaN(primaryDate.getTime())) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary: primaryDate.toISOString(),\n      secondary: cursor.secondary,\n    }\n  }\n}\n\nexport class CreatedAtDidKeyset extends TimeCidKeyset<{\n  createdAt: string\n  did: string // dids are treated identically to cids in TimeCidKeyset\n}> {\n  labelResult(result: { createdAt: string; did: string }) {\n    return { primary: result.createdAt, secondary: result.did }\n  }\n}\n\nexport class IndexedAtDidKeyset extends TimeCidKeyset<{\n  indexedAt: string\n  did: string // dids are treated identically to cids in TimeCidKeyset\n}> {\n  labelResult(result: { indexedAt: string; did: string }) {\n    return { primary: result.indexedAt, secondary: result.did }\n  }\n}\n\n/**\n * This is being deprecated. Use {@link GenericKeyset#paginate} instead.\n */\nexport const paginate = <\n  QB extends AnyQb,\n  K extends GenericKeyset<unknown, any>,\n>(\n  qb: QB,\n  opts: {\n    limit?: number\n    cursor?: string\n    direction?: 'asc' | 'desc'\n    keyset: K\n    tryIndex?: boolean\n    // By default, pg does nullsFirst\n    nullsLast?: boolean\n  },\n): QB => {\n  return opts.keyset.paginate(qb, opts)\n}\n\ntype SingleKeyCursor = {\n  primary: string\n}\n\ntype SingleKeyLabeledResult = {\n  primary: string | number\n}\n\n/**\n * GenericSingleKey is similar to {@link GenericKeyset} but for a single key cursor.\n */\nexport abstract class GenericSingleKey<R, LR extends SingleKeyLabeledResult> {\n  constructor(public primary: DbRef) {}\n  abstract labelResult(result: R): LR\n  abstract labeledResultToCursor(labeled: LR): SingleKeyCursor\n  abstract cursorToLabeledResult(cursor: SingleKeyCursor): LR\n  packFromResult(results: R | R[]): string | undefined {\n    const result = Array.isArray(results) ? results.at(-1) : results\n    if (!result) return\n    return this.pack(this.labelResult(result))\n  }\n  page<Result extends R>(\n    results: Result[],\n    limit: number,\n  ): { items: Result[]; cursor?: string } {\n    const items = results.slice(0, limit)\n    return {\n      items,\n      cursor:\n        results.length > limit && items.length\n          ? this.packFromResult(items[items.length - 1])\n          : undefined,\n    }\n  }\n  pack(labeled?: LR): string | undefined {\n    if (!labeled) return\n    const cursor = this.labeledResultToCursor(labeled)\n    return this.packCursor(cursor)\n  }\n  unpack(cursorStr?: string): LR | undefined {\n    const cursor = this.unpackCursor(cursorStr)\n    if (!cursor) return\n    return this.cursorToLabeledResult(cursor)\n  }\n  packCursor(cursor?: SingleKeyCursor): string | undefined {\n    if (!cursor) return\n    return cursor.primary\n  }\n  unpackCursor(cursorStr?: string): SingleKeyCursor | undefined {\n    if (!cursorStr) return\n    const result = cursorStr.split('__')\n    const [primary, ...others] = result\n    if (!primary || others.length > 0) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary,\n    }\n  }\n  getSql(labeled?: LR, direction?: 'asc' | 'desc') {\n    if (labeled === undefined) return\n    if (direction === 'asc') {\n      return sql<SqlBool>`${this.primary} > ${labeled.primary}`\n    }\n    return sql<SqlBool>`${this.primary} < ${labeled.primary}`\n  }\n  paginate<QB extends AnyQb>(\n    qb: QB,\n    opts: {\n      limit?: number\n      cursor?: string\n      direction?: 'asc' | 'desc'\n      // By default, pg does nullsFirst\n      nullsLast?: boolean\n    },\n  ): QB {\n    const { limit, cursor, direction = 'desc', nullsLast } = opts\n    const keySql = this.getSql(this.unpack(cursor), direction)\n    return qb\n      .$if(!!limit, (q) => q.limit((limit as number) + 1))\n      .$if(!nullsLast, (q) => q.orderBy(this.primary, direction))\n      .$if(!!nullsLast, (q) =>\n        q.orderBy(\n          direction === 'asc'\n            ? sql`${this.primary} asc nulls last`\n            : sql`${this.primary} desc nulls last`,\n        ),\n      )\n      .$if(!!keySql, (qb) => (keySql ? qb.where(keySql) : qb)) as QB\n  }\n}\n\ntype SortAtResult = { sortAt: string }\ntype TimeLabeledResult = SingleKeyCursor\n\nexport class IsoTimeKey<TimeResult = SortAtResult> extends GenericSingleKey<\n  TimeResult,\n  TimeLabeledResult\n> {\n  labelResult(result: TimeResult): TimeLabeledResult\n  labelResult<TimeResult extends SortAtResult>(result: TimeResult) {\n    return { primary: result.sortAt }\n  }\n  labeledResultToCursor(labeled: TimeLabeledResult) {\n    return {\n      primary: new Date(labeled.primary).toISOString(),\n    }\n  }\n  cursorToLabeledResult(cursor: SingleKeyCursor) {\n    const primaryDate = new Date(cursor.primary)\n    if (isNaN(primaryDate.getTime())) {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n    return {\n      primary: primaryDate.toISOString(),\n    }\n  }\n}\n\nexport class IsoSortAtKey extends IsoTimeKey<{\n  sortAt: string\n}> {\n  labelResult(result: { sortAt: string }) {\n    return { primary: result.sortAt }\n  }\n}\n\nexport class IsoUpdatedAtKey extends IsoTimeKey<{\n  updatedAt: string\n}> {\n  labelResult(result: { updatedAt: string }) {\n    return { primary: result.updatedAt }\n  }\n}\n\ntype KeyResult = { key: string }\ntype RkeyLabeledResult = SingleKeyCursor\n\nexport class RkeyKey<RkeyResult = KeyResult> extends GenericSingleKey<\n  RkeyResult,\n  RkeyLabeledResult\n> {\n  labelResult(result: RkeyResult): RkeyLabeledResult\n  labelResult<RkeyResult extends KeyResult>(result: RkeyResult) {\n    return { primary: result }\n  }\n  labeledResultToCursor(labeled: RkeyLabeledResult) {\n    return {\n      primary: labeled.primary,\n    }\n  }\n  cursorToLabeledResult(cursor: SingleKeyCursor) {\n    try {\n      ensureValidRecordKey(cursor.primary)\n      return {\n        primary: cursor.primary,\n      }\n    } catch {\n      throw new InvalidRequestError('Malformed cursor')\n    }\n  }\n}\n\nexport class StashKeyKey extends RkeyKey<{\n  key: string\n}> {\n  labelResult(result: { key: string }) {\n    return { primary: result.key }\n  }\n}\n"]}

package/dist/data-plane/server/op-thread.d.ts

@@ -0,0 +1,7 @@
+export type OpThreadReply = {
+    uri: string;
+    parentUri: string;
+    deletedAt: string | null;
+};
+export declare const resolveCanonicalOpThread: (rootUri: string, opReplies: OpThreadReply[]) => string[] | undefined;
+//# sourceMappingURL=op-thread.d.ts.map

package/dist/data-plane/server/op-thread.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"op-thread.d.ts","sourceRoot":"","sources":["../../../src/data-plane/server/op-thread.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,aAAa,GAAG;IAC1B,GAAG,EAAE,MAAM,CAAA;IACX,SAAS,EAAE,MAAM,CAAA;IACjB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;CACzB,CAAA;AAMD,eAAO,MAAM,wBAAwB,YAC1B,MAAM,aACJ,aAAa,EAAE,KACzB,MAAM,EAAE,GAAG,SAwDb,CAAA"}

package/dist/data-plane/server/op-thread.js

@@ -0,0 +1,71 @@
+import { TID } from '@atproto/common';
+import { AtUri } from '@atproto/syntax';
+// Resolves the oldest contiguous line of OP replies from a thread root.
+// A deleted reply keeps its place while replies below it survive. Once
+// nothing survives below it, only a reply written after the deletion may
+// claim the vacated slot.
+export const resolveCanonicalOpThread = (rootUri, opReplies) => {
+    const repliesByParent = new Map();
+    const deletedAtByUri = new Map();
+    for (const reply of opReplies) {
+        const siblings = repliesByParent.get(reply.parentUri) ?? [];
+        siblings.push(reply.uri);
+        repliesByParent.set(reply.parentUri, siblings);
+        if (reply.deletedAt !== null) {
+            deletedAtByUri.set(reply.uri, new Date(reply.deletedAt).getTime());
+        }
+    }
+    // Sort replies (TID enables this) for each parent so we can
+    // deterministically pick the oldest reply.
+    for (const replies of repliesByParent.values()) {
+        replies.sort();
+    }
+    // Track the current path so malformed denormalized rows cannot introduce a
+    // cycle into the result.
+    const visited = new Set([rootUri]);
+    let validOpThread = true;
+    const resolve = (parentUri) => {
+        let vacatedAt = 0;
+        for (const reply of repliesByParent.get(parentUri) ?? []) {
+            if (visited.has(reply)) {
+                validOpThread = false;
+                return [];
+            }
+            if (vacatedAt !== 0 && !isOpReplyNewerThanTimestamp(reply, vacatedAt)) {
+                continue;
+            }
+            visited.add(reply);
+            const below = resolve(reply);
+            if (!validOpThread) {
+                return [];
+            }
+            const deletedAt = deletedAtByUri.get(reply);
+            if (deletedAt === undefined || below.length > 0) {
+                return [reply, ...below];
+            }
+            visited.delete(reply);
+            if (deletedAt > vacatedAt) {
+                vacatedAt = deletedAt;
+            }
+        }
+        return [];
+    };
+    const opThread = [rootUri, ...resolve(rootUri)];
+    // A chain of one is just the root; only threads with OP replies count.
+    return validOpThread && opThread.length > 1 ? opThread : undefined;
+};
+// isOpReplyNewerThanTimestamp reports whether the post's rkey places its creation after ts.
+// The rkey is a TID, so this is the post's own claimed creation time rather
+// than when we observed it. An unparseable rkey is treated as not newer, which
+// keeps a malformed reply from claiming a vacated slot.
+const isOpReplyNewerThanTimestamp = (uri, ts) => {
+    try {
+        const tid = TID.fromStr(new AtUri(uri).rkey);
+        // TID timestamps are in microseconds.
+        return tid.timestamp() / 1000 > ts;
+    }
+    catch {
+        return false;
+    }
+};
+//# sourceMappingURL=op-thread.js.map

package/dist/data-plane/server/op-thread.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"op-thread.js","sourceRoot":"","sources":["../../../src/data-plane/server/op-thread.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAA;AACrC,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AAQvC,wEAAwE;AACxE,uEAAuE;AACvE,yEAAyE;AACzE,0BAA0B;AAC1B,MAAM,CAAC,MAAM,wBAAwB,GAAG,CACtC,OAAe,EACf,SAA0B,EACJ,EAAE;IACxB,MAAM,eAAe,GAAG,IAAI,GAAG,EAAoB,CAAA;IACnD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAA;IAChD,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAA;QAC3D,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;QACxB,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAA;QAC9C,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,EAAE,CAAC;YAC7B,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAA;QACpE,CAAC;IACH,CAAC;IAED,4DAA4D;IAC5D,2CAA2C;IAC3C,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC;QAC/C,OAAO,CAAC,IAAI,EAAE,CAAA;IAChB,CAAC;IAED,2EAA2E;IAC3E,yBAAyB;IACzB,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAA;IAClC,IAAI,aAAa,GAAG,IAAI,CAAA;IACxB,MAAM,OAAO,GAAG,CAAC,SAAiB,EAAY,EAAE;QAC9C,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,KAAK,MAAM,KAAK,IAAI,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;YACzD,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvB,aAAa,GAAG,KAAK,CAAA;gBACrB,OAAO,EAAE,CAAA;YACX,CAAC;YAED,IAAI,SAAS,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,KAAK,EAAE,SAAS,CAAC,EAAE,CAAC;gBACtE,SAAQ;YACV,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YAElB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAA;YAC5B,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,OAAO,EAAE,CAAA;YACX,CAAC;YACD,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YAC3C,IAAI,SAAS,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChD,OAAO,CAAC,KAAK,EAAE,GAAG,KAAK,CAAC,CAAA;YAC1B,CAAC;YAED,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YACrB,IAAI,SAAS,GAAG,SAAS,EAAE,CAAC;gBAC1B,SAAS,GAAG,SAAS,CAAA;YACvB,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAA;IACX,CAAC,CAAA;IAED,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;IAC/C,uEAAuE;IACvE,OAAO,aAAa,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAA;AACpE,CAAC,CAAA;AAED,4FAA4F;AAC5F,4EAA4E;AAC5E,+EAA+E;AAC/E,wDAAwD;AACxD,MAAM,2BAA2B,GAAG,CAAC,GAAW,EAAE,EAAU,EAAW,EAAE;IACvE,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAA;QAC5C,sCAAsC;QACtC,OAAO,GAAG,CAAC,SAAS,EAAE,GAAG,IAAI,GAAG,EAAE,CAAA;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;AACH,CAAC,CAAA","sourcesContent":["import { TID } from '@atproto/common'\nimport { AtUri } from '@atproto/syntax'\n\nexport type OpThreadReply = {\n  uri: string\n  parentUri: string\n  deletedAt: string | null\n}\n\n// Resolves the oldest contiguous line of OP replies from a thread root.\n// A deleted reply keeps its place while replies below it survive. Once\n// nothing survives below it, only a reply written after the deletion may\n// claim the vacated slot.\nexport const resolveCanonicalOpThread = (\n  rootUri: string,\n  opReplies: OpThreadReply[],\n): string[] | undefined => {\n  const repliesByParent = new Map<string, string[]>()\n  const deletedAtByUri = new Map<string, number>()\n  for (const reply of opReplies) {\n    const siblings = repliesByParent.get(reply.parentUri) ?? []\n    siblings.push(reply.uri)\n    repliesByParent.set(reply.parentUri, siblings)\n    if (reply.deletedAt !== null) {\n      deletedAtByUri.set(reply.uri, new Date(reply.deletedAt).getTime())\n    }\n  }\n\n  // Sort replies (TID enables this) for each parent so we can\n  // deterministically pick the oldest reply.\n  for (const replies of repliesByParent.values()) {\n    replies.sort()\n  }\n\n  // Track the current path so malformed denormalized rows cannot introduce a\n  // cycle into the result.\n  const visited = new Set([rootUri])\n  let validOpThread = true\n  const resolve = (parentUri: string): string[] => {\n    let vacatedAt = 0\n    for (const reply of repliesByParent.get(parentUri) ?? []) {\n      if (visited.has(reply)) {\n        validOpThread = false\n        return []\n      }\n\n      if (vacatedAt !== 0 && !isOpReplyNewerThanTimestamp(reply, vacatedAt)) {\n        continue\n      }\n\n      visited.add(reply)\n\n      const below = resolve(reply)\n      if (!validOpThread) {\n        return []\n      }\n      const deletedAt = deletedAtByUri.get(reply)\n      if (deletedAt === undefined || below.length > 0) {\n        return [reply, ...below]\n      }\n\n      visited.delete(reply)\n      if (deletedAt > vacatedAt) {\n        vacatedAt = deletedAt\n      }\n    }\n    return []\n  }\n\n  const opThread = [rootUri, ...resolve(rootUri)]\n  // A chain of one is just the root; only threads with OP replies count.\n  return validOpThread && opThread.length > 1 ? opThread : undefined\n}\n\n// isOpReplyNewerThanTimestamp reports whether the post's rkey places its creation after ts.\n// The rkey is a TID, so this is the post's own claimed creation time rather\n// than when we observed it. An unparseable rkey is treated as not newer, which\n// keeps a malformed reply from claiming a vacated slot.\nconst isOpReplyNewerThanTimestamp = (uri: string, ts: number): boolean => {\n  try {\n    const tid = TID.fromStr(new AtUri(uri).rkey)\n    // TID timestamps are in microseconds.\n    return tid.timestamp() / 1000 > ts\n  } catch {\n    return false\n  }\n}\n"]}

package/dist/data-plane/server/routes/activity-subscription.js

@@ -55,11 +55,11 @@ export default (db) => ({
             cursor,
             limit,
         });
-        const res = await builder.execute();
-        const dids = res.map(({ subjectDid }) => subjectDid);
+        const page = key.page(await builder.execute(), limit);
+        const dids = page.items.map(({ subjectDid }) => subjectDid);
         return {
             dids,
-            cursor: key.packFromResult(res),
+            cursor: page.cursor,
         };
     },
 });

package/dist/data-plane/server/routes/activity-subscription.js.map

@@ -1 +1 @@
-{"version":3,"file":"activity-subscription.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/activity-subscription.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AAEvC,OAAO,EAEL,kDAAkD,GACnD,MAAM,2BAA2B,CAAA;AAClC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AAEjD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,0CAA0C,CAAC,GAAG;QAClD,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,GAAG,CAAA;QACrC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAI,kDAAkD,CAAC;gBAC5D,aAAa,EAAE,EAAE;aAClB,CAAC,CAAA;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,uBAAuB,CAAC;aACnC,SAAS,EAAE;aACX,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,WAAW,CAAC;aACtC,OAAO,EAAE,CAAA;QAEZ,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAA;QAC1C,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CACnC,CAAC,GAAG,EAAsC,EAAE;YAC1C,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;YAClC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO;oBACL,QAAQ;oBACR,SAAS,EACP,UAAU,CAAC,kDAAkD;yBAC1D,KAAK;oBACV,GAAG,EAAE,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,EAAE;iBACf,CAAA;YACH,CAAC;YAED,OAAO;gBACL,QAAQ;gBACR,SAAS,EACP,UAAU,CAAC,kDAAkD,CAAC,KAAK;gBACrE,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;gBACnC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;gBACrC,UAAU,EAAE,OAAO,CAAC,UAAU;aAC/B,CAAA;QACH,CAAC,CACF,CAAA;QAED,OAAO;YACL,aAAa;SACd,CAAA;IACH,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,GAAG;QACnC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QAEvC,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,uBAAuB,CAAC;aACnC,SAAS,EAAE;aACX,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAA;QAElC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAA;QAC7D,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QACF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QACnC,MAAM,IAAI,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,CAAA;QACpD,OAAO;YACL,IAAI;YACJ,MAAM,EAAE,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC;SAChC,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import type { PlainMessage } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport { keyBy } from '@atproto/common'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport {\n  type ActivitySubscription,\n  GetActivitySubscriptionsByActorAndSubjectsResponse,\n} from '../../../proto/bsky_pb.js'\nimport { Namespaces } from '../../../stash.js'\nimport type { Database } from '../db/index.js'\nimport { StashKeyKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActivitySubscriptionsByActorAndSubjects(req) {\n    const { actorDid, subjectDids } = req\n    if (subjectDids.length === 0) {\n      return new GetActivitySubscriptionsByActorAndSubjectsResponse({\n        subscriptions: [],\n      })\n    }\n\n    const res = await db.db\n      .selectFrom('activity_subscription')\n      .selectAll()\n      .where('creator', '=', actorDid)\n      .where('subjectDid', 'in', subjectDids)\n      .execute()\n\n    const bySubject = keyBy(res, 'subjectDid')\n    const subscriptions = subjectDids.map(\n      (did): PlainMessage<ActivitySubscription> => {\n        const subject = bySubject.get(did)\n        if (!subject) {\n          return {\n            actorDid,\n            namespace:\n              Namespaces.AppBskyNotificationDefsSubjectActivitySubscription\n                .$type,\n            key: '',\n            post: undefined,\n            reply: undefined,\n            subjectDid: '',\n          }\n        }\n\n        return {\n          actorDid,\n          namespace:\n            Namespaces.AppBskyNotificationDefsSubjectActivitySubscription.$type,\n          key: subject.key,\n          post: subject.post ? {} : undefined,\n          reply: subject.reply ? {} : undefined,\n          subjectDid: subject.subjectDid,\n        }\n      },\n    )\n\n    return {\n      subscriptions,\n    }\n  },\n\n  async getActivitySubscriptionDids(req) {\n    const { actorDid, cursor, limit } = req\n\n    let builder = db.db\n      .selectFrom('activity_subscription')\n      .selectAll()\n      .where('creator', '=', actorDid)\n\n    const { ref } = db.db.dynamic\n    const key = new StashKeyKey(ref('activity_subscription.key'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n    const res = await builder.execute()\n    const dids = res.map(({ subjectDid }) => subjectDid)\n    return {\n      dids,\n      cursor: key.packFromResult(res),\n    }\n  },\n})\n"]}
+{"version":3,"file":"activity-subscription.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/activity-subscription.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AAEvC,OAAO,EAEL,kDAAkD,GACnD,MAAM,2BAA2B,CAAA;AAClC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AAEjD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,0CAA0C,CAAC,GAAG;QAClD,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,GAAG,CAAA;QACrC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAI,kDAAkD,CAAC;gBAC5D,aAAa,EAAE,EAAE;aAClB,CAAC,CAAA;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,uBAAuB,CAAC;aACnC,SAAS,EAAE;aACX,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,WAAW,CAAC;aACtC,OAAO,EAAE,CAAA;QAEZ,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAA;QAC1C,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,CACnC,CAAC,GAAG,EAAsC,EAAE;YAC1C,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;YAClC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO;oBACL,QAAQ;oBACR,SAAS,EACP,UAAU,CAAC,kDAAkD;yBAC1D,KAAK;oBACV,GAAG,EAAE,EAAE;oBACP,IAAI,EAAE,SAAS;oBACf,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,EAAE;iBACf,CAAA;YACH,CAAC;YAED,OAAO;gBACL,QAAQ;gBACR,SAAS,EACP,UAAU,CAAC,kDAAkD,CAAC,KAAK;gBACrE,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;gBACnC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;gBACrC,UAAU,EAAE,OAAO,CAAC,UAAU;aAC/B,CAAA;QACH,CAAC,CACF,CAAA;QAED,OAAO;YACL,aAAa;SACd,CAAA;IACH,CAAC;IAED,KAAK,CAAC,2BAA2B,CAAC,GAAG;QACnC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QAEvC,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,uBAAuB,CAAC;aACnC,SAAS,EAAE;aACX,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAA;QAElC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAA;QAC7D,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,CAAA;QAC3D,OAAO;YACL,IAAI;YACJ,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import type { PlainMessage } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport { keyBy } from '@atproto/common'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport {\n  type ActivitySubscription,\n  GetActivitySubscriptionsByActorAndSubjectsResponse,\n} from '../../../proto/bsky_pb.js'\nimport { Namespaces } from '../../../stash.js'\nimport type { Database } from '../db/index.js'\nimport { StashKeyKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActivitySubscriptionsByActorAndSubjects(req) {\n    const { actorDid, subjectDids } = req\n    if (subjectDids.length === 0) {\n      return new GetActivitySubscriptionsByActorAndSubjectsResponse({\n        subscriptions: [],\n      })\n    }\n\n    const res = await db.db\n      .selectFrom('activity_subscription')\n      .selectAll()\n      .where('creator', '=', actorDid)\n      .where('subjectDid', 'in', subjectDids)\n      .execute()\n\n    const bySubject = keyBy(res, 'subjectDid')\n    const subscriptions = subjectDids.map(\n      (did): PlainMessage<ActivitySubscription> => {\n        const subject = bySubject.get(did)\n        if (!subject) {\n          return {\n            actorDid,\n            namespace:\n              Namespaces.AppBskyNotificationDefsSubjectActivitySubscription\n                .$type,\n            key: '',\n            post: undefined,\n            reply: undefined,\n            subjectDid: '',\n          }\n        }\n\n        return {\n          actorDid,\n          namespace:\n            Namespaces.AppBskyNotificationDefsSubjectActivitySubscription.$type,\n          key: subject.key,\n          post: subject.post ? {} : undefined,\n          reply: subject.reply ? {} : undefined,\n          subjectDid: subject.subjectDid,\n        }\n      },\n    )\n\n    return {\n      subscriptions,\n    }\n  },\n\n  async getActivitySubscriptionDids(req) {\n    const { actorDid, cursor, limit } = req\n\n    let builder = db.db\n      .selectFrom('activity_subscription')\n      .selectAll()\n      .where('creator', '=', actorDid)\n\n    const { ref } = db.db.dynamic\n    const key = new StashKeyKey(ref('activity_subscription.key'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n    const page = key.page(await builder.execute(), limit)\n    const dids = page.items.map(({ subjectDid }) => subjectDid)\n    return {\n      dids,\n      cursor: page.cursor,\n    }\n  },\n})\n"]}

package/dist/data-plane/server/routes/blocks.js

@@ -34,10 +34,10 @@ export default (db) => ({
             cursor,
             keyset,
         });
-        const blocks = await builder.execute();
+        const page = keyset.page(await builder.execute(), limit);
         return {
-            blockUris: blocks.map((b) => b.uri),
-            cursor: keyset.packFromResult(blocks),
+            blockUris: page.items.map((b) => b.uri),
+            cursor: page.cursor,
         };
     },
     async getBidirectionalBlockViaList(req) {
@@ -92,10 +92,10 @@ export default (db) => ({
             cursor,
             keyset,
         });
-        const lists = await builder.execute();
+        const page = keyset.page(await builder.execute(), limit);
         return {
-            listUris: lists.map((l) => l.uri),
-            cursor: keyset.packFromResult(lists),
+            listUris: page.items.map((l) => l.uri),
+            cursor: page.cursor,
         };
     },
 });

package/dist/data-plane/server/routes/blocks.js.map

@@ -1 +1 @@
-{"version":3,"file":"blocks.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/blocks.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAA;AAE7D,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,qBAAqB,CAAC,GAAG;QAC7B,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAA;QACnC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,aAAa,CAAC;aACzB,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CACZ,EAAE,CAAC,EAAE,CAAC;YACJ,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE,QAAQ,CAAC;gBACxC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,SAAS,CAAC;aAC7C,CAAC;YACF,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE,SAAS,CAAC;gBACzC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC5C,CAAC;SACH,CAAC,CACH;aACA,KAAK,CAAC,CAAC,CAAC;aACR,SAAS,EAAE;aACX,gBAAgB,EAAE,CAAA;QAErB,OAAO;YACL,QAAQ,EAAE,GAAG,EAAE,GAAG;SACnB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAG;QACjB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,aAAa,CAAC;aACzB,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC3C,SAAS,EAAE,CAAA;QAEd,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,oBAAoB,CAAC,EACzB,GAAG,CAAC,iBAAiB,CAAC,CACvB,CAAA;QACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QACtC,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YACnC,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC;SACtC,CAAA;IACH,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,GAAG;QACpC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAA;QACnC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,YAAY,CAAC;aACxB,SAAS,CAAC,WAAW,EAAE,mBAAmB,EAAE,uBAAuB,CAAC;aACpE,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CACZ,EAAE,CAAC,EAAE,CAAC;YACJ,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,QAAQ,CAAC;gBACvC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,SAAS,CAAC;aAC3C,CAAC;YACF,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,SAAS,CAAC;gBACxC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC1C,CAAC;SACH,CAAC,CACH;aACA,KAAK,CAAC,CAAC,CAAC;aACR,SAAS,CAAC,YAAY,CAAC;aACvB,gBAAgB,EAAE,CAAA;QAErB,OAAO;YACL,OAAO,EAAE,GAAG,EAAE,UAAU;SACzB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,GAAG;QAChC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;QACjC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,YAAY,CAAC;aACxB,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC;aACjC,SAAS,EAAE;aACX,KAAK,CAAC,CAAC,CAAC;aACR,gBAAgB,EAAE,CAAA;QACrB,OAAO;YACL,YAAY,EAAE,GAAG,EAAE,GAAG;SACvB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,GAAG;QACjC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,MAAM,CAAC;aAClB,KAAK,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACpB,MAAM,CACJ,EAAE,CAAC,EAAE;aACF,UAAU,CAAC,YAAY,CAAC;aACxB,KAAK,CAAC,oBAAoB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC1C,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;aACvD,SAAS,EAAE,CACf,CACF;aACA,SAAS,CAAC,MAAM,CAAC,CAAA;QAEpB,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAA;QACxE,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QAErC,OAAO;YACL,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YACjC,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC;SACrC,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { Database } from '../db/index.js'\nimport { TimeCidKeyset, paginate } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getBidirectionalBlock(req) {\n    const { actorDid, targetDid } = req\n    const res = await db.db\n      .selectFrom('actor_block')\n      .where((eb) =>\n        eb.or([\n          eb.and([\n            eb('actor_block.creator', '=', actorDid),\n            eb('actor_block.subjectDid', '=', targetDid),\n          ]),\n          eb.and([\n            eb('actor_block.creator', '=', targetDid),\n            eb('actor_block.subjectDid', '=', actorDid),\n          ]),\n        ]),\n      )\n      .limit(1)\n      .selectAll()\n      .executeTakeFirst()\n\n    return {\n      blockUri: res?.uri,\n    }\n  },\n\n  async getBlocks(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('actor_block')\n      .where('actor_block.creator', '=', actorDid)\n      .selectAll()\n\n    const keyset = new TimeCidKeyset(\n      ref('actor_block.sortAt'),\n      ref('actor_block.cid'),\n    )\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n\n    const blocks = await builder.execute()\n    return {\n      blockUris: blocks.map((b) => b.uri),\n      cursor: keyset.packFromResult(blocks),\n    }\n  },\n\n  async getBidirectionalBlockViaList(req) {\n    const { actorDid, targetDid } = req\n    const res = await db.db\n      .selectFrom('list_block')\n      .innerJoin('list_item', 'list_item.listUri', 'list_block.subjectUri')\n      .where((eb) =>\n        eb.or([\n          eb.and([\n            eb('list_block.creator', '=', actorDid),\n            eb('list_item.subjectDid', '=', targetDid),\n          ]),\n          eb.and([\n            eb('list_block.creator', '=', targetDid),\n            eb('list_item.subjectDid', '=', actorDid),\n          ]),\n        ]),\n      )\n      .limit(1)\n      .selectAll('list_block')\n      .executeTakeFirst()\n\n    return {\n      listUri: res?.subjectUri,\n    }\n  },\n\n  async getBlocklistSubscription(req) {\n    const { actorDid, listUri } = req\n    const res = await db.db\n      .selectFrom('list_block')\n      .where('creator', '=', actorDid)\n      .where('subjectUri', '=', listUri)\n      .selectAll()\n      .limit(1)\n      .executeTakeFirst()\n    return {\n      listblockUri: res?.uri,\n    }\n  },\n\n  async getBlocklistSubscriptions(req) {\n    const { actorDid, limit, cursor } = req\n    const { ref } = db.db.dynamic\n    let builder = db.db\n      .selectFrom('list')\n      .where(({ exists }) =>\n        exists(\n          db.db\n            .selectFrom('list_block')\n            .where('list_block.creator', '=', actorDid)\n            .whereRef('list_block.subjectUri', '=', ref('list.uri'))\n            .selectAll(),\n        ),\n      )\n      .selectAll('list')\n\n    const keyset = new TimeCidKeyset(ref('list.createdAt'), ref('list.cid'))\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n    const lists = await builder.execute()\n\n    return {\n      listUris: lists.map((l) => l.uri),\n      cursor: keyset.packFromResult(lists),\n    }\n  },\n})\n"]}
+{"version":3,"file":"blocks.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/blocks.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAA;AAE7D,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,qBAAqB,CAAC,GAAG;QAC7B,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAA;QACnC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,aAAa,CAAC;aACzB,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CACZ,EAAE,CAAC,EAAE,CAAC;YACJ,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE,QAAQ,CAAC;gBACxC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,SAAS,CAAC;aAC7C,CAAC;YACF,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE,SAAS,CAAC;gBACzC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC5C,CAAC;SACH,CAAC,CACH;aACA,KAAK,CAAC,CAAC,CAAC;aACR,SAAS,EAAE;aACX,gBAAgB,EAAE,CAAA;QAErB,OAAO;YACL,QAAQ,EAAE,GAAG,EAAE,GAAG;SACnB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAG;QACjB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,aAAa,CAAC;aACzB,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC3C,SAAS,EAAE,CAAA;QAEd,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,oBAAoB,CAAC,EACzB,GAAG,CAAC,iBAAiB,CAAC,CACvB,CAAA;QACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QACxD,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,GAAG;QACpC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAA;QACnC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,YAAY,CAAC;aACxB,SAAS,CAAC,WAAW,EAAE,mBAAmB,EAAE,uBAAuB,CAAC;aACpE,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CACZ,EAAE,CAAC,EAAE,CAAC;YACJ,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,QAAQ,CAAC;gBACvC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,SAAS,CAAC;aAC3C,CAAC;YACF,EAAE,CAAC,GAAG,CAAC;gBACL,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,SAAS,CAAC;gBACxC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC1C,CAAC;SACH,CAAC,CACH;aACA,KAAK,CAAC,CAAC,CAAC;aACR,SAAS,CAAC,YAAY,CAAC;aACvB,gBAAgB,EAAE,CAAA;QAErB,OAAO;YACL,OAAO,EAAE,GAAG,EAAE,UAAU;SACzB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,GAAG;QAChC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;QACjC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,YAAY,CAAC;aACxB,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC;aACjC,SAAS,EAAE;aACX,KAAK,CAAC,CAAC,CAAC;aACR,gBAAgB,EAAE,CAAA;QACrB,OAAO;YACL,YAAY,EAAE,GAAG,EAAE,GAAG;SACvB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,GAAG;QACjC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,MAAM,CAAC;aAClB,KAAK,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CACpB,MAAM,CACJ,EAAE,CAAC,EAAE;aACF,UAAU,CAAC,YAAY,CAAC;aACxB,KAAK,CAAC,oBAAoB,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC1C,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;aACvD,SAAS,EAAE,CACf,CACF;aACA,SAAS,CAAC,MAAM,CAAC,CAAA;QAEpB,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAA;QACxE,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QAExD,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YACtC,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { Database } from '../db/index.js'\nimport { TimeCidKeyset, paginate } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getBidirectionalBlock(req) {\n    const { actorDid, targetDid } = req\n    const res = await db.db\n      .selectFrom('actor_block')\n      .where((eb) =>\n        eb.or([\n          eb.and([\n            eb('actor_block.creator', '=', actorDid),\n            eb('actor_block.subjectDid', '=', targetDid),\n          ]),\n          eb.and([\n            eb('actor_block.creator', '=', targetDid),\n            eb('actor_block.subjectDid', '=', actorDid),\n          ]),\n        ]),\n      )\n      .limit(1)\n      .selectAll()\n      .executeTakeFirst()\n\n    return {\n      blockUri: res?.uri,\n    }\n  },\n\n  async getBlocks(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('actor_block')\n      .where('actor_block.creator', '=', actorDid)\n      .selectAll()\n\n    const keyset = new TimeCidKeyset(\n      ref('actor_block.sortAt'),\n      ref('actor_block.cid'),\n    )\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n\n    const page = keyset.page(await builder.execute(), limit)\n    return {\n      blockUris: page.items.map((b) => b.uri),\n      cursor: page.cursor,\n    }\n  },\n\n  async getBidirectionalBlockViaList(req) {\n    const { actorDid, targetDid } = req\n    const res = await db.db\n      .selectFrom('list_block')\n      .innerJoin('list_item', 'list_item.listUri', 'list_block.subjectUri')\n      .where((eb) =>\n        eb.or([\n          eb.and([\n            eb('list_block.creator', '=', actorDid),\n            eb('list_item.subjectDid', '=', targetDid),\n          ]),\n          eb.and([\n            eb('list_block.creator', '=', targetDid),\n            eb('list_item.subjectDid', '=', actorDid),\n          ]),\n        ]),\n      )\n      .limit(1)\n      .selectAll('list_block')\n      .executeTakeFirst()\n\n    return {\n      listUri: res?.subjectUri,\n    }\n  },\n\n  async getBlocklistSubscription(req) {\n    const { actorDid, listUri } = req\n    const res = await db.db\n      .selectFrom('list_block')\n      .where('creator', '=', actorDid)\n      .where('subjectUri', '=', listUri)\n      .selectAll()\n      .limit(1)\n      .executeTakeFirst()\n    return {\n      listblockUri: res?.uri,\n    }\n  },\n\n  async getBlocklistSubscriptions(req) {\n    const { actorDid, limit, cursor } = req\n    const { ref } = db.db.dynamic\n    let builder = db.db\n      .selectFrom('list')\n      .where(({ exists }) =>\n        exists(\n          db.db\n            .selectFrom('list_block')\n            .where('list_block.creator', '=', actorDid)\n            .whereRef('list_block.subjectUri', '=', ref('list.uri'))\n            .selectAll(),\n        ),\n      )\n      .selectAll('list')\n\n    const keyset = new TimeCidKeyset(ref('list.createdAt'), ref('list.cid'))\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n    const page = keyset.page(await builder.execute(), limit)\n\n    return {\n      listUris: page.items.map((l) => l.uri),\n      cursor: page.cursor,\n    }\n  },\n})\n"]}

package/dist/data-plane/server/routes/bookmarks.js

@@ -16,13 +16,13 @@ export default (db) => ({
             cursor,
             limit,
         });
-        const res = await builder.execute();
+        const page = key.page(await builder.execute(), limit);
         return {
-            bookmarks: res.map((b) => ({
+            bookmarks: page.items.map((b) => ({
                 key: b.key,
                 subject: b.subjectUri,
             })),
-            cursor: key.packFromResult(res),
+            cursor: page.cursor,
         };
     },
     async getBookmarksByActorAndSubjects(req) {

package/dist/data-plane/server/routes/bookmarks.js.map

@@ -1 +1 @@
-{"version":3,"file":"bookmarks.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/bookmarks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAEjE,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AAEvC,OAAO,EAEL,sCAAsC,GACvC,MAAM,2BAA2B,CAAA;AAClC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AAEjD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,iBAAiB,CAAC,GAAG;QACzB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,UAAU,CAAC;aACtB,KAAK,CAAC,kBAAkB,EAAE,GAAG,EAAE,QAAQ,CAAC;aACxC,SAAS,EAAE,CAAA;QAEd,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAA;QAChD,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QACnC,OAAO;YACL,SAAS,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACzB,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,OAAO,EAAE,CAAC,CAAC,UAAU;aACtB,CAAC,CAAC;YACH,MAAM,EAAE,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC;SAChC,CAAA;IACH,CAAC;IAED,KAAK,CAAC,8BAA8B,CAAC,GAAG;QACtC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,GAAG,CAAA;QAE9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,sCAAsC,CAAC;gBAChD,SAAS,EAAE,EAAE;aACd,CAAC,CAAA;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,UAAU,CAAC;aACtB,KAAK,CAAC,kBAAkB,EAAE,GAAG,EAAE,QAAQ,CAAC;aACxC,KAAK,CAAC,qBAAqB,EAAE,IAAI,EAAE,IAAI,CAAC;aACxC,SAAS,EAAE;aACX,OAAO,EAAE,CAAA;QAEZ,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAA;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAA0B,EAAE;YACzD,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;YAC/B,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,GAAG,EAAE,SAAS;oBACd,UAAU,EAAE,EAAE;oBACd,UAAU,EAAE,EAAE;oBACd,SAAS,EAAE,SAAS;iBACrB,CAAA;YACH,CAAC;YAED,OAAO;gBACL,GAAG,EAAE;oBACH,QAAQ;oBACR,SAAS,EAAE,UAAU,CAAC,2BAA2B,CAAC,KAAK;oBACvD,GAAG,EAAE,QAAQ,CAAC,GAAG;iBAClB;gBACD,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;aAC5D,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,OAAO;YACL,SAAS;SACV,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import { type PlainMessage, Timestamp } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport { keyBy } from '@atproto/common'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport {\n  type Bookmark,\n  GetBookmarksByActorAndSubjectsResponse,\n} from '../../../proto/bsky_pb.js'\nimport { Namespaces } from '../../../stash.js'\nimport type { Database } from '../db/index.js'\nimport { StashKeyKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorBookmarks(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('bookmark')\n      .where('bookmark.creator', '=', actorDid)\n      .selectAll()\n\n    const key = new StashKeyKey(ref('bookmark.key'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n\n    const res = await builder.execute()\n    return {\n      bookmarks: res.map((b) => ({\n        key: b.key,\n        subject: b.subjectUri,\n      })),\n      cursor: key.packFromResult(res),\n    }\n  },\n\n  async getBookmarksByActorAndSubjects(req) {\n    const { actorDid, uris } = req\n\n    if (uris.length === 0) {\n      return new GetBookmarksByActorAndSubjectsResponse({\n        bookmarks: [],\n      })\n    }\n\n    const res = await db.db\n      .selectFrom('bookmark')\n      .where('bookmark.creator', '=', actorDid)\n      .where('bookmark.subjectUri', 'in', uris)\n      .selectAll()\n      .execute()\n\n    const byUri = keyBy(res, 'subjectUri')\n    const bookmarks = uris.map((did): PlainMessage<Bookmark> => {\n      const bookmark = byUri.get(did)\n      if (!bookmark) {\n        return {\n          ref: undefined,\n          subjectUri: '',\n          subjectCid: '',\n          indexedAt: undefined,\n        }\n      }\n\n      return {\n        ref: {\n          actorDid,\n          namespace: Namespaces.AppBskyBookmarkDefsBookmark.$type,\n          key: bookmark.key,\n        },\n        subjectUri: bookmark.subjectUri,\n        subjectCid: bookmark.subjectCid,\n        indexedAt: Timestamp.fromDate(new Date(bookmark.indexedAt)),\n      }\n    })\n\n    return {\n      bookmarks,\n    }\n  },\n})\n"]}
+{"version":3,"file":"bookmarks.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/bookmarks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAEjE,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AAEvC,OAAO,EAEL,sCAAsC,GACvC,MAAM,2BAA2B,CAAA;AAClC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AAEjD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,iBAAiB,CAAC,GAAG;QACzB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,UAAU,CAAC;aACtB,KAAK,CAAC,kBAAkB,EAAE,GAAG,EAAE,QAAQ,CAAC;aACxC,SAAS,EAAE,CAAA;QAEd,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAA;QAChD,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QACrD,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChC,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,OAAO,EAAE,CAAC,CAAC,UAAU;aACtB,CAAC,CAAC;YACH,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,8BAA8B,CAAC,GAAG;QACtC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,GAAG,CAAA;QAE9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,sCAAsC,CAAC;gBAChD,SAAS,EAAE,EAAE;aACd,CAAC,CAAA;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,EAAE;aACpB,UAAU,CAAC,UAAU,CAAC;aACtB,KAAK,CAAC,kBAAkB,EAAE,GAAG,EAAE,QAAQ,CAAC;aACxC,KAAK,CAAC,qBAAqB,EAAE,IAAI,EAAE,IAAI,CAAC;aACxC,SAAS,EAAE;aACX,OAAO,EAAE,CAAA;QAEZ,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAA;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAA0B,EAAE;YACzD,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;YAC/B,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;oBACL,GAAG,EAAE,SAAS;oBACd,UAAU,EAAE,EAAE;oBACd,UAAU,EAAE,EAAE;oBACd,SAAS,EAAE,SAAS;iBACrB,CAAA;YACH,CAAC;YAED,OAAO;gBACL,GAAG,EAAE;oBACH,QAAQ;oBACR,SAAS,EAAE,UAAU,CAAC,2BAA2B,CAAC,KAAK;oBACvD,GAAG,EAAE,QAAQ,CAAC,GAAG;iBAClB;gBACD,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;aAC5D,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,OAAO;YACL,SAAS;SACV,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import { type PlainMessage, Timestamp } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport { keyBy } from '@atproto/common'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport {\n  type Bookmark,\n  GetBookmarksByActorAndSubjectsResponse,\n} from '../../../proto/bsky_pb.js'\nimport { Namespaces } from '../../../stash.js'\nimport type { Database } from '../db/index.js'\nimport { StashKeyKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorBookmarks(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('bookmark')\n      .where('bookmark.creator', '=', actorDid)\n      .selectAll()\n\n    const key = new StashKeyKey(ref('bookmark.key'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n\n    const page = key.page(await builder.execute(), limit)\n    return {\n      bookmarks: page.items.map((b) => ({\n        key: b.key,\n        subject: b.subjectUri,\n      })),\n      cursor: page.cursor,\n    }\n  },\n\n  async getBookmarksByActorAndSubjects(req) {\n    const { actorDid, uris } = req\n\n    if (uris.length === 0) {\n      return new GetBookmarksByActorAndSubjectsResponse({\n        bookmarks: [],\n      })\n    }\n\n    const res = await db.db\n      .selectFrom('bookmark')\n      .where('bookmark.creator', '=', actorDid)\n      .where('bookmark.subjectUri', 'in', uris)\n      .selectAll()\n      .execute()\n\n    const byUri = keyBy(res, 'subjectUri')\n    const bookmarks = uris.map((did): PlainMessage<Bookmark> => {\n      const bookmark = byUri.get(did)\n      if (!bookmark) {\n        return {\n          ref: undefined,\n          subjectUri: '',\n          subjectCid: '',\n          indexedAt: undefined,\n        }\n      }\n\n      return {\n        ref: {\n          actorDid,\n          namespace: Namespaces.AppBskyBookmarkDefsBookmark.$type,\n          key: bookmark.key,\n        },\n        subjectUri: bookmark.subjectUri,\n        subjectCid: bookmark.subjectCid,\n        indexedAt: Timestamp.fromDate(new Date(bookmark.indexedAt)),\n      }\n    })\n\n    return {\n      bookmarks,\n    }\n  },\n})\n"]}

package/dist/data-plane/server/routes/drafts.js

@@ -13,15 +13,15 @@ export default (db) => ({
             cursor,
             limit,
         });
-        const res = await builder.execute();
+        const page = key.page(await builder.execute(), limit);
         return {
-            drafts: res.map((d) => ({
+            drafts: page.items.map((d) => ({
                 key: d.key,
                 payload: Buffer.from(d.payload),
                 createdAt: Timestamp.fromDate(new Date(d.createdAt)),
                 updatedAt: Timestamp.fromDate(new Date(d.updatedAt)),
             })),
-            cursor: key.packFromResult(res),
+            cursor: page.cursor,
         };
     },
 });

package/dist/data-plane/server/routes/drafts.js.map

@@ -1 +1 @@
-{"version":3,"file":"drafts.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/drafts.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAKjE,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAA;AAErD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,cAAc,CAAC,GAAG;QACtB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,OAAO,CAAC;aACnB,KAAK,CAAC,eAAe,EAAE,GAAG,EAAE,QAAQ,CAAC;aACrC,SAAS,EAAE,CAAA;QAEd,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAA;QACvD,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QAEF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QACnC,OAAO;YACL,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAA2B,EAAE,CAAC,CAAC;gBAC/C,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC/B,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACpD,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACrD,CAAC,CAAC;YACH,MAAM,EAAE,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC;SAChC,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import { type PlainMessage, Timestamp } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { DraftInfo } from '../../../proto/bsky_pb.js'\nimport type { Database } from '../db/index.js'\nimport { IsoUpdatedAtKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorDrafts(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('draft')\n      .where('draft.creator', '=', actorDid)\n      .selectAll()\n\n    const key = new IsoUpdatedAtKey(ref('draft.updatedAt'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n\n    const res = await builder.execute()\n    return {\n      drafts: res.map((d): PlainMessage<DraftInfo> => ({\n        key: d.key,\n        payload: Buffer.from(d.payload),\n        createdAt: Timestamp.fromDate(new Date(d.createdAt)),\n        updatedAt: Timestamp.fromDate(new Date(d.updatedAt)),\n      })),\n      cursor: key.packFromResult(res),\n    }\n  },\n})\n"]}
+{"version":3,"file":"drafts.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/drafts.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAKjE,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAA;AAErD,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,cAAc,CAAC,GAAG;QACtB,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAA;QACvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAE7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,OAAO,CAAC;aACnB,KAAK,CAAC,eAAe,EAAE,GAAG,EAAE,QAAQ,CAAC;aACrC,SAAS,EAAE,CAAA;QAEd,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAA;QACvD,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE;YAC9B,MAAM;YACN,KAAK;SACN,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QACrD,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAA2B,EAAE,CAAC,CAAC;gBACtD,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC/B,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACpD,SAAS,EAAE,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACrD,CAAC,CAAC;YACH,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;CACF,CAAC,CAAA","sourcesContent":["import { type PlainMessage, Timestamp } from '@bufbuild/protobuf'\nimport type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { DraftInfo } from '../../../proto/bsky_pb.js'\nimport type { Database } from '../db/index.js'\nimport { IsoUpdatedAtKey } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorDrafts(req) {\n    const { actorDid, cursor, limit } = req\n    const { ref } = db.db.dynamic\n\n    let builder = db.db\n      .selectFrom('draft')\n      .where('draft.creator', '=', actorDid)\n      .selectAll()\n\n    const key = new IsoUpdatedAtKey(ref('draft.updatedAt'))\n    builder = key.paginate(builder, {\n      cursor,\n      limit,\n    })\n\n    const page = key.page(await builder.execute(), limit)\n    return {\n      drafts: page.items.map((d): PlainMessage<DraftInfo> => ({\n        key: d.key,\n        payload: Buffer.from(d.payload),\n        createdAt: Timestamp.fromDate(new Date(d.createdAt)),\n        updatedAt: Timestamp.fromDate(new Date(d.updatedAt)),\n      })),\n      cursor: page.cursor,\n    }\n  },\n})\n"]}

package/dist/data-plane/server/routes/feed-gens.js

@@ -13,10 +13,10 @@ export default (db) => ({
             cursor,
             keyset,
         });
-        const feeds = await builder.execute();
+        const page = keyset.page(await builder.execute(), limit);
         return {
-            uris: feeds.map((f) => f.uri),
-            cursor: keyset.packFromResult(feeds),
+            uris: page.items.map((f) => f.uri),
+            cursor: page.cursor,
         };
     },
     async getSuggestedFeeds(req) {
@@ -24,19 +24,21 @@ export default (db) => ({
             .selectFrom('suggested_feed')
             .orderBy('suggested_feed.order', 'asc')
             .$if(!!req.cursor, (q) => q.where('order', '>', parseInt(req.cursor, 10)))
-            .limit(req.limit || 50)
+            .limit((req.limit || 50) + 1)
             .selectAll()
             .execute();
+        const limit = req.limit || 50;
+        const items = feeds.slice(0, limit);
         return {
-            uris: feeds.map((f) => f.uri),
-            cursor: feeds.at(-1)?.order.toString(),
+            uris: items.map((f) => f.uri),
+            cursor: feeds.length > limit ? items.at(-1)?.order.toString() : undefined,
         };
     },
     async searchFeedGenerators(req) {
         return searchFeedGeneratorsImpl(db, req.query, req.limit);
     },
     async searchFeedGeneratorsV2(req) {
-        const { uris, cursor } = await searchFeedGeneratorsImpl(db, req.params?.query ?? '', req.params?.limit ?? 25);
+        const { uris, cursor } = await searchFeedGeneratorsImpl(db, req.params?.query ?? '', req.params?.limit ?? 25, req.params?.cursor);
         return {
             feedGenerators: uris.map((uri) => ({ uri, score: 0 })),
             pageInfo: { cursor: cursor ?? '', hitsTotal: 0n },
@@ -46,7 +48,7 @@ export default (db) => ({
         throw new Error('unimplemented');
     },
 });
-const searchFeedGeneratorsImpl = async (db, query, limit) => {
+const searchFeedGeneratorsImpl = async (db, query, limit, cursor) => {
     const { ref } = db.db.dynamic;
     const trimmed = query.trim();
     let builder = db.db
@@ -54,11 +56,11 @@ const searchFeedGeneratorsImpl = async (db, query, limit) => {
         .$if(!!trimmed, (q) => q.where('displayName', 'ilike', `%${trimmed}%`))
         .selectAll();
     const keyset = new TimeCidKeyset(ref('feed_generator.createdAt'), ref('feed_generator.cid'));
-    builder = paginate(builder, { limit, keyset });
-    const feeds = await builder.execute();
+    builder = paginate(builder, { limit, cursor, keyset });
+    const page = keyset.page(await builder.execute(), limit);
     return {
-        uris: feeds.map((f) => f.uri),
-        cursor: keyset.packFromResult(feeds),
+        uris: page.items.map((f) => f.uri),
+        cursor: page.cursor,
     };
 };
 //# sourceMappingURL=feed-gens.js.map

package/dist/data-plane/server/routes/feed-gens.js.map

@@ -1 +1 @@
-{"version":3,"file":"feed-gens.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/feed-gens.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAA;AAE7D,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,aAAa,CAAC,GAAG;QACrB,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;QAEvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,gBAAgB,CAAC;aAC5B,SAAS,EAAE;aACX,KAAK,CAAC,wBAAwB,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAA;QAEjD,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,0BAA0B,CAAC,EAC/B,GAAG,CAAC,oBAAoB,CAAC,CAC1B,CAAA;QACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;QAErC,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAC7B,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC;SACrC,CAAA;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,GAAG;QACzB,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,EAAE;aACtB,UAAU,CAAC,gBAAgB,CAAC;aAC5B,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAAC;aACtC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;aACzE,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;aACtB,SAAS,EAAE;aACX,OAAO,EAAE,CAAA;QACZ,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAC7B,MAAM,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE;SACvC,CAAA;IACH,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,GAAG;QAC5B,OAAO,wBAAwB,CAAC,EAAE,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,GAAG;QAC9B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,wBAAwB,CACrD,EAAE,EACF,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,EACvB,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,CACxB,CAAA;QACD,OAAO;YACL,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACtD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;SAClD,CAAA;IACH,CAAC;IAED,KAAK,CAAC,sBAAsB;QAC1B,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAA;IAClC,CAAC;CACF,CAAC,CAAA;AAEF,MAAM,wBAAwB,GAAG,KAAK,EACpC,EAAY,EACZ,KAAa,EACb,KAAa,EACb,EAAE;IACF,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;IAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAA;IAC5B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;SAChB,UAAU,CAAC,gBAAgB,CAAC;SAC5B,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,EAAE,IAAI,OAAO,GAAG,CAAC,CAAC;SACtE,SAAS,EAAE,CAAA;IACd,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,0BAA0B,CAAC,EAC/B,GAAG,CAAC,oBAAoB,CAAC,CAC1B,CAAA;IACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA;IAC9C,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;IACrC,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;QAC7B,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC;KACrC,CAAA;AACH,CAAC,CAAA","sourcesContent":["import type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { Database } from '../db/index.js'\nimport { TimeCidKeyset, paginate } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorFeeds(req) {\n    const { actorDid, limit, cursor } = req\n\n    const { ref } = db.db.dynamic\n    let builder = db.db\n      .selectFrom('feed_generator')\n      .selectAll()\n      .where('feed_generator.creator', '=', actorDid)\n\n    const keyset = new TimeCidKeyset(\n      ref('feed_generator.createdAt'),\n      ref('feed_generator.cid'),\n    )\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n    const feeds = await builder.execute()\n\n    return {\n      uris: feeds.map((f) => f.uri),\n      cursor: keyset.packFromResult(feeds),\n    }\n  },\n\n  async getSuggestedFeeds(req) {\n    const feeds = await db.db\n      .selectFrom('suggested_feed')\n      .orderBy('suggested_feed.order', 'asc')\n      .$if(!!req.cursor, (q) => q.where('order', '>', parseInt(req.cursor, 10)))\n      .limit(req.limit || 50)\n      .selectAll()\n      .execute()\n    return {\n      uris: feeds.map((f) => f.uri),\n      cursor: feeds.at(-1)?.order.toString(),\n    }\n  },\n\n  async searchFeedGenerators(req) {\n    return searchFeedGeneratorsImpl(db, req.query, req.limit)\n  },\n\n  async searchFeedGeneratorsV2(req) {\n    const { uris, cursor } = await searchFeedGeneratorsImpl(\n      db,\n      req.params?.query ?? '',\n      req.params?.limit ?? 25,\n    )\n    return {\n      feedGenerators: uris.map((uri) => ({ uri, score: 0 })),\n      pageInfo: { cursor: cursor ?? '', hitsTotal: 0n },\n    }\n  },\n\n  async getFeedGeneratorStatus() {\n    throw new Error('unimplemented')\n  },\n})\n\nconst searchFeedGeneratorsImpl = async (\n  db: Database,\n  query: string,\n  limit: number,\n) => {\n  const { ref } = db.db.dynamic\n  const trimmed = query.trim()\n  let builder = db.db\n    .selectFrom('feed_generator')\n    .$if(!!trimmed, (q) => q.where('displayName', 'ilike', `%${trimmed}%`))\n    .selectAll()\n  const keyset = new TimeCidKeyset(\n    ref('feed_generator.createdAt'),\n    ref('feed_generator.cid'),\n  )\n  builder = paginate(builder, { limit, keyset })\n  const feeds = await builder.execute()\n  return {\n    uris: feeds.map((f) => f.uri),\n    cursor: keyset.packFromResult(feeds),\n  }\n}\n"]}
+{"version":3,"file":"feed-gens.js","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/feed-gens.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAA;AAE7D,eAAe,CAAC,EAAY,EAAwC,EAAE,CAAC,CAAC;IACtE,KAAK,CAAC,aAAa,CAAC,GAAG;QACrB,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,GAAG,CAAA;QAEvC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;QAC7B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;aAChB,UAAU,CAAC,gBAAgB,CAAC;aAC5B,SAAS,EAAE;aACX,KAAK,CAAC,wBAAwB,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAA;QAEjD,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,0BAA0B,CAAC,EAC/B,GAAG,CAAC,oBAAoB,CAAC,CAC1B,CAAA;QACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE;YAC1B,KAAK;YACL,MAAM;YACN,MAAM;SACP,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;QAExD,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAClC,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAA;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,GAAG;QACzB,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,EAAE;aACtB,UAAU,CAAC,gBAAgB,CAAC;aAC5B,OAAO,CAAC,sBAAsB,EAAE,KAAK,CAAC;aACtC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;aACzE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC;aAC5B,SAAS,EAAE;aACX,OAAO,EAAE,CAAA;QACZ,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,EAAE,CAAA;QAC7B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;QACnC,OAAO;YACL,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;YAC7B,MAAM,EAAE,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;SAC1E,CAAA;IACH,CAAC;IAED,KAAK,CAAC,oBAAoB,CAAC,GAAG;QAC5B,OAAO,wBAAwB,CAAC,EAAE,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;IAC3D,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,GAAG;QAC9B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,wBAAwB,CACrD,EAAE,EACF,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,EACvB,GAAG,CAAC,MAAM,EAAE,KAAK,IAAI,EAAE,EACvB,GAAG,CAAC,MAAM,EAAE,MAAM,CACnB,CAAA;QACD,OAAO;YACL,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACtD,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;SAClD,CAAA;IACH,CAAC;IAED,KAAK,CAAC,sBAAsB;QAC1B,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAA;IAClC,CAAC;CACF,CAAC,CAAA;AAEF,MAAM,wBAAwB,GAAG,KAAK,EACpC,EAAY,EACZ,KAAa,EACb,KAAa,EACb,MAAe,EACf,EAAE;IACF,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,OAAO,CAAA;IAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAA;IAC5B,IAAI,OAAO,GAAG,EAAE,CAAC,EAAE;SAChB,UAAU,CAAC,gBAAgB,CAAC;SAC5B,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,EAAE,OAAO,EAAE,IAAI,OAAO,GAAG,CAAC,CAAC;SACtE,SAAS,EAAE,CAAA;IACd,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,0BAA0B,CAAC,EAC/B,GAAG,CAAC,oBAAoB,CAAC,CAC1B,CAAA;IACD,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;IACtD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAA;IACxD,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;QAClC,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAA;AACH,CAAC,CAAA","sourcesContent":["import type { ServiceImpl } from '@connectrpc/connect'\nimport type { Service } from '../../../proto/bsky_connect.js'\nimport type { Database } from '../db/index.js'\nimport { TimeCidKeyset, paginate } from '../db/pagination.js'\n\nexport default (db: Database): Partial<ServiceImpl<typeof Service>> => ({\n  async getActorFeeds(req) {\n    const { actorDid, limit, cursor } = req\n\n    const { ref } = db.db.dynamic\n    let builder = db.db\n      .selectFrom('feed_generator')\n      .selectAll()\n      .where('feed_generator.creator', '=', actorDid)\n\n    const keyset = new TimeCidKeyset(\n      ref('feed_generator.createdAt'),\n      ref('feed_generator.cid'),\n    )\n    builder = paginate(builder, {\n      limit,\n      cursor,\n      keyset,\n    })\n    const page = keyset.page(await builder.execute(), limit)\n\n    return {\n      uris: page.items.map((f) => f.uri),\n      cursor: page.cursor,\n    }\n  },\n\n  async getSuggestedFeeds(req) {\n    const feeds = await db.db\n      .selectFrom('suggested_feed')\n      .orderBy('suggested_feed.order', 'asc')\n      .$if(!!req.cursor, (q) => q.where('order', '>', parseInt(req.cursor, 10)))\n      .limit((req.limit || 50) + 1)\n      .selectAll()\n      .execute()\n    const limit = req.limit || 50\n    const items = feeds.slice(0, limit)\n    return {\n      uris: items.map((f) => f.uri),\n      cursor: feeds.length > limit ? items.at(-1)?.order.toString() : undefined,\n    }\n  },\n\n  async searchFeedGenerators(req) {\n    return searchFeedGeneratorsImpl(db, req.query, req.limit)\n  },\n\n  async searchFeedGeneratorsV2(req) {\n    const { uris, cursor } = await searchFeedGeneratorsImpl(\n      db,\n      req.params?.query ?? '',\n      req.params?.limit ?? 25,\n      req.params?.cursor,\n    )\n    return {\n      feedGenerators: uris.map((uri) => ({ uri, score: 0 })),\n      pageInfo: { cursor: cursor ?? '', hitsTotal: 0n },\n    }\n  },\n\n  async getFeedGeneratorStatus() {\n    throw new Error('unimplemented')\n  },\n})\n\nconst searchFeedGeneratorsImpl = async (\n  db: Database,\n  query: string,\n  limit: number,\n  cursor?: string,\n) => {\n  const { ref } = db.db.dynamic\n  const trimmed = query.trim()\n  let builder = db.db\n    .selectFrom('feed_generator')\n    .$if(!!trimmed, (q) => q.where('displayName', 'ilike', `%${trimmed}%`))\n    .selectAll()\n  const keyset = new TimeCidKeyset(\n    ref('feed_generator.createdAt'),\n    ref('feed_generator.cid'),\n  )\n  builder = paginate(builder, { limit, cursor, keyset })\n  const page = keyset.page(await builder.execute(), limit)\n  return {\n    uris: page.items.map((f) => f.uri),\n    cursor: page.cursor,\n  }\n}\n"]}

package/dist/data-plane/server/routes/feeds.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"feeds.d.ts","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/feeds.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AACtD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAA;AAE7D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAA;;0BAG9B,EAAE,EAAE,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,OAAO,CAAC,CAAC"}
+{"version":3,"file":"feeds.d.ts","sourceRoot":"","sources":["../../../../src/data-plane/server/routes/feeds.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AACtD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAA;AAE7D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAA;;0BAG9B,EAAE,EAAE,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,OAAO,CAAC,CAAC"}

package/dist/data-plane/server/routes/feeds.js

@@ -1,3 +1,4 @@
+import assert from 'node:assert';
 import { FeedType } from '../../../proto/bsky_pb.js';
 import { TimeCidKeyset, paginate } from '../db/pagination.js';
 export default (db) => ({
@@ -52,10 +53,10 @@ export default (db) => ({
             cursor,
             keyset,
         });
-        const feedItems = await builder.execute();
+        const page = keyset.page(await builder.execute(), limit);
         return {
-            items: feedItems.map(feedItemFromRow),
-            cursor: keyset.packFromResult(feedItems),
+            items: page.items.map(feedItemFromRow),
+            cursor: page.cursor,
         };
     },
     async getTimeline(req) {
@@ -78,7 +79,7 @@ export default (db) => ({
             .where('feed_item.originatorDid', '=', actorDid)
             .selectAll('feed_item');
         selfQb = paginate(selfQb, {
-            limit: Math.min(limit, 10),
+            limit,
             cursor,
             keyset,
             tryIndex: true,
@@ -87,19 +88,39 @@ export default (db) => ({
             followQb.execute(),
             selfQb.execute(),
         ]);
-        const feedItems = [...followRes, ...selfRes]
-            .sort((a, b) => {
+        // Use own posts to fill space left by followed-account posts. When followed-account
+        // posts already fill the page, still allow up to 10 own posts into the merged results.
+        const selfLimit = Math.max(
+        // Allow up to 10 own posts even when followed-account posts can fill the page.
+        Math.min(limit, 10), 
+        // Allow enough own posts to fill the slots not occupied by followed-account posts.
+        limit - Math.min(followRes.length, limit));
+        // The extra own post proves that another page exists, but is not part of this page.
+        const selfHasMore = selfRes.length > selfLimit;
+        const selfItems = selfRes.slice(0, selfLimit);
+        const feedItems = [...followRes, ...selfItems].sort((a, b) => {
             if (a.sortAt > b.sortAt)
                 return -1;
             if (a.sortAt < b.sortAt)
                 return 1;
             return a.cid > b.cid ? -1 : 1;
-        })
-            .slice(0, limit);
-        return {
-            items: feedItems.map(feedItemFromRow),
-            cursor: keyset.packFromResult(feedItems),
-        };
+        });
+        const page = keyset.page(feedItems, limit);
+        if (page.items.length === 0) {
+            return { items: [], cursor: undefined };
+        }
+        const items = page.items.map(feedItemFromRow);
+        // The combined results exceeded the requested limit.
+        if (page.cursor) {
+            return { items, cursor: page.cursor };
+        }
+        // Own posts exceeded their separate cap, but the combined results did not exceed the requested limit.
+        if (selfHasMore) {
+            const lastItem = page.items.at(-1);
+            assert(lastItem);
+            return { items, cursor: keyset.packFromResult(lastItem) };
+        }
+        return { items, cursor: undefined };
     },
     async getListFeed(req) {
         const { listUri, cursor, limit } = req;
@@ -116,10 +137,10 @@ export default (db) => ({
             keyset,
             tryIndex: true,
         });
-        const feedItems = await builder.execute();
+        const page = keyset.page(await builder.execute(), limit);
         return {
-            items: feedItems.map((item) => ({ uri: item.uri, cid: item.cid })),
-            cursor: keyset.packFromResult(feedItems),
+            items: page.items.map((item) => ({ uri: item.uri, cid: item.cid })),
+            cursor: page.cursor,
         };
     },
 });