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
|
const vscode = require('vscode');
const fnRegex = /^(\t*)(.*)err\s?:?=.+?$/;
class ErrorsWrapper {
provideCodeActions(document, range) {
const editor = vscode.window.activeTextEditor;
if (!editor) {
return undefined;
}
const line = document.lineAt(editor.selection.start.line);
if (!fnRegex.test(line.text)) {
vscode.commands.executeCommand('setContext', 'allowWrapIferr', false);
return undefined;
}
vscode.commands.executeCommand('setContext', 'allowWrapIferr', true);
const action = new vscode.CodeAction('Add error checking', vscode.CodeActionKind.RefactorRewrite);
action.command = { command: 'gotools.wrap-error', title: 'Add error checking block', tooltip: '' };
return [
action,
];
}
}
const wrapError = () => {
const editor = vscode.window.activeTextEditor;
if (!editor) {
return;
}
const document = editor.document;
const line = document.lineAt(editor.selection.start.line);
const matches = line.text.match(fnRegex);
if (matches === null || matches.length === 0) {
return;
}
const extravars = matches[2].split(',').map(x => x.trim()).filter(x => x);
if (extravars.filter(x => x !== "_").length > 0) {
editor.insertSnippet(
new vscode.SnippetString(`\nif err != nil {\n\treturn \${1:nil, }\${2:err}\n}\n`),
new vscode.Position(line.range.end.line, line.range.end.character + line.firstNonWhitespaceCharacterIndex),
);
} else {
const tabs = matches[1];
const original = matches[0].trimStart()
editor.insertSnippet(
new vscode.SnippetString(`${tabs}if ${original}; err != nil {\n${tabs}\treturn \${2:nil, }\${3:err}\n${tabs}}\n`),
line.range,
{
undoStopBefore: true,
undoStopAfter: true
}
);
}
};
module.exports = {
ErrorsWrapper, wrapError
}
|
