From 88b9632911ee7a85c2b91bd938854f190eaeed21 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Thu, 16 Jul 2026 18:10:39 +0530
Subject: [PATCH] feat(zero): add documents.byIds query for document tabs

---
 surfsense_web/zero/queries/documents.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/surfsense_web/zero/queries/documents.ts b/surfsense_web/zero/queries/documents.ts
index d2f483989..6c110aee2 100644
--- a/surfsense_web/zero/queries/documents.ts
+++ b/surfsense_web/zero/queries/documents.ts
@@ -9,6 +9,9 @@ export const documentQueries = {
 		if (!canReadSpace(ctx, workspaceId)) return denySpace(query).orderBy("createdAt", "desc");
 		return constrainToAllowedSpaces(query, ctx).orderBy("createdAt", "desc");
 	}),
+	byIds: defineQuery(z.object({ ids: z.array(z.number()) }), ({ args: { ids }, ctx }) =>
+		constrainToAllowedSpaces(zql.documents, ctx).where("id", "IN", ids)
+	),
 };
 
 export const connectorQueries = {