feat(code): Aligning review page for local and cloud

apps/code/src/main/services/cloud-task/schemas.ts

@@ -56,6 +56,20 @@ export const sendCommandInput = z.object({
     "close",
     "permission_response",
     "set_config_option",
+    // Sandbox commands — operate on the sandbox filesystem/git directly
+    "git/changed_files",
+    "git/diff_cached",
+    "git/diff_unstaged",
+    "git/diff_head",
+    "git/diff_stats",
+    "git/current_branch",
+    "git/file_at_head",
+    "git/stage_files",
+    "git/unstage_files",
+    "git/discard_file",
+    "git/sync_status",
+    "git/repo_info",
+    "fs/read_file",
   ]),
   params: z.record(z.string(), z.unknown()).optional(),
 });

apps/code/src/renderer/features/code-review/components/DiffStatsBadge.tsx

@@ -1,64 +1,74 @@
 import { Tooltip } from "@components/ui/Tooltip";
 import { useGitQueries } from "@features/git-interaction/hooks/useGitQueries";
-import { computeDiffStats } from "@features/git-interaction/utils/diffStats";
 import { useCwd } from "@features/sidebar/hooks/useCwd";
-import { useCloudChangedFiles } from "@features/task-detail/hooks/useCloudChangedFiles";
-import { useWorkspace } from "@features/workspace/hooks/useWorkspace";
+import { useIsWorkspaceCloudRun } from "@features/workspace/hooks/useWorkspace";
 import { GitDiff } from "@phosphor-icons/react";
 import { Button } from "@posthog/quill";
 import { Flex, Text } from "@radix-ui/themes";
 import {
   formatHotkey,
   SHORTCUTS,
 } from "@renderer/constants/keyboard-shortcuts";
+import { useSandboxDiffStats } from "@renderer/features/code-review/hooks/useSandboxGit";
 import { useReviewNavigationStore } from "@renderer/features/code-review/stores/reviewNavigationStore";
 import type { Task } from "@shared/types";
-import { useMemo } from "react";
 
 interface DiffStatsBadgeProps {
   task: Task;
 }
 
-function useChangedFileStats(task: Task) {
-  const taskId = task.id;
-  const workspace = useWorkspace(taskId);
-  const isCloud =
-    workspace?.mode === "cloud" || task.latest_run?.environment === "cloud";
-  const repoPath = useCwd(taskId);
+export function DiffStatsBadge({ task }: DiffStatsBadgeProps) {
+  const isCloud = useIsWorkspaceCloudRun(task.id);
+  if (isCloud) {
+    return <CloudDiffStatsBadge task={task} />;
+  }
+  return <LocalDiffStatsBadge task={task} />;
+}
 
-  const { diffStats: localDiffStats } = useGitQueries(
-    isCloud ? undefined : repoPath,
+function LocalDiffStatsBadge({ task }: DiffStatsBadgeProps) {
+  const repoPath = useCwd(task.id);
+  const { diffStats } = useGitQueries(repoPath);
+  return (
+    <DiffStatsBadgeView
+      task={task}
+      filesChanged={diffStats.filesChanged}
+      linesAdded={diffStats.linesAdded}
+      linesRemoved={diffStats.linesRemoved}
+    />
   );
+}
 
-  const { reviewFiles } = useCloudChangedFiles(taskId, task);
-
-  return useMemo(() => {
-    if (isCloud) {
-      const stats = computeDiffStats(reviewFiles);
-      return {
-        filesChanged: stats.filesChanged,
-        linesAdded: stats.linesAdded,
-        linesRemoved: stats.linesRemoved,
-      };
-    }
-    return {
-      filesChanged: localDiffStats.filesChanged,
-      linesAdded: localDiffStats.linesAdded,
-      linesRemoved: localDiffStats.linesRemoved,
-    };
-  }, [isCloud, reviewFiles, localDiffStats]);
+function CloudDiffStatsBadge({ task }: DiffStatsBadgeProps) {
+  const { data: stats } = useSandboxDiffStats(task.id, {
+    refetchInterval: 10_000,
+  });
+  return (
+    <DiffStatsBadgeView
+      task={task}
+      filesChanged={stats?.filesChanged ?? 0}
+      linesAdded={stats?.linesAdded ?? 0}
+      linesRemoved={stats?.linesRemoved ?? 0}
+    />
+  );
 }
 
-export function DiffStatsBadge({ task }: DiffStatsBadgeProps) {
+function DiffStatsBadgeView({
+  task,
+  filesChanged,
+  linesAdded,
+  linesRemoved,
+}: DiffStatsBadgeProps & {
+  filesChanged: number;
+  linesAdded: number;
+  linesRemoved: number;
+}) {
   const taskId = task.id;
-  const { filesChanged, linesAdded, linesRemoved } = useChangedFileStats(task);
   const reviewMode = useReviewNavigationStore(
     (s) => s.reviewModes[taskId] ?? "closed",
   );
   const setReviewMode = useReviewNavigationStore((s) => s.setReviewMode);
 
   const hasChanges = filesChanged > 0;
-
   const isOpen = reviewMode !== "closed";
 
   const handleClick = () => {