Selection Range Provider是vscode中用于实现层级化代码选择的API,通过注册provideSelectionRanges方法,按光标位置从内到外逐层扩展选择范围,如从变量名扩展至函数体;需结合AST解析构建准确的SelectionRange链式结构以提升选择智能性。

在 VSCode 中实现选择范围提供者(Selection Range Provider),主要是为了让编辑器能够根据光标位置,智能地扩展选择范围。比如从一个变量名逐步扩展到语句、代码块、函数体甚至整个文件。这个功能在双击或连续按 Ctrl+Shift+→ 时特别有用。
什么是 Selection Range Provider?
Selection Range Provider 是 VSCode 语言扩展中的一个 API,允许你自定义代码中“层级化选择”的逻辑。它返回一个从当前位置向外逐层扩大的范围数组,每一层都比上一层更大。
要实现这个功能,你需要注册 SelectionRangeProvider 并实现其 provideSelectionRanges 方法。
如何注册 Selection Range Provider
在你的扩展的 extension.ts 文件中,使用 languages.registerSelectionRangeProvider 注册提供者:
vscode.languages.registerSelectionRangeProvider('javascript', { provideSelectionRanges(document, positions, token) { const ranges = positions.map(position => { // 返回每个 position 对应的选择范围链 return buildSelectionRanges(document, position); }); return ranges; } });
上面的 ‘javascript‘ 可以替换为你的语言标识,比如 myLang,也可以传入一个 LanguageSelector。
实现 provideSelectionRanges 方法
核心是构建从光标位置开始,逐层外扩的范围链。每层是一个 SelectionRange,包含当前范围和父级范围。
示例:简单基于括号和行边界构建层级:
function buildSelectionRanges(document: vscode.TextDocument, position: vscode.Position): vscode.SelectionRange[] { const ranges: vscode.SelectionRange[] = []; let start = position; let end = position; <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">// 向前查找最近的开始边界(如 {, (, 行首) // 向后查找结束边界(如 }, ), 行尾) // 这里简化处理:逐行扩展 while (start.line > 0 && end.line < document.lineCount - 1) { const lineRange = document.lineAt(start.line).range; const currentRange = new vscode.Range( start.translate(0, -1).isBefore(lineRange.start) ? lineRange.start : start.translate(0, -1), end.translate(0, 1).isAfter(lineRange.end) ? lineRange.end : end.translate(0, 1) ); ranges.push(new vscode.SelectionRange(currentRange)); // 扩展到更大的上下文(比如整个函数) // 实际项目中应结合 AST 解析更准确 start = start.translate(-1, 0); end = end.translate(1, 0); } // 最外层:整个文档 ranges.push(new vscode.SelectionRange(new vscode.Range( new vscode.Position(0, 0), document.lineAt(document.lineCount - 1).range.end ))); return ranges;
}
注意:SelectionRange 是链式结构,实际应通过 parent 字段连接,而不是直接返回数组。正确写法:
let parent: vscode.SelectionRange | undefined = undefined; for (const range of ranges.reverse()) { parent = new vscode.SelectionRange(range, parent); } return [parent]; // 注意返回的是最外层作为根
结合语言解析提升准确性
如果只是按行或字符扩展,效果有限。建议结合语法树(AST)来识别代码结构。
例如使用 Tree-sitter 或语言自带的 parser(如 typescript 的语言服务),找到光标所在节点,然后逐层向上返回其父节点的范围。
步骤如下:
- 解析文档生成 AST
- 找到包含光标位置的最深节点
- 沿着父节点一路向上,每个节点生成一个 Range
- 构造 SelectionRange 链并返回
这样能实现从变量 → 表达式 → 语句 → 代码块 → 函数的自然选择流程。
基本上就这些。实现 Selection Range Provider 能显著提升编辑体验,尤其对结构化语言非常实用。关键在于理解层级关系,并合理构造 parent 链。不复杂但容易忽略细节。


