实例解析:如何开发 VSCode LSP 服务

2,431 阅读9分钟

「本文已参与好文召集令活动,点击查看:后端、大前端双赛道投稿,2万元奖池等你挑战!

从一张动图说起:

上图应该大家经常使用的 错误诊断 功能,它能够在你编写代码的过程中提示,那一块代码存在什么类型的问题。

这个看似高大上的功能,从插件开发者的角度看其实特别简单,基本上就是用上一篇文章《你不知道的 VSCode 代码高亮原理》中简单介绍过的 VSCode 开发语言特性的三种方案:

  • 基于 Sematic Tokens Provider 协议的词法高亮
  • 基于 Language API 的编程式语法高亮
  • 基于 Language Server Protocol 的多进程架构语法高亮

其中, Language Server Protocol 由于性能与开发效率上的优势已经逐渐成为主流实现方案,本文接下来会基于 LSP 展开介绍各种语言特性的实现细节,解答 LSP 的通讯模型与开发模式。

示例代码

本文示例均已同步到 github,建议读者先拉下代码实际体验:

# 1. clone 示例代码
git clone git@github.com:Tecvan-fe/vscode-lsp-sample.git
# 2. 安装依赖
npm i # or yarn
# 3. 使用 vscode 打开示例代码
code ./vscode-lsp-sample
# 4. 在 vscode 中按下 F5 启动调试

顺利执行完毕后,可以看到插件的调试窗口:

核心代码有:

  • server/src/server.ts:LSP 服务端代码,提供代码补全、错误诊断、代码提示等常见语言功能的示例
  • client/src/extension.ts:提供一系列 LSP 参数,包括 Server 的调试端口、代码入口、通讯方式等。
  • packages.json:主要提供了语法插件所需要的配置信息,包括:
    • activationEvents: 声明插件的激活条件,代码中的 onLanguage:plaintext 意为打开 txt 文本文件时激活
    • main: 插件的入口文件

其中,client/src/extension.tspackages.json 都比较简单,本文过多介绍,重点在于 server/src/server.ts 文件,接下来我们逐步拆解,解析不同语言特性的实现细节。

如何编写 Language Server

Server 结构解析

示例项目的 server/src/server.ts 实现了一个小型但完整的 Language Server 应用,核心代码:


// 要素1: 初始化 LSP 连接对象
const connection = createConnection(ProposedFeatures.all);

// 要素2: 创建文档集合对象,用于映射到实际文档
const documents: TextDocuments<TextDocument> = new TextDocuments(TextDocument);

connection.onInitialize((params: InitializeParams) => {
  // 要素3: 显式声明插件支持的语言特性
  const result: InitializeResult = {
    capabilities: {
      hoverProvider: true
    },
  };
  return result;
});

// 要素4: 将文档集合对象关联到连接对象
documents.listen(connection);

// 要素5: 开始监听连接对象
connection.listen();

从示例代码可以总结出 Language Server 的 5 个必要步骤:

  • 创建 connection 对象,用于实现客户端与服务器之间的信息互通
  • 创建 documents 文档集合对象,用于映射客户端正在编辑的文件
  • connection.onInitialize 事件中,显式声明插件支持的语法特性,例如上例中返回对象包含 hoverProvider: true 声明,表示该插件能够提供代码悬停提示功能
  • documents 关联到 connection 对象
  • 调用 connection.listen 函数,开始监听客户端消息

上述 connectiondocuments 等对象定义在 npm 包:

  • vscode-languageserver/node
  • vscode-languageserver-textdocument

这是一个基本模板,主要完成了 Language Server 各种初始化操作,后续就可以使用 connection.onXXXdocuments.onXXX 监听各类交互事件,并在事件回调中返回符合 LSP 协议的结果,或者显式调用通讯函数如 connection.sendDiagnostics 发送交互信息。

接下来我们通过几个简单实例,分析各项语言特性的实现逻辑。

悬停提示

当鼠标停留在语言元素如函数、变量、符号等 token 时,VSCode 会显示 token 对应描述与帮助信息:

要实现悬停提示功能,首先需要声明插件支持 hoverProvider 特性:

connection.onInitialize((params: InitializeParams) => {
  return {
    capabilities: {
      hoverProvider: true
    },
  };
});

之后,需要监听 connection.onHover 事件,并在事件回调中返回提示信息:

connection.onHover((params: HoverParams): Promise<Hover> => {
  return Promise.resolve({
    contents: ["Hover Demo"],
  });
});

OK,这就是一个很简单的语言特性示例了,本质上就是监听事件 + 返回结果,非常简单。

代码格式化

代码格式化是一个特别有用的功能,能够帮助用户快速、自动完成代码的美化处理,实现效果如:

实现悬停提示功能,首先需要声明插件支持 documentFormattingProvider 特性:

{
    ...
    capabilities : {
        documentFormattingProvider: true
        ...
    }
}

之后,监听 onDocumentFormatting 事件:

connection.onDocumentFormatting(
  (params: DocumentFormattingParams): Promise<TextEdit[]> => {
    const { textDocument } = params;
    const doc = documents.get(textDocument.uri)!;
    const text = doc.getText();
    const pattern = /\b[A-Z]{3,}\b/g;
    let match;
    const res = [];
    // 查找连续大写字符串
    while ((match = pattern.exec(text))) {
      res.push({
        range: {
          start: doc.positionAt(match.index),
          end: doc.positionAt(match.index + match[0].length),
        },
        // 将大写字符串替换为 驼峰风格
        newText: match[0].replace(/(?<=[A-Z])[A-Z]+/, (r) => r.toLowerCase()),
      });
    }

    return Promise.resolve(res);
  }
);

示例代码中,回调函数主要实现将连续大写字符串格式化为驼峰字符串,效果如图:

函数签名

函数签名特性在用户输入函数调用语法时触发,此时 VSCode 会根据 Language Server 返回的内容,显示该函数的帮助信息。

实现函数签名功能,需要首先声明插件支持 documentFormattingProvider 特性:

{
    ...
    capabilities : {
        signatureHelpProvider: {
            triggerCharacters: ["("],
        }
        ...
    }
}

之后,监听 onSignatureHelp 事件:

connection.onSignatureHelp(
  (params: SignatureHelpParams): Promise<SignatureHelp> => {
    return Promise.resolve({
      signatures: [
        {
          label: "Signature Demo",
          documentation: "帮助文档",
          parameters: [
            {
              label: "@p1 first param",
              documentation: "参数说明",
            },
          ],
        },
      ],
      activeSignature: 0,
      activeParameter: 0,
    });
  }
);

实现效果:

错误提示

注意,错误提示的实现逻辑与上述事件 + 响应的模式有一点点不同:

  • 首先不需要通过capabilities 做额外声明;
  • 监听的是 documents.onDidChangeContent 事件,而不是 connection 对象上的事件
  • 不是在事件回调中用 return 语句返回错误信息,而是调用 connection.sendDiagnostics 发送错误消息

完整示例:

// 增量错误诊断
documents.onDidChangeContent((change) => {
  const textDocument = change.document;

  // The validator creates diagnostics for all uppercase words length 2 and more
  const text = textDocument.getText();
  const pattern = /\b[A-Z]{2,}\b/g;
  let m: RegExpExecArray | null;

  let problems = 0;
  const diagnostics: Diagnostic[] = [];
  while ((m = pattern.exec(text))) {
    problems++;
    const diagnostic: Diagnostic = {
      severity: DiagnosticSeverity.Warning,
      range: {
        start: textDocument.positionAt(m.index),
        end: textDocument.positionAt(m.index + m[0].length),
      },
      message: `${m[0]} is all uppercase.`,
      source: "Diagnostics Demo",
    };
    diagnostics.push(diagnostic);
  }

  // Send the computed diagnostics to VSCode.
  connection.sendDiagnostics({ uri: textDocument.uri, diagnostics });
});

这段逻辑诊断代码中是否存在连续大写字符串,通过 sendDiagnostics 发送相应的错误信息,实现效果:

如何识别事件与响应体

上述示例,我有意忽略大多数实现细节,更关注实现语言特性的基本框架和输入输出。授人以鱼不如授人以渔,所以接下来我们花一点点时间了解从哪里获取这些接口、参数、响应体的信息。有两个非常重要的链接:

这两个网页提供了 VSCode 所支持的所有语言特性的详细介绍,可以在这里找到你想要实现的特性的概念性描述,例如对于代码补齐:

嗯,有点复杂且太过 detail,不过还是很有必要耐心了解下,让你对即将要做的事情有一个高层概念上的理解。

此外,如果你选择使用 TS 编写 LSP,事情会变得更简单。vscode-languageserver 包提供了非常完善的 Typescript 类型定义,我们完全可以借助 ts + VSCode 的代码提示找到需要使用的监听函数:

Kapture 2021-07-08 at 23.54.33.gif

之后,根据函数签名找到参数、结果的类型定义:

Kapture 2021-07-09 at 10.25.54.gif

之后,就可以根据类型定义,有针对性地处理参数,返回对应结构的数据。

深入理解 LSP

看完示例后,我们再反过头来看看 LSP。LSP —— Language Server Protocol 本质上是一种基于 JSON-RPC 的进程间通讯协议,LSP 本身包含两大块内容:

  • 定义 client 与 server 之间的通讯模型,也就是谁、在什么时候、以什么方式向对方发送什么格式的信息,接收方又以什么方式返回响应信息
  • 定义通讯信息体,也就是以什么格式、什么字段、什么样的值表达信息状态

作为类比,HTTP 协议专门用于描述网络节点间如何传输、理解超媒体文档的网络通讯协议;而 LSP 协议则专门用于描述 IDE 中,用户行为与响应之间的通讯方式与信息结构。

总结一下,LSP 架构的工作流程如下:

  • 编辑器如 VSCode 跟踪、计算、管理用户行为模型,在发生某些特定的行为序列时,以 LSP 协议规定的通讯方式向 Language Server 发送动作与上下文参数
  • Language Server 根据这些参数异步地返回响应信息
  • 编辑器再根据响应信息处理交互反馈

简单说,编辑器负责与用户直接交互, Language Server 负责在背后默默计算如何响应用户的交互动作,两者以进程粒度分离、解耦,在 LSP 协议框架下各司其职又协作共生。就好像我们通常开发的 Web 应用中,前端负责与用户交互,服务端负责管理诸如权限、业务数据、业务状态流转等不可见的部分。

目前,LSP 协议已经发展到 3.16 版本,覆盖大多数语言特性,包括:

  • 代码补全
  • 代码高亮
  • 定义跳转
  • 类型推断
  • 错误检测
  • 等等

得益于 LSP 清晰的设计,这些语言特性的开发套路都很相似,学习曲线很平滑,开发的时候基本上只需要关心监听那个函数,返回什么格式的结构,可以说掌握上述几个示例之后就可以很简单地上手了。

过去,IDE 对语言特性的支持是集成在 IDE 或者以同构插件形式实现的,在 VSCode 中这种同构扩展能力以 Language APISematic Tokens Provider 接口方式提供,这两种方式在上一篇文章《你不知道的 VSCode 代码高亮原理》都有过介绍了,虽然架构上比较简单,容易理解,但有一些明显硬伤:

  • 插件开发者必须复用 VSCode 本身的开发语言、环境,例如 Python 语言插件就必须用 JavaScript 写
  • 同一个编程语言需要为不同 IDE 重复开发相似的扩展插件,重复投入

LSP 最大的优势就是将 IDE 客户端与实际计算交互特性的服务端隔离开来,同一个 Language Service 可以重复应用在多个不同 Language Client 中。

此外,LSP 协议下客户端、服务器分别在各自进程运行,在性能上也会有正向收益:

  • 确保 UI 进程不卡顿
  • Node 环境下,充分利用多核 CPU 能力
  • 由于不再限定 Language Server 的技术栈,开发者可以选择更高性能的语言,例如 Go

总的来说,就是很强。

总结

本文介绍了 VSCode 下,开发一款基于 LSP 的语言插件所需要具备的最最基本的技能,实际开发的时候通常还会混合另一种技术:嵌入式语法 —— Embedded Languages Server ,实现复杂的多语言复合支持,如果有人感兴趣,我们下周可以聊聊。