src/goLanguageServer.ts: intercept test and benchmark command
This CL adds middleware to intercept the `go test -run` and `go test
-bench` commands from gopls and replaces them with the `go.test.cursor`
and `go.benchmark.cursor` commands respectively.
Change-Id: I911bdd060523cd6180270440607b7bcc88cf01c3
Reviewed-on: https://go-review.googlesource.com/c/vscode-go/+/244857
Reviewed-by: Hyang-Ah Hana Kim <hyangah@gmail.com>
diff --git a/src/goLanguageServer.ts b/src/goLanguageServer.ts
index 6d8446f..0f1bcfc 100644
--- a/src/goLanguageServer.ts
+++ b/src/goLanguageServer.ts
@@ -15,8 +15,16 @@
import util = require('util');
import vscode = require('vscode');
import {
- CloseAction, CompletionItemKind, ErrorAction, HandleDiagnosticsSignature, InitializeError,
- LanguageClient, Message, ProvideCompletionItemsSignature, ProvideDocumentLinksSignature,
+ CloseAction,
+ CompletionItemKind,
+ ErrorAction,
+ ExecuteCommandSignature,
+ HandleDiagnosticsSignature,
+ InitializeError,
+ LanguageClient,
+ Message,
+ ProvideCompletionItemsSignature,
+ ProvideDocumentLinksSignature,
RevealOutputChannelOn,
} from 'vscode-languageclient';
import WebRequest = require('web-request');
@@ -249,6 +257,32 @@
},
},
middleware: {
+ provideCodeLenses: async (doc, token, next): Promise<vscode.CodeLens[]> => {
+ const codeLens = await next(doc, token);
+ return codeLens.map((lens: vscode.CodeLens) => {
+ switch (lens.command.title) {
+ case 'run test': {
+ const args = lens.command.arguments;
+ return new vscode.CodeLens(lens.range, {
+ ...lens.command,
+ command: 'go.test.cursor',
+ arguments: [{ functionName: args[args.indexOf('run') + 1] }],
+ });
+ }
+ case 'run benchmark': {
+ const args = lens.command.arguments;
+ return new vscode.CodeLens(lens.range, {
+ ...lens.command,
+ command: 'go.benchmark.cursor',
+ arguments: [{ functionName: args[args.indexOf('bench') + 1] }],
+ });
+ }
+ default: {
+ return lens;
+ }
+ }
+ });
+ },
handleDiagnostics: (
uri: vscode.Uri,
diagnostics: vscode.Diagnostic[],
