1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
"use strict";
import { runInNewContext } from "vm";
import * as path from "path";
/*
Modified code from https://github.com/hdevalke/rust-test-lens/
*/
import {
CancellationToken,
CodeLens,
CodeLensProvider,
Event,
EventEmitter,
Range,
TextDocument,
DebugConfiguration,
} from "vscode";
/*
This is not good enough. I think I have to let the language server make the lenses. Also I can't tell when the debug button is pressed, so users can end up pressing the button multiple times.
Would also be neat to see how many tests have failed since last, and also press a button to run all tests in the workspace.
*/
export class RunnableCodeLensProvider implements CodeLensProvider {
constructor(private _onDidChange: EventEmitter<void>) { }
public async provideCodeLenses(
doc: TextDocument,
token: CancellationToken
): Promise<CodeLens[]> {
if (token.isCancellationRequested) {
return [];
}
let lenses: CodeLens[] = this.testMethodLenses(doc);
return lenses;
}
get onDidChangeCodeLenses(): Event<void> {
return this._onDidChange.event;
}
public updateArgs() {
}
private testMethodLenses(doc: TextDocument) {
const text = doc.getText();
const reTest = /\@\(?test\)?/g;
const reFnTest = /\s*\w+\s*::\s*proc\s*\s*\(/g;
var testMatch: RegExpExecArray | null = null;
let lenses: CodeLens[] = [];
while ((testMatch = reTest.exec(text)) !== null) {
reFnTest.lastIndex = reTest.lastIndex;
const match = reFnTest.exec(text);
if (match === null) {
continue;
}
const fn = match[0].split(":")[0];
if (fn && fn[0]) {
const debugCodelens = this.makeDebugLens(reTest.lastIndex, testMatch[0].length, fn.trim(), doc);
if (debugCodelens !== undefined) {
lenses.push(debugCodelens);
}
const runCodelens = this.makeRunLens(reTest.lastIndex, testMatch[0].length, fn.trim(), doc);
if (runCodelens !== undefined) {
lenses.push(runCodelens);
}
}
}
return lenses;
}
private makeDebugLens(index: number, length: number, fn: string, doc: TextDocument) {
const startIdx = index - length;
const start = doc.positionAt(startIdx);
const end = doc.positionAt(index);
const range = new Range(start, end);
return new CodeLens(range, {
title: "Debug",
command: "ols.runDebugTest",
tooltip: "Debug",
arguments: [{
function: fn,
cwd: path.dirname(doc.uri.fsPath)
}]
});
}
private makeRunLens(index: number, length: number, fn: string, doc: TextDocument): any | undefined {
const startIdx = index - length;
const start = doc.positionAt(startIdx);
const end = doc.positionAt(index);
const range = new Range(start, end);
return new CodeLens(range, {
title: "Run",
command: "ols.runTest",
tooltip: "Run",
arguments: [{
function: fn,
cwd: path.dirname(doc.uri.fsPath)
}]
});
}
}
|
