手把手教你开发自己的VSCode插件

2024年 3月 19日 70.5k 0

一、前言

在VSCode市场上搜索一个PDF阅读插件,找到了下载量最高的插件。在检查了插件的源代码之后,发现它直接嵌入了pdf.js的Web界面。

图片图片

开发这样的插件并不复杂,只需要一些插件开发的知识。

接下来,将在这里与大家分享这个插件是如何开发的。虽然与源代码可能有些许不同,但基本原理是相同的。

二、定义自定义编辑器

默认情况下,VSCode不直接支持查看某些特定类型的文件。但是,你可以通过使用自定义编辑器(customerEditors)来扩展其功能。

通过customerEditors,你可以创建完全可自定义的读/写编辑器,以替换VSCode的标准文本编辑器,用于处理特定类型的资源。

例如,在编辑Markdown文件时,可以创建一个自定义编辑器,从而实时预览Markdown的渲染效果。

对于PDF文件的预览,同样也可以使用customerEditors功能。

首先,在插件的pacakge.json文件中定义它:

"contributes": {
  "customEditors": [
    {
      "viewType": "dodo-reader.pdfEditor",
      "displayName": "PDF Viewer",
      "selector": [
        {
          "filenamePattern": "*.pdf"
        }
      ]
    }
  ]
}

三、注册自定义编辑器

创建一个实现vscode.CustomEditorProvider接口的类。该接口包含了用于管理自定义编辑器(例如打开和保存)的方法。例如:

class PdfEditorProvider implements Partial {
 constructor() {
 // 可在此处进行初始化操作
 }

resolveCustomEditor(document: CustomDocument, webviewPanel: WebviewPanel, _token: CancellationToken) {
 // 基于URI创建一个自定义文档,并返回一个自定义文档对象
 }
openCustomDocument(uri: vscode.Uri, openContext: vscode.CustomDocumentOpenContext, token: vscode.CancellationToken) {
 // 将自定义文档与Webview面板关联,并处理编辑器内容与Webview之间的交互
 }
}
 // 注册你的提供程序
const myProvider = new PdfEditorProvider();
const disposable = vscode.window.registerCustomEditorProvider('dodo-reader.pdfEditor', myProvider);
context.subscriptions.push(disposable);

四、改进视图提供程序

如上所述注册自定义编辑器之后,下一步就是加强PdfEditorProvider接口,以定义视图的显示方式。

显示涉及使用网络视图,而你将使用pdf.js的Web视图程序。首先,下载预构建的程序(现代浏览器),并将其提取到你的项目目录中。

图片图片

这个程序可以直接在浏览器中访问。在目录中启动一个服务,并打开地址:

你可以通过添加查询参数?file=fileUrl来打开PDF文件。

4.1 openCustomDocument

当打开PDF文件时,首先会调用openCustomDocument方法。你可以基于传入的URI创建一个自定义文档对象。该文档对象将包含你想要编辑的内容。在这个程序中,不进行任何处理,而是直接返回一个包含URI的对象,该对象将传递给resolveCustomEditor方法的document参数。

openCustomDocument(uri: vscode.Uri, openContext: vscode.CustomDocumentOpenContext, token: vscode.CancellationToken) {
 return {
   uri,
   dispose: () => { }
 };
}

4.2 resolveCustomEditor

在resolveCustomEditor方法中,可以定义要显示的视图,程序如下。

resolveCustomEditor(document: CustomDocument, webviewPanel: WebviewPanel, _token: CancellationToken) {
   webviewPanel.webview.html = 'Hello World!';
}

要显示PDF文件,只需将下载的PDF程序的HTML内容替换为当前的webviewPanel.webview.html值。最初认为可以更改为:

// 嵌入一个iframe以查看PDF
webviewPanel.webview.html = `
  
  
   
     
   
   
     
   
  
`

这个方法是最简单的,但令人意外的是,iframe没有显示内容。这可能是由于VSCode的安全策略限制造成的。

因此,在这里考虑了另一种方法:读取PDF视图主页的HTML内容,并将其赋值给webviewPanel.webview.html。

然而,这种方法可能会遇到一个问题:如何向PDF视图程序提供文件链接。

如前所述,PDF视图程序将通过查询参数“file.”获取文件链接。如果没有提供此参数,程序将读取默认链接。以下是一个相关的源代码示例:

file = params.get("file") ?? _app_options.AppOptions.get("defaultUrl");

然而,直接赋值HTML文本不能通过URL提供“file”的值。这可能需要修改源代码。尽管修改源代码可能会带来一些不便,但一开始似乎也没有其他办法,所以不得不尝试一下。一种方法是改变获取“file”值的方式,直接从全局变量中获取。可以使用如下方法在HTML内容中添加“file”值:

window.file = 'https://...'

然后将上述源代码修改为:

file = window.file ?? _app_options.AppOptions.get("defaultUrl");

然而,就在我以为成功即将到来的时候,出现了一个红色错误:“加载PDF时出现错误。文件来源与阅览器不匹配”。

经过进一步调查,在相应的源代码中发现了一个“fileinputchange”事件监听处理程序,如下所示:

var webViewerFileInputChange = function (evt) {
 if (PDFViewerApplication.pdfViewer?.isInPresentationMode) {
   return;
 }
 const file = evt.fileInput.files[0];
 PDFViewerApplication.open({
   url: URL.createObjectURL(file),
   originalUrl: file.name
 });
};

从代码中很容易看出,一旦检测到文件发生更改,就会立即调用open方法打开文件。重要的是要注意,该open方法并不验证URL是否与当前源相匹配。目前,它使用了一个blob链接。在这种情况下,我们是否可以在初始化后直接调用open方法来打开文件呢?经过一些改造,最终证明是可行的。以下是修改后的代码示例:

resolveCustomEditor(document: CustomDocument, webviewPanel: WebviewPanel, _token: CancellationToken) {
   webviewPanel.webview.options = {
     enableScripts: true,
     localResourceRoots: [vscode.Uri.file(path.dirname(document.uri.fsPath)), this.context.extensionUri]
  };
   const base = vscode.Uri.joinPath(this.context.extensionUri, 'dist/web/pdf/web/')
   webviewPanel.webview.html = readFileSync(path.join(base.fsPath, 'viewer.html'), 'utf8').replace('', `
   
   
   
 `)
}

对上面的关键代码进行分析:

  • localResourceRoots参数:该参数的目的是定义可以通过Web URL访问的目录。在这里,需要明确定义两个目录:一个用于打开PDF文件的当前目录,另一个用于插件所在的目录。确保两者都正确定义,否则可能在访问时遇到401错误。
  • tag:通过设置tag,可以为网页内部加载资源提供基本路径。确保资源能够正确加载。
  • setTimeout函数:在调用程序打开文件时使用setTimeout的目的是在打开默认PDF文件后再打开所需的PDF文件。这样可以防止后面执行打开默认PDF的操作覆盖想要打开的PDF(尽管实际上会出现默认文件错误,但不会产生实质性影响)。

与之前提到的PDF插件实现相比,这种实现要简单得多,但原理基本相同。

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论