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 && (