✨ feat(vscode): 代码阅读的函数和类可以看调用树

vscode/src/codeReader/editor/messages.ts

@@ -5,6 +5,7 @@
 import type { ClaudeImage } from '../../cc/spawnClaude'
 import type { FileAttribute } from '../attributes'
 import type { BlockView } from '../blockView'
+import type { CallTree } from '../callTree/model'
 import type { DocumentDiff } from '../diff/previewDiff'
 import type { ShownName } from '../nameResolution'
 import type { ProfileChoice } from '../qa/askProfiles'
@@ -129,6 +130,8 @@ export interface StickyScrollSettings {
  * repoLinkBroken：这个仓库链接打不开。没有 symbol 是文件或目录没有；有 symbol 是文件在、这个符号没有；没有 path 是不带路径的引用找不到
  * repoLinkOpened：这个仓库链接打开了。路径在；带了 symbol 时这个符号也在；没有 path 是不带路径的引用找到了
  * repoImagesChecked：id 那次 checkRepoImages 的回答。src 是预览里能直接用的图片地址；没有 src 是文件不在
+ * callTreeCounts：version 这份文档里，每个函数、类落在几棵调用树里（只包括大于 0 的）。不是页面上这个版本的不理
+ * callTree：anchor 那块的调用树。trees 空着是没有调用它的地方。不是页面上这个版本的不理
  */
 export type ExtensionToWebview
   = | DocumentMessage
@@ -153,6 +156,8 @@ export type ExtensionToWebview
     | { type: 'repoLinkOpened', path?: string, symbol?: string }
     | { type: 'repoImagesChecked', id: number, images: { path: string, src?: string }[] }
     | { type: 'focusPathAsk' }
+    | { type: 'callTreeCounts', version: number, counts: Record<string, number> }
+    | { type: 'callTree', version: number, anchor: string, name: string, where: string, trees: CallTree[] }
 
 /**
  * edit：把 startLine..endLine 这几行整行替换成 text（扩展端给每行补回原来的缩进）
@@ -178,6 +183,7 @@ export type ExtensionToWebview
  *   没有 path 是不带路径的引用（`[Name]`、`[Class.member]`），按这份文件找 symbol（本文件定义的，再是导入的，见 symbolLinks.ts）
  * checkRepoLinks：预览里这批仓库路径问哪些文件或目录没有，这批不带路径的引用（symbols）问哪些找不到。id 标出是哪一次问
  * checkRepoImages：预览里这批仓库图片，问哪些文件在，在的要预览里能直接用的地址。id 标出是哪一次问
+ * openCallTree：打开 anchor 那块的调用树。version 是这块所在的文档版本，对不上就不查
  */
 export type WebviewToExtension
   = | { type: 'ready' }
@@ -201,3 +207,4 @@ export type WebviewToExtension
     | { type: 'openRepoLink', path?: string, symbol?: string }
     | { type: 'checkRepoLinks', id: number, paths: string[], symbols: string[] }
     | { type: 'checkRepoImages', id: number, paths: string[] }
+    | { type: 'openCallTree', version: number, anchor: string }

vscode/src/codeReader/editor/previewEditor.ts

@@ -23,6 +23,8 @@ import { commands, EndOfLine, Position, Range, Uri, window, workspace, Workspace
 import { logger } from '../../logging/logger'
 import { webviewHtml, webviewUiRoot } from '../../webview/html'
 import { DocumentCallees } from '../calleeKinds'
+import { loadCallEdges } from '../callTree/query'
+import { CallTreeSession } from '../callTree/session'
 import { findBlock, rewriteBlockComment } from '../commentRewrite'
 import { PreviewDiff } from '../diff/previewDiff'
 import { blockReplacement } from '../editText'
@@ -187,6 +189,20 @@ class Preview {
   private readonly rewriteStops = new Map<string, AbortController>()
   /** 最近一次发出去的文档（版本、块树、源码）：问 AI 时按块的记号在里面找那块代码，对比基准变了时拿它重新比 */
   private sent: SentDocument | undefined
+  /** 函数、类落在几棵调用树里。文档发出去之后再数，不挡住第一次预览 */
+  private readonly callTrees = new CallTreeSession(
+    message => this.post(message),
+    () => {
+      if (!this.sent || this.sent.version !== this.document.version)
+        return undefined
+      return {
+        version: this.sent.version,
+        blocks: this.sent.blocks,
+        load: (offset, cancelled) => loadCallEdges(this.document, offset, cancelled),
+      }
+    },
+  )
+
   /** 最近一次发出去的文档里语言插件自己认得出的名字（LanguageAdapter.symbols）：语言扩展查不到定义、各处出现时用 */
   private symbols: { version: number, list: LocalSymbol[] } | undefined
   /** 最近一次发出去的文档里各条导入引进来的名字：注释里不带路径的引用按它找导入的定义 */
@@ -318,6 +334,7 @@ class Preview {
     if (!this.ready)
       return
     const request = ++this.latest
+    this.callTrees.cancel()
     if (pathPreviewOf(this.document.uri)) {
       this.sent = undefined
       const reveal = this.pendingReveal
@@ -359,6 +376,7 @@ class Preview {
       this.names.resolve(version, report => resolveNames(displayNames, message.blocks, this.nameSources(), report))
       this.lookups.resolve(version, session.requests(), this.lookupSources())
       this.callees.resolve(version, callees.requests(), offset => this.calleeKind(offset))
+      this.callTrees.refresh()
     }
   }
 
@@ -455,6 +473,7 @@ class Preview {
   }
 
   dispose(): void {
+    this.callTrees.dispose()
     this.rewriteAbort.abort()
     clearTimeout(this.timer)
     this.names.dispose()
@@ -608,6 +627,9 @@ class Preview {
       case 'checkRepoImages':
         await this.checkRepoImages(msg.id, msg.paths)
         return
+      case 'openCallTree':
+        this.callTrees.open(msg.version, msg.anchor)
+        return
       case 'occurrenceNavigation':
         this.owner.navigation.update(this, msg.available)
     }