fix: pass team id to vanilla trpc client

apps/web/src/app/(dashboard)/documents/data-table-action-button.tsx

@@ -50,9 +50,16 @@ export const DataTableActionButton = ({ row, team }: DataTableActionButtonProps)
   const onDownloadClick = async () => {
     try {
       const document = !recipient
-        ? await trpcClient.document.getDocumentById.query({
+        ? await trpcClient.document.getDocumentById.query(
+            {
               documentId: row.id,
-          })
+            },
+            {
+              context: {
+                teamId: team?.id?.toString(),
+              },
+            },
+          )
         : await trpcClient.document.getDocumentByToken.query({
             token: recipient.token,
           });

packages/trpc/client/index.ts

@@ -12,10 +12,32 @@ export const trpc = createTRPCClient<AppRouter>({
       true: httpLink({
         url: `${getBaseUrl()}/api/trpc`,
         transformer: SuperJSON,
+        headers: (opts) => {
+          if (typeof opts.op.context.teamId === 'string') {
+            return {
+              'x-team-id': opts.op.context.teamId,
+            };
+          }
+
+          return {};
+        },
       }),
       false: httpBatchLink({
         url: `${getBaseUrl()}/api/trpc`,
         transformer: SuperJSON,
+        headers: (opts) => {
+          const operationWithTeamId = opts.opList.find(
+            (op) => op.context.teamId && typeof op.context.teamId === 'string',
+          );
+
+          if (operationWithTeamId && typeof operationWithTeamId.context.teamId === 'string') {
+            return {
+              'x-team-id': operationWithTeamId.context.teamId,
+            };
+          }
+
+          return {};
+        },
       }),
     }),
   ],