🐛 fix(vscode): 注释里不带路径的引用从被注释的声明起一层层往外找，同类的方法、类的成员都能链到

vscode/src/codeReader/editor/messages.ts

@@ -18,6 +18,16 @@ export type { Excluded, ShownName }
 
 export interface TranslatorOption { id: string, label: string }
 
+/**
+ * 注释里一个不带路径的引用（`[Name]`、`[Class.member]`）：symbol 是写的名字；block 是这段注释属于哪一块（块的 anchor），
+ * 从那块代码的声明起一层层往外找（见 symbolLinks.ts）。不在哪块的注释里的没有 block，只从文件顶层找。
+ * 同一个名字写在不同块的注释里，指到的可以不一样。
+ */
+export interface SymbolReference {
+  symbol: string
+  block?: string
+}
+
 /**
  * 让预览滚到包含 line 的块。definition 只在跳到定义时有：定义处那个名字在源码里的下标（Segment.offset），
  * 预览展开把它折叠住的块和属性行，再让就是这处定义的那一块或属性表里那一行闪几下。
@@ -140,9 +150,10 @@ export interface StickyScrollSettings {
  * diff：这份文件相比对比基准改了什么（审查工作副本里的、从「未提交的更改」点开的有改动的才有，见 [对比基准](vscode/src/codeReader/diff/diffBase.ts#DiffBase)）：每发一份文档之后算好了补发，
  *   基准变了（提交后 HEAD 动了、换了审查的工单、文件有了改动、这个标签页换了从哪儿打开）再发；null 是不比了（退出了审查、改动提交了、从别处点开了）
  * openedFrom：这个标签页是从哪儿打开的，页面记进自己的状态，VS Code 重启后恢复这个标签页时随 ready 报回来；ready 后发一次，开着时又从侧栏点开、换了从哪儿打开的再发
- * repoLinksChecked：id 那次 checkRepoLinks 的回答，missing 是仓库里没有的路径，unresolved 是找不到的不带路径的引用
- * repoLinkBroken：这个仓库链接打不开。没有 symbol 是文件或目录没有；有 symbol 是文件在、这个符号没有；没有 path 是不带路径的引用找不到
- * repoLinkOpened：这个仓库链接打开了。路径在；带了 symbol 时这个符号也在；没有 path 是不带路径的引用找到了
+ * repoLinksChecked：id 那次 checkRepoLinks 的回答，missing 是仓库里没有的路径，unresolved 是找不到的不带路径的引用（原样带回问的那几条）
+ * repoLinkBroken：这个仓库链接打不开。没有 symbol 是文件或目录没有；有 symbol 是文件在、这个符号没有；没有 path 是不带路径的引用找不到，
+ *   block 同 openRepoLink
+ * repoLinkOpened：这个仓库链接打开了。路径在；带了 symbol 时这个符号也在；没有 path 是不带路径的引用找到了，block 同 openRepoLink
  * repoImagesChecked：id 那次 checkRepoImages 的回答。src 是预览里能直接用的图片地址；没有 src 是文件不在
  * callTreeCounts：version 这份文档里，每个能看调用树的块（BlockView.callTree）落在几棵调用树里（只包括大于 0 的）。不是页面上这个版本的不理
  * callTree：anchor 那块的调用树。trees 空着、又没有 contractMiss 是没有调用它的地方；接口卡片在契约文件里没对上时带 contractMiss。
@@ -167,9 +178,9 @@ export type ExtensionToWebview
     | { type: 'rewriteEnd', askId: string, status: 'done' | 'failed' | 'stopped', text?: string, error?: string, entryId?: string }
     | { type: 'diff', diff: DocumentDiff | null }
     | { type: 'openedFrom', from: OpenedFrom }
-    | { type: 'repoLinksChecked', id: number, missing: string[], unresolved: string[] }
-    | { type: 'repoLinkBroken', path?: string, symbol?: string }
-    | { type: 'repoLinkOpened', path?: string, symbol?: string }
+    | { type: 'repoLinksChecked', id: number, missing: string[], unresolved: SymbolReference[] }
+    | { type: 'repoLinkBroken', path?: string, symbol?: string, block?: string }
+    | { type: 'repoLinkOpened', path?: string, symbol?: string, block?: string }
     | { type: 'repoImagesChecked', id: number, images: { path: string, src?: string }[] }
     | { type: 'focusPathAsk' }
     | { type: 'callTreeCounts', version: number, counts: Record<string, number> }
@@ -183,7 +194,7 @@ export type ExtensionToWebview
  *   version 是这份下标所属的文档版本
  * goToDefinition：⌘ / Ctrl + 点击，跳到 offset 处名字的定义
  * peekDefinition：鼠标停在带下标的名字上，要 offset 处名字的定义卡片（几处定义取第一处）；id、version 同 probeDefinition
- * peekLink：鼠标停在注释里指到符号的链接上（带 #符号 的仓库链接、不带路径的引用），要那个符号的定义卡片；path、symbol 同 openRepoLink
+ * peekLink：鼠标停在注释里指到符号的链接上（带 #符号 的仓库链接、不带路径的引用），要那个符号的定义卡片；path、symbol、block 同 openRepoLink
  * findOccurrences：单击带下标的名字，问它在这份文档里出现在哪几处；id、version 同 probeDefinition
  * setFilter：改块筛选（整份替换），扩展端记下后发 filter 给所有预览
  * setOnlyChanges：打开 / 关掉「只看改动」，扩展端记下后发 filter 给所有预览
@@ -200,8 +211,9 @@ export type ExtensionToWebview
  * occurrenceNavigation：眼下能不能用键盘在高亮着的名字的各处出现之间跳（有高亮、页面有焦点、焦点不在输入框里），变了才发；
  *   扩展端按它设 when 子句用的上下文键
  * openRepoLink：点了注释里的仓库路径链接，打开 path（从仓库根算起）。symbol 是类名、函数名或「类名.成员名」，没有就只打开文件或在代码阅读侧栏的树上定位目录；
- *   没有 path 是不带路径的引用（`[Name]`、`[Class.member]`），按这份文件找 symbol（本文件定义的，再是导入的，见 symbolLinks.ts）
- * checkRepoLinks：预览里这批仓库路径问哪些文件或目录没有，这批不带路径的引用（symbols）问哪些找不到。id 标出是哪一次问
+ *   没有 path 是不带路径的引用（`[Name]`、`[Class.member]`），按这份文件找 symbol：block 是它写在哪块的注释里（见 SymbolReference），
+ *   从那块的声明起一层层往外找本文件定义的，再是导入的（见 symbolLinks.ts）
+ * checkRepoLinks：预览里这批仓库路径问哪些文件或目录没有，这批不带路径的引用（references）问哪些找不到。id 标出是哪一次问
  * checkRepoImages：预览里这批仓库图片，问哪些文件在，在的要预览里能直接用的地址。id 标出是哪一次问
  * openCallTree：打开 anchor 那块的调用树。version 是这块所在的文档版本，对不上就不查
  * goToCallNode：点了调用树上的一个节点，打开 uri 那个文件，跳到 line 行 character 列（都从 0 起，见 CallTreeNode.jump）
@@ -220,15 +232,15 @@ export type WebviewToExtension
     | { type: 'probeDefinition', id: number, version: number, offset: number }
     | { type: 'goToDefinition', version: number, offset: number }
     | { type: 'peekDefinition', id: number, version: number, offset: number }
-    | { type: 'peekLink', id: number, path?: string, symbol: string }
+    | { type: 'peekLink', id: number, path?: string, symbol: string, block?: string }
     | { type: 'findOccurrences', id: number, version: number, offset: number }
     | { type: 'ask', askId: string, anchor: string, question: string, images: ClaudeImage[] }
     | { type: 'stopAsk', askId: string }
     | { type: 'deleteQa', anchor: string, entryId: string }
     | { type: 'selectProfile', kind: ChatKind, id: string }
     | { type: 'occurrenceNavigation', available: boolean }
-    | { type: 'openRepoLink', path?: string, symbol?: string }
-    | { type: 'checkRepoLinks', id: number, paths: string[], symbols: string[] }
+    | { type: 'openRepoLink', path?: string, symbol?: string, block?: string }
+    | { type: 'checkRepoLinks', id: number, paths: string[], references: SymbolReference[] }
     | { type: 'checkRepoImages', id: number, paths: string[] }
     | { type: 'openCallTree', version: number, anchor: string }
     | { type: 'goToCallNode', uri: string, line: number, character: number }

vscode/src/codeReader/editor/previewEditor.test.ts

@@ -1,9 +1,13 @@
+import type { BlockView } from '../blockView'
 import type { DiffBase, OpenedFrom } from '../diff/diffBase'
 import type { ProfileChoice } from '../qa/askProfiles'
 import type { RewriteRequest } from '../qa/codeQa'
+import type { SymbolNode } from '../symbolLinks'
 import type { ExtensionToWebview, WebviewToExtension } from './messages'
 import { describe, expect, it, vi } from 'vitest'
 import { window, workspace } from 'vscode'
+import { documentMessage } from './documentMessage'
+import { documentSymbolNodes } from './documentSymbols'
 import { PreviewEditorProvider } from './previewEditor'
 
 vi.mock('vscode', () => ({
@@ -40,6 +44,8 @@ vi.mock('./documentMessage', () => ({
     imports: [],
   })),
 }))
+/** 文档符号由各个测试给 */
+vi.mock('./documentSymbols', () => ({ documentSymbolNodes: vi.fn(async () => []) }))
 
 const SOURCE = ['// 旧说明', 'export function load() {}']
 
@@ -345,3 +351,40 @@ describe('问 AI 和重写注释的 profile 下拉各管各的', () => {
     expect(page.posted()).not.toContainEqual(expect.objectContaining({ type: 'profiles', kind: 'rewrite' }))
   })
 })
+
+describe('注释里不带路径的引用按它写在哪块的注释里找', () => {
+  const METHOD = 'class AgentService > function get_default_agent'
+  const HELPER = 'function helper'
+
+  /** 一块有名字的代码：名字在源码里的下标是 at */
+  function declared(anchor: string, at: number, children: BlockView[] = []): BlockView {
+    return { id: anchor, anchor, kind: 'function', startLine: 0, endLine: 0, headerEndLine: 0, code: '', codeTokens: [], defines: { startIndex: at, endIndex: at + 1 }, attributes: { kind: 'function' }, concepts: [], children, translation: [] }
+  }
+  function symbol(name: string, start: number, end: number, children: SymbolNode[] = []): SymbolNode {
+    return { name, range: { startIndex: start, endIndex: end }, selection: { startIndex: start, endIndex: start + name.length }, children }
+  }
+
+  it('同一个名字：方法说明里的找得到同类的另一个方法，顶层函数说明里的找不到；标坏链接、点了打开都按各自那块找', async () => {
+    const blocks = [declared('class AgentService', 100, [declared(METHOD, 400)]), declared(HELPER, 950)]
+    vi.mocked(documentMessage).mockImplementationOnce(async input => ({
+      message: { type: 'document', version: input.version, languageId: input.languageId, blocks, attributes: [], translators: [], translatorId: '', names: [] },
+      displayNames: [],
+      symbols: [],
+      imports: [],
+    }) as never)
+    vi.mocked(documentSymbolNodes).mockResolvedValue([
+      symbol('AgentService', 100, 900, [symbol('get_agent_admitted', 200, 300), symbol('get_default_agent', 400, 500)]),
+      symbol('helper', 950, 990),
+    ])
+    const { provider, document } = opening()
+    const page = tab()
+    provider.resolveCustomTextEditor(document as never, page.panel)
+    await page.send({ type: 'ready' })
+
+    await page.send({ type: 'checkRepoLinks', id: 1, paths: [], references: [{ symbol: 'get_agent_admitted', block: METHOD }, { symbol: 'get_agent_admitted', block: HELPER }] })
+    expect(page.posted()).toContainEqual({ type: 'repoLinksChecked', id: 1, missing: [], unresolved: [{ symbol: 'get_agent_admitted', block: HELPER }] })
+
+    await page.send({ type: 'openRepoLink', symbol: 'get_agent_admitted', block: HELPER })
+    expect(page.posted()).toContainEqual({ type: 'repoLinkBroken', symbol: 'get_agent_admitted', block: HELPER })
+  })
+})

vscode/src/codeReader/editor/previewEditor.ts

@@ -14,7 +14,7 @@ import type { TreeSitterRuntime } from '../runtime'
 import type { ImportedName, LinkTarget, SymbolNode } from '../symbolLinks'
 import type { BlockFilter } from './blockFilter'
 import type { FoundLocation } from './definitions'
-import type { ChatKind, DefinitionPeek, ExtensionToWebview, Occurrence, Reveal, WebviewToExtension } from './messages'
+import type { ChatKind, DefinitionPeek, ExtensionToWebview, Occurrence, Reveal, SymbolReference, WebviewToExtension } from './messages'
 import type { TargetSpot } from './openTarget'
 import type { TopSubject } from './pathPreview'
 import { realpath } from 'node:fs/promises'
@@ -33,7 +33,7 @@ import { DocumentNames, resolveNames } from '../nameResolution'
 import { PATH_ASK_ANCHOR } from '../qa/pathQa'
 import { askPrompt, followUpPrompt } from '../qa/prompt'
 import { DocumentLookups } from '../ruleLookups'
-import { resolveReference } from '../symbolLinks'
+import { referenceScope, resolveReference } from '../symbolLinks'
 import { definitionAction } from './definitionAction'
 import { definitionSiteNames, namesOnCard, peekBlock } from './definitionPeek'
 import { definitionKind, findDefinitions, languageDefinitions, pickDefinition, ruleDefinitions } from './definitions'
@@ -669,16 +669,16 @@ class Preview {
         await this.peekDefinition(msg.id, msg.version, msg.offset)
         return
       case 'peekLink':
-        await this.peekLink(msg.id, msg.path, msg.symbol)
+        await this.peekLink(msg.id, msg.path, msg.symbol, msg.block)
         return
       case 'findOccurrences':
         await this.findOccurrences(msg.id, msg.version, msg.offset)
         return
       case 'openRepoLink':
-        await this.openRepoLink(msg.path, msg.symbol)
+        await this.openRepoLink(msg.path, msg.symbol, msg.block)
         return
       case 'checkRepoLinks':
-        await this.checkRepoLinks(msg.id, msg.paths, msg.symbols)
+        await this.checkRepoLinks(msg.id, msg.paths, msg.references)
         return
       case 'checkRepoImages':
         await this.checkRepoImages(msg.id, msg.paths)
@@ -770,10 +770,10 @@ class Preview {
   }
 
   /** 回答 webview 第 id 次要注释里的链接指的符号的卡片：符号同点这个链接时找的那个（symbolTarget），找不到没有。 */
-  private async peekLink(id: number, rel: string | undefined, symbol: string): Promise<void> {
+  private async peekLink(id: number, rel: string | undefined, symbol: string, block: string | undefined): Promise<void> {
     let peek: DefinitionPeek | undefined
     try {
-      const target = await this.symbolTarget(rel, symbol)
+      const target = await this.symbolTarget(rel, symbol, block)
       peek = target && await this.peekAt(Uri.parse(target.uri), target.selection.startIndex)
     }
     catch (err) {
@@ -843,8 +843,8 @@ class Preview {
     }
   }
 
-  /** 预览里这批仓库路径，哪些文件或目录不在；这批不带路径的引用，哪些找不到。 */
-  private async checkRepoLinks(id: number, paths: string[], symbols: string[]): Promise<void> {
+  /** 预览里这批仓库路径，哪些文件或目录不在；这批不带路径的引用，各按写在哪块找，哪些找不到。 */
+  private async checkRepoLinks(id: number, paths: string[], references: SymbolReference[]): Promise<void> {
     const root = await repoRootOf(this.document.uri.fsPath)
     const asked = new Map<string, Promise<SymbolNode[]>>()
     const symbolsOf = (uri: string): Promise<SymbolNode[]> => {
@@ -858,10 +858,10 @@ class Preview {
           return rel
         return undefined
       })),
-      Promise.all(symbols.map(async symbol => await this.resolveSymbolLink(symbol, symbolsOf) ? undefined : symbol)),
+      Promise.all(references.map(async reference => await this.resolveSymbolLink(reference, symbolsOf) ? undefined : reference)),
     ])
     if (this.ready)
-      this.post({ type: 'repoLinksChecked', id, missing: missing.filter(rel => rel !== undefined), unresolved: unresolved.filter(symbol => symbol !== undefined) })
+      this.post({ type: 'repoLinksChecked', id, missing: missing.filter(rel => rel !== undefined), unresolved: unresolved.filter(reference => reference !== undefined) })
   }
 
   /** 仓库根放进这个预览能读的本地目录，图片地址才能加载。真实位置和 git 给的路径不一样时两个都放。 */
@@ -905,19 +905,19 @@ class Preview {
 
   /**
    * 打开注释里的仓库链接：目录在代码阅读侧栏的树上定位，文件用编辑器打开。
-   * 带了符号就按文档符号的名字跳过去；文件或符号没有就不打开。没有路径的是不带路径的引用，按这份文件找（见 resolveSymbolLink）。
+   * 带了符号就按文档符号的名字跳过去；文件或符号没有就不打开。没有路径的是不带路径的引用，按这份文件、从它写在的那块（block）找（见 resolveSymbolLink）。
    */
-  private async openRepoLink(rel: string | undefined, symbol?: string): Promise<void> {
+  private async openRepoLink(rel: string | undefined, symbol?: string, block?: string): Promise<void> {
     const link = rel ?? `[${symbol}]`
     try {
       if (symbol !== undefined) {
-        const target = await this.symbolTarget(rel, symbol)
+        const target = await this.symbolTarget(rel, symbol, block)
         if (!target) {
-          this.markRepoLinkBroken(rel, symbol)
+          this.markRepoLinkBroken(rel, symbol, block)
           return
         }
         await this.showLinkTarget(target)
-        this.markRepoLinkOpened(rel, symbol)
+        this.markRepoLinkOpened(rel, symbol, block)
         return
       }
       if (rel === undefined)
@@ -946,12 +946,12 @@ class Preview {
   }
 
   /**
-   * 链接指的符号在哪：没有 path 的是不带路径的引用，按这份文件找（resolveSymbolLink）；有 path 的在那个文件的文档符号里找（locateRepoSymbol）。
-   * 文件没有、是目录、符号没有时没有。点链接、悬停链接看定义卡片都按它找。
+   * 链接指的符号在哪：没有 path 的是不带路径的引用，按这份文件、从它写在的那块（block）找（resolveSymbolLink）；
+   * 有 path 的在那个文件的文档符号里找（locateRepoSymbol）。文件没有、是目录、符号没有时没有。点链接、悬停链接看定义卡片都按它找。
    */
-  private async symbolTarget(rel: string | undefined, symbol: string): Promise<LinkTarget | undefined> {
+  private async symbolTarget(rel: string | undefined, symbol: string, block: string | undefined): Promise<LinkTarget | undefined> {
     if (rel === undefined)
-      return this.resolveSymbolLink(symbol)
+      return this.resolveSymbolLink({ symbol, block })
     const root = await repoRootOf(this.document.uri.fsPath)
     const entry = root === undefined ? undefined : await resolveRepoEntry(root, rel)
     return entry && !entry.isDirectory ? this.locateRepoSymbol(Uri.file(entry.fsPath), symbol) : undefined
@@ -964,13 +964,15 @@ class Preview {
   }
 
   /**
-   * 注释里不带路径的引用（`[Name]`、`[Class.member]`）指到哪：本文件定义的优先，再是导入进来的（见 symbolLinks.ts）。
+   * 注释里不带路径的引用（`[Name]`、`[Class.member]`）指到哪：从它写在的那块（block，在眼下这份块树里找）的声明起一层层往外找本文件定义的，
+   * 再是导入进来的（见 symbolLinks.ts）。检查坏链接、点击、悬停看定义卡片都走这里。
    * 文档符号同仓库链接找符号那一套（symbols，一批引用一起问时共用一份），导入的名字问语言扩展它的定义（同跳到定义），定义几处取第一处。
    */
-  private async resolveSymbolLink(symbol: string, symbols: (uri: string) => Promise<SymbolNode[]> = documentSymbolNodes): Promise<LinkTarget | undefined> {
-    const { document } = this
+  private async resolveSymbolLink({ symbol, block }: SymbolReference, symbols: (uri: string) => Promise<SymbolNode[]> = documentSymbolNodes): Promise<LinkTarget | undefined> {
+    const { document, sent } = this
+    const at = block === undefined || !sent ? undefined : referenceScope(sent.blocks, block)
     try {
-      return await resolveReference(symbol, this.imports, {
+      return await resolveReference(symbol, at, this.imports, {
         uri: document.uri.toString(),
         symbols,
         definition: async (offset) => {
@@ -998,14 +1000,14 @@ class Preview {
     await this.openElsewhere(doc.uri, { range: new Range(doc.positionAt(selection.startIndex), doc.positionAt(selection.endIndex)), offset: selection.startIndex })
   }
 
-  private markRepoLinkBroken(path: string | undefined, symbol?: string): void {
+  private markRepoLinkBroken(path: string | undefined, symbol?: string, block?: string): void {
     if (this.ready)
-      this.post({ type: 'repoLinkBroken', ...(path !== undefined && { path }), ...(symbol !== undefined && { symbol }) })
+      this.post({ type: 'repoLinkBroken', ...(path !== undefined && { path }), ...(symbol !== undefined && { symbol }), ...(block !== undefined && { block }) })
   }
 
-  private markRepoLinkOpened(path: string | undefined, symbol?: string): void {
+  private markRepoLinkOpened(path: string | undefined, symbol?: string, block?: string): void {
     if (this.ready)
-      this.post({ type: 'repoLinkOpened', ...(path !== undefined && { path }), ...(symbol !== undefined && { symbol }) })
+      this.post({ type: 'repoLinkOpened', ...(path !== undefined && { path }), ...(symbol !== undefined && { symbol }), ...(block !== undefined && { block }) })
   }
 
   /**