feat: add polling to campaign query and improve cache invalidation across crowdfunding hooks

Author: 0xdevcollins

app/me/crowdfunding/[slug]/milestones/[milestoneIndex]/page.tsx

@@ -1,6 +1,7 @@
 'use client';
 
 import { use } from 'react';
+import { useRouter } from 'next/navigation';
 import { uploadMilestoneDocuments } from '@/lib/api/upload';
 import {
   useCampaign,
@@ -26,6 +27,7 @@ interface PageProps {
 export default function MilestoneDetailPage({ params }: PageProps) {
   const { slug, milestoneIndex: milestoneParam } = use(params);
 
+  const router = useRouter();
   const { data: campaign, isLoading, error } = useCampaign(slug);
 
   const milestones = campaign?.milestones ?? [];
@@ -77,6 +79,7 @@ export default function MilestoneDetailPage({ params }: PageProps) {
           toast.success(
             'Evidence submitted. The team will review your submission.'
           );
+          router.push(`/me/crowdfunding/${campaign!.slug}/milestones`);
         },
         onError: err => {
           toast.error(

features/crowdfunding/api/keys.ts

@@ -13,7 +13,11 @@ export const crowdfundingKeys = {
   mine: (page: number, limit: number) =>
     [...crowdfundingKeys.all, 'mine', { page, limit }] as const,
 
-  // Single campaign (by ID or slug - same cache slot since slug is unique)
+  // Covers ALL single-campaign entries regardless of whether they were keyed by
+  // UUID or slug. Use this for invalidation when you only have one form of the key.
+  campaignPrefix: () => [...crowdfundingKeys.all, 'campaign'] as const,
+
+  // Single campaign (by ID or slug — separate cache entries, same data)
   campaign: (idOrSlug: string) =>
     [...crowdfundingKeys.all, 'campaign', idOrSlug] as const,
 

features/crowdfunding/api/use-campaign.ts

@@ -71,6 +71,16 @@ export function useCampaign(idOrSlug: string | null | undefined) {
     queryFn: () =>
       isId ? fetchCampaignById(idOrSlug!) : fetchCampaignBySlug(idOrSlug!),
     staleTime: 30_000,
+    // Poll during transient states so the UI self-heals without a manual refresh:
+    // PUBLISHING — fast poll until the on-chain tx confirms and status flips to FUNDING.
+    // VOTING / FUNDING — slower poll to pick up vote counts and contributions from other users.
+    refetchInterval: query => {
+      const status = (query.state.data as CrowdfundingCampaign | undefined)
+        ?.v2Status;
+      if (status === 'PUBLISHING') return 5_000;
+      if (status === 'VOTING' || status === 'FUNDING') return 30_000;
+      return false;
+    },
   });
 }
 
@@ -106,10 +116,8 @@ export function useReviseAndResubmit() {
   const queryClient = useQueryClient();
   return useMutation({
     mutationFn: (id: string) => reviseAndResubmit(id),
-    onSuccess: (_data, id) => {
-      queryClient.invalidateQueries({
-        queryKey: crowdfundingKeys.campaign(id),
-      });
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: crowdfundingKeys.all });
     },
   });
 }
@@ -155,9 +163,11 @@ export function useContributeV2() {
   return useMutation({
     mutationFn: ({ id, body }: { id: string; body: ContributeV2Body }) =>
       contributeV2(id, body),
-    onSuccess: (_data, { id }) => {
+    // Invalidate all campaign entries (UUID-keyed AND slug-keyed) so both the
+    // builder page and the public page update without a manual refresh.
+    onSuccess: () => {
       queryClient.invalidateQueries({
-        queryKey: crowdfundingKeys.campaign(id),
+        queryKey: crowdfundingKeys.campaignPrefix(),
       });
     },
   });
@@ -168,12 +178,11 @@ export function useCastVote() {
   return useMutation({
     mutationFn: ({ id, choice }: { id: string; choice: 'UP' | 'DOWN' }) =>
       castVote(id, choice),
-    onSuccess: (_data, { id }) => {
-      queryClient.invalidateQueries({
-        queryKey: crowdfundingKeys.campaign(id),
-      });
+    // campaignPrefix() covers ['crowdfunding','campaign'] which matches every
+    // per-campaign entry — slug-keyed, UUID-keyed, and my-vote sub-keys alike.
+    onSuccess: () => {
       queryClient.invalidateQueries({
-        queryKey: [...crowdfundingKeys.campaign(id), 'my-vote'],
+        queryKey: crowdfundingKeys.campaignPrefix(),
       });
     },
   });

features/crowdfunding/api/use-milestone.ts

@@ -75,11 +75,16 @@ export function useSubmitMilestoneEvidence(campaignId: string) {
       });
     },
     onSuccess: (_data, { milestoneId }) => {
+      // Bust the single-milestone entry and ALL campaign entries regardless of
+      // whether they were cached by UUID or by slug (public page uses slug).
       queryClient.invalidateQueries({
         queryKey: crowdfundingKeys.milestone(campaignId, milestoneId),
       });
       queryClient.invalidateQueries({
-        queryKey: crowdfundingKeys.campaign(campaignId),
+        queryKey: crowdfundingKeys.milestones(campaignId),
+      });
+      queryClient.invalidateQueries({
+        queryKey: crowdfundingKeys.campaignPrefix(),
       });
     },
   });