如何访问VSCode原生语言服务器获取JS/TS函数列表？

嘿，我懂你想直接调用VS Code内置的JS/TS语言服务器来抓取文件里的所有函数列表——这个需求其实挺实用，但官方文档确实没把这条路子讲得太明明白白。别担心，我给你梳理下可行的实现步骤：

核心逻辑 #

VS Code里的语言服务器和客户端是通过RPC机制通信的，你可以通过扩展API获取到已注册的JS/TS语言客户端，然后向它发送标准的LSP（语言服务器协议）请求来获取符号信息。内置的JS/TS语言服务器的ID是typescript，这是我们要定位的目标。

具体实现代码 #

首先，在你的扩展代码里获取JS/TS语言客户端：

import * as vscode from 'vscode';

// 获取内置的TS/JS语言客户端
async function getTSLanguageClient(): Promise<vscode.LanguageClient | undefined> {
    // 遍历所有扩展，找到导出了getLanguageClient方法的实例
    const allClients = vscode.extensions.all
        .map(ext => ext.exports?.getLanguageClient?.())
        .filter(client => client);
    
    // 筛选出ID为typescript的客户端（对应JS/TS语言服务）
    return allClients.find(client => client.id === 'typescript') as vscode.LanguageClient | undefined;
}

接下来，发送textDocument/documentSymbol请求获取文档符号，再过滤出函数：

async function extractFunctionsFromActiveFile() {
    const client = await getTSLanguageClient();
    if (!client) {
        vscode.window.showErrorMessage('无法连接到JS/TS语言服务器');
        return;
    }

    const activeEditor = vscode.window.activeTextEditor;
    if (!activeEditor?.document) {
        vscode.window.showErrorMessage('请先打开一个JS/TS文件');
        return;
    }

    const docUri = activeEditor.document.uri;
    // 发送LSP的documentSymbol请求
    const symbols = await client.sendRequest<vscode.DocumentSymbol[]>(
        'textDocument/documentSymbol',
        { textDocument: { uri: docUri.toString() } }
    );

    // 递归遍历符号树，筛选出所有函数类型的符号
    const functionList: string[] = [];
    function traverse(symbols: vscode.DocumentSymbol[]) {
        symbols.forEach(sym => {
            if (sym.kind === vscode.SymbolKind.Function) {
                functionList.push(`${sym.name} (行${sym.range.start.line + 1})`);
            }
            if (sym.children) traverse(sym.children);
        });
    }
    traverse(symbols);

    // 输出结果，这里可以改成你需要的处理逻辑
    console.log('提取到的函数：', functionList);
    vscode.window.showInformationMessage(`找到${functionList.length}个函数：\n${functionList.join('\n')}`);
}

关键注意点 #

• 这个方法依赖VS Code内置的JS/TS语言服务，确保用户没有手动禁用它（默认是启用状态）
• textDocument/documentSymbol是LSP标准请求，返回的不仅有函数，还有类、变量、接口等符号，你可以通过SymbolKind来过滤不同类型的元素
• 如果需要更详细的函数信息（比如参数、返回类型），可以后续针对单个函数发送textDocument/hover或textDocument/definition请求，但documentSymbol已经足够满足“获取函数列表”的需求

内容的提问来源于stack exchange，提问作者F Yaqoob