diff --git a/apps/mail/components/party.tsx b/apps/mail/components/party.tsx index 954e7fff1..380daeceb 100644 --- a/apps/mail/components/party.tsx +++ b/apps/mail/components/party.tsx @@ -3,6 +3,7 @@ import { useQueryClient } from '@tanstack/react-query'; import { atom, useAtomValue, useSetAtom } from 'jotai'; import { useTRPC } from '@/providers/query-provider'; import { usePartySocket } from 'partysocket/react'; +import { useThreads } from '@/hooks/use-threads'; import { useSession } from '@/lib/auth-client'; import { useEffect } from 'react'; @@ -10,6 +11,7 @@ export const NotificationProvider = ({ headers }: { headers: Record) => { + onMessage: async (message: MessageEvent) => { const [threadId, type] = message.data.split(':'); if (type === 'end') { console.log('invalidating thread', threadId); - queryClient.invalidateQueries({ + await queryClient.invalidateQueries({ queryKey: trpc.mail.get.queryKey({ id: threadId }), }); + await refetchThreads(); } console.log('party message', threadId, type); },