From 353e5d6374c77a2744590a5bfd3329672009b281 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 22 Dec 2024 16:17:22 +0000
Subject: feat: Add basic pagination to searchBookmarks tRPC

---
 packages/trpc/routers/bookmarks.ts | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

(limited to 'packages')

diff --git a/packages/trpc/routers/bookmarks.ts b/packages/trpc/routers/bookmarks.ts
index 9708706a..8a4170cd 100644
--- a/packages/trpc/routers/bookmarks.ts
+++ b/packages/trpc/routers/bookmarks.ts
@@ -515,9 +515,25 @@ export const bookmarksAppRouter = router({
     .input(
       z.object({
         text: z.string(),
+        cursor: z
+          .object({
+            offset: z.number(),
+            limit: z.number(),
+          })
+          .nullish(),
+      }),
+    )
+    .output(
+      z.object({
+        bookmarks: z.array(zBookmarkSchema),
+        nextCursor: z
+          .object({
+            offset: z.number(),
+            limit: z.number(),
+          })
+          .nullable(),
       }),
     )
-    .output(zGetBookmarksResponseSchema)
     .query(async ({ input, ctx }) => {
       const client = await getSearchIdxClient();
       if (!client) {
@@ -531,6 +547,12 @@ export const bookmarksAppRouter = router({
         showRankingScore: true,
         attributesToRetrieve: ["id"],
         sort: ["createdAt:desc"],
+        ...(input.cursor
+          ? {
+              offset: input.cursor.offset,
+              limit: input.cursor.limit,
+            }
+          : {}),
       });
 
       if (resp.hits.length == 0) {
@@ -562,7 +584,16 @@ export const bookmarksAppRouter = router({
       });
       results.sort((a, b) => idToRank[b.id] - idToRank[a.id]);
 
-      return { bookmarks: results.map(toZodSchema), nextCursor: null };
+      return {
+        bookmarks: results.map(toZodSchema),
+        nextCursor:
+          resp.hits.length + resp.offset >= resp.estimatedTotalHits
+            ? null
+            : {
+                offset: resp.hits.length + resp.offset,
+                limit: resp.limit,
+              },
+      };
     }),
   getBookmarks: authedProcedure
     .input(zGetBookmarksRequestSchema)
-- 
cgit v1.2.3-70-g09d2