diff --git a/apps/code/src/renderer/features/message-editor/components/DiffStatsIndicator.tsx b/apps/code/src/renderer/features/message-editor/components/DiffStatsIndicator.tsx
index 00432bcfc..fbf1c997b 100644
--- a/apps/code/src/renderer/features/message-editor/components/DiffStatsIndicator.tsx
+++ b/apps/code/src/renderer/features/message-editor/components/DiffStatsIndicator.tsx
@@ -1,16 +1,19 @@
import type { DiffStats } from "@features/git-interaction/utils/diffStats";
-import { Flex, Text } from "@radix-ui/themes";
+import { Text } from "@radix-ui/themes";
+import { useReviewNavigationStore } from "@renderer/features/code-review/stores/reviewNavigationStore";
import { useTRPC } from "@renderer/trpc";
import { useQuery } from "@tanstack/react-query";
interface DiffStatsIndicatorProps {
repoPath: string | null | undefined;
overrideStats?: DiffStats | null;
+ taskId?: string;
}
export function DiffStatsIndicator({
repoPath,
overrideStats,
+ taskId,
}: DiffStatsIndicatorProps) {
const trpc = useTRPC();
const { data: localStats } = useQuery(
@@ -26,13 +29,28 @@ export function DiffStatsIndicator({
);
const diffStats = overrideStats ?? localStats;
+ const reviewMode = useReviewNavigationStore((s) =>
+ taskId ? (s.reviewModes[taskId] ?? "closed") : "closed",
+ );
+ const setReviewMode = useReviewNavigationStore((s) => s.setReviewMode);
if (!diffStats || diffStats.filesChanged === 0) {
return null;
}
+ const handleClick = () => {
+ if (taskId) {
+ setReviewMode(taskId, reviewMode !== "closed" ? "closed" : "split");
+ }
+ };
+
return (
-
+
+
);
}
diff --git a/apps/code/src/renderer/features/message-editor/components/MessageEditor.tsx b/apps/code/src/renderer/features/message-editor/components/MessageEditor.tsx
index acf980213..7203e9049 100644
--- a/apps/code/src/renderer/features/message-editor/components/MessageEditor.tsx
+++ b/apps/code/src/renderer/features/message-editor/components/MessageEditor.tsx
@@ -104,6 +104,7 @@ function ModeAndBranchRow({
{showBranchSelector && showDiffStats && (