✨ feat(vscode): 重写注释时在卡片下展示模型过程

vscode/webview-ui/src/codeReader/BlockTree.tsx

@@ -3,10 +3,11 @@ import type { DiffTree, FoldedRuns, PlacedRemoved } from './diffView'
 import type { GridSlots } from './PropertyGrid'
 import type { QaState } from './qa'
 import type { PendingImage } from './qaImages'
+import type { RewriteRun } from './rewriteRun'
 import type { BlockMarks, BlockView } from './types'
 import { ChevronDown, ChevronRight, ChevronsUpDown, EyeOff } from 'lucide-react'
 import { Fragment, useState } from 'react'
-import { AskThread } from './AskThread'
+import { AskThread, RewriteThread } from './AskThread'
 import { BlockTools } from './BlockTools'
 import { CodeEditor } from './CodeEditor'
 import { DiffText } from './DiffText'
@@ -37,9 +38,11 @@ export interface BlockTreeState {
   onToggleFold: (id: string) => void
   /** 一次记下几块、几行折叠还是展开（点子行展开一块、同一行里别的收起） */
   onSetFolds: (folds: Record<string, boolean>) => void
-  /** 正在按注释规范重写的注释块（记号） */
-  rewriting: ReadonlySet<string>
+  /** 按注释规范重写的过程，按块的记号。只在这一页上 */
+  rewrites: ReadonlyMap<string, RewriteRun>
   onRewriteComment: (anchor: string) => void
+  /** 叫停这块正在进行的重写，不写文件 */
+  onStopRewrite: (anchor: string) => void
   /** 问 AI；文件不在工作区文件夹里（问答没地方存）时没有 */
   qa?: QaTree
   /** 代码差异：新增、修改的块和删掉的块；侧栏开着「未提交的更改」或审查工单、基准里有这个文件时才有 */
@@ -149,6 +152,7 @@ export function BlockLevel({ blocks, owner, tree, slotted = false }: { blocks: B
                   <BlockCard block={block} tree={tree} />
                 </ChangeRow>
                 {tree.qa && <BlockQa key={block.anchor} anchor={block.anchor} qa={tree.qa} />}
+                <BlockRewrite anchor={block.anchor} tree={tree} />
                 {hasShownChildren(block, tree) && !tree.collapsed.has(block.id) && (
                   <BlockLevel blocks={block.children} owner={block} tree={tree} />
                 )}
@@ -180,6 +184,13 @@ function BlockQa({ anchor, qa }: { anchor: string, qa: QaTree }): ReactElement {
   )
 }
 
+function BlockRewrite({ anchor, tree }: { anchor: string, tree: BlockTreeState }): ReactElement | null {
+  const run = tree.rewrites.get(anchor)
+  if (!run)
+    return null
+  return <RewriteThread run={run} onStop={() => tree.onStopRewrite(anchor)} />
+}
+
 /**
  * 预览状态的内容：函数、类、枚举、导入、常量、变量块和库规则的卡片是属性表，其余是一行译文；
  * 译文下面依次接着摊开实参的表（跨几行写的调用）、说明。标差异时 marks 是这块自己的内容改了哪些（页面上正是差异算的那一版才给）。
@@ -279,7 +290,7 @@ function BlockCard({ block, tree }: { block: BlockView, tree: BlockTreeState }):
         onToggleEdit={() => tree.onToggleEdit(block.id)}
         asking={tree.qa?.state.open.includes(block.anchor)}
         onToggleAsk={tree.qa && (() => tree.qa?.onToggle(block.anchor))}
-        rewriting={tree.rewriting.has(block.anchor)}
+        rewriting={tree.rewrites.get(block.anchor)?.status === 'running'}
         onRewrite={block.kind === 'comment' ? () => tree.onRewriteComment(block.anchor) : undefined}
       />
     </div>