实现智能选择功能需注册selectionRangeProvider,通过解析语法结构按层级返回从内到外的选择范围数组,如变量→表达式→语句→函数,每个SelectionRange通过parent形成链式结构,顺序为内层到外层,配合expand/shrink命令实现精准选区调整。

在 vscode 中开发智能选择(Smart Select)功能,主要是通过实现 “智能选择提供者”(Selection Range Provider)来完成。这个功能允许用户通过“扩展选择”(Expand Selection)和“缩小选择”(Shrink Selection)命令,按语法结构逐层调整选区范围,比如从一个变量名扩展到表达式、语句块,再到整个函数。
实现 SelectionRangeProvider 接口
要为你的语言添加智能选择支持,需要在 VSCode 扩展中注册一个 SelectionRangeProvider。该提供者根据当前光标位置,返回一个嵌套的选择范围数组,从最内层开始,逐层向外扩展。
接口定义如下:
selectionRangeProvider: {   provideSelectionRanges(     document: TextDocument,     positions: Position[],     token: CancellationToken   ): ProviderResult<SelectionRange[]>; }
其中每个 SelectionRange 包含一个 range 字段,并可选地指向其 parent,形成链式结构。
注册智能选择提供者
在 extension.ts 的 activate 函数中注册提供者:
vscode.languages.registerSelectionRangeProvider('yourLanguage', {   provideSelectionRanges(document, positions) {     return positions.map(position => {       // 从当前位置构建选择层级       return buildSelectionRange(document, position);     });   } });
将 'yourLanguage' 替换为你支持的语言标识符,如 Javascript、myLang 等。
构建选择层级的逻辑
核心在于解析文档内容,找到光标所在位置的语法结构。你可以:
- 使用正则匹配简单结构(适合轻量语言)
- 集成 AST 解析器(如 Tree-sitter、Chevrotain、ANTLR 生成的解析器)获取精确语法节点
示例:从一个变量向外围扩展
function buildSelectionRange(document: vscode.TextDocument, position: vscode.Position) {   const text = document.lineAt(position).text;   const wordRange = document.getWordRangeAtPosition(position);   if (!wordRange) return null; <p>let currentRange = new vscode.Selection(wordRange.start, wordRange.end); const ranges = [];</p><p>// 模拟逐层扩展:词 → 表达式 → 语句 → 块 ranges.push(new vscode.SelectionRange(currentRange));</p><p>// 向前查找分号或换行,构造语句范围 const statementRange = getStatementRange(document, position); if (statementRange && !statementRange.isEqual(currentRange)) { ranges.push(new vscode.SelectionRange(statementRange, ranges[0])); }</p><p>// 更外层:函数体、类等,依此类推 const functionRange = getFunctionRange(document, position); if (functionRange) { ranges.push(new vscode.SelectionRange(functionRange, ranges[ranges.length - 1])); }</p><p>return ranges.reverse(); // VSCode 需要从外到内?实际是从内到外!保持顺序:内 → 外 }
注意:VSCode 要求返回的数组顺序是“从最内层开始,逐级 parent 指向外层”,即 [inner, middle, outer],每个的 parent 指向前一个更外层的范围。
测试与调试
使用快捷键 Shift+Alt+RightArrow(扩展选择)查看效果。打开命令面板执行 “Developer: Inspect Editor Tokens and Scopes” 可辅助定位光标处的语法信息。
确保你的语言服务器或文本处理逻辑能准确识别嵌套结构,避免跳层或遗漏。
基本上就这些。只要正确构建 selection range 链,VSCode 就能智能响应用户的选区操作。不复杂但容易忽略 parent 关联和顺序问题。


