fix: wip

packages/trpc/server/document-router/router.ts

@@ -18,11 +18,14 @@ import { findDocuments } from '@documenso/lib/server-only/document/find-document
 import { getDocumentById } from '@documenso/lib/server-only/document/get-document-by-id';
 import { getDocumentAndSenderByToken } from '@documenso/lib/server-only/document/get-document-by-token';
 import { getDocumentWithDetailsById } from '@documenso/lib/server-only/document/get-document-with-details-by-id';
+import type { GetStatsInput } from '@documenso/lib/server-only/document/get-stats';
+import { getStats } from '@documenso/lib/server-only/document/get-stats';
 import { moveDocumentToTeam } from '@documenso/lib/server-only/document/move-document-to-team';
 import { resendDocument } from '@documenso/lib/server-only/document/resend-document';
 import { searchDocumentsWithKeyword } from '@documenso/lib/server-only/document/search-documents-with-keyword';
 import { sendDocument } from '@documenso/lib/server-only/document/send-document';
 import { updateDocument } from '@documenso/lib/server-only/document/update-document';
+import { getTeamById } from '@documenso/lib/server-only/team/get-team';
 import { symmetricEncrypt } from '@documenso/lib/universal/crypto';
 import { getPresignPostUrl } from '@documenso/lib/universal/upload/server-actions';
 
@@ -39,6 +42,8 @@ import {
   ZDuplicateDocumentRequestSchema,
   ZDuplicateDocumentResponseSchema,
   ZFindDocumentAuditLogsQuerySchema,
+  ZFindDocumentsInternalRequestSchema,
+  ZFindDocumentsInternalResponseSchema,
   ZFindDocumentsRequestSchema,
   ZFindDocumentsResponseSchema,
   ZGenericSuccessResponse,
@@ -124,6 +129,82 @@ export const documentRouter = router({
       return documents;
     }),
 
+  /**
+   * Internal endpoint for /documents page to additionally return getStats.
+   *
+   * @private
+   */
+  findDocumentsInternal: authenticatedProcedure
+    .meta({
+      openapi: {
+        method: 'GET',
+        path: '/document',
+        summary: 'Find documents',
+        description: 'Find documents based on a search criteria',
+        tags: ['Document'],
+      },
+    })
+    .input(ZFindDocumentsInternalRequestSchema)
+    .output(ZFindDocumentsInternalResponseSchema)
+    .query(async ({ input, ctx }) => {
+      const { user, teamId } = ctx;
+
+      const {
+        query,
+        templateId,
+        page,
+        perPage,
+        orderByDirection,
+        orderByColumn,
+        source,
+        status,
+        period,
+        senderIds,
+      } = input;
+
+      const getStatOptions: GetStatsInput = {
+        user,
+        period,
+        search: query,
+      };
+
+      if (teamId) {
+        const team = await getTeamById({ userId: user.id, teamId });
+
+        getStatOptions.team = {
+          teamId: team.id,
+          teamEmail: team.teamEmail?.email,
+          senderIds,
+          currentTeamMemberRole: team.currentTeamMember?.role,
+          currentUserEmail: user.email,
+          userId: user.id,
+        };
+      }
+
+      const [stats, documents] = await Promise.all([
+        getStats(getStatOptions),
+        findDocuments({
+          userId: user.id,
+          teamId,
+          templateId,
+          query,
+          source,
+          status,
+          page,
+          perPage,
+          period,
+          orderBy: orderByColumn
+            ? { column: orderByColumn, direction: orderByDirection }
+            : undefined,
+        }),
+      ]);
+
+      return {
+        ...documents,
+        stats,
+      };
+    }),
+
   /**
    * @public
    *