From a73ba551db0e29725e256d5b4a9f5288c39a5f21 Mon Sep 17 00:00:00 2001
From: iliya <iliyazelenkog@gmail.com>
Date: Sun, 1 Mar 2026 13:47:30 +0200
Subject: [PATCH] fix: render new files via CodeMirror and fix root drop
 refresh
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- FileSectionDiff: show CodeMirror for new files even when contentSource
  is 'unavailable' — write-new snippets provide the content
- ReviewDiffContent: remove duplicate file header (already in FileSectionHeader)
- editorSlice: fix refreshDirectory for project root — tree is the root's
  children array, not a nested entry, so merge must replace top-level
---
 .../team/review/FileSectionDiff.tsx           |  9 ++++----
 .../team/review/ReviewDiffContent.tsx         | 13 ------------
 src/renderer/store/slices/editorSlice.ts      | 21 ++++++++++++++++++-
 3 files changed, 25 insertions(+), 18 deletions(-)

diff --git a/src/renderer/components/team/review/FileSectionDiff.tsx b/src/renderer/components/team/review/FileSectionDiff.tsx
index 701c3b68..867d2347 100644
--- a/src/renderer/components/team/review/FileSectionDiff.tsx
+++ b/src/renderer/components/team/review/FileSectionDiff.tsx
@@ -89,8 +89,9 @@ export const FileSectionDiff = ({
       return writeSnippets[writeSnippets.length - 1].newString;
     })();
 
-  const hasCodeMirrorContent =
-    fileContent && fileContent.contentSource !== 'unavailable' && resolvedModified !== null;
+  // Show CodeMirror whenever we have resolved modified content (including new files
+  // where contentSource may be 'unavailable' but write-new snippet provides the content)
+  const hasCodeMirrorContent = resolvedModified !== null;
 
   if (!hasCodeMirrorContent) {
     return (
@@ -105,12 +106,12 @@ export const FileSectionDiff = ({
     <div className="overflow-auto">
       <DiffErrorBoundary
         filePath={file.filePath}
-        oldString={fileContent.originalFullContent ?? ''}
+        oldString={fileContent?.originalFullContent ?? ''}
         newString={resolvedModified}
       >
         <CodeMirrorDiffView
           key={`${file.filePath}:${discardCounter}`}
-          original={fileContent.originalFullContent ?? ''}
+          original={fileContent?.originalFullContent ?? ''}
           modified={resolvedModified}
           fileName={file.relativePath}
           readOnly={false}
diff --git a/src/renderer/components/team/review/ReviewDiffContent.tsx b/src/renderer/components/team/review/ReviewDiffContent.tsx
index b2e57ffc..20ccf9f6 100644
--- a/src/renderer/components/team/review/ReviewDiffContent.tsx
+++ b/src/renderer/components/team/review/ReviewDiffContent.tsx
@@ -81,19 +81,6 @@ export const ReviewDiffContent = ({ file }: ReviewDiffContentProps) => {
 
   return (
     <div className="space-y-4 p-4">
-      {/* Заголовок файла */}
-      <div className="flex items-center gap-2">
-        <span className="text-sm font-medium text-text">{file.relativePath}</span>
-        {file.isNewFile && (
-          <span className="rounded bg-green-500/20 px-1.5 py-0.5 text-[10px] text-green-400">
-            NEW
-          </span>
-        )}
-        <span className="ml-auto text-xs text-text-muted">
-          {nonErrorSnippets.length} change{nonErrorSnippets.length !== 1 ? 's' : ''}
-        </span>
-      </div>
-
       {/* Snippets */}
       {nonErrorSnippets.map((snippet, index) => (
         <SnippetDiffView key={snippet.toolUseId} snippet={snippet} index={index} />
diff --git a/src/renderer/store/slices/editorSlice.ts b/src/renderer/store/slices/editorSlice.ts
index b5b8888d..1dde315a 100644
--- a/src/renderer/store/slices/editorSlice.ts
+++ b/src/renderer/store/slices/editorSlice.ts
@@ -860,7 +860,26 @@ async function refreshDirectory(
   try {
     const result = await api.editor.readDir(dirPath);
     const currentTree = get().editorFileTree;
-    if (currentTree) {
+    if (!currentTree) return;
+
+    const projectPath = get().editorProjectPath;
+    if (dirPath === projectPath) {
+      // Root refresh — tree IS the root's children, so preserve expanded subtrees
+      // by merging new entries with existing children data
+      const existingByPath = new Map<string, FileTreeEntry>();
+      for (const entry of currentTree) {
+        existingByPath.set(entry.path, entry);
+      }
+      const merged = result.entries.map((entry) => {
+        const existing = existingByPath.get(entry.path);
+        // Preserve expanded subtree children for directories that still exist
+        if (existing?.children && entry.type === 'directory') {
+          return { ...entry, children: existing.children };
+        }
+        return entry;
+      });
+      set({ editorFileTree: merged });
+    } else {
       const updatedTree = mergeChildrenIntoTree(currentTree, dirPath, result.entries);
       set({ editorFileTree: updatedTree });
     }