fix(code-review): refresh branch/PR review diff on code updates

Generated-By: PostHog Code
Task-Id: 6fa58d62-11d1-4326-aada-39af904ddb51

Author: MattPua

packages/ui/src/features/code-review/components/ReviewPage.tsx

@@ -428,13 +428,23 @@ function RemoteReviewPage({
   const taskId = task.id;
   const isBranch = effectiveSource === "branch";
 
-  const { data: branchFiles = EMPTY_CHANGED_FILES, isLoading: branchLoading } =
-    useLocalBranchChangedFiles(
-      isBranch && isReviewOpen ? repoPath : null,
-      isBranch && isReviewOpen ? branch : null,
-    );
-  const { data: prFiles = EMPTY_CHANGED_FILES, isLoading: prLoading } =
-    usePrChangedFiles(!isBranch && isReviewOpen ? prUrl : null);
+  const {
+    data: branchFiles = EMPTY_CHANGED_FILES,
+    isLoading: branchLoading,
+    refetch: refetchBranch,
+  } = useLocalBranchChangedFiles(
+    isBranch && isReviewOpen ? repoPath : null,
+    isBranch && isReviewOpen ? branch : null,
+  );
+  const {
+    data: prFiles = EMPTY_CHANGED_FILES,
+    isLoading: prLoading,
+    refetch: refetchPr,
+  } = usePrChangedFiles(!isBranch && isReviewOpen ? prUrl : null);
+
+  const onRefresh = useCallback(() => {
+    void (isBranch ? refetchBranch() : refetchPr());
+  }, [isBranch, refetchBranch, refetchPr]);
 
   const files = isBranch ? branchFiles : prFiles;
   const isLoading = isBranch
@@ -479,6 +489,7 @@ function RemoteReviewPage({
       onExpandAll={reviewState.expandAll}
       onCollapseAll={reviewState.collapseAll}
       onUncollapseFile={reviewState.uncollapseFile}
+      onRefresh={onRefresh}
       effectiveSource={effectiveSource}
       branchSourceAvailable={branchSourceAvailable}
       prSourceAvailable={prSourceAvailable}

packages/ui/src/features/git-interaction/gitCacheKeys.ts

@@ -52,6 +52,10 @@ export function invalidateGitBranchQueries(repoPath: string) {
   queryClient.invalidateQueries(
     provider.gitPathFilter("getLocalBranchChangedFiles"),
   );
+  queryClient.invalidateQueries(provider.gitPathFilter("getPrChangedFiles"));
+  queryClient.invalidateQueries(
+    provider.gitPathFilter("getBranchChangedFiles"),
+  );
 }
 
 export function clearGitReviewQueries() {