feat: Add ability to filter by bookmark type

2 files changed, 58 insertions, 0 deletions

diff --git a/packages/trpc/lib/__tests__/search.test.ts b/packages/trpc/lib/__tests__/search.test.ts
index 468aef83..8a6f1949 100644
--- a/packages/trpc/lib/__tests__/search.test.ts
+++ b/packages/trpc/lib/__tests__/search.test.ts
@@ -280,6 +280,54 @@ describe("getBookmarkIdsFromMatcher", () => {
     expect(result).toEqual(["b1", "b2"]);
   });
 
+  it("should handle type matcher", async () => {
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.LINK,
+        inverse: false,
+      }),
+    ).toEqual(["b1", "b2", "b4"]);
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.TEXT,
+        inverse: false,
+      }),
+    ).toEqual(["b3", "b5"]);
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.ASSET,
+        inverse: false,
+      }),
+    ).toEqual(["b6"]);
+  });
+
+  it("should handle type matcher with inverse=true", async () => {
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.LINK,
+        inverse: true,
+      }),
+    ).toEqual(["b3", "b5", "b6"]);
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.TEXT,
+        inverse: true,
+      }),
+    ).toEqual(["b1", "b2", "b4", "b6"]);
+    expect(
+      await getBookmarkIdsFromMatcher(mockCtx, {
+        type: "type",
+        typeName: BookmarkTypes.ASSET,
+        inverse: true,
+      }),
+    ).toEqual(["b1", "b2", "b3", "b4", "b5"]);
+  });
+
   it("should handle dateBefore matcher with inverse=true", async () => {
     const matcher: Matcher = {
       type: "dateBefore",
diff --git a/packages/trpc/lib/search.ts b/packages/trpc/lib/search.ts
index 74aaf51b..eac323df 100644
--- a/packages/trpc/lib/search.ts
+++ b/packages/trpc/lib/search.ts
@@ -7,6 +7,7 @@ import {
   like,
   lt,
   lte,
+  ne,
   notExists,
   notLike,
 } from "drizzle-orm";
@@ -233,6 +234,15 @@ async function getIds(
           ),
         );
     }
+    case "type": {
+      const comp = matcher.inverse
+        ? ne(bookmarks.type, matcher.typeName)
+        : eq(bookmarks.type, matcher.typeName);
+      return db
+        .select({ id: bookmarks.id })
+        .from(bookmarks)
+        .where(and(eq(bookmarks.userId, userId), comp));
+    }
     case "and": {
       const vals = await Promise.all(
         matcher.matchers.map((m) => getIds(db, userId, m)),